Golang gRPC双向流数据处理实践

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

在使用 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
    }
  }
}

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

客户端实现并发读写

客户端同样使用 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 通知服务端客户端已结束发送。

错误处理与连接状态管理

双向流一旦任一端出错，整个 stream 就会中断。建议在 Recv 或 Send 返回非 nil 错误时退出循环，并做必要清理。可通过 context 控制超时或取消：

• Recv 返回 io.EOF 表示对方关闭了写入，属于正常结束
• Send 失败通常意味着网络问题或对端已关闭
• 设置合理的 Keepalive 参数可检测长时间空闲连接是否有效

基本上就这些。只要保证读写分离、合理处理 EOF 和异常，gRPC 双向流在 Go 中使用起来非常直接。关键是设计好消息边界和通信协议，避免粘包或顺序错乱问题。