rust - grpc 实现

TLNR;

之前研究过go和java版本的grpc实现,在第一步就是针对 proto文件进行相关语言代码的生成处理,这一过程不能说很愉快.
今天体验了rust语言下的grpc一种实现后,真心体验到此语言带来的快捷与便利.
本文记录了rust下grpc基础功能实现,通过实现一个单参数接口服务,来体验一下rust的grpc处理.

先来效果图:

服务端

客户端

0x01 进入正题

进入代码讲解之前, 先看一下代码结构全貌:

src

0x02 定义proto

首先, 在项目根目录下,创建文件 proto/lua.proto, 内容如下:

syntax = "proto3";
package com.lua;

service LuaService {
    rpc dataToJson(LuaRequest) returns (LuaReply);
}

message LuaRequest {
    string bin = 1;
}

message LuaReply {
    string json = 1;
}

这是一个很基础的接口定义, 给出了入参,出参,还有接口方法名称信息.

0x03 关于proto文件的编译

开篇讲到rust中对proto的处理比起go/java要方便一些, 具体体现在哪里呢?
在根目录中添加一个 build.rs

fn main() {
    tonic_build::compile_protos("proto/lua.proto").unwrap();
}

此文件的编译不是由用户手动执行, 而是在编译其他目标时,自动执行编译.
它的实现,还需要一个Cargo.toml配置:

[build-dependencies]
tonic-build = {version = "0.2", features = ["prost"]}

0x04 服务端代码实现

use tonic::{transport::Server, Request, Response, Status};

use lua::lua_service_server::{LuaService, LuaServiceServer};
use lua::{LuaReply, LuaRequest};

pub mod lua {
    tonic::include_proto!("com.lua");
}

#[derive(Default)]
pub struct LuaRpc {}

#[tonic::async_trait]
impl LuaService for LuaRpc {
    async fn data_to_json(
        &self,
        request: Request<LuaRequest>,
    ) -> Result<Response<LuaReply>, Status> {
        println!("Got a request from {:?}", request.remote_addr());

        let reply = lua::LuaReply {
            json: format!("Hello {}!", request.into_inner().bin),
        };
        Ok(Response::new(reply))
    }
}

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let address = "[::1]:50052".parse().unwrap();
    let greeter = LuaRpc::default();

    println!("LuaServiceServer listening on {}", address);

    Server::builder()
        .add_service(LuaServiceServer::new(greeter))
        .serve(address)
        .await?;

    Ok(())
}

这里有些大小写转换的机制只能自己体会了, 另一方面, 美中不足的地方是: 这个代码中关于proto的地方全是盲打出来, 目前还没有找到代码提示的方式.
不过问题不大, 代码内容很简单:
相当于 proto 定义了一个 trait, 我们在server端写一个struct, 并让它实现这个trait就OK了.

最后, 在main方法中, 使用tonic::Server来启动服务.

0x05 客户端代码实现

use lua::lua_service_client::LuaServiceClient;
use lua::LuaRequest;

pub mod lua {
    tonic::include_proto!("com.lua");
}

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let mut client = LuaServiceClient::connect("http://[::1]:50052").await?;

    let request = tonic::Request::new(LuaRequest {
        bin: "hello".into(),
    });

    let response = client.data_to_json(request).await?;

    println!("RESPONSE={:?}", response);

    Ok(())
}

相比而言, 客户端代码简单很多, 因为只需要调用即可 .
这里不再赘述.

0x06 最后关于 cargo.toml

[package]
name = "rust-std"
version = "0.1.0"
authors = ["Gorey"]
edition = "2018"

# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

[[bin]]
name = "lua-server"
path = "src/luaGrpc/luaServer.rs"
edition = "2018"

[[bin]]
name = "lua-client"
path = "src/luaGrpc/luaClient.rs"
edition = "2018"

[dependencies]
tonic = "0.2"
prost = "0.6"
tokio = { version = "0.2", features = ["macros"] }

[build-dependencies]
tonic-build = {version = "0.2", features = ["prost"]}

这里主要定义了rpc要使用的依赖, 同时定义了两个 bin 编译目标.

一切准备就绪后, 就可以在 terminal里依次启动两个目标:

cargo run --bin lua-server
cargo run --bin lua-client

0x07 写点废话

昨天升级了 fedora32 ,终于在vmware下不花屏了. 神清气爽, 哈哈哈

之前基于go实现了一波grpc,但是实际运行时,cpu占用率还挺搞的, 这次用rust再造一波轮子,看看能否把CPU资源降低一些. god bless me~