chore(mempool_node): add component replacer for servers

commit-id:f81de1c2

crates/mempool_infra/src/component_server/definitions.rs

@@ -1,5 +1,6 @@
 use std::any::type_name;
 
+use thiserror::Error;
 use tokio::sync::mpsc::Receiver;
 use tracing::info;
 
@@ -26,3 +27,13 @@ pub async fn request_response_loop<Request, Response, Component>(
         tx.send(res).await.expect("Response connection should be open.");
     }
 }
+
+#[derive(Clone, Debug, Error)]
+pub enum ReplaceComponentError {
+    #[error("Internal error.")]
+    InternalError,
+}
+
+pub trait ComponentReplacer<Component> {
+    fn replace(&mut self, component: Component) -> Result<(), ReplaceComponentError>;
+}

crates/mempool_infra/src/component_server/empty_component_server.rs

@@ -3,6 +3,7 @@ use std::any::type_name;
 use async_trait::async_trait;
 use tracing::info;
 
+use crate::component_server::{ComponentReplacer, ReplaceComponentError};
 use crate::errors::{ComponentError, ComponentServerError};
 use crate::starters::Startable;
 
@@ -29,3 +30,10 @@ impl<Component: Startable<ComponentError> + Send + Sync> Startable<ComponentServ
 pub fn create_empty_server<Component: Send + Sync>(component: Component) -> EmptyServer<Component> {
     EmptyServer::new(component)
 }
+
+impl<Component> ComponentReplacer<Component> for EmptyServer<Component> {
+    fn replace(&mut self, component: Component) -> Result<(), ReplaceComponentError> {
+        self.component = component;
+        Ok(())
+    }
+}

crates/mempool_infra/src/component_server/local_component_server.rs

@@ -6,6 +6,7 @@ use tracing::error;
 
 use super::definitions::request_response_loop;
 use crate::component_definitions::{ComponentRequestAndResponseSender, ComponentRequestHandler};
+use crate::component_server::{ComponentReplacer, ReplaceComponentError};
 use crate::errors::{ComponentError, ComponentServerError};
 use crate::starters::Startable;
 
@@ -181,3 +182,16 @@ where
         Self { component, rx, _local_server_type: PhantomData }
     }
 }
+
+impl<Component, Request, Response, LocalServerType> ComponentReplacer<Component>
+    for BaseLocalComponentServer<Component, Request, Response, LocalServerType>
+where
+    Component: ComponentRequestHandler<Request, Response>,
+    Request: Send + Sync,
+    Response: Send + Sync,
+{
+    fn replace(&mut self, component: Component) -> Result<(), ReplaceComponentError> {
+        self.component = component;
+        Ok(())
+    }
+}