From 7f4c2289bdf8f10c0589d112ba792086e40346e8 Mon Sep 17 00:00:00 2001
From: zhengpeng <847850277@qq.com>
Date: Thu, 28 Nov 2024 11:15:58 +0800
Subject: [PATCH] [ISSUES#]2721 health_check_provider.

---
 crates/torii/core/src/engine.rs |  3 ++-
 crates/torii/core/src/utils.rs  | 18 ++++++++++++++++--
 2 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/crates/torii/core/src/engine.rs b/crates/torii/core/src/engine.rs
index 86730665f3..5285d71f45 100644
--- a/crates/torii/core/src/engine.rs
+++ b/crates/torii/core/src/engine.rs
@@ -45,6 +45,7 @@ use crate::processors::{
 };
 use crate::sql::{Cursors, Sql};
 use crate::types::{Contract, ContractType};
+use crate::utils::health_check_provider;
 
 type EventProcessorMap<P> = HashMap<Felt, Vec<Box<dyn EventProcessor<P>>>>;
 
@@ -241,6 +242,7 @@ impl<P: Provider + Send + Sync + std::fmt::Debug + 'static> Engine<P> {
     }
 
     pub async fn start(&mut self) -> Result<()> {
+        health_check_provider(self.provider.clone()).await;
         let mut backoff_delay = Duration::from_secs(1);
         let max_backoff_delay = Duration::from_secs(60);
 
@@ -300,7 +302,6 @@ impl<P: Provider + Send + Sync + std::fmt::Debug + 'static> Engine<P> {
     // TODO: since we now process blocks in chunks we can parallelize the fetching of data
     pub async fn fetch_data(&mut self, cursors: &Cursors) -> Result<FetchDataResult> {
         let latest_block = self.provider.block_hash_and_number().await?;
-
         let from = cursors.head.unwrap_or(0);
         let total_remaining_blocks = latest_block.block_number - from;
         let blocks_to_process = total_remaining_blocks.min(self.config.blocks_chunk_size);
diff --git a/crates/torii/core/src/utils.rs b/crates/torii/core/src/utils.rs
index 516e739e8a..665168cf64 100644
--- a/crates/torii/core/src/utils.rs
+++ b/crates/torii/core/src/utils.rs
@@ -1,11 +1,14 @@
+use std::sync::Arc;
 use std::time::Duration;
 
 use anyhow::Result;
 use chrono::{DateTime, Utc};
 use futures_util::TryStreamExt;
 use ipfs_api_backend_hyper::{IpfsApi, IpfsClient, TryFromUri};
+use starknet::core::types::{BlockId, BlockTag};
+use starknet::providers::Provider;
 use tokio_util::bytes::Bytes;
-use tracing::info;
+use tracing::{error, info};
 
 use crate::constants::{
     IPFS_CLIENT_MAX_RETRY, IPFS_CLIENT_PASSWORD, IPFS_CLIENT_URL, IPFS_CLIENT_USERNAME,
@@ -20,7 +23,6 @@ pub fn must_utc_datetime_from_timestamp(timestamp: u64) -> DateTime<Utc> {
 pub fn utc_dt_string_from_timestamp(timestamp: u64) -> String {
     must_utc_datetime_from_timestamp(timestamp).to_rfc3339()
 }
-
 pub async fn fetch_content_from_ipfs(cid: &str, mut retries: u8) -> Result<Bytes> {
     let client = IpfsClient::from_str(IPFS_CLIENT_URL)?
         .with_credentials(IPFS_CLIENT_USERNAME, IPFS_CLIENT_PASSWORD);
@@ -46,6 +48,18 @@ pub async fn fetch_content_from_ipfs(cid: &str, mut retries: u8) -> Result<Bytes
         IPFS_CLIENT_MAX_RETRY, cid
     )))
 }
+
+pub async fn health_check_provider<P: Provider + Sync + std::fmt::Debug + 'static>(
+    provider: Arc<P>,
+) {
+    let latest_block = provider.get_block_with_tx_hashes(BlockId::Tag(BlockTag::Latest)).await;
+    if let Ok(latest_block) = latest_block {
+        info!("Provider health_check latest_block is: {:?}", latest_block);
+    } else {
+        error!("Provider: {:?} is unhealthy. please check you config!", provider);
+    }
+}
+
 // tests
 #[cfg(test)]
 mod tests {