如何在Golang中实现gRPC双向流-Golang-PHP中文网

如何在Golang中实现gRPC双向流

P粉602998670

发布： 2025-10-27 13:13:02

原创

711人浏览过

在Golang中实现gRPC双向流需先在.proto文件定义stream类型的rpc方法，如Chat(stream ChatMessage) returns (stream ChatMessage)；接着使用protoc生成Go代码，得到服务端ChatService_ChatServer和客户端ChatService_ChatClient接口；服务端通过Recv()接收并用Send()发送消息，循环处理实现全双工通信；客户端则通过goroutine并发调用Send()和Recv()，实现双向数据传输。

如何在golang中实现grpc双向流

在Golang中实现gRPC双向流，核心是定义一个服务方法，其请求和响应都带有stream关键字，然后在客户端和服务端同时读写数据流。整个过程基于Protocol Buffers定义接口，并使用gRPC运行时处理通信。

定义.proto文件中的双向流方法

要在gRPC中启用双向流，首先需要在.proto文件中定义一个使用stream的rpc方法。这个方法的请求和响应类型都会携带stream标识。

syntax = "proto3";

package example;

message ChatMessage {
    string user = 1;
    string message = 2;
}

service ChatService {
    rpc Chat(stream ChatMessage) returns (stream ChatMessage);
}

登录后复制

上述定义表示Chat方法允许客户端和服务端持续发送消息，实现全双工通信。

生成Go代码

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

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

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

登录后复制

会生成chat.pb.go文件，其中包含服务接口和客户端存根，重点关注生成的ChatService_ChatServer和ChatService_ChatClient类型。

实现服务端逻辑

服务端需循环读取客户端消息，同时可随时向客户端发送响应。

如知AI笔记

如知笔记——支持markdown的在线笔记，支持ai智能写作、AI搜索，支持DeepseekR1满血大模型

查看详情

示例实现：

type ChatServer struct{}

func (s *ChatServer) Chat(stream pb.ChatService_ChatServer) error {
    for {
        msg, err := stream.Recv()
        if err != nil {
            return err
        }

        // 处理收到的消息
        response := &pb.ChatMessage{
            user:    "server",
            message: "echo: " + msg.message,
        }

        // 发送响应
        if err := stream.Send(response); err != nil {
            return err
        }
    }
}

登录后复制

服务端通过Recv()接收流消息，Send()发送消息，直到连接关闭或发生错误。

实现客户端逻辑

客户端同样可以并发地发送和接收消息。

示例代码：

conn, _ := grpc.Dial("localhost:50051", grpc.WithInsecure())
client := pb.NewChatServiceClient(conn)
stream, _ := client.Chat(context.Background())

// 启动goroutine接收服务端消息
go func() {
    for {
        msg, err := stream.Recv()
        if err != nil {
            log.Println(err)
            return
        }
        log.Printf("[%s] %s", msg.user, msg.message)
    }
}()

// 发送消息
for i := 0; i < 5; i++ {
    msg := &pb.ChatMessage{
        user:    "client",
        message: fmt.Sprintf("hello %d", i),
    }
    stream.Send(msg)
    time.Sleep(time.Second)
}

stream.CloseSend()

登录后复制

客户端使用Send()发送，Recv()接收，通常用单独的goroutine处理接收逻辑以避免阻塞。

基本上就这些。只要.proto定义正确，生成代码后按流式接口编写收发逻辑，就能实现稳定的双向通信。注意控制上下文超时、错误处理和资源释放，避免连接泄漏。

以上就是如何在Golang中实现gRPC双向流的详细内容，更多请关注php中文网其它相关文章！