gRPC 四模式之 客户端流RPC模式

客户端流RPC模式

在客户端流 RPC 模式中，客户端会发送多个请求给服务器端，而不再是单个请求。服务器端则会发送一个响应给客户端。但是，服务器端不一定要等到从客户端接收到所有消息后才发送响应。基于这样的逻辑，我们可以在接收到流中的一条消息或几条消息之后就发送响应，也可以在读取完流中的所有消息之后再发送响应。

C++的实现

在C++中，gRPC的客户端流模式的实现大致如下：

首先，你需要在.proto文件中定义服务：

syntax = "proto3";

service YourService {
  rpc YourRpc(stream YourRequest) returns (YourResponse);
}

message YourRequest {
  // Your request fields
}

message YourResponse {
  // Your response fields
}

然后，你可以使用gRPC的protobuf插件生成C++代码。生成的代码中会包含一个YourService::Stub类，你可以使用这个类来调用RPC。

在客户端，你可以这样使用这个stub：

#include <grpcpp/grpcpp.h>
#include "your_service.grpc.pb.h"

void Run() {
  auto channel = grpc::CreateChannel("localhost:50051", grpc::InsecureChannelCredentials());
  auto stub = YourService::NewStub(channel);

  grpc::ClientContext context;
  YourResponse response;

  // 这个writer用于向服务器发送请求
  auto writer = stub->YourRpc(&context, &response);

  for (int i = 0; i < 10; ++i) {
    YourRequest request;
    // 设置你的请求字段

    // 使用writer向服务器发送请求
    if (!writer->Write(request)) {
      // 如果写入失败，那么可能是连接已经断开
      break;
    }
  }

  // 告诉服务器我们已经完成了写入
  writer->WritesDone();

  // 等待服务器的响应
  grpc::Status status = writer->Finish();

  if (status.ok()) {
    // 处理响应
  } else {
    // 处理错误
  }
}

代码分析
这是一个使用gRPC库编写的简单客户端程序，它连接到gRPC服务器并发送大量请求。以下是代码的主要部分及其功能：

1. 引入必要的头文件：

• <grpcpp/grpcpp.h>：gRPC库的头文件。
• "your_service.grpc.pb.h"：你的.proto文件中定义的服务和消息类型头文件。

1. Run()函数：

• 创建一个gRPC通道，指定服务器的地址和端口，以及是否使用安全连接。
• 创建一个服务器的远程代理（stub），用于调用服务器的RPC方法。

1. 创建一个客户端上下文（context）和一个响应对象（response）。
2. 创建一个数据写入器（writer），用于将请求发送到服务器。
3. 循环10次，每次创建一个请求对象（request），设置请求字段，并使用数据写入器向服务器发送请求。
4. 调用WritesDone()方法，向服务器发送一个信号，表示已经完成了写入操作。
5. 调用Finish()方法，等待服务器的响应。如果响应成功，则处理响应；否则，处理错误。

这个程序的作用是连接到gRPC服务器，发送大量请求，并处理服务器的响应。你可以根据需要修改代码以适应你的具体需求。

---

分享一个有趣的 学习链接：https://xxetb.xet.tech/s/HY8za