0

0

如何用Golang实现双向流式RPC 深入gRPC流式通信模式

P粉602998670

P粉602998670

发布时间:2025-07-22 10:18:02

|

824人浏览过

|

来源于php中文网

原创

双向流式rpc在golang中通过客户端和服务端持续发送多个消息实现灵活通信。定义proto接口时,使用stream关键字声明参数和返回值,如rpc bidirectionalchat(stream messagerequest) returns (stream messageresponse)。服务端逻辑需处理recv()接收客户端消息并send()回应。客户端则创建流对象后用goroutine监听服务端响应,主线程发送消息。注意上下文控制、错误处理及测试调试以确保稳定交互。

如何用Golang实现双向流式RPC 深入gRPC流式通信模式

在使用 Golang 实现 gRPC 的时候,很多人会接触到“流式 RPC”这个概念。其中双向流式 RPC 是最灵活、也最具挑战的一种通信模式。它允许客户端和服务端各自持续发送多个消息,形成一个双向流动的通道。这种机制非常适合实时性要求高、交互频繁的场景,比如聊天应用、实时数据推送等。

如何用Golang实现双向流式RPC 深入gRPC流式通信模式

要实现这样的通信方式,关键在于定义好 proto 接口,并在服务端和客户端正确处理流的读写操作。

如何用Golang实现双向流式RPC 深入gRPC流式通信模式

定义 proto 文件:双向流式接口的基础

gRPC 的四种调用方式中,双向流式是通过 rpc 方法中参数和返回值都使用 stream 关键字来声明的。例如:

立即学习go语言免费学习笔记(深入)”;

syntax = "proto3";

package chat;

service ChatService {
  rpc BidirectionalChat(stream MessageRequest) returns (stream MessageResponse);
}

message MessageRequest {
  string content = 1;
}

message MessageResponse {
  string reply = 1;
}

上面的例子中,BidirectionalChat 就是一个典型的双向流方法。客户端可以持续发送 MessageRequest,服务端也能持续返回 MessageResponse。proto 文件定义清楚后,就可以用 protoc 工具生成对应的服务代码。

如何用Golang实现双向流式RPC 深入gRPC流式通信模式
注意:确保你安装了 protoc-gen-go-grpc 插件,否则无法生成正确的流式接口代码。

编写服务端逻辑:接收请求并响应流

在 Golang 中实现服务端时,gRPC 自动生成的代码会提供一个带有两个流参数的方法签名。你需要在这个方法内部处理输入流和输出流。

举个例子:

TextIn Tools
TextIn Tools

是一款免费在线OCR工具,包含文字识别、表格识别,PDF转文件,文件转PDF、其他格式转换,识别率高,体验好,免费。

下载
func (s *chatServer) BidirectionalChat(stream pb.ChatService_BidirectionalChatServer) error {
    for {
        req, err := stream.Recv()
        if err == io.EOF {
            return nil
        }
        if err != nil {
            return err
        }

        // 处理客户端发来的消息
        log.Printf("收到消息: %s", req.Content)

        // 回复客户端
        resp := &pb.MessageResponse{
            Reply: fmt.Sprintf("服务器收到了: %s", req.Content),
        }
        if err := stream.Send(resp); err != nil {
            return err
        }
    }
}

这段代码的核心是不断从客户端读取消息(Recv),然后根据内容做出回应(Send)。需要注意的是,一旦客户端关闭连接(如遇到 EOF),服务端也应该优雅退出该函数。

客户端实现:双向发送与接收消息

客户端方面,使用 gRPC 生成的 BidirectionalChatClient 类型创建一个流对象,之后就可以同时发送和接收消息了。

示例代码如下:

stream, err := client.BidirectionalChat(context.Background())
if err != nil {
    log.Fatalf("无法建立流: %v", err)
}

// 单独开一个 goroutine 来监听服务端消息
go func() {
    for {
        resp, err := stream.Recv()
        if err == io.EOF {
            break
        }
        if err != nil {
            log.Fatalf("接收失败: %v", err)
        }
        fmt.Println("收到回复:", resp.Reply)
    }
}()

// 主线程循环发送消息
for i := 0; i < 5; i++ {
    msg := fmt.Sprintf("第 %d 条消息", i+1)
    if err := stream.Send(&pb.MessageRequest{Content: msg}); err != nil {
        log.Fatalf("发送失败: %v", err)
    }
    time.Sleep(time.Second)
}

// 发送完后关闭流
stream.CloseSend()

这里用了 goroutine 来并发处理接收服务端的消息,而主线程则负责定时发送消息。这种方式适用于需要长时间保持连接并双向通信的场景。

此外,要注意上下文(context)控制和错误处理,避免因为网络问题或对方断开导致程序卡住。

常见注意事项和调试技巧

  • 上下文控制:为防止流挂起太久,建议设置合理的超时时间。
  • 错误处理:不管是 Send 还是 Recv,都要判断是否有错误,特别是 io.EOF 和 connection reset 等常见错误。
  • 测试工具:可以用 evans 或者自定义客户端/服务端进行联调。
  • 性能压测:对于高并发场景,建议做些压力测试,观察流式通信在大量消息下的表现。

总的来说,Golang 对 gRPC 流式通信的支持已经很完善,只要理解好 proto 定义和服务端/客户端的交互流程,实现起来并不难。

基本上就这些。

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

173

2024.02.23

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

224

2024.02.23

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

335

2024.02.23

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

206

2024.03.05

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

388

2024.05.21

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

193

2025.06.09

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

187

2025.06.10

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

191

2025.06.17

桌面文件位置介绍
桌面文件位置介绍

本专题整合了桌面文件相关教程,阅读专题下面的文章了解更多内容。

0

2025.12.30

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
golang socket 编程
golang socket 编程

共2课时 | 0.1万人学习

nginx浅谈
nginx浅谈

共15课时 | 0.8万人学习

golang和swoole核心底层分析
golang和swoole核心底层分析

共3课时 | 0.1万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号