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

在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_ChatServer和ChatService_ChatClient类型。
实现服务端逻辑
服务端需循环读取客户端消息,同时可随时向客户端发送响应。
示例实现:
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定义正确,生成代码后按流式接口编写收发逻辑,就能实现稳定的双向通信。注意控制上下文超时、错误处理和资源释放,避免连接泄漏。


