< 返回版块

chrislearn 发表于 2021-04-07 12:49

Salvo 是一个简单易用的 Rust Web 后端框架. 目标是让 Rust 下的 Web 后端开发能像 Go 等其他语言里的一样简单.

🎯 功能特色

  • 基于hyper, tokio 的异步 Web 后端框架;
  • 支持 Websocket;
  • 统一的中间件和句柄接口, 中间件系统支持在句柄之前或者之后运行;
  • 简单易用的路由系统, 支持路由嵌套, 在任何嵌套层都可以添加中间件;
  • 内置 multipart 表单处理, 处理上传文件变得非常简单;
  • 支持从多个本地目录映射成一个虚拟目录提供服务.

⚡️ 快速开始

你可以查看实例代码, 或者访问网站.

创建一个全新的项目:

cargo new hello_salvo --bin

添加依赖项到 Cargo.toml

[dependencies]
salvo = "0.10"
tokio = { version = "1", features = ["full"] }

main.rs 中创建一个简单的函数句柄, 命名为hello_world, 这个函数只是简单地打印文本 "Hello World".

use salvo::prelude::*;

#[fn_handler]
async fn hello_world(_req: &mut Request, _depot: &mut Depot, res: &mut Response) {
    res.render_plain_text("Hello World");
}

对于 fn_handler, 可以根据需求和喜好有不同种写法.

  • 可以将一些没有用到的参数省略掉, 比如这里的 _req, _depot.

    #[fn_handler]
    async fn hello_world(res: &mut Response) {
        res.render_plain_text("Hello World");
    }
    
  • 对于任何实现 Writer 的类型都是可以直接作为函数返回值. 比如, &str 实现了 Writer, 会直接按纯文本输出:

    #[fn_handler]
    async fn hello_world(res: &mut Response) -> &'static str {
        "Hello World"
    }
    
  • 更常见的情况是, 我们需要通过返回一个 Result<T, E> 来简化程序中的错误处理. 如果 Result<T, E>TE 都实现 Writer, 则 Result<T, E> 可以直接作为函数返回类型:

    #[fn_handler]
    async fn hello_world(res: &mut Response) -> Result<&'static str, ()> {
        Ok("Hello World")
    }
    

main 函数中, 我们需要首先创建一个根路由, 然后创建一个 Server 并且调用它的 bind 函数:

use salvo::prelude::*;

#[fn_handler]
async fn hello_world() -> &'static str {
    "Hello World"
}
#[tokio::main]
async fn main() {
    let router = Router::new().get(hello_world);
    let server = Server::new(router);
    server.bind(([0, 0, 0, 0], 7878)).await;
}

中间件

Salvo 中的中间件其实就是 Handler, 没有其他任何特别之处.

树状路由系统

正常情况下我们是这样写路由的:

Router::new().path("articles").get(list_articles).post(create_article);
Router::new()
    .path("articles/<id>")
    .get(show_article)
    .patch(edit_article)
    .delete(delete_article);

往往查看文章和文章列表是不需要用户登录的, 但是创建, 编辑, 删除文章等需要用户登录认证权限才可以. Salvo 中支持嵌套的路由系统可以很好地满足这种需求. 我们可以把不需要用户登录的路由写到一起:

Router::new()
    .path("articles")
    .get(list_articles)
    .push(Router::new().path("<id>").get(show_article));

然后把需要用户登录的路由写到一起, 并且使用相应的中间件验证用户是否登录:

Router::new()
    .path("articles")
    .before(auth_check)
    .post(list_articles)
    .push(Router::new().path("<id>").patch(edit_article).delete(delete_article));

虽然这两个路由都有这同样的 path("articles"), 然而它们依然可以被同时添加到同一个父路由, 所以最后的路由长成了这个样子:

Router::new()
    .push(
        Router::new()
            .path("articles")
            .get(list_articles)
            .push(Router::new().path("<id>").get(show_article)),
    )
    .push(
        Router::new()
            .path("articles")
            .before(auth_check)
            .post(list_articles)
            .push(Router::new().path("<id>").patch(edit_article).delete(delete_article)),
    );

<id>匹配了路径中的一个片段, 正常情况下文章的 id 只是一个数字, 这是我们可以使用正则表达式限制 id 的匹配规则, r"<id:/\d+/>".

对于这种数字类型, 还有一种更简单的方法是使用 <id:num>, 具体写法为:

  • <id:num>, 匹配任意多个数字字符;
  • <id:num[10]>, 只匹配固定特定数量的数字字符,这里的 10 代表匹配仅仅匹配 10 个数字字符;
  • <id:num(..10)> 代表匹配 1 到 9 个数字字符;
  • <id:num(3..10)> 代表匹配 3 到 9 个数字字符;
  • <id:num(..=10)> 代表匹配 1 到 10 个数字字符;
  • <id:num(3..=10)> 代表匹配 3 到 10 个数字字符;
  • <id:num(10..)> 代表匹配至少 10 个数字字符.

还可以通过 <*> 或者 <**> 匹配所有剩余的路径片段. 为了代码易读性性强些, 也可以添加适合的名字, 让路径语义更清晰, 比如: <**file_path>.

允许组合使用多个表达式匹配同一个路径片段, 比如 /articles/article_<id:num>/.

文件上传

可以通过 Request 中的 get_file 异步获取上传的文件:

#[fn_handler]
async fn upload(req: &mut Request, res: &mut Response) {
    let file = req.get_file("file").await;
    if let Some(file) = file {
        let dest = format!("temp/{}", file.filename().unwrap_or_else(|| "file".into()));
        if let Err(e) = std::fs::copy(&file.path, Path::new(&dest)) {
            res.set_status_code(StatusCode::INTERNAL_SERVER_ERROR);
        } else {
            res.render_plain_text("Ok");
        }
    } else {
        res.set_status_code(StatusCode::BAD_REQUEST);
    }
}

多文件上传也是非常容易处理的:

#[fn_handler]
async fn upload(req: &mut Request, res: &mut Response) {
    let files = req.get_files("files").await;
    if let Some(files) = files {
        let mut msgs = Vec::with_capacity(files.len());
        for file in files {
            let dest = format!("temp/{}", file.filename().unwrap_or_else(|| "file".into()));
            if let Err(e) = std::fs::copy(&file.path, Path::new(&dest)) {
                res.set_status_code(StatusCode::INTERNAL_SERVER_ERROR);
                res.render_plain_text(&format!("file not found in request: {}", e.to_string()));
            } else {
                msgs.push(dest);
            }
        }
        res.render_plain_text(&format!("Files uploaded:\n\n{}", msgs.join("\n")));
    } else {
        res.set_status_code(StatusCode::BAD_REQUEST);
        res.render_plain_text("file not found in request");
    }
}

更多示例

您可以从 examples 文件夹下查看更多示例代码:

部分代码和示例移植自 warp, multipart-async, mime-multipartactix-web.

☕ 支持

Salvo是一个开源项目, 如果想支持本项目, 可以 ☕ 在这里买一杯咖啡.

⚠️ 开源协议

Salvo 项目采用 MIT License (LICENSE-MIT or http://opensource.org/licenses/MIT)

评论区

写评论
Mike Tang 2021-04-07 14:10

6666

1 共 1 条评论, 1 页