add more instruments

Author: cctdaniel

src/agent/pyth/rpc/get_all_products.rs

@@ -1,8 +1,10 @@
 use {
     crate::agent::state,
     anyhow::Result,
+    tracing::instrument,
 };
 
+#[instrument(skip_all)]
 pub async fn get_all_products<S>(state: &S) -> Result<serde_json::Value>
 where
     S: state::Prices,

src/agent/pyth/rpc/get_product.rs

@@ -12,8 +12,10 @@ use {
         Request,
         Value,
     },
+    tracing::instrument,
 };
 
+#[instrument(skip_all, fields(account))]
 pub async fn get_product<S>(
     state: &S,
     request: &Request<Method, Value>,
@@ -27,6 +29,8 @@ where
     }?;
 
     let account = params.account.parse::<solana_sdk::pubkey::Pubkey>()?;
+    tracing::Span::current().record("account", account.to_string());
+
     let product = state.get_product(&account).await?;
     Ok(serde_json::to_value(product)?)
 }

src/agent/pyth/rpc/get_product_list.rs

@@ -1,8 +1,10 @@
 use {
     crate::agent::state,
     anyhow::Result,
+    tracing::instrument,
 };
 
+#[instrument(skip_all)]
 pub async fn get_product_list<S>(state: &S) -> Result<serde_json::Value>
 where
     S: state::Prices,

src/agent/pyth/rpc/subscribe_price.rs

@@ -15,8 +15,10 @@ use {
         Value,
     },
     tokio::sync::mpsc,
+    tracing::instrument,
 };
 
+#[instrument(skip_all, fields(account))]
 pub async fn subscribe_price<S>(
     state: &S,
     notify_price_tx: &mpsc::Sender<NotifyPrice>,
@@ -33,6 +35,8 @@ where
     )?;
 
     let account = params.account.parse::<solana_sdk::pubkey::Pubkey>()?;
+    tracing::Span::current().record("account", account.to_string());
+
     let subscription = state
         .subscribe_price(&account, notify_price_tx.clone())
         .await;

src/agent/pyth/rpc/subscribe_price_sched.rs

@@ -15,8 +15,10 @@ use {
         Value,
     },
     tokio::sync::mpsc,
+    tracing::instrument,
 };
 
+#[instrument(skip_all, fields(account))]
 pub async fn subscribe_price_sched<S>(
     state: &S,
     notify_price_sched_tx: &mpsc::Sender<NotifyPriceSched>,
@@ -33,6 +35,8 @@ where
     )?;
 
     let account = params.account.parse::<solana_sdk::pubkey::Pubkey>()?;
+    tracing::Span::current().record("account", account.to_string());
+
     let subscription = state
         .subscribe_price_sched(&account, notify_price_sched_tx.clone())
         .await;