Rust中接收GET /请求时关闭Warp Web服务器问题排查

我正在学习Rust，想要实现的功能是：收到访问GET /的请求时，终止或关闭运行的Web服务器。
不清楚是Warp框架本身不支持该操作，还是代码实现存在问题。当前编写的代码完全无法响应任何HTTP请求，代码如下：

pub async fn perform_oauth_flow(&self) {
        let (tx, rx) = channel::unbounded();

        let routes = warp::path::end().map(move || {
            println!("handling");
            tx.send("kill");
            Ok(warp::reply::with_status("OK", http::StatusCode::CREATED))
        });

        println!("Spawning server");
        let webserver_thread = thread::spawn(|| async {
            spawn(warp::serve(routes).bind(([127, 0, 0, 1], 3000)))
                .await
                .unwrap();
        });

        println!("waiting for result");
        let result = rx.recv().unwrap();
        println!("Got result");
        if result == "kill" {
            webserver_thread.join().unwrap().await;
        }
    }

你的代码有两个核心错误，导致服务根本没启动，自然无法响应任何请求：

1. 异步运行时使用完全错误：std::thread::spawn创建的系统线程只能执行同步代码，你传入的闭包返回了一个async块，但没有任何运行时去轮询（poll）这个Future，相当于线程启动后只创建了一个Future对象就直接结束了，Warp服务从头到尾都没跑起来。
2. 没有使用Warp原生提供的关停接口：你自己实现的通道+线程join的关停逻辑本身就不成立，就算服务跑起来了，服务进程没有收到终止信号的话，会一直阻塞监听端口，join操作会永久卡住。

Warp本身基于Tokio构建，原生支持优雅关停，不需要自己额外实现复杂的线程、通道逻辑。

直接使用Warp提供的bind_with_graceful_shutdown接口，配合Tokio的单触发通道传递关停信号即可，不需要额外开系统线程：

use warp::http::StatusCode;
use tokio::sync::oneshot;

// 该函数需要在Tokio运行时上下文中调用，比如用#[tokio::main]标注程序入口
pub async fn perform_oauth_flow() {
    // oneshot通道专门用于单次通知场景，比unbounded通道更轻量，适合传递关停信号
    let (shutdown_tx, shutdown_rx) = oneshot::channel();

    let routes = warp::path::end().map(move || {
        println!("收到根路径请求，准备关停服务");
        // 发送关停信号，忽略接收端已丢弃的边缘错误
        let _ = shutdown_tx.send(());
        warp::reply::with_status("OK，服务即将关闭", StatusCode::CREATED)
    });

    println!("启动Web服务，监听127.0.0.1:3000");
    // 绑定服务同时注册关停监听逻辑
    let (listen_addr, server) = warp::serve(routes)
        .bind_with_graceful_shutdown(
            ([127, 0, 0, 1], 3000),
            async move {
                shutdown_rx.await.ok();
                println!("收到关停信号，开始停止服务");
            }
        );
    println!("服务已启动，访问地址：http://{}", listen_addr);

    // 等待服务运行，直到关停完成
    server.await;
    println!("服务已完全退出");
}

• 如果你确实有特殊需求要把Warp服务跑在单独的系统线程里，必须在线程内手动启动Tokio运行时，不能直接把async块扔给std::thread::spawn，示例写法：

let webserver_thread = std::thread::spawn(|| {
    // 手动在新线程创建Tokio运行时
    let rt = tokio::runtime::Builder::new_multi_thread()
        .enable_all()
        .build()
        .unwrap();
    // 阻塞运行服务逻辑
    rt.block_on(async {
        // 这里放上面的warp服务启动、路由定义逻辑即可
    })
});
// 后续等待线程结束即可
webserver_thread.join().unwrap();

• 不要混用同步线程接口和异步Future，所有异步逻辑必须依托对应的异步运行时才能执行，这是Rust异步编程最基础的规则。

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