feat: 钉钉告警可配置

src/main/java/com/xydl/cac/task/RuleCheckTask.java

@@ -28,6 +28,8 @@ public class RuleCheckTask {
     DataService dataService;
     @Resource
     WarningRepository warningRepository;
+    @Resource
+    DingTalkPushUtil dingTalkPushUtil;
 
     @Scheduled(initialDelay = 90000, fixedDelay = 60000)
     private void checkAll() {
@@ -88,7 +90,7 @@ public class RuleCheckTask {
         warningRepository.save(warning);
         String str = warning.getZsbName() + "-" + warning.getWarnDesc()
                 + ", 采集时间:" + dateStr + ", 当前值:" + value + " " + warning.getTriggerDesc();
-        DingTalkPushUtil.pushText(str);
+        dingTalkPushUtil.pushText(str);
     }
 
 }

src/main/java/com/xydl/cac/util/DingTalkPushUtil.java

@@ -6,6 +6,8 @@ import com.dingtalk.api.request.OapiRobotSendRequest;
 import com.dingtalk.api.response.OapiRobotSendResponse;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.codec.binary.Base64;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Component;
 
 import javax.crypto.Mac;
 import javax.crypto.spec.SecretKeySpec;
@@ -14,20 +16,26 @@ import java.net.URLEncoder;
 /**
  * 钉钉推送工具类
  */
+@Component
 @Slf4j
 public class DingTalkPushUtil {
 
-    public static final String CUSTOM_ROBOT_TOKEN = "e65e730cba22e320e16926fd4ff19ce787fa2162d065792bb6562c6d4a4cf328";
-    public static final String SECRET = "SEC72e5fb1b4ce7f9fed55386040d599035c50f8d2a181ad66bd1277549f0716124";
-    public static final String KEY_WORD = "运维告警: ";
+    @Value("${cac.dingding.send}")
+    public boolean send;
+    @Value("${cac.dingding.token}")
+    public String token;
+    @Value("${cac.dingding.secret}")
+    public String secret;
+    public String KEY_WORD = "运维告警: ";
 
 
-    public static void pushText(String content) {
+    public void pushText(String content) {
+        if (send) {
             try {
                 Long timestamp = System.currentTimeMillis();
-            String stringToSign = timestamp + "\n" + SECRET;
+                String stringToSign = timestamp + "\n" + secret;
                 Mac mac = Mac.getInstance("HmacSHA256");
-            mac.init(new SecretKeySpec(SECRET.getBytes("UTF-8"), "HmacSHA256"));
+                mac.init(new SecretKeySpec(secret.getBytes("UTF-8"), "HmacSHA256"));
                 byte[] signData = mac.doFinal(stringToSign.getBytes("UTF-8"));
                 String sign = URLEncoder.encode(new String(Base64.encodeBase64(signData)), "UTF-8");
 
@@ -43,11 +51,12 @@ public class DingTalkPushUtil {
                 //设置消息类型
                 req.setMsgtype("text");
                 req.setText(text);
-            OapiRobotSendResponse rsp = client.execute(req, CUSTOM_ROBOT_TOKEN);
+                OapiRobotSendResponse rsp = client.execute(req, token);
             } catch (Exception e) {
                 log.error("钉钉发送失败", e);
             }
         }
+    }
 
 }
 

src/main/resources/application-dev.yml

@@ -26,4 +26,8 @@ spring:
       max-request-size: 100MB
 
 cac:
+  dingding:
+    send: true
+    token: e65e730cba22e320e16926fd4ff19ce787fa2162d065792bb6562c6d4a4cf328
+    secret: SEC72e5fb1b4ce7f9fed55386040d599035c50f8d2a181ad66bd1277549f0716124
   rsakey: MIIBVAIBADANBgkqhkiG9w0BAQEFAASCAT4wggE6AgEAAkEAqhHyZfSsYourNxaY7Nt+PrgrxkiA50efORdI5U5lsW79MmFnusUA355oaSXcLhu5xxB38SMSyP2KvuKNPuH3owIDAQABAkAfoiLyL+Z4lf4Myxk6xUDgLaWGximj20CUf+5BKKnlrK+Ed8gAkM0HqoTt2UZwA5E2MzS4EI2gjfQhz5X28uqxAiEA3wNFxfrCZlSZHb0gn2zDpWowcSxQAgiCstxGUoOqlW8CIQDDOerGKH5OmCJ4Z21v+F25WaHYPxCFMvwxpcw99EcvDQIgIdhDTIqD2jfYjPTY8Jj3EDGPbH2HHuffvflECt3Ek60CIQCFRlCkHpi7hthhYhovyloRYsM+IS9h/0BzlEAuO0ktMQIgSPT3aFAgJYwKpqRYKlLDVcflZFCKY7u3UP8iWi1Qw0Y=

src/main/resources/application-prod.yml

@@ -26,4 +26,8 @@ spring:
       max-request-size: 100MB
 
 cac:
+  dingding:
+    send: false
+    token: e65e730cba22e320e16926fd4ff19ce787fa2162d065792bb6562c6d4a4cf328
+    secret: SEC72e5fb1b4ce7f9fed55386040d599035c50f8d2a181ad66bd1277549f0716124
   rsakey: MIIBVAIBADANBgkqhkiG9w0BAQEFAASCAT4wggE6AgEAAkEAqhHyZfSsYourNxaY7Nt+PrgrxkiA50efORdI5U5lsW79MmFnusUA355oaSXcLhu5xxB38SMSyP2KvuKNPuH3owIDAQABAkAfoiLyL+Z4lf4Myxk6xUDgLaWGximj20CUf+5BKKnlrK+Ed8gAkM0HqoTt2UZwA5E2MzS4EI2gjfQhz5X28uqxAiEA3wNFxfrCZlSZHb0gn2zDpWowcSxQAgiCstxGUoOqlW8CIQDDOerGKH5OmCJ4Z21v+F25WaHYPxCFMvwxpcw99EcvDQIgIdhDTIqD2jfYjPTY8Jj3EDGPbH2HHuffvflECt3Ek60CIQCFRlCkHpi7hthhYhovyloRYsM+IS9h/0BzlEAuO0ktMQIgSPT3aFAgJYwKpqRYKlLDVcflZFCKY7u3UP8iWi1Qw0Y=