本文将通过poem框架所提供的grpc功能来实现一个简单token provider服务
goem介绍
poem 是一个功能齐全且易于使用的 Web 框架，采用 Rust 编程语言。
（git源码地址：https://github.com/poem-web/poem）
gRPC介绍
gRPC 是Google发起的一个开源远程过程调用系统。该系统基于HTTP/2协议传输，使用Protocol Buffers 作为接口描述语言。 其他功能： 认证 双向流 流控制 超时 最常见的应用场景是： 微服务框架下，多种语言服务之间的高效交互。 将手机服务、浏览器连接至后台 产生高效的客户端库。
实践项目：token服务
token服务经常被设计为微服务平台中通用的一个服务，常常是用来进行相关校验，如接口的访问权限校验、用户身份的有效性校验等。
本项目将只提供两个简单的grpc接口：
- token生成
- token校验
创建项目
cargo init token_provider
添加依赖
Cargo.toml文件内容如下：
[package]
name = "token_provider"
version = "0.1.0"
edition = "2021"
[dependencies]
prost = "0.11.0"
serde = {version = "1.0.139", features = ["derive"]}
serde_json = "1.0.82"
tokio = {version = "1.19.2", features = ["macros", "rt-multi-thread"]}
poem = "1.3.47"
poem-grpc = "0.2.11"
[build-dependencies]
poem-grpc-build = "0.2.4"
添加proto文件
在项目根目录：
mkdir proto
echo '' > token_service.proto
token_service.proto文件内容如下：
syntax = "proto3";
package token_service;
service TokenService {
rpc GetToken(GetTokenRequest) returns (GetTokenReply);
rpc CheckToken(CheckTokenRequest) returns (CheckTokenReply);
}
//
message GetTokenRequest{
string source_code=1;//用来成token的源内容，可以是用户名、角色等的组合
}
message GetTokenReply{
string token=1;
}
//
message CheckTokenRequest{
string token = 1;
}
message CheckTokenReply{
string source_code=1;
}
添加build.rs文件
在项目根目录：
echo '' > build.rs
build.rs文件内容如下：
use std::io::Result;
use poem_grpc_build::compile_protos;
fn main() -> Result<()> {
compile_protos(
&[
"./proto/token_service.proto",
],
&["./proto"],
)
}
编译proto
cargo build
注：要保证本机已经安装了protobuf，否则编译的时候会提示找不到protoc二进制文件，我的是Mac系统，执行 brew install protobuf
如果编译能够顺利完成，在目录 target/debug/build下可以发现token_provider-xxxx的两个文件夹，其中一个文件夹内容out文件夹及通过proto自动生成的对应源码toker_service.rs文件，对应的接口框架已经生成
接下来将具体实现接口内容
实现接口
打开main.rs文件并添加如下内容：
use poem::{listener::TcpListener, Server};
use poem_grpc::{Code, Request, Response, RouteGrpc, Status};
//必须导入proto，否则无法获取对应的已经生成的接口
mod token_service {
poem_grpc::include_proto!("token_service");
}
use token_service::{
CheckTokenReply, CheckTokenRequest, GetTokenReply, GetTokenRequest, TokenService,
TokenServiceServer,
};
//服务的地址信息
const ADDR: &str = "127.0.0.1:5100";
//对生成的接口进行具体实现
struct RTokenService;
#[poem::async_trait]
impl TokenService for RTokenService {
///获取token
async fn get_token(
&self,
request: Request<GetTokenRequest>,
) -> Result<Response<GetTokenReply>, Status> {
let source_code = request.into_inner().source_code;
if !source_code.is_empty() {
let token = String::from(""); //根据自己的具体逻辑生成token
return Ok(Response::new(GetTokenReply { token }));
} else {
return Err(Status::new(Code::InvalidArgument).with_message("empty source code"));
}
}
///校验token，并返回source_code
async fn check_token(
&self,
request: Request<CheckTokenRequest>,
) -> Result<Response<CheckTokenReply>, Status> {
//
let token = request.into_inner().token;
let source_code = String::from("..."); //从token中提取出source_code，并根据自己的逻辑进行验证
Ok(Response::new(CheckTokenReply { source_code }))
}
}
#[tokio::main]
async fn main() -> Result<(), std::io::Error> {
let route = RouteGrpc::new().add_service(TokenServiceServer::new(RTokenService));
Server::new(TcpListener::bind(ADDR)).run(route).await
}
运行服务
cargo run
测试服务接口
echo '' > token_test_client.rs
文件内容如下：
use poem_grpc::{ClientConfig, Request};
mod constant;
poem_grpc::include_proto!("token_service");
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync + 'static>> {
let client = TokenServiceClient::new(
ClientConfig::builder()
.uri(["http://", "127.0.0.1:5100"].concat())
.build()
.unwrap(),
);
//生成token
let request = Request::new(GetTokenRequest {
source_code: "111".into(),
});
let response = client.get_token(request).await?;
println!("RESPONSE={:?}", response);
//验证token
let request = Request::new(CheckTokenRequest {
token: "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJleHAiOjE2NjA4ODkyMTksImlhdCI6MTY2MDgwMjgxOSwic3ViIjoiY29tbW9uIHRva2VuIiwic291cmNlX2NvZGUiOiIxMTEifQ.R382aJ0P6D1YkMIjNCw8ePTo5Pji-hg6uXqX6QXdux8".into(),
});
let response = client.check_token(request).await?;
println!("RESPONSE={:?}", response);
Ok(())
}
总结
至此，token的grpc服务已经完成。
下一篇将用poem创建一个简单的rest api服务（包括友好的swanger）来演示与grpc服务的交互。
