Golang如何实现gRPC双向流通信

定义.proto文件中的双向流方法:使用stream关键字声明输入和输出流,如rpc Chat(stream Message) returns (stream Message);2. 生成Go代码:通过protoc命令配合插件生成chat.pb.go和chat_grpc.pb.go文件;3. 实现服务端逻辑:在循环中调用Recv()接收客户端消息,并用Send()回传消息;4. 编写客户端代码:通过goroutine分离接收与发送逻辑,实现并发通信。该模式适用于实时聊天、监控等需持续交互的场景。

golang如何实现grpc双向流通信

在Go语言中实现gRPC双向流通信,核心在于定义一个返回和接收都是流式的数据接口。这种方式适用于客户端和服务端需要持续交换数据的场景,比如实时聊天、日志推送或监控系统。

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

要启用双向流,必须在.proto文件中声明一个stream输入和stream输出的方法。例如:

syntax = "proto3";package chat;service ChatService {  rpc Chat(stream Message) returns (stream Message);}message Message {  string content = 1;  string sender = 2;}

这里的Chat方法接受一个消息流,并返回一个消息流,表示客户端和服务端都可以持续发送消息。

2. 生成Go代码

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

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

确保安装了protoc-gen-goprotoc-gen-go-grpc 执行命令生成代码:

protoc --go_out=. --go-grpc_out=. chat.proto

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

3. 实现服务端逻辑

服务端通过读取客户端的流并同时向其写入数据来实现双向通信:

func (s *ChatServer) Chat(stream chat.ChatService_ChatServer) error {    for {        msg, err := stream.Recv()        if err != nil {            return err        }        // 处理收到的消息        log.Printf("Received: %s from %s", msg.Content, msg.Sender)        // 回复消息(也可以广播给其他客户端)        response := &chat.Message{            Content: "Echo: " + msg.Content,            Sender:  "server",        }        if err := stream.Send(response); err != nil {            return err        }    }}

Recv()用于接收客户端消息,Send()用于向客户端发送消息,二者可在同一个循环中交替调用。

4. 编写客户端代码

客户端同样使用流进行收发:

client := chat.NewChatServiceClient(conn)stream, _ := client.Chat(context.Background())// 开启goroutine接收服务端消息go func() {    for {        msg, err := stream.Recv()        if err != nil {            log.Fatal(err)        }        log.Printf("Server: %s", msg.Content)    }}()// 发送消息for i := 0; i < 5; i++ {    stream.Send(&chat.Message{        Content: fmt.Sprintf("Hello %d", i),        Sender:  "client",    })    time.Sleep(time.Second)}

注意:客户端也需要用独立的goroutine处理接收,避免阻塞发送逻辑。

基本上就这些。只要.proto定义正确,服务端和客户端都使用stream.Send()和stream.Recv(),就能实现完整的双向通信。这种模式灵活高效,适合高频率小数据量的实时交互场景。

以上就是Golang如何实现gRPC双向流通信的详细内容,更多请关注创想鸟其它相关文章!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1422881.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2025年12月16日 17:16:11
下一篇 2025年12月16日 17:16:16

相关推荐

发表回复

登录后才能评论
关注微信