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泄漏。
