信息发布→ 登录 注册 退出

Golang如何使用gRPC实现双向流通信_Golang gRPC双向流通信实践详解

发布时间:2025-11-25

点击量:
首先定义支持双向流的.proto文件,生成Go代码后分别实现服务器和客户端逻辑,通过stream.Send和stream.Recv实现全双工通信。

在Go语言中使用gRPC实现双向流通信,可以高效地处理客户端与服务器之间持续交互的场景,比如实时聊天、日志推送、监控系统等。gRPC基于HTTP/2协议,天然支持双向流(Bidirectional Streaming),允许客户端和服务器同时发送多个消息,形*双工通信。

定义.proto文件支持双向流

要实现双向流通信,首先需要在.proto文件中定义一个服务方法,其请求和响应都使用stream关键字。

示例:chat.proto

syntax = "proto3";

package chat;

service ChatService {
rpc ChatStream(stream Message) returns (stream Message);
}

message Message {
string user = 1;
string content = 2;
}

这里ChatStream方法接收一个流式的Message,并返回一个流式的Message,表示客户端和服务器都可以连续发送消息。

生成Go代码

使用protoc配合gRPC插件生成Go代码:

protoc -I . --go_out=plugins=grpc:. chat.proto

会生成chat.pb.go文件,包含服务接口和数据结构定义。

实现服务器端逻辑

服务器需实现服务接口,并在处理函数中通过stream.Recv()接收消息,用stream.Send()发送消息。

示例服务器代码:

func (s *ChatServer) ChatStream(stream chat.ChatService_ChatStreamServer) error {
  for {
    msg, err := stream.Recv()
    if err == io.EOF {
      return nil
    }
    if err != nil {
      return err
    }

    // 处理消息,可广播给其他客户端
    response := &chat.Message{
      User: "Server",
      Content: "Echo: " + msg.Content,
    }

    stream.Send(response)
  }
}

注意:服务器在Recv遇到io.EOF时表示客户端关闭了发送流。

实现客户端逻辑

客户端通过调用ChatStream获得一个流对象,使用SendRecv与服务器通信。

示例客户端代码:

stream, _ := client.ChatStream(context.Background())

// 开启goroutine接收服务器消息
go func() {
  for {
    msg, err := stream.Recv()
    if err == io.EOF {
      break
    }
    if err != nil {
      log.Fatal(err)
    }
    fmt.Printf("收到: %s - %s\n", msg.User, msg.Content)
  }
}()

// 发送消息
for i := 0; i   msg := &chat.Message{User: "Client", Content: fmt.Sprintf("消息 %d", i)}
  stream.Send(msg)
  time.Sleep(time.Second)
}

stream.CloseSend()

客户端使用独立的goroutine接收消息,避免阻塞发送流程。调用CloseSend通知服务器不再发送数据。

基本上就这些。只要.proto定义正确,收发逻辑清晰,gRPC的双向流在Go中使用非常直接。关键是理解流的生命周期和EOF的处理。实际项目中可结合上下文取消、超时控制和连接复用提升稳定性。

标签:# go  # golang  # go语言  # stream  # 实时聊天  # echo  # EOF  # String  # if  # for  # Error  # printf  # break  # 数据结构  # 接口  
在线客服
服务热线

服务热线

4008888355

微信咨询
二维码
返回顶部
×二维码

截屏,微信识别二维码

打开微信

微信号已复制,请打开微信添加咨询详情!