Golang如何使用gRPC实现双向流通信_Golang gRPC双向流应用示例-Golang-PHP中文网

Golang如何使用gRPC实现双向流通信_Golang gRPC双向流应用示例

P粉602998670

发布： 2025-11-22 18:56:02

原创

359人浏览过

Go语言通过gRPC实现双向流通信，适用于聊天系统等实时场景。1. 定义chat.proto文件声明消息结构和BidirectionalChat服务，使用stream关键字指定双向流；2. 用protoc生成Go代码，得到.pb.go和.grpc.pb.go文件；3. 服务端实现ChatServer结构体，重写BidirectionalChat方法，循环接收客户端消息并发送响应；4. 客户端创建流后启goroutine异步接收服务端消息，同时循环发送消息；5. 运行服务端和客户端，可观察到双方持续收发消息，完成双向通信。需注意流关闭与错误处理，防止资源泄漏。

golang如何使用grpc实现双向流通信_golang grpc双向流应用示例

在Go语言中使用gRPC实现双向流通信，可以实现在客户端和服务端之间持续发送和接收消息。这种模式适用于聊天系统、实时数据推送等场景。下面通过一个简单示例展示如何使用gRPC定义并实现双向流通信。

1. 定义.proto文件

首先需要定义gRPC服务的接口和消息格式。创建一个chat.proto文件：

syntax = "proto3";
<p>package chat;</p><p>// 消息结构
message Message {
string content = 1;
string user = 2;
}</p><p>// 双向流服务
service ChatService {
rpc BidirectionalChat(stream Message) returns (stream Message);
}

登录后复制

这个定义表示BidirectionalChat方法接收一个消息流，并返回一个消息流，即双向流。

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

2. 生成gRPC代码

使用Protocol Buffer编译器protoc配合Go插件生成代码：

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

登录后复制

执行后会生成chat.pb.go和chat_grpc.pb.go两个文件，包含服务接口和数据结构。

3. 实现服务端逻辑

编写gRPC服务端，处理双向流请求：

Fliki

高效帮用户创建视频，具有文本转语音功能

151

查看详情

package main
<p>import (
"context"
"log"
"net"</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">"google.golang.org/grpc"
pb "your-module-path/chat" // 替换为你的模块路径

登录后复制

)

type ChatServer struct { pb.UnimplementedChatServiceServer }

// BidirectionalChat 处理双向流 func (s *ChatServer) BidirectionalChat(stream pb.ChatService_BidirectionalChatServer) error { for { // 接收客户端消息 msg, err := stream.Recv() if err != nil { return err }

    log.Printf("收到消息: %s (用户: %s)", msg.Content, msg.User)

    // 构造响应消息
    response := &pb.Message{
        Content: "已收到: " + msg.Content,
        User:    "Server",
    }

    // 发送回客户端
    if err := stream.Send(response); err != nil {
        return err
    }
}

登录后复制

}

func main() { lis, err := net.Listen("tcp", ":50051") if err != nil { log.Fatalf("监听失败: %v", err) }

grpcServer := grpc.NewServer()
pb.RegisterChatServiceServer(grpcServer, &ChatServer{})

log.Println("gRPC服务器启动在 :50051")
if err := grpcServer.Serve(lis); err != nil {
    log.Fatalf("启动失败: %v", err)
}

登录后复制

}

4. 编写客户端代码

客户端也通过流发送和接收消息：

package main
<p>import (
"context"
"log"
"time"</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">"google.golang.org/grpc"
pb "your-module-path/chat"

登录后复制

)

func main() { conn, err := grpc.Dial("localhost:50051", grpc.WithInsecure()) if err != nil { log.Fatalf("连接失败: %v", err) } defer conn.Close()

client := pb.NewChatServiceClient(conn)

// 建立双向流
stream, err := client.BidirectionalChat(context.Background())
if err != nil {
    log.Fatalf("创建流失败: %v", err)
}

// 启动goroutine接收服务端消息
go func() {
    for {
        msg, err := stream.Recv()
        if err != nil {
            log.Printf("接收错误: %v", err)
            return
        }
        log.Printf("<< %s (%s)", msg.Content, msg.User)
    }
}()

// 发送消息
users := []string{"Alice", "Bob"}
for i := 0; i < 5; i++ {
    msg := &pb.Message{
        Content: "Hello " + string(rune('A'+i)),
        User:    users[i%2],
    }
    if err := stream.Send(msg); err != nil {
        log.Printf("发送失败: %v", err)
        break
    }
    time.Sleep(time.Second)
}

// 关闭发送流
stream.CloseSend()
time.Sleep(2 * time.Second) // 等待接收剩余消息

登录后复制

}