rust/hyperswitch
1
0
Fork 0
mirror of https://github.com/juspay/hyperswitch.git synced 2025-11-03 21:37:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add tick interval to drainer (#517)

Browse Source
This commit is contained in:
Nishant Joshi
2023-02-07 20:20:51 +05:30
committed by GitHub
parent ac30313ff1
commit 3292960996
7 changed files with 25 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
crates/drainer/src/lib.rs
View File

@ -19,12 +19,15 @@ pub async fn start_drainer(
number_of_streams: u8,
max_read_count: u64,
shutdown_interval: u32,
loop_interval: u32,
) -> errors::DrainerResult<()> {
let mut stream_index: u8 = 0;
let mut jobs_picked: u8 = 0;
let mut shutdown_interval =
tokio::time::interval(std::time::Duration::from_millis(shutdown_interval.into()));
let mut loop_interval =
tokio::time::interval(std::time::Duration::from_millis(loop_interval.into()));
let signal =
get_allowed_signals()
@ -49,8 +52,12 @@ pub async fn start_drainer(
));
jobs_picked += 1;
}
(stream_index, jobs_picked) =
utils::increment_stream_index((stream_index, jobs_picked), number_of_streams);
(stream_index, jobs_picked) = utils::increment_stream_index(
(stream_index, jobs_picked),
number_of_streams,
&mut loop_interval,
)
.await;
}
Ok(()) | Err(oneshot::error::TryRecvError::Closed) => {
logger::info!("Awaiting shutdown!");