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

fix(drainer): increase jobs picked only when stream is not empty (#2958)

Browse Source
This commit is contained in:
Kartikeya Hegde
2023-11-23 17:22:20 +05:30
committed by GitHub
parent 35a44ed253
commit 42eedf3a8c
3 changed files with 28 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,4 +1,7 @@
use std::{collections::HashMap, sync::Arc};
use std::{
collections::HashMap,
sync::{atomic, Arc},
};
use error_stack::IntoReport;
use redis_interface as redis;
@ -127,19 +130,20 @@ 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 async fn increment_stream_index(
(index, jobs_picked): (u8, u8),
(index, jobs_picked): (u8, Arc<atomic::AtomicU8>),
total_streams: u8,
interval: &mut tokio::time::Interval,
) -> (u8, u8) {
) -> u8 {
if index == total_streams - 1 {
interval.tick().await;
match jobs_picked {
match jobs_picked.load(atomic::Ordering::SeqCst) {
0 => metrics::CYCLES_COMPLETED_UNSUCCESSFULLY.add(&metrics::CONTEXT, 1, &[]),
_ => metrics::CYCLES_COMPLETED_SUCCESSFULLY.add(&metrics::CONTEXT, 1, &[]),
}
(0, 0)
jobs_picked.store(0, atomic::Ordering::SeqCst);
0
} else {
(index + 1, jobs_picked)
index + 1
}
}