Rust Axum中间件鉴权实战:从JWT到RBAC的6种生产模式

编程语言

Rust Web服务的鉴权,为什么总是踩坑

你写了个Axum接口,裸奔上线;你加了JWT校验,发现token过期后用户直接401;你想做RBAC权限控制,发现Axum的中间件里拿不到用户角色;你加了限流,发现未认证请求也在消耗配额。2026年,Axum 0.8已经提供了FromRequestParts提取器、Tower Layer中间件、以及类型安全的状态管理——但鉴权这件事,从来不是框架能帮你做完的

本文将从JWT校验出发,带你完成JWT认证→API Key认证→RBAC权限控制→限流防护→会话管理→生产级鉴权服务的6种实战模式,让Axum的鉴权从"能跑"变成"能抗"。


核心概念

概念 说明
FromRequestParts Axum提取器trait,从请求Parts中提取认证信息
JWT (JSON Web Token) 无状态令牌,包含用户身份与过期时间的签名数据
API Key 通过Header传递的密钥,适合服务间调用
RBAC 基于角色的访问控制,用户→角色→权限三级模型
Tower Layer 中间件抽象层,用于组合认证、限流等横切关注点
Rate Limiting 限流,防止接口被恶意刷量
Session 有状态会话,服务端存储登录状态(Redis等)
Claims JWT载荷中的声明数据(sub/exp/role等)

鉴权请求流程

请求鉴权流程:
1. 客户端发送请求,携带Authorization Header或API Key
2. 中间件/Extractor提取认证凭据
3. 验证凭据有效性(签名校验/过期检查/密钥匹配)
4. 构建用户上下文(UserId/Role/Permissions)
5. RBAC中间件检查用户是否有权访问当前路由
6. 限流中间件检查请求频率
7. Handler执行业务逻辑,可通过State访问用户上下文
8. 响应返回客户端

问题分析:Axum鉴权开发的5大挑战

  1. JWT校验与用户上下文脱节:中间件里校验了token,但Handler里拿不到用户信息,只能重新解析一遍
  2. 多种认证方式难以共存:JWT和API Key要同时支持,中间件写成了if-else面条代码
  3. RBAC权限模型设计混乱:角色和权限用字符串硬编码,新增权限要改十几个地方
  4. 限流与认证顺序冲突:限流在认证之前,未认证请求浪费限流配额;限流在认证之后,恶意请求直接打到认证层
  5. 会话管理缺乏生产方案:JWT无状态但无法主动吊销,Session有状态但Redis连接池和过期清理都是坑

分步实操:6种生产级鉴权模式

模式1:JWT认证中间件与FromRequestParts

use axum::extract::{FromRequestParts, Request};
use axum::http::request::Parts;
use axum::response::{IntoResponse, Response};
use axum::middleware::{Next, from_fn};
use axum::{Json, Router, middleware, routing::get};
use jsonwebtoken::{decode, encode, DecodingKey, EncodingKey, Header, Validation};
use serde::{Deserialize, Serialize};
use chrono::{Utc, Duration};
use std::sync::Arc;

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct Claims {
    pub sub: String,
    pub role: String,
    pub exp: i64,
    pub iat: i64,
}

#[derive(Clone)]
pub struct AuthConfig {
    pub jwt_secret: String,
    pub jwt_expiration_hours: i64,
}

pub struct AppState {
    pub auth_config: AuthConfig,
    pub db: DbPool,
}

pub type SharedState = Arc<AppState>;

impl Claims {
    pub fn new(user_id: &str, role: &str, expiration_hours: i64) -> Self {
        let now = Utc::now();
        Self {
            sub: user_id.to_string(),
            role: role.to_string(),
            iat: now.timestamp(),
            exp: (now + Duration::hours(expiration_hours)).timestamp(),
        }
    }

    pub fn encode(&self, secret: &str) -> Result<String, jsonwebtoken::errors::Error> {
        encode(
            &Header::default(),
            self,
            &EncodingKey::from_secret(secret.as_bytes()),
        )
    }

    pub fn decode(token: &str, secret: &str) -> Result<Self, jsonwebtoken::errors::Error> {
        let token_data = decode::<Claims>(
            token,
            &DecodingKey::from_secret(secret.as_bytes()),
            &Validation::default(),
        )?;
        Ok(token_data.claims)
    }
}
use axum::extract::FromRequestParts;
use axum::http::StatusCode;

pub struct AuthUser {
    pub user_id: String,
    pub role: String,
}

#[axum::async_trait]
impl FromRequestParts<SharedState> for AuthUser {
    type Rejection = AuthError;

    async fn from_request_parts(
        parts: &mut Parts,
        state: &SharedState,
    ) -> Result<Self, Self::Rejection> {
        let auth_header = parts
            .headers
            .get("authorization")
            .and_then(|v| v.to_str().ok())
            .ok_or(AuthError::MissingToken)?;

        let token = auth_header
            .strip_prefix("Bearer ")
            .ok_or(AuthError::InvalidTokenFormat)?;

        let claims = Claims::decode(token, &state.auth_config.jwt_secret)
            .map_err(|_| AuthError::InvalidToken)?;

        Ok(AuthUser {
            user_id: claims.sub,
            role: claims.role,
        })
    }
}

async fn protected_handler(
    auth_user: AuthUser,
) -> Result<Json<serde_json::Value>, AuthError> {
    Ok(Json(serde_json::json!({
        "user_id": auth_user.user_id,
        "role": auth_user.role,
    })))
}

pub fn auth_router() -> Router<SharedState> {
    Router::new()
        .route("/me", get(protected_handler))
        .route("/refresh", get(refresh_token))
}

async fn refresh_token(
    auth_user: AuthUser,
    State(state): State<SharedState>,
) -> Result<Json<serde_json::Value>, AuthError> {
    let claims = Claims::new(
        &auth_user.user_id,
        &auth_user.role,
        state.auth_config.jwt_expiration_hours,
    );
    let token = claims.encode(&state.auth_config.jwt_secret)?;
    Ok(Json(serde_json::json!({ "token": token })))
}

模式2:API Key认证与Header提取

use axum::extract::FromRequestParts;
use axum::http::request::Parts;
use std::collections::HashMap;
use tokio::sync::RwLock;

#[derive(Debug, Clone)]
pub struct ApiKeyInfo {
    pub key_id: String,
    pub client_name: String,
    pub permissions: Vec<String>,
    pub rate_limit: u32,
}

pub struct ApiKeyState {
    pub keys: Arc<RwLock<HashMap<String, ApiKeyInfo>>>,
}

pub struct AuthenticatedApi {
    pub key_info: ApiKeyInfo,
}

#[axum::async_trait]
impl FromRequestParts<SharedState> for AuthenticatedApi {
    type Rejection = AuthError;

    async fn from_request_parts(
        parts: &mut Parts,
        state: &SharedState,
    ) -> Result<Self, Self::Rejection> {
        let api_key = parts
            .headers
            .get("x-api-key")
            .and_then(|v| v.to_str().ok())
            .ok_or(AuthError::MissingApiKey)?;

        let keys = state.api_key_state.keys.read().await;
        let key_info = keys
            .get(api_key)
            .ok_or(AuthError::InvalidApiKey)?
            .clone();

        Ok(AuthenticatedApi { key_info })
    }
}

async fn api_endpoint(
    auth: AuthenticatedApi,
) -> Result<Json<serde_json::Value>, AuthError> {
    Ok(Json(serde_json::json!({
        "client": auth.key_info.client_name,
        "permissions": auth.key_info.permissions,
    })))
}
use axum::extract::FromRequestParts;
use axum::http::request::Parts;

pub enum AuthMethod {
    Jwt(AuthUser),
    ApiKey(AuthenticatedApi),
}

pub struct MultiAuth;

#[axum::async_trait]
impl FromRequestParts<SharedState> for AuthMethod {
    type Rejection = AuthError;

    async fn from_request_parts(
        parts: &mut Parts,
        state: &SharedState,
    ) -> Result<Self, Self::Rejection> {
        if parts.headers.contains_key("x-api-key") {
            let api_auth = AuthenticatedApi::from_request_parts(parts, state).await?;
            Ok(AuthMethod::ApiKey(api_auth))
        } else if parts.headers.contains_key("authorization") {
            let jwt_auth = AuthUser::from_request_parts(parts, state).await?;
            Ok(AuthMethod::Jwt(jwt_auth))
        } else {
            Err(AuthError::NoAuthMethod)
        }
    }
}

async fn multi_auth_handler(
    auth: AuthMethod,
) -> Json<serde_json::Value> {
    match auth {
        AuthMethod::Jwt(user) => Json(serde_json::json!({
            "method": "jwt",
            "user_id": user.user_id,
        })),
        AuthMethod::ApiKey(api) => Json(serde_json::json!({
            "method": "api_key",
            "client": api.key_info.client_name,
        })),
    }
}

模式3:RBAC角色权限控制与枚举权限

use strum::{Display, EnumString};
use std::collections::HashSet;

#[derive(Debug, Clone, PartialEq, Eq, Hash, Display, EnumString)]
pub enum Permission {
    #[strum(to_string = "users:read")]
    UsersRead,
    #[strum(to_string = "users:write")]
    UsersWrite,
    #[strum(to_string = "users:delete")]
    UsersDelete,
    #[strum(to_string = "products:read")]
    ProductsRead,
    #[strum(to_string = "products:write")]
    ProductsWrite,
    #[strum(to_string = "orders:read")]
    OrdersRead,
    #[strum(to_string = "orders:write")]
    OrdersWrite,
    #[strum(to_string = "admin:full")]
    AdminFull,
}

#[derive(Debug, Clone, PartialEq, Eq, Display, EnumString)]
pub enum Role {
    #[strum(to_string = "viewer")]
    Viewer,
    #[strum(to_string = "editor")]
    Editor,
    #[strum(to_string = "admin")]
    Admin,
    #[strum(to_string = "superadmin")]
    SuperAdmin,
}

impl Role {
    pub fn permissions(&self) -> HashSet<Permission> {
        match self {
            Role::Viewer => HashSet::from([
                Permission::UsersRead,
                Permission::ProductsRead,
                Permission::OrdersRead,
            ]),
            Role::Editor => HashSet::from([
                Permission::UsersRead,
                Permission::ProductsRead,
                Permission::ProductsWrite,
                Permission::OrdersRead,
                Permission::OrdersWrite,
            ]),
            Role::Admin => HashSet::from([
                Permission::UsersRead,
                Permission::UsersWrite,
                Permission::ProductsRead,
                Permission::ProductsWrite,
                Permission::OrdersRead,
                Permission::OrdersWrite,
            ]),
            Role::SuperAdmin => HashSet::from([Permission::AdminFull]),
        }
    }

    pub fn has_permission(&self, permission: &Permission) -> bool {
        let perms = self.permissions();
        perms.contains(&Permission::AdminFull) || perms.contains(permission)
    }
}
use axum::extract::{FromRequestParts, Path};
use axum::http::request::Parts;

pub struct RequirePermission(pub Permission);

#[axum::async_trait]
impl FromRequestParts<SharedState> for RequirePermission {
    type Rejection = AuthError;

    async fn from_request_parts(
        parts: &mut Parts,
        state: &SharedState,
    ) -> Result<Self, Self::Rejection> {
        let auth_user = AuthUser::from_request_parts(parts, state).await?;
        let role: Role = auth_user.role.parse().map_err(|_| AuthError::InvalidRole)?;

        let required = Permission::UsersWrite;
        if !role.has_permission(&required) {
            return Err(AuthError::Forbidden(
                format!("Missing permission: {}", required),
            ));
        }

        Ok(RequirePermission(required))
    }
}

pub fn require_permission(permission: Permission) -> impl FnMut(Request, Next) -> std::pin::Pin<Box<dyn std::future::Future<Output = Result<Response, AuthError>> + Send>> + Clone {
    move |request: Request, next: Next| {
        let perm = permission.clone();
        Box::pin(async move {
            let (mut parts, body) = request.into_parts();
            let auth_user = AuthUser::from_request_parts(&mut parts, &state).await?;

            let role: Role = auth_user.role.parse().map_err(|_| AuthError::InvalidRole)?;
            if !role.has_permission(&perm) {
                return Err(AuthError::Forbidden(
                    format!("Missing permission: {}", perm),
                ));
            }

            let request = Request::from_parts(parts, body);
            Ok(next.run(request).await)
        })
    }
}

async fn delete_user_handler(
    auth_user: AuthUser,
    Path(user_id): Path<String>,
) -> Result<StatusCode, AuthError> {
    let role: Role = auth_user.role.parse().map_err(|_| AuthError::InvalidRole)?;
    if !role.has_permission(&Permission::UsersDelete) {
        return Err(AuthError::Forbidden("Missing users:delete permission".into()));
    }
    tracing::info!("Deleting user: {}", user_id);
    Ok(StatusCode::NO_CONTENT)
}

模式4:限流中间件与Tower Layer

use axum::extract::Request;
use axum::middleware::Next;
use axum::response::Response;
use std::collections::HashMap;
use std::sync::Arc;
use std::time::Instant;
use tokio::sync::RwLock;

#[derive(Clone)]
pub struct RateLimitConfig {
    pub max_requests: u32,
    pub window_secs: u64,
}

#[derive(Debug)]
struct RateLimitEntry {
    count: u32,
    window_start: Instant,
}

#[derive(Clone)]
pub struct RateLimitState {
    pub entries: Arc<RwLock<HashMap<String, RateLimitEntry>>>,
    pub config: RateLimitConfig,
}

impl RateLimitState {
    pub fn new(config: RateLimitConfig) -> Self {
        Self {
            entries: Arc::new(RwLock::new(HashMap::new())),
            config,
        }
    }

    pub async fn check_rate_limit(&self, key: &str) -> bool {
        let mut entries = self.entries.write().await;
        let now = Instant::now();

        let entry = entries.entry(key.to_string()).or_insert(RateLimitEntry {
            count: 0,
            window_start: now,
        });

        if now.duration_since(entry.window_start).as_secs() > self.config.window_secs {
            entry.count = 0;
            entry.window_start = now;
        }

        entry.count += 1;
        entry.count <= self.config.max_requests
    }
}

pub async fn rate_limit_middleware(
    request: Request,
    next: Next,
) -> Result<Response, AuthError> {
    let state = request
        .extensions()
        .get::<RateLimitState>()
        .cloned()
        .ok_or(AuthError::InternalServerError("Rate limit state not found".into()))?;

    let key = request
        .headers()
        .get("x-api-key")
        .and_then(|v| v.to_str().ok())
        .or_else(|| {
            request.headers()
                .get("x-forwarded-for")
                .and_then(|v| v.to_str().ok())
        })
        .unwrap_or("anonymous")
        .to_string();

    if !state.check_rate_limit(&key).await {
        return Err(AuthError::RateLimited);
    }

    Ok(next.run(request).await)
}
use axum::Router;
use axum::middleware;
use tower::ServiceBuilder;
use tower_http::limit::RateLimitLayer;
use std::time::Duration;

pub fn create_rate_limited_router(state: SharedState) -> Router<SharedState> {
    let rate_limit_state = RateLimitState::new(RateLimitConfig {
        max_requests: 100,
        window_secs: 60,
    });

    Router::new()
        .route("/api/data", get(data_handler))
        .layer(middleware::from_fn(rate_limit_middleware))
        .layer(middleware::from_fn(auth_middleware))
        .with_state(state)
}

pub fn create_tower_rate_limited_router(state: SharedState) -> Router<SharedState> {
    Router::new()
        .route("/api/data", get(data_handler))
        .layer(
            ServiceBuilder::new()
                .layer(RateLimitLayer::new(100, Duration::from_secs(60)))
                .into_inner(),
        )
        .with_state(state)
}

async fn data_handler() -> &'static str {
    "OK"
}

模式5:会话管理与Redis后端

use redis::AsyncCommands;
use serde::{Deserialize, Serialize};
use uuid::Uuid;

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Session {
    pub session_id: String,
    pub user_id: String,
    pub role: String,
    pub created_at: i64,
    pub expires_at: i64,
    pub ip_address: Option<String>,
    pub user_agent: Option<String>,
}

impl Session {
    pub fn new(user_id: &str, role: &str, ttl_secs: i64, ip: Option<&str>, ua: Option<&str>) -> Self {
        let now = Utc::now().timestamp();
        Self {
            session_id: Uuid::new_v4().to_string(),
            user_id: user_id.to_string(),
            role: role.to_string(),
            created_at: now,
            expires_at: now + ttl_secs,
            ip_address: ip.map(|s| s.to_string()),
            user_agent: ua.map(|s| s.to_string()),
        }
    }

    pub fn is_expired(&self) -> bool {
        Utc::now().timestamp() > self.expires_at
    }
}

#[derive(Clone)]
pub struct SessionStore {
    pub client: redis::Client,
    pub ttl_secs: i64,
    pub key_prefix: String,
}

impl SessionStore {
    pub fn new(redis_url: &str, ttl_secs: i64) -> Result<Self, redis::RedisError> {
        Ok(Self {
            client: redis::Client::open(redis_url)?,
            ttl_secs,
            key_prefix: "session:".to_string(),
        })
    }

    pub async fn create_session(&self, session: &Session) -> Result<(), redis::RedisError> {
        let mut conn = self.client.get_multiplexed_async_connection().await?;
        let key = format!("{}{}", self.key_prefix, session.session_id);
        let value = serde_json::to_string(session).unwrap();
        conn.set_ex(&key, value, self.ttl_secs as u64).await?;
        Ok(())
    }

    pub async fn get_session(&self, session_id: &str) -> Result<Option<Session>, redis::RedisError> {
        let mut conn = self.client.get_multiplexed_async_connection().await?;
        let key = format!("{}{}", self.key_prefix, session_id);
        let value: Option<String> = conn.get(&key).await?;
        match value {
            Some(v) => Ok(Some(serde_json::from_str(&v).unwrap())),
            None => Ok(None),
        }
    }

    pub async fn delete_session(&self, session_id: &str) -> Result<(), redis::RedisError> {
        let mut conn = self.client.get_multiplexed_async_connection().await?;
        let key = format!("{}{}", self.key_prefix, session_id);
        conn.del(&key).await?;
        Ok(())
    }

    pub async fn refresh_session(&self, session_id: &str) -> Result<(), redis::RedisError> {
        let mut conn = self.client.get_multiplexed_async_connection().await?;
        let key = format!("{}{}", self.key_prefix, session_id);
        conn.expire(&key, self.ttl_secs as i64).await?;
        Ok(())
    }
}
use axum::extract::{FromRequestParts, Request};
use axum::http::request::Parts;
use axum::middleware::Next;
use axum::response::Response;
use axum::Extension;

pub struct SessionUser {
    pub session: Session,
}

#[axum::async_trait]
impl FromRequestParts<SharedState> for SessionUser {
    type Rejection = AuthError;

    async fn from_request_parts(
        parts: &mut Parts,
        state: &SharedState,
    ) -> Result<Self, Self::Rejection> {
        let cookie_header = parts
            .headers
            .get("cookie")
            .and_then(|v| v.to_str().ok())
            .ok_or(AuthError::MissingSession)?;

        let session_id = extract_session_id(cookie_header)
            .ok_or(AuthError::MissingSession)?;

        let session = state
            .session_store
            .get_session(&session_id)
            .await
            .map_err(|_| AuthError::InternalServerError("Redis error".into()))?
            .ok_or(AuthError::SessionExpired)?;

        if session.is_expired() {
            state.session_store.delete_session(&session_id).await.ok();
            return Err(AuthError::SessionExpired);
        }

        state.session_store.refresh_session(&session_id).await.ok();

        Ok(SessionUser { session })
    }
}

fn extract_session_id(cookie_header: &str) -> Option<String> {
    cookie_header
        .split(';')
        .find_map(|cookie| {
            let mut parts = cookie.trim().splitn(2, '=');
            let name = parts.next()?.trim();
            if name == "session_id" {
                Some(parts.next()?.trim().to_string())
            } else {
                None
            }
        })
}

async fn login_handler(
    State(state): State<SharedState>,
    Json(payload): Json<LoginRequest>,
) -> Result<Json<serde_json::Value>, AuthError> {
    let user = state.db.verify_user(&payload.username, &payload.password).await
        .ok_or(AuthError::InvalidCredentials)?;

    let session = Session::new(
        &user.id,
        &user.role,
        state.session_store.ttl_secs,
        None,
        None,
    );

    state.session_store.create_session(&session).await
        .map_err(|_| AuthError::InternalServerError("Failed to create session".into()))?;

    Ok(Json(serde_json::json!({
        "session_id": session.session_id,
        "expires_at": session.expires_at,
    })))
}

async fn logout_handler(
    session_user: SessionUser,
    State(state): State<SharedState>,
) -> Result<StatusCode, AuthError> {
    state.session_store
        .delete_session(&session_user.session.session_id)
        .await
        .map_err(|_| AuthError::InternalServerError("Failed to delete session".into()))?;
    Ok(StatusCode::NO_CONTENT)
}

模式6:生产级鉴权服务组合

use axum::{Router, routing::{get, post}, middleware, extract::State};
use tower::ServiceBuilder;
use tower_http::cors::{CorsLayer, Any};
use tower_http::trace::TraceLayer;
use std::time::Duration;

pub struct AppState {
    pub auth_config: AuthConfig,
    pub api_key_state: ApiKeyState,
    pub rate_limit_state: RateLimitState,
    pub session_store: SessionStore,
    pub db: DbPool,
}

pub type SharedState = Arc<AppState>;

pub fn create_production_router(state: SharedState) -> Router {
    let public_routes = Router::new()
        .route("/auth/login", post(login_handler))
        .route("/auth/register", post(register_handler))
        .route("/health", get(health_check));

    let jwt_protected = Router::new()
        .route("/users/me", get(get_profile))
        .route("/users/me", post(update_profile))
        .route("/auth/refresh", get(refresh_token))
        .layer(middleware::from_fn_with_state(
            state.clone(),
            jwt_auth_middleware,
        ));

    let api_key_routes = Router::new()
        .route("/api/v1/data", get(data_handler))
        .route("/api/v1/reports", get(reports_handler))
        .layer(middleware::from_fn_with_state(
            state.clone(),
            api_key_auth_middleware,
        ));

    let admin_routes = Router::new()
        .route("/admin/users", get(list_users_handler))
        .route("/admin/users/{id}", post(delete_user_handler))
        .layer(middleware::from_fn_with_state(
            state.clone(),
            admin_auth_middleware,
        ));

    let session_routes = Router::new()
        .route("/dashboard", get(dashboard_handler))
        .route("/auth/logout", post(logout_handler))
        .layer(middleware::from_fn_with_state(
            state.clone(),
            session_auth_middleware,
        ));

    Router::new()
        .merge(public_routes)
        .nest("/api", jwt_protected)
        .nest("/external", api_key_routes)
        .nest("/manage", admin_routes)
        .nest("/web", session_routes)
        .layer(
            ServiceBuilder::new()
                .layer(TraceLayer::new_for_http())
                .layer(CorsLayer::new().allow_origin(Any).allow_methods(Any).allow_headers(Any))
                .into_inner(),
        )
        .with_state(state)
}

async fn jwt_auth_middleware(
    State(state): State<SharedState>,
    mut request: Request,
    next: Next,
) -> Result<Response, AuthError> {
    let auth_header = request.headers()
        .get("authorization")
        .and_then(|v| v.to_str().ok())
        .ok_or(AuthError::MissingToken)?;

    let token = auth_header
        .strip_prefix("Bearer ")
        .ok_or(AuthError::InvalidTokenFormat)?;

    let claims = Claims::decode(token, &state.auth_config.jwt_secret)
        .map_err(|_| AuthError::InvalidToken)?;

    request.extensions_mut().insert(AuthUser {
        user_id: claims.sub.clone(),
        role: claims.role.clone(),
    });

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

async fn api_key_auth_middleware(
    State(state): State<SharedState>,
    mut request: Request,
    next: Next,
) -> Result<Response, AuthError> {
    let api_key = request.headers()
        .get("x-api-key")
        .and_then(|v| v.to_str().ok())
        .ok_or(AuthError::MissingApiKey)?;

    let keys = state.api_key_state.keys.read().await;
    let key_info = keys.get(api_key).ok_or(AuthError::InvalidApiKey)?.clone();

    if !state.rate_limit_state.check_rate_limit(&key_info.key_id).await {
        return Err(AuthError::RateLimited);
    }

    request.extensions_mut().insert(key_info);
    Ok(next.run(request).await)
}

async fn admin_auth_middleware(
    State(state): State<SharedState>,
    mut request: Request,
    next: Next,
) -> Result<Response, AuthError> {
    let auth_header = request.headers()
        .get("authorization")
        .and_then(|v| v.to_str().ok())
        .ok_or(AuthError::MissingToken)?;

    let token = auth_header.strip_prefix("Bearer ").ok_or(AuthError::InvalidTokenFormat)?;
    let claims = Claims::decode(token, &state.auth_config.jwt_secret)
        .map_err(|_| AuthError::InvalidToken)?;

    let role: Role = claims.role.parse().map_err(|_| AuthError::InvalidRole)?;
    if !role.has_permission(&Permission::UsersWrite) {
        return Err(AuthError::Forbidden("Admin access required".into()));
    }

    request.extensions_mut().insert(AuthUser {
        user_id: claims.sub,
        role: claims.role,
    });

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

async fn session_auth_middleware(
    State(state): State<SharedState>,
    mut request: Request,
    next: Next,
) -> Result<Response, AuthError> {
    let cookie_header = request.headers()
        .get("cookie")
        .and_then(|v| v.to_str().ok())
        .ok_or(AuthError::MissingSession)?;

    let session_id = extract_session_id(cookie_header).ok_or(AuthError::MissingSession)?;
    let session = state.session_store.get_session(&session_id).await
        .map_err(|_| AuthError::InternalServerError("Redis error".into()))?
        .ok_or(AuthError::SessionExpired)?;

    if session.is_expired() {
        state.session_store.delete_session(&session_id).await.ok();
        return Err(AuthError::SessionExpired);
    }

    state.session_store.refresh_session(&session_id).await.ok();
    request.extensions_mut().insert(session);
    Ok(next.run(request).await)
}

避坑指南

坑1:JWT密钥硬编码

// ❌ 错误:密钥硬编码在代码中
let secret = "my-super-secret-key-123";

// ✅ 正确:从环境变量读取,启动时校验
let secret = std::env::var("JWT_SECRET")
    .expect("JWT_SECRET must be set");
if secret.len() < 32 {
    panic!("JWT_SECRET must be at least 32 characters");
}

坑2:中间件中State类型不匹配

// ❌ 错误:from_fn无法访问State,编译报错
.layer(middleware::from_fn(my_auth_middleware))

async fn my_auth_middleware(
    State(state): State<SharedState>, // from_fn不支持State参数!
    request: Request,
    next: Next,
) -> Result<Response, AuthError> { ... }

// ✅ 正确:使用from_fn_with_state
.layer(middleware::from_fn_with_state(state.clone(), my_auth_middleware))

async fn my_auth_middleware(
    State(state): State<SharedState>,
    request: Request,
    next: Next,
) -> Result<Response, AuthError> { ... }

坑3:JWT Claims缺少过期校验

// ❌ 错误:手动创建Validation但没启用exp检查
let validation = Validation::new(jsonwebtoken::Algorithm::HS256);
// 默认会检查exp,但如果手动构造可能遗漏

// ✅ 正确:显式配置Validation
let mut validation = Validation::new(jsonwebtoken::Algorithm::HS256);
validation.leeway = 60; // 允许60秒时钟偏移
validation.validate_exp = true;
validation.validate_nbf = true;
let token_data = decode::<Claims>(token, &key, &validation)?;

坑4:RBAC用字符串匹配权限

// ❌ 错误:字符串硬编码,容易拼写错误且无法编译时检查
if user.role == "admin" || user.permissions.contains(&"users:write".to_string()) {
    // 拼写错误不会被编译器发现
}

// ✅ 正确:使用枚举+strum,编译时保证权限名正确
#[derive(EnumString)]
pub enum Permission {
    #[strum(to_string = "users:write")]
    UsersWrite,
}

let perm = Permission::UsersWrite;
if role.has_permission(&perm) {
    // 编译时安全
}

坑5:Redis连接未复用

// ❌ 错误:每次请求创建新连接
async fn get_session(session_id: &str) -> Option<Session> {
    let client = redis::Client::open("redis://localhost").unwrap();
    let mut conn = client.get_async_connection().await.unwrap(); // 每次新建!
    conn.get(session_id).await.ok()
}

// ✅ 正确:使用multiplexed连接复用
pub struct SessionStore {
    client: redis::Client,
}

impl SessionStore {
    pub async fn get_session(&self, session_id: &str) -> Result<Option<Session>, redis::RedisError> {
        let mut conn = self.client.get_multiplexed_async_connection().await?;
        let value: Option<String> = conn.get(session_id).await?;
        // ...
    }
}

报错排查

序号 报错信息 原因 解决方法
1 the trait FromRequestParts is not implemented for AuthUser 未实现FromRequestParts trait 使用#[axum::async_trait]实现trait,确保Rejection类型实现IntoResponse
2 mismatched types: expected State<X>, found State<Y> 中间件和Router的State类型不一致 统一使用type SharedState = Arc<AppState>别名
3 JWT decode error: InvalidToken token格式错误或密钥不匹配 检查Bearer前缀、密钥一致性、算法匹配
4 JWT decode error: ExpiredSignature token已过期 实现refresh token机制,前端自动续期
5 future cannot be sent between threads safely Redis连接非Send类型跨await 使用get_multiplexed_async_connection()替代get_async_connection()
6 missing field exp in Claims Claims结构体缺少exp字段 确保Claims包含exp和iat字段,Validation默认检查
7 Rate limit state not found in extensions Extension未注入 在中间件或Router层通过.layer(Extension(state))注入
8 Redis: Connection refused Redis服务未启动 检查Redis服务状态和连接URL配置
9 handler has too many arguments Handler参数超过4个Extractor 用结构体合并或使用Extension传递认证信息
10 Cannot drop a runtime in a context that is already inside a runtime 在async函数中同步创建Redis连接 使用get_multiplexed_async_connection().await异步连接

进阶优化

1. JWT黑名单与主动吊销

use std::collections::HashSet;
use tokio::sync::RwLock;

#[derive(Clone)]
pub struct JwtBlacklist {
    pub revoked_tokens: Arc<RwLock<HashSet<String>>>,
    pub redis_client: Option<redis::Client>,
}

impl JwtBlacklist {
    pub fn new(redis_url: Option<&str>) -> Result<Self, redis::RedisError> {
        let client = redis_url
            .map(redis::Client::open)
            .transpose()?;
        Ok(Self {
            revoked_tokens: Arc::new(RwLock::new(HashSet::new())),
            redis_client: client,
        })
    }

    pub async fn revoke_token(&self, jti: &str, exp_secs: i64) -> Result<(), AuthError> {
        if let Some(client) = &self.redis_client {
            let mut conn = client.get_multiplexed_async_connection().await
                .map_err(|_| AuthError::InternalServerError("Redis connection failed".into()))?;
            let key = format!("jwt:blacklist:{}", jti);
            redis::cmd("SET")
                .arg(&key)
                .arg("1")
                .arg("EX")
                .arg(exp_secs)
                .exec_async(&mut conn)
                .await
                .map_err(|_| AuthError::InternalServerError("Redis SET failed".into()))?;
        } else {
            let mut tokens = self.revoked_tokens.write().await;
            tokens.insert(jti.to_string());
        }
        Ok(())
    }

    pub async fn is_revoked(&self, jti: &str) -> bool {
        if let Some(client) = &self.redis_client {
            if let Ok(mut conn) = client.get_multiplexed_async_connection().await {
                let key = format!("jwt:blacklist:{}", jti);
                if let Ok(exists) = redis::cmd("EXISTS")
                    .arg(&key)
                    .query_async::<i32>(&mut conn)
                    .await
                {
                    return exists > 0;
                }
            }
        }
        let tokens = self.revoked_tokens.read().await;
        tokens.contains(jti)
    }
}

2. 认证中间件性能优化

use axum::extract::Request;
use axum::middleware::Next;
use axum::response::Response;
use moka::sync::Cache;
use std::sync::Arc;
use std::time::Duration;

#[derive(Clone)]
pub struct AuthCache {
    pub token_cache: Cache<String, AuthUser>,
    pub api_key_cache: Cache<String, ApiKeyInfo>,
}

impl AuthCache {
    pub fn new(max_entries: usize, ttl_secs: u64) -> Self {
        Self {
            token_cache: Cache::builder()
                .max_capacity(max_entries as u64)
                .time_to_live(Duration::from_secs(ttl_secs))
                .build(),
            api_key_cache: Cache::builder()
                .max_capacity(max_entries as u64)
                .time_to_live(Duration::from_secs(ttl_secs))
                .build(),
        }
    }
}

pub async fn cached_jwt_auth_middleware(
    State(state): State<SharedState>,
    mut request: Request,
    next: Next,
) -> Result<Response, AuthError> {
    let auth_header = request.headers()
        .get("authorization")
        .and_then(|v| v.to_str().ok())
        .ok_or(AuthError::MissingToken)?;

    let token = auth_header.strip_prefix("Bearer ").ok_or(AuthError::InvalidTokenFormat)?;

    if let Some(cached_user) = state.auth_cache.token_cache.get(token) {
        request.extensions_mut().insert(cached_user);
        return Ok(next.run(request).await);
    }

    let claims = Claims::decode(token, &state.auth_config.jwt_secret)
        .map_err(|_| AuthError::InvalidToken)?;

    let auth_user = AuthUser {
        user_id: claims.sub.clone(),
        role: claims.role.clone(),
    };

    state.auth_cache.token_cache.insert(token.to_string(), auth_user.clone());
    request.extensions_mut().insert(auth_user);
    Ok(next.run(request).await)
}

3. OpenTelemetry可观测性集成

use axum::extract::Request;
use axum::middleware::Next;
use axum::response::Response;
use opentelemetry::trace::{Span, Tracer};
use opentelemetry::KeyValue;

pub async fn observability_middleware(
    request: Request,
    next: Next,
) -> Response {
    let method = request.method().clone();
    let path = request.uri().path().to_string();
    let auth_method = if request.headers().contains_key("x-api-key") {
        "api_key"
    } else if request.headers().contains_key("authorization") {
        "jwt"
    } else if request.headers().contains_key("cookie") {
        "session"
    } else {
        "none"
    };

    let tracer = opentelemetry::global::tracer("auth-service");
    let mut span = tracer.start(format!("{} {}", method, path));
    span.set_attribute(KeyValue::new("auth.method", auth_method.to_string()));
    span.set_attribute(KeyValue::new("http.method", method.to_string()));
    span.set_attribute(KeyValue::new("http.path", path.clone()));

    let response = next.run(request).await;

    span.set_attribute(KeyValue::new("http.status_code", response.status().as_u16() as i64));
    span.end();

    response
}

对比分析

维度 Axum+Tower Actix-web Guard Go Gin Java Spring Security
认证模型 FromRequestParts+Layer Guard trait+Extractor 中间件函数 Filter链+SecurityContext
类型安全 ✅编译时 ⚠️部分运行时 ❌运行时 ❌运行时
中间件组合 ✅Tower ServiceBuilder ⚠️手动嵌套 ✅中间件链 ✅Filter链
RBAC支持 需自行实现 需自行实现 casbin等库 ✅内置@PreAuthorize
JWT生态 jsonwebtoken jsonwebtoken golang-jwt jjwt/nimbus
会话管理 需自行实现 需自行实现 gorilla/sessions ✅内置Session
性能 ⭐极高 ⭐极高 ⭐高 ⭐中
学习曲线 ⭐陡 ⭐陡 ⭐平缓 ⭐极陡
限流集成 tower-http 自行实现 tollbooth等 bucket4j等
生产就绪度 ⭐高 ⭐高 ⭐极高 ⭐极高

总结:Axum鉴权的核心优势在于类型安全的提取器——FromRequestParts让认证信息像普通参数一样注入Handler,编译时就能发现类型不匹配。2026年的生产实践:用FromRequestParts实现JWT/API Key提取→枚举+strum构建RBAC权限模型→from_fn_with_state编写带状态的中间件→Redis管理会话与JWT黑名单→moka缓存加速token校验→Tower ServiceBuilder组合中间件管道。关键是要理解Axum的提取器模型——认证不是"拦截器",而是"类型提取"。


在线工具推荐

本站提供浏览器本地工具,免注册即可试用 →

#Rust#Axum#中间件#JWT#认证鉴权#2026#Tower