0

0

如何在Golang中实现gRPC双向流

P粉602998670

P粉602998670

发布时间:2025-10-27 13:13:02

|

741人浏览过

|

来源于php中文网

原创

在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_ChatServerChatService_ChatClient类型。

实现服务端逻辑

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

Bika.ai
Bika.ai

打造您的AI智能体员工团队

下载

示例实现:

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定义正确,生成代码后按流式接口编写收发逻辑,就能实现稳定的双向通信。注意控制上下文超时、错误处理和资源释放,避免连接泄漏。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

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

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

182

2024.02.23

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

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

229

2024.02.23

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

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

343

2024.02.23

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

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

209

2024.03.05

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

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

394

2024.05.21

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

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

220

2025.06.09

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

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

193

2025.06.10

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

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

398

2025.06.17

俄罗斯Yandex引擎入口
俄罗斯Yandex引擎入口

2026年俄罗斯Yandex搜索引擎最新入口汇总,涵盖免登录、多语言支持、无广告视频播放及本地化服务等核心功能。阅读专题下面的文章了解更多详细内容。

158

2026.01.28

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Go 教程
Go 教程

共32课时 | 4.3万人学习

Go语言实战之 GraphQL
Go语言实战之 GraphQL

共10课时 | 0.8万人学习

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

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