Remove default hot pool size cap

This commit is contained in:
fawney19
2026-05-28 17:01:03 +08:00
parent 47886abd2b
commit ef2a96bcc4
2 changed files with 7 additions and 2 deletions
@@ -36,7 +36,7 @@ use super::pool_score_rebuild::ensure_provider_key_pool_scores_for_keys;
const POOL_QUOTA_PROBE_REDIS_PREFIX: &str = "ap:quota_probe:last";
const POOL_QUOTA_PROBE_DEFAULT_SCAN_INTERVAL_SECONDS: u64 = 60;
const POOL_QUOTA_PROBE_MIN_SCAN_INTERVAL_SECONDS: u64 = 15;
const POOL_QUOTA_PROBE_DEFAULT_MAX_KEYS_PER_PROVIDER: usize = 50;
const POOL_QUOTA_PROBE_UNLIMITED_MAX_KEYS_PER_PROVIDER: usize = usize::MAX;
const POOL_QUOTA_PROBE_PROVIDER_SCORE_READ_LIMIT: usize = 100_000;
const POOL_QUOTA_PROBE_DEFAULT_GLOBAL_CONCURRENCY: usize = 16;
const POOL_QUOTA_PROBE_PROVIDER_LOCK_TTL_MS: u64 = 30_000;
@@ -106,7 +106,7 @@ impl PoolQuotaProbeWorkerConfig {
.max(POOL_QUOTA_PROBE_MIN_SCAN_INTERVAL_SECONDS);
let max_keys_per_provider = env_usize(
"POOL_QUOTA_PROBE_MAX_KEYS_PER_PROVIDER",
POOL_QUOTA_PROBE_DEFAULT_MAX_KEYS_PER_PROVIDER,
POOL_QUOTA_PROBE_UNLIMITED_MAX_KEYS_PER_PROVIDER,
);
let global_concurrency = env_usize(
"POOL_QUOTA_PROBE_GLOBAL_CONCURRENCY",
@@ -413,4 +413,9 @@ mod tests {
assert!(low.desired_hot >= PROVIDER_POOL_DEMAND_FLOOR);
assert!(low.desired_hot < high.desired_hot);
}
#[test]
fn desired_hot_without_provider_limit_is_capped_by_active_keys() {
assert_eq!(provider_pool_desired_hot(100, 100.0, 37, usize::MAX), 37);
}
}