refactor: lazy pool key scheduling

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

@@ -14,6 +14,10 @@ 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

@@ -0,0 +1,27 @@
+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,
+    lease: &RuntimeLockLease,
+) -> Result<bool, DataLayerError> {
+    runtime.lock_release(lease).await
+}

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

@@ -1,9 +1,14 @@
 mod keys;
+mod leases;
 mod mutations;
 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::mutations::{
     clear_admin_provider_pool_cooldown, reset_admin_provider_pool_cost,
 };