run metrics

jito-foundation · Aug 2, 2024 · b60d4f1 · b60d4f1
1 parent 8a95460
commit b60d4f1
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 5 deletions.
diff --git a/keepers/validator-keeper/src/main.rs b/keepers/validator-keeper/src/main.rs
@@ -48,6 +48,9 @@ fn set_run_flags(args: &Args) -> u32 {
     if args.run_steward {
         run_flags = set_flag(run_flags, KeeperOperations::Steward);
     }
+    if args.run_emit_metrics {
+        run_flags = set_flag(run_flags, KeeperOperations::EmitMetrics);
+    }
 
     run_flags
 }

diff --git a/keepers/validator-keeper/src/operations/keeper_operations.rs b/keepers/validator-keeper/src/operations/keeper_operations.rs
@@ -37,7 +37,7 @@ pub enum KeeperOperations {
     MevEarned,
     MevCommission,
     Steward,
-    EmitHistory,
+    EmitMetrics,
 }
 
 pub fn set_flag(run_flags: u32, flag: KeeperOperations) -> u32 {
@@ -217,17 +217,17 @@ impl KeeperOperations {
             // EMIT HISTORY
             (
                 "num-emit-metrics-runs",
-                runs_for_epoch[KeeperOperations::EmitHistory as usize],
+                runs_for_epoch[KeeperOperations::EmitMetrics as usize],
                 i64
             ),
             (
                 "num-emit-metrics-errors",
-                errors_for_epoch[KeeperOperations::EmitHistory as usize],
+                errors_for_epoch[KeeperOperations::EmitMetrics as usize],
                 i64
             ),
             (
                 "num-emit-metrics-txs",
-                txs_for_epoch[KeeperOperations::EmitHistory as usize],
+                txs_for_epoch[KeeperOperations::EmitMetrics as usize],
                 i64
             ),
             // STEWARD

diff --git a/keepers/validator-keeper/src/operations/metrics_emit.rs b/keepers/validator-keeper/src/operations/metrics_emit.rs
@@ -16,7 +16,7 @@ use validator_history::ValidatorHistoryEntry;
 use super::keeper_operations::{check_flag, KeeperOperations};
 
 fn _get_operation() -> KeeperOperations {
-    KeeperOperations::EmitHistory
+    KeeperOperations::EmitMetrics
 }
 
 fn _should_run() -> bool {

diff --git a/keepers/validator-keeper/src/state/keeper_config.rs b/keepers/validator-keeper/src/state/keeper_config.rs
@@ -128,6 +128,10 @@ pub struct Args {
     #[arg(long, env, default_value = "true")]
     pub run_steward: bool,
 
+    /// Run emit metrics
+    #[arg(long, env, default_value = "true")]
+    pub run_emit_metrics: bool,
+
     /// Run with the startup flag set to true
     #[arg(long, env, default_value = "true")]
     pub full_startup: bool,
@@ -171,6 +175,7 @@ impl fmt::Display for Args {
             Run Stake Upload: {}\n\
             Run Gossip Upload: {}\n\
             Run Steward: {}\n\
+            Run Emit Metrics: {}\n\
             Full Startup: {}\n\
             No Pack: {}\n\
             Pay for New Accounts: {}\n\
@@ -196,6 +201,7 @@ impl fmt::Display for Args {
             self.run_stake_upload,
             self.run_gossip_upload,
             self.run_steward,
+            self.run_emit_metrics,
             self.full_startup,
             self.no_pack,
             self.pay_for_new_accounts,