在Go中实现gRPC服务端流，需在.proto文件定义返回stream的接口，服务端使用Send发送多条数据，客户端通过Recv循环接收，适用于日志推送等场景。
在Go语言中实现gRPC服务端流处理，核心是让服务器在接收到客户端请求后，持续向客户端发送多个数据项。这种模式适用于日志推送、实时通知或数据订阅等场景。
定义.proto文件中的服务接口
服务端流式gRPC需要在.proto文件中声明返回类型为流的接口。客户端发送一个请求，服务端通过流返回多个响应。
示例如下：
syntax = “proto3”;
message Request {
string query = 1;
}
message Response {
string data = 1;
}
service StreamService {
rpc ServerStream(Request) returns (stream Response);
}
这里stream Response表示服务端会连续发送多个Response消息。
立即学习“go语言免费学习笔记（深入）”；
生成Go代码并实现服务逻辑
使用protoc配合gRPC插件生成Go代码后，需实现对应的服务结构体和方法。
服务端实现如下：
type server struct{}
func (s *server) ServerStream(req *Request, stream StreamService_ServerStreamServer) error {
// 发送多个响应
for i := 0; i response := &Response{
Data: fmt.Sprintf(“Message %d for %s”, i, req.Query),
}
// 使用Send方法发送
if err := stream.Send(response); err != nil {
return err
}
time.Sleep(time.Second) // 模拟间隔
}
return nil
}
注意参数StreamService_ServerStreamServer是gRPC生成的流接口类型，提供Send方法用于发送数据。
启动gRPC服务器
标准的gRPC服务器启动流程即可：
func main() {
lis, err := net.Listen(“tcp”, “:50051”)
if err != nil {
log.Fatalf(“failed to listen: %v”, err)
}
s := grpc.NewServer()
RegisterStreamServiceServer(s, &server{})
if err := s.Serve(lis); err != nil {
log.Fatalf(“failed to serve: %v”, err)
}
}
客户端接收流数据
客户端调用时获取流对象，并循环调用Recv接收消息：
conn, _ := grpc.Dial(“:50051”, grpc.WithInsecure())
client := NewStreamServiceClient(conn)
stream, _ := client.ServerStream(context.Background(), &Request{Query: “hello”})
for {
resp, err := stream.Recv()
if err == io.EOF {
break // 流结束
}
if err != nil {
log.Fatal(err)
}
fmt.Println(resp.Data)
}
Recv方法阻塞等待服务端发送数据，直到流关闭或发生错误。
基本上就这些。关键点在于.proto中定义stream返回类型，服务端使用Send发送多条数据，客户端用Recv循环接收。整个过程由gRPC框架管理连接和序列化，开发者只需关注业务逻辑。不复杂但容易忽略细节，比如上下文控制和错误处理。
以上就是Golang如何实现gRPC服务端流处理的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1419787.html
微信扫一扫 
支付宝扫一扫 
