有办法关闭tokio::runtime::Runtime吗？

Question

问题在于使用Tokio的微服务，其中连接到db和其他组件异步，但当某些连接失败时，微服务不会结束工作。当你需要这个案子的时候真是太棒了，但是当连接丢失的时候我需要结束这个微服务的工作.那么你能帮我如何安全关闭程序吗？

src/main.rs

use tokio; // 1.0.0+

fn main() {
    let rt = tokio::runtime::Builder::new_multi_thread()
        .worker_threads(workers_number)
        .enable_all()
        .build()
        .unwrap();
    rt.block_on(async {
    // health cheacker connection
    let health_checker = match HealthChecker::new(some_configuration).await?;
    // some connection to db
    // ...
    // transport client connection
    // ...

    // so when connection failed or lost I need to 
    // end process like `std::process::abort()`
    // but I cant use it, because its unsafe

    let mut task_handler = vec![];

    // create some task

    join_all(task_handler).await;
  });
}

有人有什么想法吗？

Answer 1

您可以调用任何Runtime关闭方法shutdown_timeout或shutdown_background。

如果需要等待，您可以使用tokio::sync::oneshot生成一个任务，在发出信号时触发关机。

use core::time::Duration;
use crate::tokio::time::sleep;
use tokio;

fn main() {
    let rt = tokio::runtime::Builder::new_multi_thread()
        .enable_all()
        .build()
        .unwrap();
        
    let handle = rt.handle().clone();
    
    let (s, r) = tokio::sync::oneshot::channel();
    
    rt.spawn(async move {
        sleep(Duration::from_secs(1)).await;
        s.send(0);
    });
    
    handle.block_on(async move {
        r.await;
        rt.shutdown_background();
    });
}

游乐场