feat: make executable operable on mutable references

commit-id:f73ba6d2
starkware-libs · May 20, 2024 · 597efc4 · 597efc4
1 parent 4febada
commit 597efc4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/crates/mempool_infra/src/component_server.rs b/crates/mempool_infra/src/component_server.rs
@@ -3,7 +3,7 @@ use tokio::sync::mpsc::{Receiver, Sender};
 
 #[async_trait]
 pub trait ComponentMessageExecutor<M, R> {
-    async fn execute(&self, message: M) -> R;
+    async fn execute(&mut self, message: M) -> R;
 }
 
 pub struct MessageAndResponseSender<M, R>

diff --git a/crates/mempool_infra/src/component_server_client_test.rs b/crates/mempool_infra/src/component_server_client_test.rs
@@ -65,7 +65,7 @@ impl ComponentATrait for ComponentClient<ComponentAMessages, ComponentAResponses
 
 #[async_trait]
 impl ComponentMessageExecutor<ComponentAMessages, ComponentAResponses> for ComponentA {
-    async fn execute(&self, message: ComponentAMessages) -> ComponentAResponses {
+    async fn execute(&mut self, message: ComponentAMessages) -> ComponentAResponses {
         match message {
             ComponentAMessages::AGetValue => ComponentAResponses::Value(self.a_get_value().await),
         }
@@ -112,7 +112,7 @@ impl ComponentBTrait for ComponentClient<ComponentBMessages, ComponentBResponses
 
 #[async_trait]
 impl ComponentMessageExecutor<ComponentBMessages, ComponentBResponses> for ComponentB {
-    async fn execute(&self, message: ComponentBMessages) -> ComponentBResponses {
+    async fn execute(&mut self, message: ComponentBMessages) -> ComponentBResponses {
         match message {
             ComponentBMessages::BGetValue => ComponentBResponses::Value(self.b_get_value().await),
         }