Grpc的四种通信方式

rpc

我们知道对于net/rpc来说 , 我们需要注册一个函数那么就需要满足

func (t *T) MethodName(argType T1 ,replyType *T2) error

很明显这个是一个请求对应一个响应 , 因此即是简单的rpc通信

grpc通信类别

grpc 分为以下四种通信方式 :

简单RPC : 一个请求 , 返回一个响应
服务端流式RPC : 一个请求 , 返回多个响应
客户端流式RPC : 多个请求 , 返回一个响应
双向流式RPC : 多个请求 , 返回多个响应

服务端流模式

我们只需要在proto中更改一下即可

业务场景 :
客户端向服务端发送一个股票代码 , 服务端就把该股票的实时数据源源不断的返回给客户端

提问1 : 为什么我不能用简单RPC模式返回一个切片的形式代替服务端流模式呢
其实可以代替 , 但是对于服务器如果要生成大量数据传输给客户端,那么简单的RPC模式就要等待所有数据都生成完毕才能收到响应。这会使得客户端等待时间过长,甚至导致网络超时

service YourService {
  rpc serverStreamHello(Person) returns (stream Result) {}
}

处理流数据

// 调用serverStreamHello方法并获取响应流
stream, err := client.ServerStreamHello(context.Background(), person)
if err != nil {
    log.Fatalf("Failed to call RPC method: %v", err)
}

// 遍历响应流并处理每个Result消息
for {
    result, err := stream.Recv()
    if err == io.EOF {
        break
    }
    if err != nil {
        log.Fatalf("Failed to receive message: %v", err)
    }

    // 处理Result消息
    log.Printf("Received result message: %v", result.GetMessage())
}

客户端流模式

业务场景 :

在上传大型多媒体文件的时候,客户端一般通过流模式逐个发送文件快,无需等待整个文件上传完成

proto

service ExampleService {
  rpc SendData (stream DataRequest) returns (DataResponse) {}
}

服务端处理

func (s *server) SendData(stream pb.ExampleService_SendDataServer) error {
    for {
        data, err := stream.Recv()
        if err == io.EOF {
            return stream.SendAndClose(&pb.DataResponse{Success: true})
        }
        if err != nil {
            return err
        }
        log.Println("Received data:", data.Data)
    }
}

双向流模式

业务场景 :
实时聊天程序
实时多人协作应用程序

proto

service ExampleService {
  rpc StreamData (stream DataRequest) returns (stream DataResponse) {}
}

服务端

func (s *server) StreamData(stream pb.ExampleService_StreamDataServer) error {
    for {
        data, err := stream.Recv()
        if err == io.EOF {
            return nil
        }
        if err != nil {
            return err
        }
        log.Println("Received data:", data.Data)
        if err := stream.Send(&pb.DataResponse{Success: true}); err != nil {
            return err
        }
    }
}

客户端

stream, err := client.StreamData(context.Background())
if err != nil {
    log.Fatalf("failed to call StreamData: %v", err)
}
defer stream.CloseSend()

for _, data := range dataSlice {
    req := &pb.DataRequest{Data: data}
    if err := stream.Send(req); err != nil {
        log.Fatalf("failed to send request: %v", err)
    }
}