grpc流式服务小demo1

gRPC 流式服务是一种 gRPC 支持的高级特性，允许客户端和服务器之间在单个连接上通过流式传输数据。这种数据传输可以是单向流，也可以是双向流。gRPC 支持四种类型的流式服务：

1. 服务器流式 RPC（Server Streaming RPC）： 服务器流式 RPC 允许客户端发送一个请求，而服务器则通过流回多个响应。客户端从返回的流中读取数据，直到流结束。这使得服务器可以在长时间运行的操作中返回多个结果。
2. 客户端流式 RPC（Client Streaming RPC）： 客户端流式 RPC 允许客户端通过流发送多个请求，而服务器则以单一响应进行回复。客户端通过流发送的请求可以是一个流，服务器返回的响应是一个响应。
3. 双向流式 RPC（Bidirectional Streaming RPC）： 双向流式 RPC 允许客户端和服务器之间通过流发送多个消息。客户端和服务器可以同时读取和写入数据。这种方式适用于需要交互的场景，其中客户端和服务器可以异步地发送和接收数据。
4. 简单 RPC： 这是最简单的形式，其中客户端发送一个请求并等待服务器的单一响应。与上述三种流式服务不同，简单 RPC 是一次性的，即客户端发送一个请求，然后等待服务器响应，请求和响应之间没有进一步的数据传输。

流式服务对于需要高效传输大量数据、实时通信、长时间运行的操作等场景非常有用。例如，视频流、日志流、实时通知等可以通过流式服务更有效地处理。

服务器流式 RPC

proto代码：

syntax = "proto3";
package stream_demo; // 指定默认包名
// 指定 golang 包名(存放路径,  ./表示存放在当前文件夹)
option go_package = "./pb";

//定义rpc服务
service Simple {
  // 定义函数
  rpc Fun (Request) returns (Response){};
}
//  请求函数内容
message Request {
  string name= 1;
}
// 响应AddRequest内容
message Response{
  string Text=1;
}
service StreamService{
  rpc Fun(Request)returns(stream Response){}
}

//运行  ：protoc --go_out=.  --go-grpc_out=. ./stream.proto

服务端代码：
package main

import (
“fmt”
“go_rpc_grpc/stream_demo/pb”
“google.golang.org/grpc”
“net”
)

type StreamService struct {
pb.UnimplementedStreamServiceServer
}

func (s *StreamService) Fun(request *pb.Request, stream pb.StreamService_FunServer) error {
fmt.Println(request)
for i := 0; i < 10; i++ {
stream.Send(&pb.Response{
Text: fmt.Sprintf(“第%d轮数据”, i),
})
}
return nil
}

func main() {
listener, err := net.Listen(“tcp”, “:8888”)
if err != nil {
}
//创建服务实例
srv := grpc.NewServer()
//注册接口的实现
pb.RegisterStreamServiceServer(srv, &StreamService{})
//启动grpc服务
if err := srv.Serve(listener); err != nil {
fmt.Println(“err:”, err)
}
}
客户端代码：
package main

import (
“context”
“fmt”
“go_rpc_grpc/stream_demo/pb”
“google.golang.org/grpc”
“google.golang.org/grpc/credentials/insecure”
)

func main() {
conn, err := grpc.Dial(“:8888”, grpc.WithTransportCredentials(insecure.NewCredentials()))
if err != nil {
}
//关闭连接
defer conn.Close()
//创建客户端
client := pb.NewStreamServiceClient(conn)
//调用add
stream, err := client.Fun(context.Background(), &pb.Request{
Name: “哈哈哈哈哈”,
})
if err != nil {
}
for i := 0; i < 10; i++ {
response, err := stream.Recv()
fmt.Println(response, err)
}
}

上述代码表示客户端想服务器发送一次“哈哈哈哈”请求，服务端相应10次
注：如果客户端不知道服务器传输到多久结束，那么可以使用一个死循环

for {
		response, err := stream.Recv()
		if err == io.EOF {
			break
		}
		fmt.Println(response)
	}

参考：
https://www.fengfengzhidao.com/article/uNliH4sBEG4v2tWkkWsW