Skip to content
This repository was archived by the owner on Apr 17, 2024. It is now read-only.

Commit

Permalink
Merge pull request #80 from Levana-Protocol/log-queries
Browse files Browse the repository at this point in the history
Optionally log requests
  • Loading branch information
lvn-rusty-dragon authored Mar 7, 2024
2 parents 452b6f1 + e20d166 commit 7d04238
Show file tree
Hide file tree
Showing 2 changed files with 17 additions and 0 deletions.
3 changes: 3 additions & 0 deletions packages/cosmos/src/client.rs
Original file line number Diff line number Diff line change
Expand Up @@ -308,6 +308,9 @@ impl Cosmos {
),
Ok(mut guard) => {
let cosmos_inner = guard.get_inner_mut();
if self.pool.builder.get_log_requests() {
tracing::info!("{action}");
}
match self.perform_query_inner(req.clone(), cosmos_inner).await {
Ok(x) => {
cosmos_inner.log_query_result(QueryResult::Success);
Expand Down
14 changes: 14 additions & 0 deletions packages/cosmos/src/cosmos_builder.rs
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,7 @@ pub struct CosmosBuilder {
osmosis_gas_price_too_old_seconds: Option<u64>,
max_price: Option<f64>,
rate_limit_per_second: Option<u64>,
log_requests: Option<bool>,
}

impl CosmosBuilder {
Expand Down Expand Up @@ -82,6 +83,7 @@ impl CosmosBuilder {
max_price: None,
rate_limit_per_second: None,
is_fast_chain: matches!(hrp.as_str(), "sei" | "inj"),
log_requests: None,
}
}

Expand Down Expand Up @@ -425,6 +427,18 @@ impl CosmosBuilder {
pub fn set_osmosis_gas_price_too_old_seconds(&mut self, secs: u64) {
self.osmosis_gas_price_too_old_seconds = Some(secs);
}

/// Should we log Cosmos requests made?
///
/// Default: false
pub fn get_log_requests(&self) -> bool {
self.log_requests.unwrap_or_default()
}

/// See [Self::get_log_requests]
pub fn set_log_requests(&mut self, log_requests: bool) {
self.log_requests = Some(log_requests);
}
}

#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
Expand Down

0 comments on commit 7d04238

Please sign in to comment.