From 42b6618acf787d5cfa5eb7cd6e5fe342ba54e22e Mon Sep 17 00:00:00 2001
From: wangli <274027703@qq.com>
Date: Mon, 28 Oct 2024 21:24:21 +0800
Subject: [PATCH] =?UTF-8?q?feat(REQ-3114)=20-=20=E5=8A=A8=E6=80=81?=
 =?UTF-8?q?=E6=8E=A7=E5=88=B6=20Stream=20=E7=8A=B6=E6=80=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../riven/dingtalk/controller/DingtalkController.java | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/riven-dingtalk/src/main/java/cn/axzo/riven/dingtalk/controller/DingtalkController.java b/riven-dingtalk/src/main/java/cn/axzo/riven/dingtalk/controller/DingtalkController.java
index 510c0be..d855ef1 100644
--- a/riven-dingtalk/src/main/java/cn/axzo/riven/dingtalk/controller/DingtalkController.java
+++ b/riven-dingtalk/src/main/java/cn/axzo/riven/dingtalk/controller/DingtalkController.java
@@ -39,18 +39,17 @@ public class DingtalkController {
     @PostMapping("/enabled")
     public CommonResponse<String> changeStream(@RequestParam String appId, @RequestParam Boolean enable) {
         OpenDingTalkClient client = clientMap.getOrDefault(appId, null);
-        if (Objects.isNull(client)) {
-            return CommonResponse.error("动态调整 Stream 的连接");
-        }
         try {
-            if (enable) {
+            if (enable && Objects.isNull(client)) {
                 ThirdApplicationReq query = new ThirdApplicationReq();
                 query.setAppId(appId);
                 List<ThirdApplication> applications = thirdApplicationService.genericQuery(query);
                 autoConnector.startStream(applications);
             } else {
-                client.stop();
-                clientMap.remove(appId);
+                if(Objects.nonNull(client)) {
+                    client.stop();
+                    clientMap.remove(appId);
+                }
             }
         } catch (Exception e) {
             return CommonResponse.error(e.getMessage());