如何在Actix-web Rust框架中捕获客户端请求中止并通知后端？

在Actix-web里并没有直接的on_client_cancel API，但可以通过以下两种核心方式实现你的需求：

1. 监听客户端连接关闭事件 #

Actix-web的HttpRequest提供了when_closed()方法，它返回一个Future，当*客户端连接关闭（包括fetch请求被AbortSignal中止）*时会完成。你可以将这个监听逻辑和后端查询任务结合，一旦检测到连接关闭就触发取消操作：

use actix_web::{HttpRequest, Data, HttpResponse, Error};
use tokio::sync::CancellationToken;
use tokio::select;

/// 执行后端查询，客户端提前中止时同步终止后端任务
#[route("/foo")]
async fn get_data(
    req: HttpRequest,
    state: Data<AppState>,
) -> Result<HttpResponse, Error> {
    let cancel_token = CancellationToken::new();
    let cancel_clone = cancel_token.clone();

    // 启动独立任务监听客户端连接关闭
    let close_listener = async {
        req.when_closed().await;
        cancel_clone.abort();
    };

    // 同时等待查询完成或连接关闭信号
    let query_result = select! {
        res = backend.query(..., cancel_token) => res,
        _ = close_listener => {
            return Err(Error::from("客户端已断开连接"));
        }
    };

    Ok(HttpResponse::Ok().json(query_result))
}

2. 结合异步任务的取消机制 #

当客户端断开连接时，Actix-web会自动取消处理该请求的异步任务。如果你的后端查询支持取消逻辑（比如基于tokio的CancellationToken），可以直接将任务取消与后端查询绑定：

use actix_web::{HttpRequest, Data, HttpResponse, Error};
use tokio::sync::CancellationToken;
use futures::future::{Abortable, AbortHandle};

#[route("/foo")]
async fn get_data(
    req: HttpRequest,
    state: Data<AppState>,
) -> Result<HttpResponse, Error> {
    let cancel_token = CancellationToken::new();
    let (abort_handle, abort_registration) = AbortHandle::new_pair();

    // 将后端查询包装为可中止的Future
    let abortable_query = Abortable::new(
        backend.query(..., cancel_token.clone()),
        abort_registration,
    );

    // 监听连接关闭，触发查询中止
    tokio::spawn(async move {
        req.when_closed().await;
        abort_handle.abort();
        cancel_token.abort();
    });

    match abortable_query.await {
        Ok(result) => Ok(HttpResponse::Ok().json(result)),
        Err(_) => Err(Error::from("因客户端断开，查询已中止")),
    }
}

针对PostgreSQL的优化说明 #

如果使用tokio-postgres库，它原生提供了CancelToken类型，可直接用于终止PostgreSQL端的查询，无需自定义取消令牌：

use tokio_postgres::CancelToken;

// 从数据库客户端获取CancelToken
let cancel_token = client.cancel_token();

// 客户端断开时触发PostgreSQL查询取消
tokio::spawn(async move {
    req.when_closed().await;
    // 忽略取消操作的错误（比如查询已完成）
    cancel_token.cancel().await.ok();
});

注意：PostgreSQL的取消操作需要额外的数据库连接，因此你的连接池需要预留足够的连接资源用于处理这类请求。

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