From 33b513e09e0c30e9ae7d85af220daf4761114af3 Mon Sep 17 00:00:00 2001
From: lihongjie0209 <lihongjie0209@gmail.com>
Date: Sun, 19 May 2024 21:58:32 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E5=96=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/DataCollectorController.java   | 15 +++++++++++
 .../coal/dataCollector/dto/GetXmlRequest.java |  3 +++
 .../service/DataCollectorService.java         | 27 +++++++++++++++++++
 3 files changed, 45 insertions(+)
 create mode 100644 src/main/java/cn/lihongjie/coal/dataCollector/dto/GetXmlRequest.java

diff --git a/src/main/java/cn/lihongjie/coal/dataCollector/controller/DataCollectorController.java b/src/main/java/cn/lihongjie/coal/dataCollector/controller/DataCollectorController.java
index 74fc2cf9..eb5f2577 100644
--- a/src/main/java/cn/lihongjie/coal/dataCollector/controller/DataCollectorController.java
+++ b/src/main/java/cn/lihongjie/coal/dataCollector/controller/DataCollectorController.java
@@ -81,6 +81,21 @@ public class DataCollectorController {
     }
 
 
+    @PostMapping("/getXml")
+    public Object getXml(@RequestBody GetXmlRequest request) {
+        return this.service.getXml(request);
+    }
 
 
+    @PostMapping("/getVersion")
+    public Object getVersion(@RequestBody GetXmlRequest request) {
+        return this.service.getVersion(request);
+    }
+
+
+    @PostMapping("/restart")
+    public Object restart(@RequestBody GetXmlRequest request) {
+        return this.service.restart(request);
+    }
+
 }
diff --git a/src/main/java/cn/lihongjie/coal/dataCollector/dto/GetXmlRequest.java b/src/main/java/cn/lihongjie/coal/dataCollector/dto/GetXmlRequest.java
new file mode 100644
index 00000000..c745152f
--- /dev/null
+++ b/src/main/java/cn/lihongjie/coal/dataCollector/dto/GetXmlRequest.java
@@ -0,0 +1,3 @@
+package cn.lihongjie.coal.dataCollector.dto;
+
+public record GetXmlRequest(String appKey){}
\ No newline at end of file
diff --git a/src/main/java/cn/lihongjie/coal/dataCollector/service/DataCollectorService.java b/src/main/java/cn/lihongjie/coal/dataCollector/service/DataCollectorService.java
index 7cf22822..84b82839 100644
--- a/src/main/java/cn/lihongjie/coal/dataCollector/service/DataCollectorService.java
+++ b/src/main/java/cn/lihongjie/coal/dataCollector/service/DataCollectorService.java
@@ -101,6 +101,33 @@ public class DataCollectorService
 
     }
 
+
+    public Object getXml(GetXmlRequest request) {
+
+        return rabbitTemplate.convertSendAndReceive(
+                "dataCollector." + request.appKey(), Map.of("action", "getXML")
+        );
+
+    }
+
+
+    public Object getVersion(GetXmlRequest request) {
+
+        return rabbitTemplate.convertSendAndReceive(
+                "dataCollector." + request.appKey(), Map.of("action", "getVersion")
+        );
+
+    }
+
+
+    public Object restart(GetXmlRequest request) {
+
+        return rabbitTemplate.convertSendAndReceive(
+                "dataCollector." + request.appKey(), Map.of("action", "restart")
+        );
+
+    }
+
     public Object sendMsg(String id, String action, Object data, Map<String, String> headers) {
 
         DataCollectorEntity collector = get(id);