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~
网友评论