在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定义正确，生成代码后按流式接口编写收发逻辑，就能实现稳定的双向通信。注意控制上下文超时、错误处理和资源释放，避免连接泄漏。
以上就是如何在Golang中实现gRPC双向流的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1417870.html
微信扫一扫 
支付宝扫一扫 
