Add genesis-chunked CLI

informalsystems · Jun 27, 2024 · d23bc9b · d23bc9b
1 parent 52fbc03
commit d23bc9b
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/rpc/src/client/bin/main.rs b/rpc/src/client/bin/main.rs
@@ -136,6 +136,8 @@ enum ClientRequest {
     ConsensusState,
     /// Get the node's genesis data.
     Genesis,
+    /// Get the node's genesis data by chunks
+    GenesisChunked,
     /// Get the node's health.
     Health,
     /// Request the latest block.
@@ -413,6 +415,20 @@ where
             serde_json::to_string_pretty(&client.genesis::<serde_json::Value>().await?)
                 .map_err(Error::serde)?
         },
+        ClientRequest::GenesisChunked => {
+            let mut data = Vec::new();
+            let mut chunks = client.genesis_chunked_stream().await;
+
+            while let Some(chunk) = chunks.next().await {
+                let mut chunk = chunk?;
+                data.append(&mut chunk);
+            }
+
+            let genesis: tendermint::genesis::Genesis<serde_json::Value> =
+                serde_json::from_slice(&data).map_err(Error::serde)?;
+
+            serde_json::to_string_pretty(&genesis).map_err(Error::serde)?
+        },
         ClientRequest::Health => {
             serde_json::to_string_pretty(&client.health().await?).map_err(Error::serde)?
         },