Improve gateway transport and usage runtime

apps/aether-gateway/src/state/runtime/billing/admin.rs

@@ -514,4 +514,23 @@ impl AppState {
             .await
             .map_err(data_error)
     }
+
+    pub(crate) async fn find_user_daily_quota_availability_for_auth(
+        &self,
+        user_id: &str,
+    ) -> Result<Option<UserDailyQuotaAvailabilityRecord>, GatewayError> {
+        let user_id = user_id.trim();
+        if user_id.is_empty() {
+            return Ok(None);
+        }
+        let ttl = self.frontdoor_runtime_guards.auth_capacity_cache_ttl;
+        if ttl.is_zero() {
+            return self.find_user_daily_quota_availability(user_id).await;
+        }
+        self.auth_daily_quota_availability_cache
+            .get_or_load(user_id.to_string(), ttl, || async move {
+                self.find_user_daily_quota_availability(user_id).await
+            })
+            .await
+    }
 }

apps/aether-gateway/src/state/runtime/wallet/reads.rs

@@ -42,19 +42,70 @@ impl AppState {
         api_key_id: &str,
         api_key_is_standalone: bool,
     ) -> Result<Option<aether_data::repository::wallet::StoredWalletSnapshot>, GatewayError> {
+        let user_id = user_id.trim();
+        let api_key_id = api_key_id.trim();
         let lookup = if api_key_is_standalone {
-            if api_key_id.trim().is_empty() {
+            if api_key_id.is_empty() {
+                None
+            } else {
+                Some((
+                    format!("api_key:{api_key_id}"),
+                    aether_data::repository::wallet::WalletLookupKey::ApiKeyId(api_key_id),
+                ))
+            }
+        } else if !user_id.is_empty() {
+            Some((
+                format!("user:{user_id}"),
+                aether_data::repository::wallet::WalletLookupKey::UserId(user_id),
+            ))
+        } else if !api_key_id.is_empty() {
+            Some((
+                format!("api_key:{api_key_id}"),
+                aether_data::repository::wallet::WalletLookupKey::ApiKeyId(api_key_id),
+            ))
+        } else {
+            None
+        };
+
+        let Some((cache_key, lookup)) = lookup else {
+            return Ok(None);
+        };
+
+        let ttl = self.frontdoor_runtime_guards.auth_capacity_cache_ttl;
+        if ttl.is_zero() {
+            return self.find_wallet(lookup).await;
+        }
+
+        self.auth_wallet_snapshot_cache
+            .get_or_load(
+                cache_key,
+                ttl,
+                || async move { self.find_wallet(lookup).await },
+            )
+            .await
+    }
+
+    pub(crate) async fn read_wallet_snapshot_for_auth_uncached(
+        &self,
+        user_id: &str,
+        api_key_id: &str,
+        api_key_is_standalone: bool,
+    ) -> Result<Option<aether_data::repository::wallet::StoredWalletSnapshot>, GatewayError> {
+        let user_id = user_id.trim();
+        let api_key_id = api_key_id.trim();
+        let lookup = if api_key_is_standalone {
+            if api_key_id.is_empty() {
                 None
             } else {
                 Some(aether_data::repository::wallet::WalletLookupKey::ApiKeyId(
                     api_key_id,
                 ))
             }
-        } else if !user_id.trim().is_empty() {
+        } else if !user_id.is_empty() {
             Some(aether_data::repository::wallet::WalletLookupKey::UserId(
                 user_id,
             ))
-        } else if !api_key_id.trim().is_empty() {
+        } else if !api_key_id.is_empty() {
             Some(aether_data::repository::wallet::WalletLookupKey::ApiKeyId(
                 api_key_id,
             ))