feat(o11y): adding Tenderly provider monitoring (#873)

reown-com · Dec 17, 2024 · dc44fba · dc44fba
1 parent 455629e
commit dc44fba
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 2 deletions.
diff --git a/src/handlers/chain_agnostic/mod.rs b/src/handlers/chain_agnostic/mod.rs
@@ -7,6 +7,7 @@ use {
             SimulationProvider,
         },
         utils::crypto::get_erc20_contract_balance,
+        Metrics,
     },
     alloy::primitives::{address, Address, B256, U256},
     ethers::{types::H160 as EthersH160, utils::keccak256},
@@ -203,6 +204,7 @@ pub struct Erc20AssetChange {
 pub async fn get_assets_changes_from_simulation(
     simulation_provider: Arc<dyn SimulationProvider>,
     transaction: &Transaction,
+    metrics: Arc<Metrics>,
 ) -> Result<(Vec<Erc20AssetChange>, u64), RpcError> {
     // Fill the state overrides for the source address for each of the supported
     // assets on the initial tx chain
@@ -233,6 +235,7 @@ pub async fn get_assets_changes_from_simulation(
             transaction.to,
             transaction.data.clone(),
             state_overrides,
+            metrics,
         )
         .await?;
     let gas_used = simulation_result.transaction.gas_used;

diff --git a/src/handlers/chain_agnostic/route.rs b/src/handlers/chain_agnostic/route.rs
@@ -131,6 +131,7 @@ async fn handler_internal(
                         let (_, simulated_gas_used) = get_assets_changes_from_simulation(
                             state.providers.simulation_provider.clone(),
                             &initial_transaction,
+                            state.metrics.clone(),
                         )
                         .await?;
                         state
@@ -156,6 +157,7 @@ async fn handler_internal(
                 let (simulation_assets_changes, gas_used) = get_assets_changes_from_simulation(
                     state.providers.simulation_provider.clone(),
                     &initial_transaction,
+                    state.metrics.clone(),
                 )
                 .await?;
                 let mut asset_transfer_value = U256::ZERO;

diff --git a/src/providers/mod.rs b/src/providers/mod.rs
@@ -878,6 +878,7 @@ pub trait SimulationProvider: Send + Sync {
         to: Address,
         input: Bytes,
         state_overrides: HashMap<Address, HashMap<B256, B256>>,
+        metrics: Arc<Metrics>,
     ) -> Result<tenderly::SimulationResponse, RpcError>;
 
     /// Get the cached gas estimation for ERC20 transfer

diff --git a/src/providers/tenderly.rs b/src/providers/tenderly.rs
@@ -1,14 +1,17 @@
 use {
     crate::{
-        error::RpcError, providers::SimulationProvider, storage::error::StorageError,
+        error::RpcError,
+        providers::{ProviderKind, SimulationProvider},
+        storage::error::StorageError,
         utils::crypto::disassemble_caip2,
+        Metrics,
     },
     alloy::primitives::{Address, Bytes, B256, U256},
     async_trait::async_trait,
     deadpool_redis::{redis::AsyncCommands, Pool},
     reqwest::Url,
     serde::{Deserialize, Serialize},
-    std::{collections::HashMap, sync::Arc},
+    std::{collections::HashMap, sync::Arc, time::SystemTime},
     tracing::error,
 };
 
@@ -79,6 +82,7 @@ pub enum TokenStandard {
 }
 
 pub struct TenderlyProvider {
+    provider_kind: ProviderKind,
     api_key: String,
     base_api_url: String,
     http_client: reqwest::Client,
@@ -98,6 +102,7 @@ impl TenderlyProvider {
         );
         let http_client = reqwest::Client::new();
         Self {
+            provider_kind: ProviderKind::Tenderly,
             api_key,
             base_api_url,
             http_client,
@@ -166,6 +171,7 @@ impl SimulationProvider for TenderlyProvider {
         to: Address,
         input: Bytes,
         state_overrides: HashMap<Address, HashMap<B256, B256>>,
+        metrics: Arc<Metrics>,
     ) -> Result<SimulationResponse, RpcError> {
         let url = Url::parse(format!("{}/simulate", &self.base_api_url).as_str())
             .map_err(|_| RpcError::ConversionParseURLError)?;
@@ -183,6 +189,7 @@ impl SimulationProvider for TenderlyProvider {
             state_objects.insert(address, account_state);
         }
 
+        let latency_start = SystemTime::now();
         let response = self
             .send_post_request(
                 url,
@@ -197,6 +204,14 @@ impl SimulationProvider for TenderlyProvider {
                 },
             )
             .await?;
+        metrics.add_latency_and_status_code_for_provider(
+            self.provider_kind,
+            response.status().into(),
+            latency_start,
+            None,
+            Some("simulate".to_string()),
+        );
+
         if !response.status().is_success() {
             error!(
                 "Failed to get the transaction simulation response from Tenderly with status: {}",

diff --git a/terraform/monitoring/dashboard.jsonnet b/terraform/monitoring/dashboard.jsonnet
@@ -139,13 +139,15 @@ dashboard.new(
     panels.status.provider(ds, vars, 'OneInch')      { gridPos: pos._4 },
     panels.status.provider(ds, vars, 'Coinbase')     { gridPos: pos._4 },
     panels.status.provider(ds, vars, 'Bungee')       { gridPos: pos._4 },
+    panels.status.provider(ds, vars, 'Tenderly')     { gridPos: pos._4 },
 
   row.new('Non-RPC providers Latency'),
     panels.non_rpc.endpoints_latency(ds, vars, 'Zerion')       { gridPos: pos._4 },
     panels.non_rpc.endpoints_latency(ds, vars, 'SolScan')      { gridPos: pos._4 },
     panels.non_rpc.endpoints_latency(ds, vars, 'OneInch')      { gridPos: pos._4 },
     panels.non_rpc.endpoints_latency(ds, vars, 'Coinbase')     { gridPos: pos._4 },
     panels.non_rpc.endpoints_latency(ds, vars, 'Bungee')       { gridPos: pos._4 },
+    panels.non_rpc.endpoints_latency(ds, vars, 'Tenderly')     { gridPos: pos._4 },
 
   row.new('Non-RPC providers Cache'),
     panels.non_rpc.cache_latency(ds, vars)      { gridPos: pos._2 },