feat: Add graceful shutdown in drainer (#498)

This commit is contained in:
Nishant Joshi
2023-02-07 12:44:24 +05:30
committed by GitHub
parent dfd1e5e254
commit 0239c422b6
11 changed files with 138 additions and 26 deletions

View File

@ -18,12 +18,19 @@ async fn main() -> DrainerResult<()> {
let number_of_streams = store.config.drainer_num_partitions;
let max_read_count = conf.drainer.max_read_count;
let shutdown_intervals = conf.drainer.shutdown_interval;
let _guard = logger::setup(&conf.log).change_context(errors::DrainerError::MetricsError)?;
logger::info!("Drainer started [{:?}] [{:?}]", conf.drainer, conf.log);
start_drainer(store.clone(), number_of_streams, max_read_count).await?;
start_drainer(
store.clone(),
number_of_streams,
max_read_count,
shutdown_intervals,
)
.await?;
store.close().await;
Ok(())