如何在Golang中实现gRPC双向流

golang中实现grpc双向流需先在.proto文件定义stream类型的rpc方法,如Chat(stream ChatMessage) returns (stream ChatMessage);接着使用protoc生成Go代码,得到服务端ChatService_ChatServer和客户端ChatService_ChatClient接口;服务端通过Recv()接收并用Send()发送消息,循环处理实现全双工通信;客户端则通过goroutine并发调用Send()和Recv(),实现双向数据传输。

如何在Golang中实现gRPC双向流

golang中实现gRPC双向流,核心是定义一个服务方法,其请求和响应都带有stream关键字,然后在客户端和服务端同时读写数据流。整个过程基于Protocol Buffers定义接口,并使用gRPC运行时处理通信。

定义.proto文件中的双向流方法

要在gRPC中启用双向流,首先需要在.proto文件中定义一个使用stream的rpc方法。这个方法的请求和响应类型都会携带stream标识。

 syntax = "proto3";  package example;  message ChatMessage {     string user = 1;     string message = 2; }  service ChatService {     rpc Chat(stream ChatMessage) returns (stream ChatMessage); } 

上述定义表示Chat方法允许客户端和服务端持续发送消息,实现全双工通信。

生成Go代码

使用protoc配合gRPC插件生成Go代码:

立即学习go语言免费学习笔记(深入)”;

 protoc -I . --go_out=plugins=grpc:. chat.proto 

会生成chat.pb.go文件,其中包含服务接口和客户端存根,重点关注生成的ChatService_ChatServerChatService_ChatClient类型。

实现服务端逻辑

服务端需循环读取客户端消息,同时可随时向客户端发送响应。

如何在Golang中实现gRPC双向流

如知AI笔记

如知笔记——支持markdown的在线笔记,支持ai智能写作、AI搜索,支持DeepseekR1满血大模型

如何在Golang中实现gRPC双向流27

查看详情 如何在Golang中实现gRPC双向流

示例实现:

 type ChatServer struct{}  func (s *ChatServer) Chat(stream pb.ChatService_ChatServer) error {     for {         msg, err := stream.Recv()         if err != nil {             return err         }          // 处理收到的消息         response := &pb.ChatMessage{             user:    "server",             message: "echo: " + msg.message,         }          // 发送响应         if err := stream.Send(response); err != nil {             return err         }     } } 

服务端通过Recv()接收流消息,Send()发送消息,直到连接关闭或发生错误。

实现客户端逻辑

客户端同样可以并发地发送和接收消息。

示例代码:

 conn, _ := grpc.Dial("localhost:50051", grpc.WithInsecure()) client := pb.NewChatServiceClient(conn) stream, _ := client.Chat(context.Background())  // 启动goroutine接收服务端消息 go func() {     for {         msg, err := stream.Recv()         if err != nil {             log.Println(err)             return         }         log.Printf("[%s] %s", msg.user, msg.message)     } }()  // 发送消息 for i := 0; i < 5; i++ {     msg := &pb.ChatMessage{         user:    "client",         message: fmt.Sprintf("hello %d", i),     }     stream.Send(msg)     time.Sleep(time.Second) }  stream.CloseSend() 

客户端使用Send()发送,Recv()接收,通常用单独的goroutine处理接收逻辑以避免阻塞。

基本上就这些。只要.proto定义正确,生成代码后按流式接口编写收发逻辑,就能实现稳定的双向通信。注意控制上下文超时、错误处理和资源释放,避免连接泄漏。

上一篇
下一篇
text=ZqhQzanResources