Merge pull request #140 from propeller-heads/lp/update-load-all-tokens

feat: add filter args to load_all_tokens

examples/price_printer/main.rs

@@ -58,6 +58,8 @@ async fn main() {
             false,
             Some(tycho_api_key.as_str()),
             Chain::Ethereum,
+            None,
+            None,
         )
         .await;
         let tvl_filter = ComponentFilter::with_tvl_range(cli.tvl_threshold, cli.tvl_threshold);

examples/quickstart/main.rs

@@ -36,9 +36,15 @@ async fn main() {
     let tvl_threshold = 10_000.0;
     let tvl_filter = ComponentFilter::with_tvl_range(tvl_threshold, tvl_threshold);
 
-    let all_tokens =
-        load_all_tokens(tycho_url.as_str(), false, Some(tycho_api_key.as_str()), Chain::Ethereum)
-            .await;
+    let all_tokens = load_all_tokens(
+        tycho_url.as_str(),
+        false,
+        Some(tycho_api_key.as_str()),
+        Chain::Ethereum,
+        None,
+        None,
+    )
+    .await;
     let mut pairs: HashMap<String, Vec<Token>> = HashMap::new();
 
     let mut protocol_stream = ProtocolStreamBuilder::new(&tycho_url, Chain::Ethereum)

src/utils.rs

@@ -1,5 +1,6 @@
 use std::collections::HashMap;
 
+use tracing::info;
 use tycho_client::{rpc::RPCClient, HttpRPCClient};
 use tycho_core::{dto::Chain, Bytes};
 
@@ -36,19 +37,43 @@ pub fn hexstring_to_vec(hexstring: &str) -> Result<Vec<u8>, SimulationError> {
 }
 
 /// Loads all tokens from Tycho and returns them as a Hashmap of address->Token.
+///
+/// # Arguments
+///
+/// * `tycho_url` - The URL of the Tycho RPC (do not include the url prefix e.g. 'https://').
+/// * `no_tls` - Whether to use HTTP instead of HTTPS.
+/// * `auth_key` - The API key to use for authentication.
+/// * `chain` - The chain to load tokens from.
+/// * `min_quality` - The minimum quality of tokens to load. Defaults to 100 if not provided.
+/// * `max_days_since_last_trade` - The max number of days since the token was last traded. Defaults
+///   are chain specific and applied if not provided.
 pub async fn load_all_tokens(
     tycho_url: &str,
     no_tls: bool,
     auth_key: Option<&str>,
     chain: Chain,
+    min_quality: Option<i32>,
+    max_days_since_last_trade: Option<u64>,
 ) -> HashMap<Bytes, Token> {
+    info!("Loading tokens from Tycho...");
     let rpc_url =
         if no_tls { format!("http://{tycho_url}") } else { format!("https://{tycho_url}") };
     let rpc_client = HttpRPCClient::new(rpc_url.as_str(), auth_key).unwrap();
 
+    // Chain specific defaults for special case chains. Otherwise defaults to 42 days.
+    let default_min_days = HashMap::from([(Chain::Base, 10_u64)]);
+
     #[allow(clippy::mutable_key_type)]
     rpc_client
-        .get_all_tokens(chain, Some(100), Some(42), 3_000)
+        .get_all_tokens(
+            chain,
+            min_quality.or(Some(100)),
+            max_days_since_last_trade.or(default_min_days
+                .get(&chain)
+                .or(Some(&42))
+                .copied()),
+            3_000,
+        )
         .await
         .expect("Unable to load tokens")
         .into_iter()