feat: update gateway pool and usage flows

apps/aether-gateway/src/handlers/admin/provider/pool/runtime/keys.rs

@@ -14,10 +14,6 @@ pub(super) fn pool_cooldown_key(provider_id: &str, key_id: &str) -> String {
     format!("ap:{provider_id}:cooldown:{key_id}")
 }
 
-pub(super) fn pool_lease_key(provider_id: &str, key_id: &str) -> String {
-    format!("ap:{provider_id}:lease:{key_id}")
-}
-
 pub(super) fn pool_cooldown_index_key(provider_id: &str) -> String {
     format!("ap:{provider_id}:cooldown_idx")
 }

apps/aether-gateway/src/handlers/admin/provider/pool/runtime/leases.rs

@@ -1,23 +1,4 @@
-use super::keys::pool_lease_key;
 use aether_runtime_state::{DataLayerError, RuntimeLockLease, RuntimeState};
-use std::time::Duration;
-
-pub(crate) const ADMIN_PROVIDER_POOL_KEY_LEASE_TTL_MS: u64 = 15 * 60 * 1000;
-
-pub(crate) async fn try_claim_admin_provider_pool_key(
-    runtime: &RuntimeState,
-    provider_id: &str,
-    key_id: &str,
-    owner: &str,
-) -> Result<Option<RuntimeLockLease>, DataLayerError> {
-    runtime
-        .lock_try_acquire(
-            &pool_lease_key(provider_id, key_id),
-            owner,
-            Duration::from_millis(ADMIN_PROVIDER_POOL_KEY_LEASE_TTL_MS),
-        )
-        .await
-}
 
 pub(crate) async fn release_admin_provider_pool_key_lease(
     runtime: &RuntimeState,

apps/aether-gateway/src/handlers/admin/provider/pool/runtime/mod.rs

@@ -5,16 +5,14 @@ mod reads;
 mod status;
 mod writes;
 
-pub(crate) use self::leases::{
-    release_admin_provider_pool_key_lease, try_claim_admin_provider_pool_key,
-    ADMIN_PROVIDER_POOL_KEY_LEASE_TTL_MS,
-};
+pub(crate) use self::leases::release_admin_provider_pool_key_lease;
 pub(crate) use self::mutations::{
     clear_admin_provider_pool_cooldown, reset_admin_provider_pool_cost,
 };
 pub(crate) use self::reads::{
     read_admin_provider_pool_cooldown_count, read_admin_provider_pool_cooldown_counts,
-    read_admin_provider_pool_cooldown_key_ids, read_admin_provider_pool_runtime_state,
+    read_admin_provider_pool_cooldown_key_ids, read_admin_provider_pool_key_cooldown_reason,
+    read_admin_provider_pool_runtime_state,
 };
 pub(crate) use self::status::build_admin_provider_pool_status_payload;
 pub(crate) use self::writes::{

apps/aether-gateway/src/handlers/admin/provider/pool/runtime/reads.rs

@@ -7,7 +7,7 @@ use crate::handlers::admin::provider::pool::config::admin_provider_pool_cache_af
 use crate::handlers::admin::provider::shared::support::{
     AdminProviderPoolConfig, AdminProviderPoolRuntimeState,
 };
-use aether_runtime_state::RuntimeState;
+use aether_runtime_state::{DataLayerError, RuntimeState};
 use std::collections::BTreeMap;
 use std::time::{SystemTime, UNIX_EPOCH};
 use tracing::warn;
@@ -202,3 +202,13 @@ pub(crate) async fn read_admin_provider_pool_cooldown_key_ids(
         .await
         .unwrap_or_default()
 }
+
+pub(crate) async fn read_admin_provider_pool_key_cooldown_reason(
+    runtime: &RuntimeState,
+    provider_id: &str,
+    key_id: &str,
+) -> Result<Option<String>, DataLayerError> {
+    runtime
+        .kv_get(&pool_cooldown_key(provider_id, key_id))
+        .await
+}