jleony/nanoreth
1
0
Fork 0
mirror of https://github.com/hl-archive-node/nanoreth.git synced 2025-12-06 10:59:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: support graceful shutdown with SIGTERM (#2070)

Browse Source
This commit is contained in:
Dan Cline
2023-04-01 02:29:47 -04:00
committed by GitHub
parent 643ee5226c
commit fa62514bf5
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
bin/reth/src/runner.rs
View File

@ -3,6 +3,7 @@
use futures::pin_mut;
use reth_tasks::{TaskExecutor, TaskManager};
use std::future::Future;
use tokio::signal::unix::SignalKind;
use tracing::trace;
/// Used to execute cli commands
@ -116,16 +117,23 @@ where
async fn run_until_ctrl_c<F, E>(fut: F) -> Result<(), E>
where
F: Future<Output = Result<(), E>>,
E: Send + Sync + 'static,
E: Send + Sync + 'static + From<std::io::Error>,
{
let mut stream = tokio::signal::unix::signal(SignalKind::terminate())?;
let sigterm = stream.recv();
let ctrl_c = tokio::signal::ctrl_c();
pin_mut!(ctrl_c, fut);
pin_mut!(sigterm, fut);
tokio::select! {
_ = ctrl_c => {
trace!(target: "reth::cli", "Received ctrl-c");
},
_ = sigterm => {
trace!(target: "reth::cli", "Received SIGTERM");
},
res = fut => res?,
}