rust之poem框架实践系列（一）grpc

本文将通过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