如何在Golang中使用gRPC进行双向流通信

在Golang中实现gRPC双向流通信需定义支持stream的Protobuf接口,服务端和客户端通过Recv和Send方法在循环中持续收发消息,实现全双工通信。

如何在golang中使用grpc进行双向流通信

在Golang中使用gRPC实现双向流通信,关键在于定义支持双向流的Protobuf接口,并在服务端和客户端分别实现流的读写逻辑。整个过程包括定义proto文件、生成代码、编写服务端处理函数以及客户端持续发送和接收消息。

定义双向流的Protobuf接口

要启用双向流,需在.proto文件中将RPC方法的请求和响应类型都声明为stream。

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

这个定义表示Chat方法允许客户端和服务端同时发送多个消息,形成全双工通信。

生成gRPC代码

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

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

确保安装了protoc编译器和Go插件 运行命令:protoc –go_out=. –go-grpc_out=. chat.proto

生成的代码包含客户端和服务端所需的接口和结构体,如ChatServiceClient和ChatServiceServer。

Otter.ai Otter.ai

一个自动的会议记录和笔记工具,会议内容生成和实时转录

Otter.ai 91 查看详情 Otter.ai

实现服务端逻辑

服务端需要通过流接收客户端消息,同时可随时向客户端推送数据。

func (s *server) Chat(stream pb.ChatService_ChatServer) error {    for {        msg, err := stream.Recv()        if err != nil {            return err        }        // 处理收到的消息        log.Printf("Received: %s", msg.Content)        // 回复消息        response := &pb.Message{Content: "Echo: " + msg.Content}        if err := stream.Send(response); err != nil {            return err        }    }}

Recv()用于接收客户端消息,Send()用于向客户端发送消息。循环处理即可维持双向通信。

编写客户端代码

客户端同样通过流发送和接收消息,与服务端对等交互。

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

注意使用独立的goroutine处理接收逻辑,避免阻塞发送。调用CloseSend()通知服务端发送结束。

基本上就这些。只要proto定义正确,服务端和客户端都能通过同一个流收发消息,实现真正的双向实时通信。关键是理解stream接口的使用方式,以及避免在单个goroutine中同时读写导致死锁。

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

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

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2025年12月2日 13:37:57
下一篇 2025年12月2日 13:38:18

相关推荐

发表回复

登录后才能评论
关注微信