当前位置：首页 > 文章列表 > Golang > Go教程 > GolanggRPC双向流开发教程

GolanggRPC双向流开发教程

2025-10-23 16:39:36 0浏览收藏

**Golang gRPC 双向流实战教程：构建实时通信应用的利器** 本文深入探讨 Golang 中 gRPC 双向流的实现与应用。gRPC 双向流允许客户端和服务器之间建立持久连接，双方可以同时发送和接收消息，非常适合构建实时消息推送、日志收集和聊天系统等高并发应用。教程将详细讲解如何在 .proto 文件中定义双向流接口，服务端如何通过 Recv 和 Send 循环处理数据，以及客户端如何实现并发读写，避免阻塞和死锁。同时，还将重点介绍错误处理、连接状态管理，以及如何通过 context 控制超时和取消，确保 gRPC 双向流的稳定性和可靠性。掌握 gRPC 双向流技术，将为你的 Golang 应用带来更强大的实时通信能力。

答案：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 后要检查错误：

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