前言
在c++rpc框架中,brpc简直越用越爽,平时工作中也常用到brpc,一直没来得及总结,抽空写点,也供自己查阅用。下附几个常用学习地址:
brpc官网开源地址:
https://github.com/luozesong/brpc/blob/master/docs/cn/redis_client.md
protobuf官方文档:
https://protobuf.dev/programming-guides/proto2/
brpc使用流程
1、定义proto
一般包含request、response、echoservice三种proto,echoservice在里面会定义一个rpc(stub)桩函数,proto编译器会生成一个与定义的stub接口同名的抽象接口,在用户只需要继承echoservice实现自己的service类之后,重写该stub函数即可。如果不定义自己的echoservice接口,一些场景也可使用brpc内部已经定义好的service接口,像nsheadservice,用户直接继承该类,实现自己的ProcessNsheadRequest()接口函数即可。
下面来看两种proto定义的例子:
- 自定义rpc service
cpp
# Tell protoc to generate base classes for C++ Service. modify to java_generic_services or py_generic_services for java or python.
option cc_generic_services = true;
message EchoRequest {
required string message = 1;
};
message EchoResponse {
required string message = 1;
};
service EchoService {
rpc Echo(EchoRequest) returns (EchoResponse);
};
上面我们在proto里面定义了EchoService服务,我们在下面定义自己的MyEchoService继承该EchoService类,重写proto自动编译生成的抽象接口Echo即可。
cpp
class MyEchoService : public EchoService {
public:
void Echo(::google::protobuf::RpcController* cntl_base,
const ::example::EchoRequest* request,
::example::EchoResponse* response,
::google::protobuf::Closure* done) {
// This RAII object calls done->Run() automatically at exit.
brpc::ClosureGuard done_guard(done);
brpc::Controller* cntl = static_cast<brpc::Controller*>(cntl_base);
// fill response
response->set_message(request->message());
}
};
继承brpc内部service接口(以nshead为例):
cpp
class Fw2NsheadService : public ::baidu::rpc::NsheadService {
public:
Fw2NsheadService(App* app);
virtual ~Fw2NsheadService();
void ProcessNsheadRequest(const baidu::rpc::Server& server,
baidu::rpc::Controller* cntl,
const baidu::rpc::NsheadMessage& request,
baidu::rpc::NsheadMessage* response,
baidu::rpc::NsheadClosure* done) {
bus::log::LogClosureGuard done_guard(cntl, done);
int ret = HTTP_STATUS_SERVER_ERROR;
if (cntl->Failed()) {
LOG(FATAL) << "Controller Failed Before Process, Reason:" << cntl->ErrorText();
BUS_SET_ERRNO(ret);
return;
}
ret = _app->Execute(request.body, response->body);
BUS_SET_ERRNO(ret);
}
private:
App* _app;
};
2、proto定义好之后,需要实现生成的service接口(上面已讲过)
需实现自己的service响应函数。另外我们通常需要在自己的sevice函数中初始化一个done_guard或自己手动调用done->run():
brpc::ClosureGuard done_guard(done);
done由框架创建,递给服务回调,包含了调用服务回调后的后续动作,包括检查response正确性,序列化,打包,发送等逻辑。done_guard就是为了保证退出服务自动调用,释放资源。如果我们需要实现异步服务,除了手动调用done->run()外,还可以使用done_guard.release()
3、定义服务对象brpc::Server server;
默认构造后的Server不包含任何服务,也不会对外提供服务,仅仅是一个对象。
4、给服务对象添加服务实例
通过如下方法插入你的Service实例。
c++
int AddService(google::protobuf::Service* service, ServiceOwnership ownership);
若ownership参数为SERVER_OWNS_SERVICE,Server在析构时会一并删除Service,意味着我们自己定义的服务实例也会被删除,如果还需要该service实例,应设为SERVER_DOESNT_OWN_SERVICE。
插入MyEchoService代码如下:
c++
brpc::Server server;
MyEchoService my_echo_service;
if (server.AddService(&my_echo_service, brpc::SERVER_DOESNT_OWN_SERVICE) != 0) {
LOG(FATAL) << "Fail to add my_echo_service";
return -1;
}
5、启动服务
c++
brpc::ServerOptions options; // 包含了默认值
options.xxx = yyy;
...
server.Start(..., &options);
start的接口有多种:
c++
int Start(const char* ip_and_port_str, const ServerOptions* opt);
int Start(EndPoint ip_and_port, const ServerOptions* opt);
int Start(int port, const ServerOptions* opt);
int Start(const char *ip_str, PortRange port_range, const ServerOptions *opt); // r32009后增加
options
为NULL时所有参数取默认值。一个服务只能监听一个端口,如果要监听多个端口需要起多个服务。
6、停止服务
c++
server.Stop(closewait_ms); // closewait_ms实际无效,出于历史原因未删
server.Join();
Stop()不会阻塞,Join()会。分成两个函数的原因在于当多个Server需要退出时,可以先全部Stop再一起Join,如果一个个Stop/Join,可能得花费Server个数倍的等待时间。
不管closewait_ms是什么值,server在退出时会等待所有正在被处理的请求完成,同时对新请求立刻回复ELOGOFF错误以防止新请求加入。这么做的原因在于只要server退出时仍有处理线程运行,就有访问到已释放内存的风险。如果你的server"退不掉",很有可能是由于某个检索线程没结束或忘记调用done了。
当client看到ELOGOFF时,会跳过对应的server,并在其他server上重试对应的请求。所以在一般情况下brpc总是"优雅退出"的,重启或上线时几乎不会或只会丢失很少量的流量。
RunUntilAskedToQuit()函数可以在大部分情况下简化server的运转和停止代码。在server.Start后,只需如下代码即会让server运行直到按到Ctrl-C。
c++
// Wait until Ctrl-C is pressed, then Stop() and Join() the server.
server.RunUntilAskedToQuit();
// server已经停止了,这里可以写释放资源的代码。
Join()完成后可以修改其中的Service,并重新Start。