feat: adding component start(initialization) to the server start

commit-id:77f186d4

crates/mempool/src/communication.rs

@@ -1,5 +1,6 @@
 use async_trait::async_trait;
 use starknet_mempool_infra::component_definitions::ComponentRequestHandler;
+use starknet_mempool_infra::component_runner::ComponentRunner;
 use starknet_mempool_infra::component_server::ComponentServer;
 use starknet_mempool_types::communication::{
     MempoolRequest, MempoolRequestAndResponseSender, MempoolResponse,
@@ -52,3 +53,6 @@ impl ComponentRequestHandler<MempoolRequest, MempoolResponse> for MempoolCommuni
         }
     }
 }
+
+#[async_trait]
+impl ComponentRunner for MempoolCommunicationWrapper {}

crates/mempool_infra/src/component_runner.rs

@@ -11,6 +11,8 @@ pub enum ComponentStartError {
 /// Interface to start components.
 #[async_trait]
 pub trait ComponentRunner {
-    /// Start the component. Normally this function should never return.
-    async fn start(&mut self) -> Result<(), ComponentStartError>;
+    /// Start the component. By default do nothing.
+    async fn start(&mut self) -> Result<(), ComponentStartError> {
+        Ok(())
+    }
 }

crates/mempool_infra/src/component_server.rs

@@ -21,7 +21,7 @@ use crate::component_runner::ComponentRunner;
 
 pub struct ComponentServer<Component, Request, Response>
 where
-    Component: ComponentRequestHandler<Request, Response>,
+    Component: ComponentRequestHandler<Request, Response> + ComponentRunner,
     Request: Send + Sync,
     Response: Send + Sync,
 {
@@ -31,7 +31,7 @@ where
 
 impl<Component, Request, Response> ComponentServer<Component, Request, Response>
 where
-    Component: ComponentRequestHandler<Request, Response>,
+    Component: ComponentRequestHandler<Request, Response> + ComponentRunner,
     Request: Send + Sync,
     Response: Send + Sync,
 {
@@ -52,11 +52,19 @@ pub trait ComponentServerStarter: Send + Sync {
 impl<Component, Request, Response> ComponentServerStarter
     for ComponentServer<Component, Request, Response>
 where
-    Component: ComponentRequestHandler<Request, Response> + Send + Sync,
+    Component: ComponentRequestHandler<Request, Response> + ComponentRunner + Send + Sync,
     Request: Send + Sync,
     Response: Send + Sync,
 {
     async fn start(&mut self) {
+        match self.component.start().await {
+            Ok(_) => info!("ComponentServer::start() completed."),
+            Err(err) => {
+                error!("ComponentServer::start() failed: {:?}", err);
+                return;
+            }
+        }
+
         while let Some(request_and_res_tx) = self.rx.recv().await {
             let request = request_and_res_tx.request;
             let tx = request_and_res_tx.tx;

crates/mempool_infra/tests/common/mod.rs

@@ -1,7 +1,7 @@
 use async_trait::async_trait;
 use serde::{Deserialize, Serialize};
 use starknet_mempool_infra::component_client::ClientResult;
-
+use starknet_mempool_infra::component_runner::ComponentRunner;
 pub(crate) type ValueA = u32;
 pub(crate) type ValueB = u8;
 
@@ -55,6 +55,9 @@ impl ComponentA {
     }
 }
 
+#[async_trait]
+impl ComponentRunner for ComponentA {}
+
 pub(crate) struct ComponentB {
     value: ValueB,
     _a: Box<dyn ComponentAClientTrait>,
@@ -69,3 +72,7 @@ impl ComponentB {
         self.value
     }
 }
+
+#[async_trait]
+impl ComponentRunner for ComponentB {}
+