diff --git a/lite/src/backend/store.rs b/lite/src/backend/store.rs index 4ee0d2c2..7e209ac8 100644 --- a/lite/src/backend/store.rs +++ b/lite/src/backend/store.rs @@ -9,8 +9,11 @@ use super::Backend; use crate::backend::{error::StorageError, kv, stream_id::StreamId}; impl Backend { - pub fn db_status(&self) -> Result<(), slatedb::Error> { - self.db.status() + // TODO: switch to `self.db.status()` once slatedb releases with + // https://github.com/slatedb/slatedb/pull/1234 + pub async fn db_status(&self) -> Result<(), slatedb::Error> { + let _ = self.db.get(b"ping").await?; + Ok(()) } pub(super) async fn db_get + Send, V>( diff --git a/lite/src/handlers/mod.rs b/lite/src/handlers/mod.rs index 88531ea5..bcedc406 100644 --- a/lite/src/handlers/mod.rs +++ b/lite/src/handlers/mod.rs @@ -17,7 +17,7 @@ pub fn router() -> axum::Router { } async fn health(State(backend): State) -> Response { - match backend.db_status() { + match backend.db_status().await { Ok(()) => "OK".into_response(), Err(err) => (StatusCode::SERVICE_UNAVAILABLE, format!("{err:?}")).into_response(), }