首页 > Golang > Go教程

Golang实现gRPC服务端流教程

时间：2025-11-01 12:09:35 220浏览收藏

Golang小白一枚，正在不断学习积累知识，现将学习到的知识记录一下，也是将我的所得分享给大家！而今天这篇文章《Golang实现gRPC服务端流详解》带大家来了解一下##content_title##，希望对大家的知识积累有所帮助，从而弥补自己的不足，助力实战开发！

在Go中实现gRPC服务端流，需在.proto文件定义返回stream的接口，服务端使用Send发送多条数据，客户端通过Recv循环接收，适用于日志推送等场景。

Golang如何实现gRPC服务端流处理

在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代码并实现服务逻辑

使用protoc配合gRPC插件生成Go代码后，需实现对应的服务结构体和方法。

服务端实现如下：

type server struct{}

func (s *server) ServerStream(req *Request, stream StreamService_ServerStreamServer) error {
  // 发送多个响应
  for i := 0; i < 5; 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服务端流教程》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！