feat: add is_ready method to Cache trait and update ready endpoint

Co-Authored-By: Tejas Badadare <[email protected]>

Author: devin-ai-integration[bot]

apps/hermes/server/src/api/rest/ready.rs

@@ -1,11 +1,12 @@
 use {
-    crate::{api::ApiState, state::aggregate::Aggregates},
+    crate::{api::ApiState, state::{aggregate::Aggregates, cache::Cache}},
     axum::{
         extract::State,
         http::StatusCode,
         response::{IntoResponse, Response},
         Json,
     },
+    serde_json::json,
 };
 
 /// Endpoint that returns OK (200) only when the cache is fully hydrated.
@@ -19,11 +20,20 @@ use {
 /// along with detailed metadata about the readiness state.
 pub async fn ready<S>(State(state): State<ApiState<S>>) -> Response
 where
-    S: Aggregates,
+    S: Aggregates + Cache,
 {
     let state = &*state.state;
-    match Aggregates::is_ready(state).await {
-        (true, _) => (StatusCode::OK, "OK").into_response(),
-        (false, metadata) => (StatusCode::SERVICE_UNAVAILABLE, Json(metadata)).into_response(),
+    let (aggregates_ready, metadata) = Aggregates::is_ready(state).await;
+    let cache_ready = Cache::is_ready(state).await;
+    
+    if aggregates_ready && cache_ready {
+        (StatusCode::OK, "OK").into_response()
+    } else {
+        let response_metadata = json!({
+            "aggregates_ready": aggregates_ready,
+            "cache_ready": cache_ready,
+            "details": metadata
+        });
+        (StatusCode::SERVICE_UNAVAILABLE, Json(response_metadata)).into_response()
     }
 }

apps/hermes/server/src/state/cache.rs

@@ -136,6 +136,8 @@ pub trait Cache {
         request_time: RequestTime,
         filter: MessageStateFilter,
     ) -> Result<Vec<MessageState>>;
+    /// 
+    async fn is_ready(&self) -> bool;
 }
 
 #[async_trait::async_trait]
@@ -274,6 +276,12 @@ where
         let cache = self.into().wormhole_merkle_state_cache.read().await;
         Ok(cache.get(&slot).cloned())
     }
+
+    /// 
+    async fn is_ready(&self) -> bool {
+        let message_cache = self.into().message_cache.read().await;
+        !message_cache.is_empty()
+    }
 }
 
 async fn retrieve_message_state(