feat: added config update message handling
This commit is contained in:
@@ -45,6 +45,7 @@ pub struct MasterHandlerImpl<MMH>
|
||||
where
|
||||
MMH: MasterMessageHandler + ?Sized,
|
||||
{
|
||||
agent_id: String,
|
||||
connector: Arc<MasterConnector>,
|
||||
message_handler: Arc<MMH>,
|
||||
message_handle_lock: tokio::sync::RwLock<Option<MessageHandleInfo>>,
|
||||
@@ -54,8 +55,9 @@ impl<MMH> MasterHandlerImpl<MMH>
|
||||
where
|
||||
MMH: MasterMessageHandler + ?Sized,
|
||||
{
|
||||
pub fn new(connector: Arc<MasterConnector>, message_handler: Arc<MMH>) -> Self {
|
||||
pub fn new(agent_id: &str, connector: Arc<MasterConnector>, message_handler: Arc<MMH>) -> Self {
|
||||
Self {
|
||||
agent_id: agent_id.to_string(),
|
||||
connector,
|
||||
message_handler,
|
||||
message_handle_lock: tokio::sync::RwLock::new(None),
|
||||
@@ -136,7 +138,7 @@ where
|
||||
message = stream.message() => {
|
||||
match message {
|
||||
Ok(Some(msg)) => {
|
||||
if let Err(e) = self.message_handler.handle_master_message(msg).await {
|
||||
if let Err(e) = self.message_handler.handle_master_message(&self.agent_id, msg).await {
|
||||
error!("Failed to handle master message: {:?}", e);
|
||||
}
|
||||
continue;
|
||||
|
||||
Reference in New Issue
Block a user