Axum中如何将Spawn任务的JoinHandle存入State以便后续终止？

你的代码存在几个关键问题，下面是修正后的方案：

核心问题分析 #

1. JoinHandle无法克隆：你给AppState派生了Clone，但tokio::task::JoinHandle并没有实现Clone trait，编译会直接报错。
2. 共享状态修改无效：Axum的State提取器默认会克隆状态实例（当类型实现Clone时），所以你在spawn_task里修改的只是局部副本，全局状态不会更新。
3. 未处理空指针风险：abort_task里直接调用unwrap()，如果没有已启动的任务会触发panic。

修正后的完整代码 #

use std::sync::{Arc, Mutex};
use std::time::Duration;
use axum::{extract::State, routing::get, Router};
use tokio::task::JoinHandle;
use tokio::time::sleep;

// 使用Arc<Mutex>实现线程安全的共享可变状态
struct AppState {
    join_handle: Mutex<Option<JoinHandle<()>>>,
}

#[tokio::main]
async fn main() {
    // 用Arc包裹状态，让多个请求能共享同一个实例
    let state = Arc::new(AppState {
        join_handle: Mutex::new(None),
    });

    let app = Router::new()
        .route("/spawn_task", get(spawn_task))
        .route("/abort_task", get(abort_task))
        .with_state(state);

    axum::Server::bind(&"0.0.0.0:3000".parse().unwrap())
        .serve(app.into_make_service())
        .await
        .unwrap();
}

pub async fn spawn_task(State(state): State<Arc<AppState>>) {
    // 锁定互斥量，修改共享状态
    let mut handle_guard = state.join_handle.lock().unwrap();
    // 如果已有任务在运行，先终止旧任务
    if let Some(handle) = handle_guard.take() {
        handle.abort();
    }
    // 生成新任务并保存句柄
    *handle_guard = Some(tokio::spawn(async move {
        println!("任务已启动");
        sleep(Duration::from_millis(10000)).await;
        println!("任务执行完成（10秒后）");
    }));
}

pub async fn abort_task(State(state): State<Arc<AppState>>) {
    let mut handle_guard = state.join_handle.lock().unwrap();
    // 安全处理无任务的情况
    if let Some(handle) = handle_guard.take() {
        handle.abort();
        println!("任务已终止");
    } else {
        println!("没有正在运行的任务");
    }
}

关键修改说明 #

• Arc<Mutex>共享状态：用Arc实现多线程共享所有权，Mutex保证同一时间只有一个线程能修改内部的JoinHandle，确保线程安全。
• 避免Clone派生：移除AppState的#[derive(Clone)]，改用Arc来共享实例。
• 安全的状态操作：在修改或访问join_handle时，通过lock()获取互斥锁，同时处理None的情况，避免panic。
• 任务覆盖处理：在启动新任务前，先检查是否有旧任务在运行，若有则终止，避免多个任务同时存在。

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