carve out limiters submodule

capture/src/capture.rs

@@ -15,8 +15,8 @@ use metrics::counter;
 use time::OffsetDateTime;
 use tracing::instrument;
 
-use crate::billing_limits::QuotaResource;
 use crate::event::{Compression, ProcessingContext};
+use crate::limiters::billing_limits::QuotaResource;
 use crate::prometheus::report_dropped_events;
 use crate::token::validate_token;
 use crate::{

capture/src/lib.rs

@@ -1,10 +1,9 @@
 pub mod api;
-pub mod billing_limits;
 pub mod capture;
 pub mod config;
 pub mod event;
 pub mod health;
-pub mod partition_limits;
+pub mod limiters;
 pub mod prometheus;
 pub mod redis;
 pub mod router;

capture/src/billing_limits.rs → capture/src/limiters/billing_limits.rs

@@ -166,7 +166,7 @@ mod tests {
     use time::Duration;
 
     use crate::{
-        billing_limits::{BillingLimiter, QuotaResource},
+        limiters::billing_limits::{BillingLimiter, QuotaResource},
         redis::MockRedisClient,
     };
 

capture/src/limiters/mod.rs

@@ -0,0 +1,2 @@
+pub mod billing_limits;
+pub mod partition_limits;

capture/src/partition_limits.rs → capture/src/limiters/partition_limits.rs

@@ -53,7 +53,7 @@ impl PartitionLimiter {
 
 #[cfg(test)]
 mod tests {
-    use crate::partition_limits::PartitionLimiter;
+    use crate::limiters::partition_limits::PartitionLimiter;
     use std::num::NonZeroU32;
 
     #[tokio::test]

capture/src/router.rs

@@ -10,7 +10,9 @@ use tower_http::cors::{AllowHeaders, AllowOrigin, CorsLayer};
 use tower_http::trace::TraceLayer;
 
 use crate::health::HealthRegistry;
-use crate::{billing_limits::BillingLimiter, capture, redis::Client, sink, time::TimeSource};
+use crate::{
+    capture, limiters::billing_limits::BillingLimiter, redis::Client, sink, time::TimeSource,
+};
 
 use crate::prometheus::{setup_metrics_recorder, track_metrics};
 

capture/src/server.rs

@@ -4,10 +4,10 @@ use std::sync::Arc;
 
 use time::Duration;
 
-use crate::billing_limits::BillingLimiter;
 use crate::config::Config;
 use crate::health::{ComponentStatus, HealthRegistry};
-use crate::partition_limits::PartitionLimiter;
+use crate::limiters::billing_limits::BillingLimiter;
+use crate::limiters::partition_limits::PartitionLimiter;
 use crate::redis::RedisClient;
 use crate::{router, sink};
 

capture/src/sink.rs

@@ -15,7 +15,7 @@ use crate::api::CaptureError;
 use crate::config::KafkaConfig;
 use crate::event::ProcessedEvent;
 use crate::health::HealthHandle;
-use crate::partition_limits::PartitionLimiter;
+use crate::limiters::partition_limits::PartitionLimiter;
 use crate::prometheus::report_dropped_events;
 
 #[async_trait]
@@ -294,7 +294,7 @@ mod tests {
     use crate::config;
     use crate::event::ProcessedEvent;
     use crate::health::HealthRegistry;
-    use crate::partition_limits::PartitionLimiter;
+    use crate::limiters::partition_limits::PartitionLimiter;
     use crate::sink::{EventSink, KafkaSink};
     use crate::utils::uuid_v7;
     use rand::distributions::Alphanumeric;

capture/tests/django_compat.rs

@@ -5,9 +5,9 @@ use axum_test_helper::TestClient;
 use base64::engine::general_purpose;
 use base64::Engine;
 use capture::api::{CaptureError, CaptureResponse, CaptureResponseCode};
-use capture::billing_limits::BillingLimiter;
 use capture::event::ProcessedEvent;
 use capture::health::HealthRegistry;
+use capture::limiters::billing_limits::BillingLimiter;
 use capture::redis::MockRedisClient;
 use capture::router::router;
 use capture::sink::EventSink;