feat: move state sync read operations in consensus app

dashpay · Jan 9, 2025 · 134855b · 134855b
1 parent 2d60300
commit 134855b
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/packages/rs-drive-abci/src/abci/app/consensus.rs b/packages/rs-drive-abci/src/abci/app/consensus.rs
@@ -15,6 +15,7 @@ use drive::grovedb::Transaction;
 use std::fmt::Debug;
 use std::sync::RwLock;
 use tenderdash_abci::proto::abci as proto;
+use dapi_grpc::tonic;
 
 /// AbciApp is an implementation of ABCI Application, as defined by Tenderdash.
 ///
@@ -201,4 +202,18 @@ where
     ) -> Result<proto::ResponseApplySnapshotChunk, proto::ResponseException> {
         handler::apply_snapshot_chunk(self, request).map_err(error_into_exception)
     }
+
+    fn list_snapshots(
+        &self,
+        request: proto::RequestListSnapshots,
+    ) -> Result<proto::ResponseListSnapshots, proto::ResponseException> {
+        handler::list_snapshots(self, request).map_err(error_into_exception)
+    }
+
+    fn load_snapshot_chunk(
+        &self,
+        request: proto::RequestLoadSnapshotChunk,
+    ) -> Result<proto::ResponseLoadSnapshotChunk, proto::ResponseException> {
+        handler::load_snapshot_chunk(self, request).map_err(error_into_exception)
+    }
 }
diff --git a/packages/rs-drive-abci/src/server.rs b/packages/rs-drive-abci/src/server.rs
@@ -40,11 +40,6 @@ pub fn start(
             tenderdash_abci::proto::abci::abci_application_server::AbciApplicationServer::new(
                 check_tx_service,
             ),
-        )
-        .add_service(
-            tenderdash_abci::proto::abci::abci_application_server::AbciApplicationServer::new(
-                state_source_service,
-            ),
         );
 
     let grpc_server_cancel = cancel.clone();