Merge pull request #96 from anoma/feat/redelegations

feat: add redelegation info

Author: mateuszjasiuk

chain/src/main.rs

@@ -8,7 +8,7 @@ use chain::config::AppConfig;
 use chain::repository;
 use chain::services::namada::{
     query_all_balances, query_all_bonds_and_unbonds, query_all_proposals,
-    query_bonds, query_last_block_height, query_tokens,
+    query_bonds, query_last_block_height, query_redelegations, query_tokens,
 };
 use chain::services::{
     db as db_service, namada as namada_service,
@@ -96,7 +96,7 @@ async fn main() -> Result<(), MainError> {
             })
         }
         (None, Some(crawler_state)) => {
-            tracing::info!(
+            tracing::debug!(
                 "Found chain crawler state, attempting initial crawl at block \
                  {}...",
                 crawler_state.last_processed_block
@@ -128,7 +128,7 @@ async fn main() -> Result<(), MainError> {
                     // If any other type of error occurred, we should not
                     // increment last_processed_block but
                     // crawl from there without initial_query
-                    tracing::info!(
+                    tracing::debug!(
                         "Initial crawl had an error (not RpcError), \
                          continuing from block {}...",
                         crawler_state.last_processed_block
@@ -139,7 +139,7 @@ async fn main() -> Result<(), MainError> {
                     // If the crawl was successful, increment last_processed
                     // block and continue from there.
                     let next_block = crawler_state.last_processed_block + 1;
-                    tracing::info!(
+                    tracing::debug!(
                         "Initial crawl was successful, continuing from block \
                          {}...",
                         next_block
@@ -152,7 +152,7 @@ async fn main() -> Result<(), MainError> {
             }
         }
         (None, None) => {
-            tracing::info!(
+            tracing::debug!(
                 "No chain crawler state found, starting from initial_query..."
             );
             None
@@ -373,12 +373,16 @@ async fn crawling_fn(
     );
 
     let addresses = block.bond_addresses();
-    let bonds = query_bonds(&client, addresses).await.into_rpc_error()?;
+    let bonds = query_bonds(&client, &addresses).await.into_rpc_error()?;
     tracing::debug!(
         block = block_height,
         "Updating bonds for {} addresses",
         bonds.len()
     );
+    let redelegations = query_redelegations(&client, &addresses)
+        .await
+        .into_rpc_error()?;
+    tracing::debug!("Updating redelegations for {} addresses", bonds.len());
 
     let bonds_updates = bonds
         .iter()
@@ -437,7 +441,7 @@ async fn crawling_fn(
 
     let first_checkpoint = Instant::now();
 
-    tracing::info!(
+    tracing::debug!(
         txs = block.transactions.len(),
         ibc_tokens = ibc_tokens.len(),
         balance_changes = balances.len(),
@@ -514,6 +518,10 @@ async fn crawling_fn(
                 repository::pos::insert_bonds(transaction_conn, bonds_updates)?;
 
                 repository::pos::insert_unbonds(transaction_conn, unbonds)?;
+                repository::pos::insert_redelegations(
+                    transaction_conn,
+                    redelegations,
+                )?;
                 repository::pos::remove_withdraws(
                     transaction_conn,
                     epoch,
@@ -553,7 +561,7 @@ async fn crawling_fn(
 
     let second_checkpoint = Instant::now();
 
-    tracing::info!(
+    tracing::debug!(
         block = block_height,
         time_taken = second_checkpoint
             .duration_since(first_checkpoint)
@@ -588,7 +596,6 @@ async fn try_initial_query(
     tracing::debug!("Querying initial data...");
     let block_height =
         query_last_block_height(client).await.into_rpc_error()?;
-
     let first_block_in_epoch = namada_service::get_first_block_in_epoch(client)
         .await
         .into_rpc_error()?;
@@ -644,7 +651,18 @@ async fn try_initial_query(
     .await
     .into_rpc_error()?;
 
-    tracing::debug!(block = block_height, "Querying bonds and unbonds...",);
+    let validators_set =
+        namada_service::get_validator_addresses_at_epoch(client, epoch)
+            .await
+            .into_rpc_error()?;
+
+    tracing::debug!("Querying redelegations...");
+    let redelegations =
+        namada_service::query_all_redelegations(client, validators_set)
+            .await
+            .into_rpc_error()?;
+
+    tracing::debug!("Querying bonds and unbonds...");
     let (bonds, unbonds) = query_all_bonds_and_unbonds(client, None, None)
         .await
         .into_rpc_error()?;
@@ -672,7 +690,7 @@ async fn try_initial_query(
         timestamp,
     };
 
-    tracing::info!(block = block_height, "Inserting initial data...");
+    tracing::debug!(block = block_height, "Inserting initial data...");
 
     conn.interact(move |conn| {
         conn.build_transaction()
@@ -723,6 +741,10 @@ async fn try_initial_query(
 
                 repository::pos::insert_bonds(transaction_conn, bonds)?;
                 repository::pos::insert_unbonds(transaction_conn, unbonds)?;
+                repository::pos::insert_redelegations(
+                    transaction_conn,
+                    redelegations,
+                )?;
 
                 repository::crawler_state::upsert_crawler_state(
                     transaction_conn,
@@ -812,7 +834,7 @@ async fn get_block(
     .await
     .into_rpc_error()?;
 
-    tracing::info!(
+    tracing::debug!(
         block = block_height,
         tm_address = tm_block_response.block.header.proposer_address.to_string(),
         namada_address = ?proposer_address_namada,