chore: TODO cleanups

anoma · Aug 8, 2024 · 4acd1c6 · 4acd1c6
1 parent aa91715
commit 4acd1c6
Show file tree

Hide file tree

Showing 5 changed files with 32 additions and 31 deletions.
diff --git a/chain/run.sh b/chain/run.sh
@@ -1,4 +1,4 @@
 . ../.env
 export TENDERMINT_URL
 export DATABASE_URL
-cargo run -- --initial-query-retry-time 5
+cargo run
diff --git a/chain/src/config.rs b/chain/src/config.rs
@@ -23,9 +23,6 @@ pub struct AppConfig {
     #[clap(long, env)]
     pub database_url: String,
 
-    #[clap(long, env)]
-    pub initial_query_retry_time: u64,
-
     #[command(flatten)]
     pub verbosity: Verbosity<InfoLevel>,
 }
diff --git a/chain/src/main.rs b/chain/src/main.rs
@@ -183,7 +183,7 @@ async fn crawling_fn(
     let redelegations = query_redelegations(&client, addresses)
         .await
         .into_rpc_error()?;
-    tracing::info!("Updating bonds for {} addresses", bonds.len());
+    tracing::info!("Updating redelegations for {} addresses", bonds.len());
 
     let addresses = block.unbond_addresses();
     let unbonds = namada_service::query_unbonds(&client, addresses)

diff --git a/chain/src/services/namada.rs b/chain/src/services/namada.rs
@@ -556,30 +556,35 @@ pub async fn query_all_redelegations(
             let delegations_iter =
                 query_storage_prefix::<NamadaSdkEpoch>(client, &key)
                     .await
-                    .context("Failed to query all delegations")
-                    // TODO:
-                    .unwrap()?;
-
-            let delegations = delegations_iter
-                .filter_map(|a| {
-                    let (key, epoch) = a;
-                    let delegator =
-                        key.segments.last().expect("Delegator address");
-
-                    let delegator = match delegator {
-                        DbKeySeg::AddressSeg(delegator) => Some(delegator),
-                        _ => None,
-                    };
-
-                    delegator.map(|delegator| Redelegation {
-                        delegator: Id::from(delegator.clone()),
-                        validator: validator_address.clone(),
-                        epoch: epoch.0 as Epoch,
-                    })
-                })
-                .collect::<Vec<_>>();
-
-            Some(delegations)
+                    .expect("Failed to query all delegations");
+
+            match delegations_iter {
+                Some(delegations_iter) => {
+                    let delegations = delegations_iter
+                        .filter_map(|a| {
+                            let (key, epoch) = a;
+                            let delegator =
+                                key.segments.last().expect("Delegator address");
+
+                            let delegator = match delegator {
+                                DbKeySeg::AddressSeg(delegator) => {
+                                    Some(delegator)
+                                }
+                                _ => None,
+                            };
+
+                            delegator.map(|delegator| Redelegation {
+                                delegator: Id::from(delegator.clone()),
+                                validator: validator_address.clone(),
+                                epoch: epoch.0 as Epoch,
+                            })
+                        })
+                        .collect::<Vec<_>>();
+
+                    Some(delegations)
+                }
+                None => None,
+            }
         })
         .map(futures::future::ready)
         .buffer_unordered(20)

diff --git a/docker-compose.yml b/docker-compose.yml
@@ -25,8 +25,7 @@ services:
     image: namada/chain-indexer
     command: "./service \
       --tendermint-url ${TENDERMINT_URL} \
-      --database-url ${DATABASE_URL} \
-      --initial-query-retry-time=15"
+      --database-url ${DATABASE_URL}
     build:
       <<: *build
       args: