Go?gRPC服務(wù)雙向流式RPC教程

更新時(shí)間：2022年06月16日 09:03:18 作者：煙花易冷人憔悴

這篇文章主要為大家介紹了Go?gRPC服務(wù)雙向流式RPC教程示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

前言

上一篇介紹了客戶端流式RPC，客戶端不斷的向服務(wù)端發(fā)送數(shù)據(jù)流，在發(fā)送結(jié)束或流關(guān)閉后，由服務(wù)端返回一個(gè)響應(yīng)。本篇將介紹雙向流式RPC。

雙向流式RPC：客戶端和服務(wù)端雙方使用讀寫流去發(fā)送一個(gè)消息序列，兩個(gè)流獨(dú)立操作，雙方可以同時(shí)發(fā)送和同時(shí)接收。

情景模擬

雙方對話（可以一問一答、一問多答、多問一答，形式靈活）。

新建proto文件

新建both_stream.proto文件

1.定義發(fā)送信息

// 定義流式請求信息
message StreamRequest{
    //流請求參數(shù)
    string question = 1;
}

2.定義接收信息

// 定義流式響應(yīng)信息 message StreamResponse{ //流響應(yīng)數(shù)據(jù) string answer = 1; } // 定義流式響應(yīng)信息
message StreamResponse{
    //流響應(yīng)數(shù)據(jù)
    string answer = 1;
}

3.定義服務(wù)方法Conversations

雙向流式rpc，只要在請求的參數(shù)前和響應(yīng)參數(shù)前都添加stream即可

service Stream{
    // 雙向流式rpc，同時(shí)在請求參數(shù)前和響應(yīng)參數(shù)前加上stream
    rpc Conversations(stream StreamRequest) returns(stream StreamResponse){};
}

4.編譯proto文件

進(jìn)入both_stream.proto所在目錄，運(yùn)行指令:

protoc --go_out=plugins=grpc:./ ./both_stream.proto

創(chuàng)建Server端

1.定義我們的服務(wù)，并實(shí)現(xiàn)RouteList方法

這里簡單實(shí)現(xiàn)對話中一問一答的形式

// StreamService 定義我們的服務(wù)
type StreamService struct{}
// Conversations 實(shí)現(xiàn)Conversations方法
func (s *StreamService) Conversations(srv pb.Stream_ConversationsServer) error {
	n := 1
	for {
		req, err := srv.Recv()
		if err == io.EOF {
			return nil
		}
		if err != nil {
			return err
		}
		err = srv.Send(&pb.StreamResponse{
			Answer: "from stream server answer: the " + strconv.Itoa(n) + " question is " + req.Question,
		})
		if err != nil {
			return err
		}
		n++
		log.Printf("from stream client question: %s", req.Question)
	}
}

2.啟動(dòng)gRPC服務(wù)器

const (
	// Address 監(jiān)聽地址
	Address string = ":8000"
	// Network 網(wǎng)絡(luò)通信協(xié)議
	Network string = "tcp"
)

func main() {
	// 監(jiān)聽本地端口
	listener, err := net.Listen(Network, Address)
	if err != nil {
		log.Fatalf("net.Listen err: %v", err)
	}
	log.Println(Address + " net.Listing...")
	// 新建gRPC服務(wù)器實(shí)例
	grpcServer := grpc.NewServer()
	// 在gRPC服務(wù)器注冊我們的服務(wù)
	pb.RegisterStreamServer(grpcServer, &StreamService{})

	//用服務(wù)器 Serve() 方法以及我們的端口信息區(qū)實(shí)現(xiàn)阻塞等待，直到進(jìn)程被殺死或者 Stop() 被調(diào)用
	err = grpcServer.Serve(listener)
	if err != nil {
		log.Fatalf("grpcServer.Serve err: %v", err)
	}
}

運(yùn)行服務(wù)端

go run server.go
:8000 net.Listing...

創(chuàng)建Client端

1.創(chuàng)建調(diào)用服務(wù)端Conversations方法

// conversations 調(diào)用服務(wù)端的Conversations方法
func conversations() {
	//調(diào)用服務(wù)端的Conversations方法，獲取流
	stream, err := streamClient.Conversations(context.Background())
	if err != nil {
		log.Fatalf("get conversations stream err: %v", err)
	}
	for n := 0; n < 5; n++ {
		err := stream.Send(&pb.StreamRequest{Question: "stream client rpc " + strconv.Itoa(n)})
		if err != nil {
			log.Fatalf("stream request err: %v", err)
		}
		res, err := stream.Recv()
		if err == io.EOF {
			break
		}
		if err != nil {
			log.Fatalf("Conversations get stream err: %v", err)
		}
		// 打印返回值
		log.Println(res.Answer)
	}
	//最后關(guān)閉流
	err = stream.CloseSend()
	if err != nil {
		log.Fatalf("Conversations close stream err: %v", err)
	}
}

2.啟動(dòng)gRPC客戶端

// Address 連接地址
const Address string = ":8000"

var streamClient pb.StreamClient

func main() {
	// 連接服務(wù)器
	conn, err := grpc.Dial(Address, grpc.WithInsecure())
	if err != nil {
		log.Fatalf("net.Connect err: %v", err)
	}
	defer conn.Close()

	// 建立gRPC連接
	streamClient = pb.NewStreamClient(conn)
	conversations()
}

運(yùn)行客戶端，獲取到服務(wù)端的應(yīng)答

go run client.go
from stream server answer: the 1 question is stream client rpc 0
from stream server answer: the 2 question is stream client rpc 1
from stream server answer: the 3 question is stream client rpc 2
from stream server answer: the 4 question is stream client rpc 3
from stream server answer: the 5 question is stream client rpc 4

服務(wù)端獲取到來自客戶端的提問

from stream client question: stream client rpc 0
from stream client question: stream client rpc 1
from stream client question: stream client rpc 2
from stream client question: stream client rpc 3
from stream client question: stream client rpc 4