andwati/telecom-platform
1
0
Fork 0
mirror of https://github.com/andwati/telecom-platform.git synced 2026-11-15 16:35:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: Update rate limiter to use StateInformationMiddleware and QuantaClock

Browse Source 
- Change create_rate_limiter return type from InMemoryState/DefaultClock to StateInformationMiddleware with QuantaClock
- Add StateInformationMiddleware import from governor::middleware
- Add Arc import from std::sync
- Remove explicit key_extractor field from GovernorLayer initialization
This commit is contained in:
nutcas3
2026-04-26 00:44:00 +03:00
parent b4b0966937
commit b6bcad067b
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
apps/charging-engine/src/rate_limit.rs
+3 -2
View File
@@ -6,14 +6,16 @@ use governor::{
clock::DefaultClock, clock::DefaultClock,
state::InMemoryState, state::InMemoryState,
Jitter, Quota, RateLimiter, Jitter, Quota, RateLimiter,
middleware::StateInformationMiddleware,
}; };
use std::num::NonZeroU32; use std::num::NonZeroU32;
use std::sync::Arc;
use tower_governor::{ use tower_governor::{
governor::config::GovernorConfigBuilder, governor::config::GovernorConfigBuilder,
key_extractor::SmartIpKeyExtractor, key_extractor::SmartIpKeyExtractor,
}; };
pub fn create_rate_limiter() -> tower_governor::GovernorLayer<SmartIpKeyExtractor, governor::state::InMemoryState, governor::clock::DefaultClock> { pub fn create_rate_limiter() -> tower_governor::GovernorLayer<SmartIpKeyExtractor, StateInformationMiddleware<governor::clock::QuantaClock>> {
let requests_per_second: u32 = std::env::var("RATE_LIMIT_RPS") let requests_per_second: u32 = std::env::var("RATE_LIMIT_RPS")
.ok() .ok()
.and_then(|s| s.parse().ok()) .and_then(|s| s.parse().ok())
@@ -34,7 +36,6 @@ pub fn create_rate_limiter() -> tower_governor::GovernorLayer<SmartIpKeyExtracto
tower_governor::GovernorLayer { tower_governor::GovernorLayer {
config: Arc::new(config), config: Arc::new(config),
key_extractor: SmartIpKeyExtractor,
} }
} }