Go语言通过gRPC实现双向流通信，适用于聊天系统等实时场景。1. 定义chat.proto文件声明消息结构和BidirectionalChat服务，使用stream关键字指定双向流；2. 用protoc生成Go代码，得到.pb.go和.grpc.pb.go文件；3. 服务端实现ChatServer结构体，重写BidirectionalChat方法，循环接收客户端消息并发送响应；4. 客户端创建流后启goroutine异步接收服务端消息，同时循环发送消息；5. 运行服务端和客户端，可观察到双方持续收发消息，完成双向通信。需注意流关闭与错误处理，防止资源泄漏。
在Go语言中使用gRPC实现双向流通信，可以实现在客户端和服务端之间持续发送和接收消息。这种模式适用于聊天系统、实时数据推送等场景。下面通过一个简单示例展示如何使用gRPC定义并实现双向流通信。
1. 定义.proto文件
首先需要定义gRPC服务的接口和消息格式。创建一个chat.proto文件：
syntax = "proto3";package chat;
// 消息结构message Message {string content = 1;string user = 2;}
// 双向流服务service ChatService {rpc BidirectionalChat(stream Message) returns (stream Message);}
这个定义表示BidirectionalChat方法接收一个消息流，并返回一个消息流，即双向流。
立即学习“go语言免费学习笔记（深入）”；
2. 生成gRPC代码
使用Protocol Buffer编译器protoc配合Go插件生成代码：
protoc --go_out=. --go-grpc_out=. chat.proto
执行后会生成chat.pb.go和chat_grpc.pb.go两个文件，包含服务接口和数据结构。
3. 实现服务端逻辑
编写gRPC服务端，处理双向流请求：
package mainimport ("context""log""net"
"google.golang.org/grpc"pb "your-module-path/chat" // 替换为你的模块路径
)
type ChatServer struct {pb.UnimplementedChatServiceServer}
// BidirectionalChat 处理双向流func (s *ChatServer) BidirectionalChat(stream pb.ChatService_BidirectionalChatServer) error {for {// 接收客户端消息msg, err := stream.Recv()if err != nil {return err}
log.Printf("收到消息: %s (用户: %s)", msg.Content, msg.User) // 构造响应消息 response := &pb.Message{ Content: "已收到: " + msg.Content, User: "Server", } // 发送回客户端 if err := stream.Send(response); err != nil { return err }}
}
func main() {lis, err := net.Listen("tcp", ":50051")if err != nil {log.Fatalf("监听失败: %v", err)}
grpcServer := grpc.NewServer()pb.RegisterChatServiceServer(grpcServer, &ChatServer{})log.Println("gRPC服务器启动在 :50051")if err := grpcServer.Serve(lis); err != nil { log.Fatalf("启动失败: %v", err)}
}
4. 编写客户端代码
客户端也通过流发送和接收消息：
package mainimport ("context""log""time"
"google.golang.org/grpc"pb "your-module-path/chat"
)
func main() {conn, err := grpc.Dial("localhost:50051", grpc.WithInsecure())if err != nil {log.Fatalf("连接失败: %v", err)}defer conn.Close()
client := pb.NewChatServiceClient(conn)// 建立双向流stream, err := client.BidirectionalChat(context.Background())if err != nil { log.Fatalf("创建流失败: %v", err)}// 启动goroutine接收服务端消息go func() { for { msg, err := stream.Recv() if err != nil { log.Printf("接收错误: %v", err) return } log.Printf("<< %s (%s)", msg.Content, msg.User) }}()// 发送消息users := []string{"Alice", "Bob"}for i := 0; i < 5; i++ { msg := &pb.Message{ Content: "Hello " + string(rune('A'+i)), User: users[i%2], } if err := stream.Send(msg); err != nil { log.Printf("发送失败: %v", err) break } time.Sleep(time.Second)}// 关闭发送流stream.CloseSend()time.Sleep(2 * time.Second) // 等待接收剩余消息
}
5. 运行与测试
依次执行以下步骤：
启动服务端：go run server.go运行客户端：go run client.go
观察输出，可以看到客户端和服务端交替收发消息，实现真正的双向通信。
基本上就这些。只要定义好proto接口，正确实现流读写逻辑，就能轻松构建高效的gRPC双向通信应用。注意处理好流的生命周期和错误，避免goroutine泄漏。
以上就是Golang如何使用gRPC实现双向流通信_Golang gRPC双向流应用示例的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1426518.html
微信扫一扫 
支付宝扫一扫 
