Rust Rocket简单入门
简介
Rust中最知名的两个web框架要数Rocket和Actix了，Rocket更注重易用性，Actix则更注重性能。这里只是了解一下Rust下的WebAPI开发流程，就学一下最简单的 Rocket。
Rocket 是一个用于 Rust 的异步 Web 框架，专注于可用性、安全性、可扩展性和速度:
github：https://github.com/rwf2/Rocket/tree/v0.5
官网：https://rocket.rs
hello world
需要最新版本的 Rust 来运行 Rocket 应用程序，运行以下命令确保安装了最新的工具链：
rustup default stable
创建一个新的基于二进制的 Cargo 项目并切换到新目录：
cargo new hello-rocket --bin
cd hello-rocket
执行以下命令，添加 Rocket 依赖项：
cargo add rocket
在 src/main.rs 文件中添加以下代码：
#[macro_use] extern crate rocket;
#[get("/")]
fn index() -> &'static str {
"Hello, world!"
}
#[launch]
fn rocket() -> _ {
rocket::build().mount("/", routes![index])
}
上面hello world示例没有main函数，main函数由launch宏生成，可以通过源码看出：
pub fn launch(args: TokenStream, input: TokenStream) -> TokenStream {
emit!(attribute::entry::launch_attribute(args, input))
}
//...
async_entry!(launch_attribute, launch::Launch, quote!(fn main() {}));
运行程序，访问 http://localhost:8000 以查看应用，VS终端输出如下：
程序带的有彩色输出，如果在文件夹手动打开后没有彩色输出，说明系统不支持ANSI转义序列。
常用功能
动态路径
动态路径比较常见的场景是动态id场景，可以传N个动态类型即动态路径有多层，只要这个类型实现了FromParam：
//访问链接示例：http://localhost:8000/hello/张三/25/true
#[get("/hello/<name>/<age>/<is_male>")]
fn hello(name: &str, age: u8, is_male: bool) -> String {
if is_male {
format!("姓名 {} ，年龄 {}, 性别 男!", name, age)
} else {
format!("姓名 {} ，年龄 {}, 性别 女!", name, age)
}
}
这个路由会匹配所有/hello/为基础路径的路由，然后将它匹配到的动态路径作为参数传递给处理器，Rocket默认给标准库里的一些常见类型以及Rocket自身的一些特殊类型实现了FromParam trait。
多个片段(segments)
可以通过<param..>的方式来匹配多个动态路径，这种类型的参数一般被叫做分段防护装置(segments guards)，都必须先实现FromSegments这个trait。
use std::path::PathBuf;
//访问链接示例：http://localhost:8000/page/foo/bar
#[get("/page/<path..>")]
fn get_page(path: PathBuf) -> String {
let mut output = String::new();
for part in path.iter() {
let part_str = part.to_string_lossy();
println!("路径参数: {}", part_str);
output.push_str(&format!("路径参数: {}\n", part_str));
}
output
}
PathBuf实现了FromSegments这个trait，所以不用担心/page或者/page//导致的解析失败，也不用担心路径遍历攻击(path traversal attacks)。
静态文件服务器
基于 分段防护装置(segments guards)，可以简单的实现一个安全的静态文件服务器：
use std::path::{Path, PathBuf};
use rocket::fs::NamedFile;
#[get("public/<file..>")]
async fn files(file: PathBuf) -> Option<NamedFile> {
NamedFile::open(Path::new("static/").join(file)).await.ok()
}
也可以使用 FileServer，只需一行代码即可：
//引入FileServer结构体
use rocket::fs::FileServer;
//将/public作为URI前缀，并将static/作为文件路径
rocket.mount("/public", FileServer::from("static/"))
在项目根目录下创建一个名为static的文件夹，并将静态文件 example.txt 放在其中，通过以下uri访问文件：
http://localhost:8000/public/example.txt
在发布项目时，可以将静态文件夹放在与可执行文件相同的目录中，或者根据部署需求将其放在其他位置。
简单WebAPI示例
下面使用Rocket实现一个简单的WebAPI，这里的示例只实现Post方法，不涉及JWT鉴权。
添加依赖
执行以下命令添加 serde 依赖：
cargo add serde --features "derive"
再运行一遍以下命令，打开 json 功能标志：
cargo add rocket --features "json"
实现接口
在 src/main.rs 文件中实现以下代码：
#[macro_use] extern crate rocket;
use rocket::serde::{Deserialize, Serialize,json::Json};
#[derive(Debug, Deserialize)]
#[serde(crate = "rocket::serde")]
struct TaskRequest {
description: String,
complete: bool
}
#[derive(Debug, Serialize)]
#[serde(crate = "rocket::serde")]
struct TaskResponse {
description: String,
complete: bool
}
#[post("/todo", data = "<task>")]
fn my_function(task: Json<TaskRequest>) -> Json<TaskResponse> {
// 处理接收到的任务
println!("Received task: {:?}", task);
// 返回处理后的任务
Json(TaskResponse {
description: task.description.clone(),
complete: task.complete,
})
}
#[launch]
fn rocket() -> _ {
rocket::build().mount("/", routes![my_function])
}
接口测试
使用 curl 测试一下接口，在cmd中执行以下命令：
curl -X POST -H "Content-Type: application/json" -d "{\"description\":\"Task 1\",\"complete\":true}" http://localhost:8000/todo
测试结果：