|
6 | 6 | //! - Transaction status checks |
7 | 7 | //! - Notifications |
8 | 8 | //! - Solana swap requests |
9 | | -use std::sync::Arc; |
| 9 | +use std::{env, sync::Arc}; |
10 | 10 |
|
11 | 11 | use apalis_redis::{Config, ConnectionManager, RedisStorage}; |
12 | 12 | use color_eyre::{eyre, Result}; |
@@ -58,19 +58,35 @@ impl Queue { |
58 | 58 | }; |
59 | 59 |
|
60 | 60 | 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(); |
61 | 67 | 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?, |
64 | 73 | 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), |
66 | 85 | shared.clone(), |
67 | 86 | ) |
68 | 87 | .await?, |
69 | | - transaction_status_queue: Self::storage("transaction_status_queue", shared.clone()) |
70 | | - .await?, |
71 | | - notification_queue: Self::storage("notification_queue", shared.clone()).await?, |
72 | 88 | solana_token_swap_request_queue: Self::storage( |
73 | | - "solana_token_swap_request_queue", |
| 89 | + &format!("{}solana_token_swap_request_queue", redis_key_prefix), |
74 | 90 | shared.clone(), |
75 | 91 | ) |
76 | 92 | .await?, |
|
0 commit comments