Learn Rust Coding · 课时

中间件

Q: 「中间件」这节课中我会学到什么？

Tower 层 你通过在浏览器中直接运行的动手代码来练习 Learn Rust Coding，全天候 AI 导师会在你学习这节课的过程中回答你的问题。

Tower 层

中间件 是 CoddyKit 上的免费 Learn Rust Coding 课时。这是第 4 节课，共 4 节。你可以在下方免费阅读本课时的完整内容 — 然后在浏览器中使用内置代码编辑器和全天候 AI 导师进行实践。这是 Learn Rust Coding 学习路径的一部分，你的进度在网页和 CoddyKit 应用中同步。 Learn Rust Coding 课程共包含 4 节课。

什么是中间件

中间件会在处理程序之前和之后运行代码，包裹整个请求/响应周期。它适用于日志记录、身份验证、压缩和超时控制。

Axum 构建于 tower 生态系统之上
中间件以层的形式添加

Service 特征

Tower 定义了一个 Service：它接收请求，并异步生成响应。层会用一个服务包裹另一个服务。

添加层

使用 .layer() 附加中间件。这里我们从 tower_http 添加请求跟踪。

use axum::{Router, routing::get};
use tower_http::trace::TraceLayer;

fn app() -> Router {
    Router::new()
        .route("/", get(|| async { "ok" }))
        .layer(TraceLayer::new_for_http())
}

层的顺序

层从底部向上包裹：最后添加的层位于最外层。请求从外层流向内层，响应则从内层流向外层。

use axum::{Router, routing::get};
use tower_http::{trace::TraceLayer, compression::CompressionLayer};

fn app() -> Router {
    Router::new()
        .route("/", get(|| async { "ok" }))
        .layer(CompressionLayer::new())
        .layer(TraceLayer::new_for_http())
}

使用 from_fn 编写自定义中间件

middleware::from_fn 会将异步函数转换为中间件。它接收请求和一个用于调用其余链的 Next。

use axum::{middleware::Next, response::Response, extract::Request};

async fn log_mw(req: Request, next: Next) -> Response {
    println!("{} {}", req.method(), req.uri());
    next.run(req).await
}

连接自定义中间件

使用 from_fn 包装您的函数，并将其作为一层添加。

use axum::{Router, routing::get, middleware};

fn app() -> Router {
    Router::new()
        .route("/", get(|| async { "ok" }))
        .layer(middleware::from_fn(log_mw))
}

提前终止请求

中间件可以不调用 next 就提前返回，这对身份验证很有用。返回响应后，调用链就会停止。

use axum::{middleware::Next, response::Response, extract::Request, http::StatusCode};

async fn auth(req: Request, next: Next) -> Result<Response, StatusCode> {
    if req.headers().contains_key("authorization") {
        Ok(next.run(req).await)
    } else {
        Err(StatusCode::UNAUTHORIZED)
    }
}

模拟中间件链

链本质上就是相互调用下一个函数的一组函数。这个可运行示例模拟了前后包裹过程。

fn handler() -> String {
    "response".to_string()
}

fn logging<F: Fn() -> String>(next: F) -> String {
    println!("-> request in");
    let res = next();
    println!("<- response out");
    res
}

fn main() {
    let out = logging(handler);
    println!("body: {}", out);
}

按路由应用中间件

在方法路由器上调用 .layer()，而不是在整个应用上调用，即可将一层应用到单个路由。

use axum::{Router, routing::get, middleware};

fn app() -> Router {
    Router::new().route(
        "/admin",
        get(|| async { "admin" }).layer(middleware::from_fn(auth)),
    )
}

常用的 tower_http 层

tower_http crate 提供了现成的中间件：

TraceLayer 用于日志记录
CompressionLayer 用于 gzip 压缩
CorsLayer 用于 CORS
TimeoutLayer 用于限制请求时间

use tower_http::cors::CorsLayer;

fn cors() -> CorsLayer {
    CorsLayer::permissive()
}

使用 ServiceBuilder 组合层

使用 ServiceBuilder 可以整洁地组合多个层。它会从上到下应用这些层，因此顺序是可预测的。

use tower::ServiceBuilder;
use tower_http::{trace::TraceLayer, compression::CompressionLayer};

fn stack() -> ServiceBuilder<impl Sized> {
    ServiceBuilder::new()
        .layer(TraceLayer::new_for_http())
        .layer(CompressionLayer::new())
}

快速检查

检验您对中间件的理解。

回顾

您已经学习了使用 tower 层实现中间件：

通过 .layer() 添加中间件；最后添加的层位于最外层
from_fn 使用 Next 构建自定义中间件
提前返回即可终止调用链，例如用于身份验证
tower_http 提供跟踪、压缩、CORS 和超时层

Axum 课程到此结束。

常见问题解答

「中间件」课时是免费的吗？

是的 — 「中间件」的完整文本可在网页上免费阅读。要进行交互式练习（内置代码编辑器和全天候 AI 导师）并解锁 Learn Rust Coding 课程的其余内容，请升级到 CoddyKit PRO。 Learn Rust Coding 课程共包含 4 节课。

「中间件」这节课中我会学到什么？

Tower 层你通过在浏览器中直接运行的动手代码来练习 Learn Rust Coding，全天候 AI 导师会在你学习这节课的过程中回答你的问题。

学习 Learn Rust Coding 需要有经验吗？

无需任何先前经验。CoddyKit 上的 Learn Rust Coding 课程适合初学者到高级学习者，你可以从这里开始或从头开始，按照自己的节奏学习。这是第 4 节课，共 4 节。

「中间件」课时需要多长时间？

大多数 CoddyKit 课程大约需要 5–10 分钟。每节课都很精短且互动，所以你能稳步进步，并在网页和应用中从离开的地方继续。

我能在这节 Learn Rust Coding 课中编写并运行代码吗？

能。每节 Learn Rust Coding 课都包含内置代码编辑器，你可以在浏览器中直接编写并运行真实代码，并获得即时 AI 反馈 — 无需本地设置。

此课程中的所有课时

← 返回 Learn Rust Coding