使用from_fn_with_state时如何修复Axum的Service trait错误？

你的代码中存在状态类型不匹配的问题，导致编译器无法满足Service<Request> trait约束。具体来说：

1. 你调用from_fn_with_state时传入的是AppState实例，但中间件函数auth_guard期望提取的是State<Arc<AppState>>。两者类型不一致，破坏了Axum中间件的类型绑定规则。
2. from_fn_with_state要求中间件函数的第一个参数类型必须与传入的状态类型完全匹配——你传入AppState，中间件就该接收State<AppState>；你要接收State<Arc<AppState>>，就得传入Arc<AppState>作为状态。

修正步骤 #

1. 统一使用Arc包裹的状态 #

先创建一次Arc<AppState>，再将其克隆（Arc的克隆是轻量操作）传递给中间件和路由：

pub fn get_router(state: AppState) -> axum::Router {
    let shared_state = Arc::new(state); // 只创建一次Arc包裹
    Router::new()
        .route("/*path", any(catch_any))
        .layer(from_fn_with_state(shared_state.clone(), auth_guard)) // 克隆Arc传递给中间件
        .with_state(shared_state) // 使用同一个Arc实例
}

2. 保持中间件函数不变 #

你的auth_guard函数本身是正确的，因为它期望的State<Arc<AppState>>现在和传入的状态类型完全匹配：

pub async fn auth_guard(
    State(state): State<Arc<AppState>>,
    mut req: Request,
    next: Next,
) -> Result<impl IntoResponse, JsonResponse> {
    // jwt解码与数据库查询逻辑...

    let auth_user = AuthUser::JWT((user.sso, user.user_id));
    req.extensions_mut().insert(auth_user);

    Ok(next.run(req).await)
}

为什么这样能解决问题？ #

• from_fn_with_state会将你传入的状态类型与中间件函数的第一个提取器类型做严格匹配，类型一致后，Axum就能自动推导并实现所需的Service trait。
• 使用Arc包裹状态可以避免频繁克隆整个AppState结构体，提升性能，同时保证路由和中间件共享同一个状态实例。

内容的提问来源于stack exchange，提问作者Lacrosse343