Golang如何使用gRPC双向流进行通信_Golang gRPC双向流通信实践-Golang-PHP中文网

Golang如何使用gRPC双向流进行通信_Golang gRPC双向流通信实践

P粉602998670

发布： 2025-12-01 10:32:58

原创

329人浏览过

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

golang如何使用grpc双向流进行通信_golang grpc双向流通信实践

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，包含服务接口和数据结构定义。

实现服务端逻辑

服务端需要实现生成的接口，在流中循环读取客户端消息，同时可以随时发送消息：

Shrink.media

Shrink.media是当今市场上最快、最直观、最智能的图像文件缩减工具

123

查看详情

func (s *ChatServer) ChatStream(stream pb.ChatService_ChatStreamServer) error { for { msg, err := stream.Recv() if err != nil { return err } // 处理消息并广播给所有连接的客户端（简化为回写） response := &pb.Message{ User: "server", Content: "echo: " + msg.Content, } if err := stream.Send(response); err != nil { return err } } }

实际应用中可维护一个客户端连接池，实现群聊或广播功能。

实现客户端逻辑

客户端同样通过流发送和接收消息：

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 中，避免阻塞发送。