基于Protobuf实现一个简单的RPC框架


使用Protobuf实现RPC框架大致可以分为以下几步:

1、编写proto文件定义服务,并使用Protoc工具生成RPC对象接口模型。

2、编写RPC服务处理主逻辑(继承框架代码中的接口类实现服务的主逻辑)

3、编写RPC服务框架

  • 作用:其作用主要包括注册的RPC Service(第1步定义、第2步实现),接收RPC客户端请求交由对应的RPC Service处理,处理完毕后对客户端进行回应

4、编写RPC客户端代码

  • 本质上就是实现RpcChannel和RpcController两个类(实际客户端调用逻辑proto工具已经帮我们在stub类中封装好了,Stub与RpcChannel和RpcController交互的逻辑在后面会有讲解)。
  • RpcChannel本质上是用于客户端与服务端之间交互的一条通道,其负责实现客户端向服务端请求时的一些数据和网络处理。一般情况下这个类中都会至少包括序列化和发包到服务端这两个操作。
  • RpcController这个类主要用于记录一次rpc调用的上下文,包括这次调用的方法以及执行结果等,在客户端和服务端都要用到这个类,最常见的就是服务端设置failed状态,客户端读取服务端执行的状态。

5、编写客户端使用样例。

​ 本文将会对上述每一个步骤进行讲解,并会在分模块讲解完之后将所有的涉及到的模块进行串联起来进行理解,最后为了对实现的RPC框架有一个直观的理解,还会给出RPC客户端请求RPC服务端时候的整个调用链路。

1、Proto定义RPC服务

1.1 编写proto文件

package goya.rpc.echo;
option cc_generic_services = true;

message EchoRequest {
   
  optional string message = 1;
}

message EchoResponse {
   
  optional string message = 1;
}

service EchoServer {
   
  rpc Echo(EchoRequest) returns(EchoResponse);
}

1.2 使用protoc工具生成RPC对象接口模型

$ protoc --cpp_out=./ echo_service.proto 
$ ls
echo_service.pb.cc  echo_service.pb.h  echo_service.proto

生成的echo_service.pb.h文件中,包含两个主要和RPC相关的类,分别是接口类EchoServer和该接口的一个实现EchoServer_Stub。

​ 其中EchoServer类是一个接口类,它继承::google::protobuf::Service抽象类,不允许实例化(默认构造函数声明为protected),这个接口中生成了一个与proto文件中的方法Echo对应的函数。

class EchoServer_Stub;

class EchoServer : public ::google::protobuf::Service {
   
	// 声明为protected,不允许显式实例化
 protected:
  // This class should be treated as an abstract interface.
  inline EchoServer() {
   };
  
 public:
  virtual ~EchoServer();
  typedef EchoServer_Stub Stub;
  // 获取当前service的属性descriptor,service的属性中又存在其中的method属性
  // 其中属性包括service->name(), service->method(), service->methmod(0)->name()的等
  static const ::google::protobuf::ServiceDescriptor* descriptor();
  // 与proto文件中对应的调用方法Echo
  virtual void Echo(::google::protobuf::RpcController* controller,
                       const ::goya::rpc::echo::EchoRequest* request,
                       ::goya::rpc::echo::EchoResponse* response,
                       ::google::protobuf::Closure* done);
  
  // implements Service ----------------------------------------------
  // 获取当前Service的descriptor
  const ::google::protobuf::ServiceDescriptor* GetDescriptor();
  
  // 实际调用方法,对于每一个远程调用方法,实际上都是通过调用该函数实现,后面后介绍
  void CallMethod(const ::google::protobuf::MethodDescriptor* method,
                  ::google::protobuf::RpcController* controller,
                  const ::google::protobuf::Message* request,
                  ::google::protobuf::Message* response,
                  ::google::protobuf::Closure* done);
  // 获取调用方法method的请求类型(如Echo方法的请求类型为EchoRequest)
  const ::google::protobuf::Message& GetRequestPrototype(
    const ::google::protobuf::MethodDescriptor* method) const;
  // 获取调用方法method的应答类型(如Echo方法的请求类型为EchoResponse)
  const ::google::protobuf::Message& GetResponsePrototype(
    const ::google::protobuf::MethodDescriptor* method) const;

 private:
  GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(EchoServer);
};

class EchoServer_Stub : public EchoServer {
   
 public:
  EchoServer_Stub(::google::protobuf::RpcChannel* channel);
  EchoServer_Stub(::google::protobuf::RpcChannel* channel,
                   ::google::protobuf::Service::ChannelOwnership ownership);
  ~EchoServer_Stub();

  inline ::google::protobuf::RpcChannel* channel() {
    return channel_; }

  // implements EchoServer ------------------------------------------
  void Echo(::google::protobuf::RpcController* controller,
                       const ::goya::rpc::echo::EchoRequest* request,
                       ::goya::rpc::echo::EchoResponse* response,
                       ::google::protobuf::Closure* done);
 private:
  // 客户端与服务端交互的通道
  ::google::protobuf::RpcChannel* channel_;
  bool owns_channel_;
  GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(EchoServer_Stub);
};

EchoServer的作用有两个:

  • 作为RPC服务端逻辑处理类(EchoServerImpl)的基类,服务逻辑处理类通过实现RPC定义的方法来处理相关逻辑,即实现Echo()函数。
  • 作为EchoServer_Stub类(上面已实现)的基类,EchoServer_Stub是客户端访问RPC服务的一个包装类

EchoServer_Stub作用如下:

​ EchoServer_Stub封装了RpcChannel,RpcChannel是客户端与服务通信的通道,由使用者自定义完成,将RPC调用本地化

2、编写RPC服务逻辑

从1.2 中可以知道protobuf为我们生成了EchoServer基类用于完成服务端处理的实际逻辑。所以我们通过编写EchoServer的实现类即可实现服务端方法的处理逻辑。

// 继承EchoServer实现类EchoServerImpl,并填充实现逻辑即可
class EchoServerImpl : public goya
  • 1
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值