Skip to content

Commit 971ab9b

Browse files
zeljkoXtirumerla
andauthored
feat: Add background job redis collections prefix support (#484)
* feat: Add background job redis collections prefix support * fix: Handle empty strings --------- Co-authored-by: tirumerla <[email protected]>
1 parent 9886d84 commit 971ab9b

File tree

1 file changed

+24
-8
lines changed

1 file changed

+24
-8
lines changed

src/jobs/queue.rs

Lines changed: 24 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
//! - Transaction status checks
77
//! - Notifications
88
//! - Solana swap requests
9-
use std::sync::Arc;
9+
use std::{env, sync::Arc};
1010

1111
use apalis_redis::{Config, ConnectionManager, RedisStorage};
1212
use color_eyre::{eyre, Result};
@@ -58,19 +58,35 @@ impl Queue {
5858
};
5959

6060
let shared = Arc::new(conn);
61+
// use REDIS_KEY_PREFIX only if set, otherwise do not use it
62+
let redis_key_prefix = env::var("REDIS_KEY_PREFIX")
63+
.ok()
64+
.filter(|v| !v.is_empty())
65+
.map(|value| format!("{value}:queue:"))
66+
.unwrap_or_default();
6167
Ok(Self {
62-
transaction_request_queue: Self::storage("transaction_request_queue", shared.clone())
63-
.await?,
68+
transaction_request_queue: Self::storage(
69+
&format!("{}transaction_request_queue", redis_key_prefix),
70+
shared.clone(),
71+
)
72+
.await?,
6473
transaction_submission_queue: Self::storage(
65-
"transaction_submission_queue",
74+
&format!("{}transaction_submission_queue", redis_key_prefix),
75+
shared.clone(),
76+
)
77+
.await?,
78+
transaction_status_queue: Self::storage(
79+
&format!("{}transaction_status_queue", redis_key_prefix),
80+
shared.clone(),
81+
)
82+
.await?,
83+
notification_queue: Self::storage(
84+
&format!("{}notification_queue", redis_key_prefix),
6685
shared.clone(),
6786
)
6887
.await?,
69-
transaction_status_queue: Self::storage("transaction_status_queue", shared.clone())
70-
.await?,
71-
notification_queue: Self::storage("notification_queue", shared.clone()).await?,
7288
solana_token_swap_request_queue: Self::storage(
73-
"solana_token_swap_request_queue",
89+
&format!("{}solana_token_swap_request_queue", redis_key_prefix),
7490
shared.clone(),
7591
)
7692
.await?,

0 commit comments

Comments
 (0)