feat: metrics for when blocks are submitted [ENG-702] (#33)

Author: rswanson

Cargo.toml

@@ -50,3 +50,5 @@ tracing-subscriber = "0.3.18"
 
 async-trait = "0.1.80"
 oauth2 = "4.4.2"
+metrics = "0.24.1"
+metrics-exporter-prometheus = "0.16.0"

bin/builder.rs

@@ -5,6 +5,7 @@ use builder::service::serve_builder_with_span;
 use builder::tasks::block::BlockBuilder;
 use builder::tasks::oauth::Authenticator;
 use builder::tasks::submit::SubmitTask;
+use metrics_exporter_prometheus::PrometheusBuilder;
 
 use tokio::select;
 
@@ -17,6 +18,8 @@ async fn main() -> eyre::Result<()> {
     let provider = config.connect_provider().await?;
     let authenticator = Authenticator::new(&config);
 
+    PrometheusBuilder::new().install().expect("failed to install prometheus exporter");
+
     tracing::debug!(rpc_url = config.host_rpc_url.as_ref(), "instantiated provider");
 
     let sequencer_signer = config.connect_sequencer_signer().await?;

src/tasks/submit.rs

@@ -17,7 +17,9 @@ use alloy::{
 use alloy_primitives::{FixedBytes, U256};
 use alloy_sol_types::SolError;
 use eyre::{bail, eyre};
+use metrics::{counter, histogram};
 use oauth2::TokenResponse;
+use std::time::Instant;
 use tokio::{sync::mpsc, task::JoinHandle};
 use tracing::{debug, error, instrument, trace};
 use zenith_types::{
@@ -170,7 +172,6 @@ impl SubmitTask {
 
             return Ok(ControlFlow::Skip);
         }
-
         self.send_transaction(resp, tx).await
     }
 
@@ -252,6 +253,7 @@ impl SubmitTask {
                 sig = hex::encode(resp.sig.as_bytes()),
                 "acquired signature from quincey"
             );
+            counter!("builder.quincey_signature_acquired").increment(1);
             resp
         };
 
@@ -264,12 +266,16 @@ impl SubmitTask {
         let handle = tokio::spawn(async move {
             loop {
                 if let Some(in_progress) = inbound.recv().await {
+                    let building_start_time = Instant::now();
                     let mut retries = 0;
                     loop {
                         match self.handle_inbound(&in_progress).await {
                             Ok(ControlFlow::Retry) => {
                                 retries += 1;
                                 if retries > 3 {
+                                    counter!("builder.building_too_many_retries").increment(1);
+                                    histogram!("builder.block_build_time")
+                                        .record(building_start_time.elapsed().as_millis() as f64);
                                     tracing::error!(
                                         "error handling inbound block: too many retries"
                                     );
@@ -279,10 +285,16 @@ impl SubmitTask {
                                 tokio::time::sleep(tokio::time::Duration::from_secs(2)).await;
                             }
                             Ok(ControlFlow::Skip) => {
+                                histogram!("builder.block_build_time")
+                                    .record(building_start_time.elapsed().as_millis() as f64);
+                                counter!("builder.skipped_blocks").increment(1);
                                 tracing::info!("skipping block");
                                 break;
                             }
                             Ok(ControlFlow::Done) => {
+                                histogram!("builder.block_build_time")
+                                    .record(building_start_time.elapsed().as_millis() as f64);
+                                counter!("builder.submitted_successful_blocks").increment(1);
                                 tracing::info!("block landed successfully");
                                 break;
                             }

src/tasks/tx_poller.rs

@@ -11,6 +11,8 @@ use serde_json::from_slice;
 
 pub use crate::config::BuilderConfig;
 
+use metrics::counter;
+
 #[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct TxPoolResponse {
     transactions: Vec<TxEnvelope>,
@@ -54,6 +56,7 @@ impl TxPoller {
         self.seen_txns.entry(*tx.tx_hash()).or_insert_with(|| {
             // add to unique transactions
             unique.push(tx.clone());
+            counter!("builder.unique_txs").increment(1);
             // expiry is now + cache_duration
             time::Instant::now() + Duration::from_secs(self.config.tx_pool_cache_duration)
         });
@@ -77,6 +80,7 @@ impl TxPoller {
 
         for key in expired_keys {
             self.seen_txns.remove(&key);
+            counter!("builder.evicted_txs").increment(1);
         }
     }
 }