Skip to content

Commit cd9d0c6

Browse files
committed
refactor: simplify rate limiter to always use SmartIpKeyExtractor
1 parent e55fdcf commit cd9d0c6

1 file changed

Lines changed: 2 additions & 14 deletions

File tree

backend/src/main.rs

Lines changed: 2 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -259,18 +259,12 @@ async fn main() {
259259
tracing::info!("Proxy headers will be stripped before rate limiting to prevent spoofing");
260260
}
261261

262-
let login_key_extractor = if trust_proxy_ip_headers {
263-
SmartIpKeyExtractor
264-
} else {
265-
PeerIpKeyExtractor
266-
};
267-
268262
// Configure rate limiting (average 1 request/sec for login, burst up to 5)
269263
let rate_limit_config = std::sync::Arc::new(
270264
GovernorConfigBuilder::default()
271265
.per_second(1)
272266
.burst_size(5)
273-
.key_extractor(login_key_extractor)
267+
.key_extractor(SmartIpKeyExtractor)
274268
.finish()
275269
.expect("Failed to build governor config"),
276270
);
@@ -283,17 +277,11 @@ async fn main() {
283277
.layer(RequestBodyLimitLayer::new(LOGIN_BODY_LIMIT))
284278
.layer(GovernorLayer::new(rate_limit_config));
285279

286-
let admin_key_extractor = if trust_proxy_ip_headers {
287-
SmartIpKeyExtractor
288-
} else {
289-
PeerIpKeyExtractor
290-
};
291-
292280
let admin_rate_limit_config = std::sync::Arc::new(
293281
GovernorConfigBuilder::default()
294282
.per_second(1)
295283
.burst_size(3)
296-
.key_extractor(admin_key_extractor)
284+
.key_extractor(SmartIpKeyExtractor)
297285
.finish()
298286
.expect("Failed to build governor config for write routes"),
299287
);

0 commit comments

Comments
 (0)