双向流通信模式支持客户端和服务器持续互发消息,适用于聊天系统等实时场景。通过在.proto文件中使用stream关键字定义请求和响应类型,实现双方全双工通信;利用protoc生成Go代码后,服务端和客户端分别通过Recv()和Send()方法在循环中处理消息收发,可扩展为广播或多用户交互系统。

gRPC 支持四种通信模式,其中双向流(Bidirectional Streaming)是最灵活的一种。在这种模式下,客户端和服务器都可以连续发送多个消息,彼此独立读写,适用于实时通信场景,比如聊天系统、实时数据推送等。
定义双向流的 Protobuf 接口
要使用 gRPC 双向流,首先要定义服务接口。在 .proto 文件中,使用 stream 关键字标记请求和响应类型:
syntax = "proto3"; package chat; service ChatService { rpc ChatStream(stream Message) returns (stream Message); } message Message { string user = 1; string content = 2; }这里 ChatStream 方法接收一个消息流,并返回一个消息流,表示客户端和服务器都能持续发送和接收消息。
生成 gRPC 代码
使用 protoc 和插件生成 Go 代码:
立即学习“go语言免费学习笔记(深入)”;
protoc --go_out=. --go-grpc_out=. chat.proto会生成两个文件:chat.pb.go 和 chat_grpc.pb.go,包含服务接口和数据结构定义。
实现服务端逻辑
服务端需要实现生成的接口,在流中循环读取客户端消息,同时可以随时发送消息:
实际应用中可维护一个客户端连接池,实现群聊或广播功能。
实现客户端逻辑
客户端同样通过流发送和接收消息:
conn, _ := grpc.Dial("localhost:50051", grpc.WithInsecure()) client := pb.NewChatServiceClient(conn) stream, _ := client.ChatStream(context.Background()) // 启动 goroutine 接收服务器消息 go func() { for { msg, err := stream.Recv() if err != nil { log.Fatal(err) } log.Printf("%s: %s", msg.User, msg.Content) } }() // 发送消息 for i := 1; i注意:接收消息必须放在独立 goroutine 中,避免阻塞发送。
运行与测试
启动服务端监听,再运行多个客户端,即可看到双向通信效果。每个客户端发送的消息会被服务端处理并返回,客户端也能实时接收到服务器推送的内容。
基本上就这些。双向流的核心在于并发控制和连接管理,理解好 Send 和 Recv 的使用时机,就能构建出高效的实时通信系统。










