chore(gateway): unify client implementations

commit-id:38f81a29
starkware-libs · Nov 17, 2024 · c132efb · c132efb
1 parent 63f6b70
commit c132efb
Showing 1 changed file with 4 additions and 11 deletions.
diff --git a/crates/starknet_gateway_types/src/communication.rs b/crates/starknet_gateway_types/src/communication.rs
@@ -55,17 +55,10 @@ pub enum GatewayClientError {
 }
 
 #[async_trait]
-impl GatewayClient for LocalGatewayClient {
-    #[instrument(skip(self))]
-    async fn add_tx(&self, gateway_input: GatewayInput) -> GatewayClientResult<TransactionHash> {
-        let request = GatewayRequest::AddTransaction(gateway_input);
-        let response = self.send(request).await;
-        handle_response_variants!(GatewayResponse, AddTransaction, GatewayClientError, GatewayError)
-    }
-}
-
-#[async_trait]
-impl GatewayClient for RemoteGatewayClient {
+impl<T> GatewayClient for T
+where
+    T: Send + Sync + ComponentClient<GatewayRequest, GatewayResponse>,
+{
     #[instrument(skip(self))]
     async fn add_tx(&self, gateway_input: GatewayInput) -> GatewayClientResult<TransactionHash> {
         let request = GatewayRequest::AddTransaction(gateway_input);