chore(mempool_p2p_types): unify client implementations

commit-id:c52611c9
starkware-libs · Nov 17, 2024 · 5509967 · 5509967
1 parent 61d5549
commit 5509967
Showing 1 changed file with 4 additions and 32 deletions.
diff --git a/crates/starknet_mempool_p2p_types/src/communication.rs b/crates/starknet_mempool_p2p_types/src/communication.rs
@@ -66,38 +66,10 @@ pub enum MempoolP2pPropagatorClientError {
 }
 
 #[async_trait]
-impl MempoolP2pPropagatorClient for LocalMempoolP2pPropagatorClient {
-    async fn add_transaction(
-        &self,
-        transaction: RpcTransaction,
-    ) -> MempoolP2pPropagatorClientResult<()> {
-        let request = MempoolP2pPropagatorRequest::AddTransaction(transaction);
-        let response = self.send(request).await;
-        handle_response_variants!(
-            MempoolP2pPropagatorResponse,
-            AddTransaction,
-            MempoolP2pPropagatorClientError,
-            MempoolP2pPropagatorError
-        )
-    }
-
-    async fn continue_propagation(
-        &self,
-        propagation_metadata: BroadcastedMessageMetadata,
-    ) -> MempoolP2pPropagatorClientResult<()> {
-        let request = MempoolP2pPropagatorRequest::ContinuePropagation(propagation_metadata);
-        let response = self.send(request).await;
-        handle_response_variants!(
-            MempoolP2pPropagatorResponse,
-            ContinuePropagation,
-            MempoolP2pPropagatorClientError,
-            MempoolP2pPropagatorError
-        )
-    }
-}
-
-#[async_trait]
-impl MempoolP2pPropagatorClient for RemoteMempoolP2pPropagatorClient {
+impl<T> MempoolP2pPropagatorClient for T
+where
+    T: Send + Sync + ComponentClient<MempoolP2pPropagatorRequest, MempoolP2pPropagatorResponse>,
+{
     async fn add_transaction(
         &self,
         transaction: RpcTransaction,