andwati/telecom-platform
1
0
Fork 0
mirror of https://github.com/andwati/telecom-platform.git synced 2026-10-23 21:22:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Add configurable expiration times for credit and block operations via environment variables

Browse Source 
- Add CREDIT_EXPIRATION_SECONDS environment variable to configure credit key expiration (defaults to 86400)
- Add BLOCK_EXPIRATION_SECONDS environment variable to configure block key expiration (defaults to 86400)
- Replace hardcoded 86400 expiration values with configurable variables in add_credit and block_user
- Update block_user warning message to include expiration duration
This commit is contained in:
nutcas3
2026-04-25 03:06:37 +03:00
parent ff33263df3
commit 2dd6c3304d
1 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
apps/charging-engine/src/charging/credit_management.rs
+13 -5
View File
@@ -24,8 +24,12 @@ impl crate::charging::ChargingEngine {
let new_balance: u64 = redis::AsyncCommands::incr(&mut conn, &key, bytes_to_add).await
.map_err(|e| crate::errors::ChargingError::RedisOperation(e.to_string()))?;
// Set expiration if this is a new key
let _: () = redis::AsyncCommands::expire(&mut conn, &key, 86400).await.unwrap_or(());
// Set expiration if this is a new key (configurable via CREDIT_EXPIRATION_SECONDS, defaults to 24 hours)
let credit_expiration: u64 = std::env::var("CREDIT_EXPIRATION_SECONDS")
.ok()
.and_then(|s| s.parse().ok())
.unwrap_or(86400);
let _: () = redis::AsyncCommands::expire(&mut conn, &key, credit_expiration as i64).await.unwrap_or(());
info!("Added {} bytes credit to IP: {}, new balance: {}", bytes_to_add, ip, new_balance);
Ok(new_balance)
@@ -75,10 +79,14 @@ impl crate::charging::ChargingEngine {
let _: () = redis::AsyncCommands::set(&mut conn, &key, "blocked").await
.map_err(|e| crate::errors::ChargingError::RedisOperation(e.to_string()))?;
// Set expiration for block (24 hours)
let _: () = redis::AsyncCommands::expire(&mut conn, &key, 86400).await.unwrap_or(());
// Set expiration for block (configurable via BLOCK_EXPIRATION_SECONDS, defaults to 24 hours)
let block_expiration: u64 = std::env::var("BLOCK_EXPIRATION_SECONDS")
.ok()
.and_then(|s| s.parse().ok())
.unwrap_or(86400);
let _: () = redis::AsyncCommands::expire(&mut conn, &key, block_expiration as i64).await.unwrap_or(());
warn!("User IP: {} has been blocked", ip);
warn!("User IP: {} has been blocked for {} seconds", ip, block_expiration);
Ok(())
}