Golang gRPC双向流数据处理实践-Golang-PHP中文网

Golang gRPC双向流数据处理实践

P粉602998670

发布： 2025-10-04 15:14:02

原创

763人浏览过

答案：gRPC双向流需在proto中定义双stream方法，服务端和客户端通过Recv和Send循环收发消息，分别处理读写，注意关闭发送端及EOF判断，避免并发调用Send，合理管理错误与连接状态。

golang grpc双向流数据处理实践

在使用 Golang 实现 gRPC 双向流时，核心在于理解 stream 的读写并发控制 以及如何高效处理连续的数据交互。双向流适用于需要客户端和服务器持续通信的场景，比如实时消息推送、日志收集或聊天系统。

定义双向流接口

在 .proto 文件中声明 streaming 方法，双方都可以发送和接收消息：

syntax = "proto3";

message StreamRequest {
  string data = 1;
}

message StreamResponse {
  string result = 1;
}

service DataService {
  rpc BidirectionalStream(stream StreamRequest) returns (stream StreamResponse);
}

这个定义表示客户端可以持续发送请求，服务端也可以持续返回响应，两者独立进行。

服务端实现数据处理逻辑

服务端通过 stream.Recv() 接收客户端消息，同时用 stream.Send() 发送响应。注意每次调用 Recv 后要检查错误：

立即学习“go语言免费学习笔记（深入）”；

func (s *DataServiceServer) BidirectionalStream(stream pb.DataService_BidirectionalStreamServer) error {
  for {
    req, err := stream.Recv()
    if err == io.EOF {
      return nil // 客户端关闭写入
    }
    if err != nil {
      return err
    }

    // 处理数据并返回响应
    response := &pb.StreamResponse{
      result: "processed: " + req.Data,
    }
    if err := stream.Send(response); err != nil {
      return err
    }
  }
}

这里的关键是循环读取与非阻塞发送可同时进行，不需要等待客户端完成所有发送。

怪兽AI数字人

数字人短视频创作，数字人直播，实时驱动数字人

查看详情

客户端实现并发读写

客户端同样使用 stream.Send() 和 stream.Recv()。由于 Send 是异步的，需注意不要在多个 goroutine 中并发调用，除非加锁或串行化：

stream, _ := client.BidirectionalStream(context.Background())
go func() {
  for i := 0; i < 5; i++ {
    stream.Send(&pb.StreamRequest{Data: fmt.Sprintf("msg %d", i)})
    time.Sleep(time.Second)
  }
  stream.CloseSend() // 显式关闭发送端
}()

// 主协程接收响应
for {
  resp, err := stream.Recv()
  if err == io.EOF { break }
  if err != nil { log.Fatal(err) }
  fmt.Println(resp.Result)
}

通常将发送和接收放在不同协程中处理，避免因阻塞导致死锁。CloseSend 通知服务端客户端已结束发送。