From ab909d3e07e3ef2182d837f308b2b2fc75ea1c52 Mon Sep 17 00:00:00 2001 From: Trezy Date: Wed, 18 Mar 2026 16:47:55 -0500 Subject: [PATCH] fix: update types for rate limits --- src/admin/rate_limits.rs | 4 ++-- src/admin/types.rs | 2 +- src/rate_limit.rs | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/admin/rate_limits.rs b/src/admin/rate_limits.rs index 5019d94..5eeca35 100644 --- a/src/admin/rate_limits.rs +++ b/src/admin/rate_limits.rs @@ -36,7 +36,7 @@ pub(super) async fn list( "SELECT capacity, refill_rate, default_query_cost, default_procedure_cost, default_proxy_cost FROM rate_limits WHERE method IS NULL", backend, ); - let row: Option<(i32, f32, i32, i32, i32)> = sqlx::query_as(&limits_sql) + let row: Option<(i32, f64, i32, i32, i32)> = sqlx::query_as(&limits_sql) .fetch_optional(&state.db) .await .map_err(|e| AppError::Internal(format!("failed to read rate limits: {e}")))?; @@ -100,7 +100,7 @@ pub(super) async fn upsert( ); sqlx::query(&sql) .bind(body.capacity as i32) - .bind(body.refill_rate as f32) + .bind(body.refill_rate) .bind(body.default_query_cost as i32) .bind(body.default_procedure_cost as i32) .bind(body.default_proxy_cost as i32) diff --git a/src/admin/types.rs b/src/admin/types.rs index b1d3a51..426a77b 100644 --- a/src/admin/types.rs +++ b/src/admin/types.rs @@ -239,7 +239,7 @@ pub(super) struct AddAllowlistBody { pub(super) struct RateLimitsResponse { pub(super) enabled: bool, pub(super) capacity: i32, - pub(super) refill_rate: f32, + pub(super) refill_rate: f64, pub(super) default_query_cost: i32, pub(super) default_procedure_cost: i32, pub(super) default_proxy_cost: i32, diff --git a/src/rate_limit.rs b/src/rate_limit.rs index b633606..1990ba5 100644 --- a/src/rate_limit.rs +++ b/src/rate_limit.rs @@ -184,7 +184,7 @@ impl RateLimiter { .unwrap_or(true); // Load global rate limit config (method IS NULL row) - let row: Option<(i32, f32, i32, i32, i32)> = sqlx::query_as( + let row: Option<(i32, f64, i32, i32, i32)> = sqlx::query_as( "SELECT capacity, refill_rate, default_query_cost, default_procedure_cost, default_proxy_cost FROM rate_limits WHERE method IS NULL", ) .fetch_optional(db) @@ -195,7 +195,7 @@ impl RateLimiter { Some((capacity, refill_rate, query_cost, procedure_cost, proxy_cost)) => { RateLimitConfig { capacity: capacity as u32, - refill_rate: refill_rate as f64, + refill_rate, default_query_cost: query_cost as u32, default_procedure_cost: procedure_cost as u32, default_proxy_cost: proxy_cost as u32, -- 2.51.2