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

refactor(redis): add redis commands required for success rate based routing (#5610)

Browse Source
This commit is contained in:
Chethan Rao
2024-08-14 18:22:33 +05:30
committed by GitHub
parent 27096e71e4
commit 340714ce1b
2 changed files with 73 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
crates/redis_interface/src/commands.rs
View File

@ -15,7 +15,7 @@ use common_utils::{
};
use error_stack::{report, ResultExt};
use fred::{
interfaces::{HashesInterface, KeysInterface, SetsInterface, StreamsInterface},
interfaces::{HashesInterface, KeysInterface, ListInterface, SetsInterface, StreamsInterface},
prelude::RedisErrorKind,
types::{
Expiration, FromRedis, MultipleIDs, MultipleKeys, MultipleOrderedPairs, MultipleStrings,
@ -371,6 +371,19 @@ impl super::RedisConnectionPool {
Ok(hsetnx)
}
#[instrument(level = "DEBUG", skip(self))]
pub async fn increment_field_in_hash(
&self,
key: &str,
field: &str,
increment: i64,
) -> CustomResult<usize, errors::RedisError> {
self.pool
.hincrby(self.add_prefix(key), field, increment)
.await
.change_context(errors::RedisError::IncrementHashFieldFailed)
}
#[instrument(level = "DEBUG", skip(self))]
pub async fn hscan(
&self,
@ -630,6 +643,55 @@ impl super::RedisConnectionPool {
})
}
#[instrument(level = "DEBUG", skip(self))]
pub async fn append_elements_to_list<V>(
&self,
key: &str,
elements: V,
) -> CustomResult<(), errors::RedisError>
where
V: TryInto<MultipleValues> + Debug + Send,
V::Error: Into<fred::error::RedisError> + Send,
{
self.pool
.rpush(self.add_prefix(key), elements)
.await
.change_context(errors::RedisError::AppendElementsToListFailed)
}
#[instrument(level = "DEBUG", skip(self))]
pub async fn get_list_elements(
&self,
key: &str,
start: i64,
stop: i64,
) -> CustomResult<Vec<String>, errors::RedisError> {
self.pool
.lrange(self.add_prefix(key), start, stop)
.await
.change_context(errors::RedisError::GetListElementsFailed)
}
#[instrument(level = "DEBUG", skip(self))]
pub async fn get_list_length(&self, key: &str) -> CustomResult<usize, errors::RedisError> {
self.pool
.llen(self.add_prefix(key))
.await
.change_context(errors::RedisError::GetListLengthFailed)
}
#[instrument(level = "DEBUG", skip(self))]
pub async fn lpop_list_elements(
&self,
key: &str,
count: Option<usize>,
) -> CustomResult<Vec<String>, errors::RedisError> {
self.pool
.lpop(self.add_prefix(key), count)
.await
.change_context(errors::RedisError::PopListElementsFailed)
}
// Consumer Group API
#[instrument(level = "DEBUG", skip(self))]