chore: refactor main initialization

cloudwalk · Aug 12, 2024 · f2d7c50 · f2d7c50
1 parent 2c8febf
commit f2d7c50
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/src/globals.rs b/src/globals.rs
@@ -8,6 +8,7 @@ use tokio::runtime::Runtime;
 use tokio_util::sync::CancellationToken;
 
 use crate::config;
+use crate::config::StratusConfig;
 use crate::config::WithCommonConfig;
 use crate::ext::spawn_signal_handler;
 use crate::infra::tracing::warn_task_cancellation;
@@ -187,6 +188,12 @@ impl GlobalState {
     // Node Mode
     // -------------------------------------------------------------------------
 
+    /// Initializes the node mode based on the StratusConfig.
+    pub fn initialize_node_mode(config: &StratusConfig) {
+        let mode = if config.follower { NodeMode::Follower } else { NodeMode::Leader };
+        Self::set_node_mode(mode);
+    }
+
     /// Sets the current node mode.
     pub fn set_node_mode(mode: NodeMode) {
         IS_LEADER.store(matches!(mode, NodeMode::Leader), Ordering::Relaxed);

diff --git a/src/main.rs b/src/main.rs
@@ -4,15 +4,10 @@ use stratus::config::StratusConfig;
 use stratus::eth::rpc::serve_rpc;
 use stratus::GlobalServices;
 use stratus::GlobalState;
-use stratus::NodeMode;
 
 fn main() -> anyhow::Result<()> {
     let global_services = GlobalServices::<StratusConfig>::init();
-    GlobalState::set_node_mode(if global_services.config.follower {
-        NodeMode::Follower
-    } else {
-        NodeMode::Leader
-    });
+    GlobalState::initialize_node_mode(&global_services.config);
     global_services.runtime.block_on(run(global_services.config))
 }