feat: add shutdown logs (#770)

Author: taco-paco

magicblock-api/src/magic_validator.rs

@@ -279,6 +279,9 @@ impl MagicValidator {
                 .build()
                 .expect("failed to bulid async runtime for rpc service");
             runtime.block_on(rpc.run());
+
+            drop(runtime);
+            info!("rpc runtime shutdown!");
         });
 
         let task_scheduler_db_path =
@@ -546,6 +549,7 @@ impl MagicValidator {
         .map_err(|err| {
             ApiError::FailedToUnregisterValidatorOnChain(format!("{err:#}"))
         })
+        .inspect(|_| info!("Unregistered validator on chain!"))
     }
 
     async fn ensure_validator_funded_on_chain(&self) -> ApiResult<()> {
@@ -687,6 +691,8 @@ impl MagicValidator {
         if let Err(err) = self.ledger_truncator.join() {
             error!("Ledger truncator did not gracefully exit: {:?}", err);
         }
+
+        info!("MagicValidator shutdown!");
     }
 
     pub fn ledger(&self) -> &Ledger {

magicblock-api/src/tickers.rs

@@ -108,36 +108,6 @@ pub fn init_system_metrics_ticker(
     accountsdb: &Arc<AccountsDb>,
     token: CancellationToken,
 ) -> tokio::task::JoinHandle<()> {
-    fn try_set_ledger_counts(ledger: &Ledger) {
-        macro_rules! try_set_ledger_count {
-            ($name:ident) => {
-                paste::paste! {
-                    match ledger.[< count_ $name >]() {
-                        Ok(count) => {
-                            metrics::[< set_ledger_ $name _count >](count);
-                        }
-                        Err(err) => warn!(
-                            "Failed to get ledger {} count: {:?}",
-                            stringify!($name),
-                            err
-                        ),
-                    }
-                }
-            };
-        }
-        try_set_ledger_count!(block_times);
-        try_set_ledger_count!(blockhashes);
-        try_set_ledger_count!(slot_signatures);
-        try_set_ledger_count!(address_signatures);
-        try_set_ledger_count!(transaction_status);
-        try_set_ledger_count!(transaction_successful_status);
-        try_set_ledger_count!(transaction_failed_status);
-        try_set_ledger_count!(transactions);
-        try_set_ledger_count!(transaction_memos);
-        try_set_ledger_count!(perf_samples);
-        try_set_ledger_count!(account_mod_data);
-    }
-
     fn try_set_ledger_storage_size(ledger: &Ledger) {
         match ledger.storage_size() {
             Ok(byte_size) => metrics::set_ledger_size(byte_size),
@@ -165,13 +135,14 @@ pub fn init_system_metrics_ticker(
                 _ = tokio::time::sleep(tick_duration) => {
                     try_set_ledger_storage_size(&ledger);
                     set_accounts_storage_size(&bank);
-                    metrics::observe_columns_count_duration(|| try_set_ledger_counts(&ledger));
                     set_accounts_count(&bank);
                 },
                 _ = token.cancelled() => {
                     break;
                 }
             }
         }
+
+        info!("System metrics ticker shutdown!");
     })
 }

magicblock-committor-service/src/service.rs

@@ -245,6 +245,8 @@ impl CommittorActor {
                 }
             }
         }
+
+        info!("CommittorActor shutdown!");
     }
 }
 

magicblock-ledger/src/ledger_truncator.rs

@@ -475,12 +475,15 @@ impl LedgerTruncator {
             self.stop();
         }
 
-        if let ServiceState::Stopped(worker_handle) = self.state {
+        let res = if let ServiceState::Stopped(worker_handle) = self.state {
             worker_handle.join()
         } else {
             warn!("LedgerTruncator was not running, nothing to stop");
             Ok(())
-        }
+        };
+        info!("LedgerTruncator shutdown!");
+
+        res
     }
 }
 

magicblock-metrics/src/service.rs

@@ -87,6 +87,8 @@ impl MetricsService {
                     }
                 );
             }
+
+            info!("MetricsService shutdown!");
         })
     }
 }

magicblock-task-scheduler/src/service.rs

@@ -289,6 +289,7 @@ impl TaskSchedulerService {
             }
         }
 
+        info!("TaskSchedulerService shutdown!");
         Ok(())
     }
 

magicblock-validator/src/main.rs

@@ -59,6 +59,9 @@ fn main() {
         .build()
         .expect("failed to build async runtime");
     runtime.block_on(run());
+    drop(runtime);
+
+    info!("main runtime shutdown!");
 }
 
 async fn run() {