apollo_proof_manager: component request handler

Author: einat-starkware

Cargo.lock

@@ -14,6 +14,7 @@ workspace = true
 [dependencies]
 apollo_proof_manager_config.workspace = true
 apollo_infra.workspace = true
+apollo_proof_manager_types.workspace = true
 async-trait.workspace = true
 hex.workspace = true
 starknet-types-core = { workspace = true, features = ["hash"] }

crates/apollo_proof_manager/Cargo.toml

@@ -0,0 +1,31 @@
+use apollo_infra::component_definitions::ComponentRequestHandler;
+use apollo_infra::component_server::{ConcurrentLocalComponentServer, RemoteComponentServer};
+use apollo_proof_manager_types::{ProofManagerError, ProofManagerRequest, ProofManagerResponse};
+use async_trait::async_trait;
+
+use crate::proof_storage::ProofStorage;
+use crate::ProofManager;
+pub type LocalProofManagerServer =
+    ConcurrentLocalComponentServer<ProofManager, ProofManagerRequest, ProofManagerResponse>;
+pub type RemoteProofManagerServer =
+    RemoteComponentServer<ProofManagerRequest, ProofManagerResponse>;
+
+#[async_trait]
+impl ComponentRequestHandler<ProofManagerRequest, ProofManagerResponse> for ProofManager {
+    async fn handle_request(&mut self, request: ProofManagerRequest) -> ProofManagerResponse {
+        match request {
+            ProofManagerRequest::SetProof(facts_hash, proof) => ProofManagerResponse::SetProof(
+                self.set_proof(facts_hash, proof)
+                    .map_err(|e| ProofManagerError::ProofStorage(e.to_string())),
+            ),
+            ProofManagerRequest::GetProof(facts_hash) => ProofManagerResponse::GetProof(
+                self.get_proof(facts_hash)
+                    .map_err(|e| ProofManagerError::ProofStorage(e.to_string())),
+            ),
+            ProofManagerRequest::ContainsProof(facts_hash) => ProofManagerResponse::ContainsProof(
+                self.contains_proof(facts_hash)
+                    .map_err(|e| ProofManagerError::ProofStorage(e.to_string())),
+            ),
+        }
+    }
+}

crates/apollo_proof_manager/src/communication.rs

@@ -1,3 +1,4 @@
+pub mod communication;
 pub mod proof_manager;
 pub mod proof_storage;