Golang实现gRPC服务端流详解

type server struct{}
<p>func (s <em>server) GetData(req </em>example.Request, stream example.DataService_GetDataServer) error {
for i := 0; i < 5; i++ {
res := &example.Response{
Message: fmt.Sprintf("Message %d for %s", i+1, req.Query),
}
// 向客户端发送一条消息
if err := stream.Send(res); err != nil {
return err
}
time.Sleep(500 * time.Millisecond) // 模拟延迟
}
return nil
}
</p>

func main() {
    lis, err := net.Listen("tcp", ":50051")
    if err != nil {
        log.Fatalf("failed to listen: %v", err)
    }
<pre class="brush:php;toolbar:false"><code>s := grpc.NewServer()
example.RegisterDataServiceServer(s, &amp;server{})

log.Println("gRPC server running on :50051")
if err := s.Serve(lis); err != nil {
    log.Fatalf("failed to serve: %v", err)
}</code>

}

conn, err := grpc.Dial("localhost:50051", grpc.WithInsecure())
if err != nil {
    log.Fatalf("connect failed: %v", err)
}
defer conn.Close()
<p>client := example.NewDataServiceClient(conn)
req := &example.Request{Query: "test"}</p><p>stream, err := client.GetData(context.Background(), req)
if err != nil {
log.Fatalf("request failed: %v", err)
}</p><p>for {
res, err := stream.Recv()
if err == io.EOF {
break // 流结束
}
if err != nil {
log.Fatalf("receive error: %v", err)
}
fmt.Println("Received:", res.Message)
}
</p>