rust/hyperswitch
1
0
Fork 0
mirror of https://github.com/juspay/hyperswitch.git synced 2025-10-30 09:38:33 +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

7
crates/drainer/src/utils.rs
View File

@ -124,8 +124,13 @@ pub fn parse_stream_entries<'a>(
// Here the output is in the format (stream_index, jobs_picked),
// similar to the first argument of the function
pub fn increment_stream_index((index, jobs_picked): (u8, u8), total_streams: u8) -> (u8, u8) {
pub async fn increment_stream_index(
(index, jobs_picked): (u8, u8),
total_streams: u8,
interval: &mut tokio::time::Interval,
) -> (u8, u8) {
if index == total_streams - 1 {
interval.tick().await;
match jobs_picked {
0 => metrics::CYCLES_COMPLETED_UNSUCCESSFULLY.add(&metrics::CONTEXT, 1, &[]),
_ => metrics::CYCLES_COMPLETED_SUCCESSFULLY.add(&metrics::CONTEXT, 1, &[]),