brpc-rs: Rust语言的Apache BRPC库

Apache BRPC是一个用于构建可靠和高性能服务的工业级RPC框架。brpc-rs允许使用Rust编程语言实现BRPC客户端和服务器。

状态

该项目目前正在积极开发中，是一个原型。许多API缺失；提供的API在1.0版本之前可能不稳定。

仓库结构

• 项目根目录
  • brpc-build
    • brpc-build 包，构建时代码生成器
  • brpc-protoc-plugin
    • brpc-protoc-plugin 包，Google Protobuf编译器的插件
  • brpc-sys
    • brpc-sys 包，Apache BRPC的FFI绑定
  • 示例
    • 示例
  • src
    • brpc-rs 包，brpc-rs的Rust API

此图说明了这些包如何协同工作

快速入门

先决条件

首先，根据getting_started.md中的说明构建和安装Apache BRPC。

或者，您可以使用Apache BRPC 0.9.6的预构建deb软件包为Ubuntu 16.04/18.04。这些不是官方软件包。

确保这些依赖项已经安装

$ sudo apt-get install libprotobuf-dev libprotoc-dev protobuf-compiler
$ sudo apt-get install libssl-dev libgflags-dev libleveldb-dev

从crates.io安装brpc-protoc-plugin。

$ cargo install brpc-protoc-plugin --version 0.1.0-alpha4

现在我们已准备好开始一个brpc-rs项目。

Cargo.toml

让我们创建一个名为echo_service的小包，它包括一个echo_client和一个echo_server。

$ cargo new echo_service && cd echo_service

在Cargo.toml中，将brpc-rs、prost和bytes添加到[dependencies]部分；将brpc-build添加到[build-dependencies]部分。例如，

[build-dependencies]
brpc-build = "0.1.0"

[dependencies]
brpc-rs = "0.1.0"
prost = "0.5.0"
bytes = "0.4.12"

在Cargo.toml中定义两个二进制文件：echo_client和echo_server

[[bin]]
name = "echo_client"
path = "src/client.rs"

[[bin]]
name = "echo_server"
path = "src/server.rs"

build.rs

将protobuf文件echo.proto放入src目录。此文件定义了EchoRequest、EchoResponse和EchoService。

syntax="proto2";
package example;
message EchoRequest {
      required string message = 1;
};
message EchoResponse {
      required string message = 1;
};
service EchoService {
      rpc echo(EchoRequest) returns (EchoResponse);
};

在Cargo.toml文件的[package]部分添加一行build = "build.rs"。然后，创建一个名为build.rs的文件，从src/echo.proto生成绑定。

fn main() {
    brpc_build::compile_protos(&["src/echo.proto"],
                               &["src"]).unwrap();
}

注意echo.proto中的package名称是example。因此，build.rs将生成两个文件，分别命名为example.rs和example.brpc.rs。

src/server.rs

接下来让我们实现echo服务器。创建如下所示的src/server.rs：

use brpc_rs::{Server, ServerOptions, ServiceOwnership};

pub mod echo {
    include!(concat!(env!("OUT_DIR"), "/example.rs"));
    include!(concat!(env!("OUT_DIR"), "/example.brpc.rs"));
}

fn main() {
    let mut service = echo::EchoService::new();
    service.set_echo_handler(&mut move |request, mut response| {
        response.message = request.message.clone();
        Ok(())
    });

    let mut server = Server::new();
    let mut options = ServerOptions::new();
    options.set_idle_timeout_ms(1000);
    server
        .add_service(&service, ServiceOwnership::ServerDoesntOwnService)
        .expect("Failed to add service");
    server.start(50000, &options).expect("Failed to start service");
    server.run(); // Run until CTRL-C
}

因为EchoService在echo.proto中定义了一个名为echo()的函数，所以brpc-protoc-plugin为EchoService生成了set_echo_handler()的Rust定义。set_echo_handler()接受一个闭包，该闭包处理来自客户端的EchoRequest并返回相同消息的EchoResponse。接下来的几行创建了一个监听在0.0.0.0:50000的服务器。

src/client.rs

use brpc_rs::{Channel, ChannelOptions};

pub mod echo {
    include!(concat!(env!("OUT_DIR"), "/example.rs"));
    include!(concat!(env!("OUT_DIR"), "/example.brpc.rs"));
}

fn main() {
    let mut options = ChannelOptions::new();
    options.set_timeout_ms(100);
    let addr = "127.0.0.1:50000".parse().expect("Invalid socket address");
    let ch = Channel::with_options(&addr, &options);
    let client = echo::EchoServiceStub::with_channel(&ch);
    let request = echo::EchoRequest {
        message: "hello".to_owned(),
    };
    match client.echo(&request) {
        Ok(r) => println!("Response: {:?}", r),
        Err(e) => eprintln!("Error: {:?}", e),
    }
}

客户端首先创建一个Channel并使用该通道初始化一个service_stub。然后，客户端调用service_stub.echo()发送请求。

运行客户端和服务器

$ cargo run --bin echo_server &
$ cargo run --bin echo_client
Response: EchoResponse { message: "hello" }

维护者

• Yiming Jing <jingyiming@baidu.com> @kevinis

许可

brpc-rs根据Apache License，版本2.0提供。请参阅LICENSE文件以获取副本。