From 4c90fa9089b30f76125504d0d793beb7ab310711 Mon Sep 17 00:00:00 2001
From: qiminbao <qaz001122>
Date: Sat, 13 Sep 2025 11:51:04 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=85=B3=E9=97=AD=E9=87=87?=
 =?UTF-8?q?=E9=9B=86=E6=8C=87=E4=BB=A4=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../agent/server/core/enums/MsgEnum.java      | 16 +++++------
 .../server/netty/enpoint/AgentEndpoint.java   | 28 +++++++++++++++----
 2 files changed, 30 insertions(+), 14 deletions(-)

diff --git a/src/main/java/com/tongran/agent/server/core/enums/MsgEnum.java b/src/main/java/com/tongran/agent/server/core/enums/MsgEnum.java
index b35e04f..3a338b4 100644
--- a/src/main/java/com/tongran/agent/server/core/enums/MsgEnum.java
+++ b/src/main/java/com/tongran/agent/server/core/enums/MsgEnum.java
@@ -40,21 +40,21 @@ public enum MsgEnum {
 
     交换机上报("SWITCHBOARD"),
 
-    开启系统采集("SYSTEM_COLLECT_START"),
+    开启或更新系统采集("SYSTEM_COLLECT_START"),
 
-    开启系统采集应答("SYSTEM_COLLECT_START_RSP"),
+    开启或更新系统采集应答("SYSTEM_COLLECT_START_RSP"),
 
-    关闭系统采集("SYSTEM_COLLECT_STOP"),
+    关闭所有系统采集("SYSTEM_COLLECT_STOP"),
 
-    关闭系统采集应答("SYSTEM_COLLECT_STOP_RSP"),
+    关闭所有系统采集应答("SYSTEM_COLLECT_STOP_RSP"),
 
-    开启交换机采集("SWITCH_COLLECT_START"),
+    开启或更新交换机采集("SWITCH_COLLECT_START"),
 
-    开启交换机采集应答("SWITCH_COLLECT_START_RSP"),
+    开启或更新交换机采集应答("SWITCH_COLLECT_START_RSP"),
 
-    关闭交换机采集("SWITCH_COLLECT_STOP"),
+    关闭所有交换机采集("SWITCH_COLLECT_STOP"),
 
-    关闭交换机采集应答("SWITCH_COLLECT_STOP_RSP"),
+    关闭所有交换机采集应答("SWITCH_COLLECT_STOP_RSP"),
 
     告警设置("ALARM_SET"),
 
diff --git a/src/main/java/com/tongran/agent/server/netty/enpoint/AgentEndpoint.java b/src/main/java/com/tongran/agent/server/netty/enpoint/AgentEndpoint.java
index 2b2b854..d34138d 100644
--- a/src/main/java/com/tongran/agent/server/netty/enpoint/AgentEndpoint.java
+++ b/src/main/java/com/tongran/agent/server/netty/enpoint/AgentEndpoint.java
@@ -199,7 +199,7 @@ public class AgentEndpoint {
         }
     }
 
-    @AgentDispatcher(msgId = MsgEnum.开启系统采集)
+    @AgentDispatcher(msgId = MsgEnum.开启或更新系统采集)
     public class SystemCollectStartHandler implements AgentHandler {
         @Override
         public String downHandle(String data, String clientId) {
@@ -207,13 +207,13 @@ public class AgentEndpoint {
         }
     }
 
-    @AgentDispatcher(msgId = MsgEnum.开启系统采集应答)
+    @AgentDispatcher(msgId = MsgEnum.开启或更新系统采集应答)
     public class SystemCollectStartRspHandler implements AgentHandler {
         @Override
         public UpMsgResponse upHandle(String data, String clientId) {
             JSONObject jsonObject = new JSONObject();
             jsonObject.put("clientId", clientId);
-            jsonObject.put("dataType", MsgEnum.开启系统采集应答.getValue());
+            jsonObject.put("dataType", MsgEnum.开启或更新系统采集应答.getValue());
             jsonObject.put("data", data);
             MqMsg mqMsg = MqMsg.builder().topic(agentMqConfig.getAgentTopic()).content(jsonObject.toString()).build();
             agentProducer.asyncSend(mqMsg);
@@ -221,7 +221,15 @@ public class AgentEndpoint {
         }
     }
 
-    @AgentDispatcher(msgId = MsgEnum.开启交换机采集)
+    @AgentDispatcher(msgId = MsgEnum.关闭所有系统采集)
+    public class SystemCollectClosetHandler implements AgentHandler {
+        @Override
+        public String downHandle(String data, String clientId) {
+            return data;
+        }
+    }
+
+    @AgentDispatcher(msgId = MsgEnum.开启或更新交换机采集)
     public class SwitchCollectStartHandler implements AgentHandler {
         @Override
         public String downHandle(String data, String clientId) {
@@ -229,13 +237,13 @@ public class AgentEndpoint {
         }
     }
 
-    @AgentDispatcher(msgId = MsgEnum.开启交换机采集应答)
+    @AgentDispatcher(msgId = MsgEnum.开启或更新交换机采集应答)
     public class SwitchCollectStartRspHandler implements AgentHandler {
         @Override
         public UpMsgResponse upHandle(String data, String clientId) {
             JSONObject jsonObject = new JSONObject();
             jsonObject.put("clientId", clientId);
-            jsonObject.put("dataType", MsgEnum.开启交换机采集应答.getValue());
+            jsonObject.put("dataType", MsgEnum.开启或更新交换机采集应答.getValue());
             jsonObject.put("data", data);
             MqMsg mqMsg = MqMsg.builder().topic(agentMqConfig.getAgentTopic()).content(jsonObject.toString()).build();
             agentProducer.asyncSend(mqMsg);
@@ -243,6 +251,14 @@ public class AgentEndpoint {
         }
     }
 
+    @AgentDispatcher(msgId = MsgEnum.关闭所有交换机采集)
+    public class SwitchCollectClosetHandler implements AgentHandler {
+        @Override
+        public String downHandle(String data, String clientId) {
+            return data;
+        }
+    }
+
     @AgentDispatcher(msgId = MsgEnum.告警设置)
     public class AlarmSetHandler implements AgentHandler {
         @Override