소스 검색

新增监听加密配置

tsurumure 10 달 전
부모
커밋
fcd292d41d

+ 6 - 2
src/main/java/com/backendsys/modules/log/controller/LogStreamController.java

@@ -1,6 +1,7 @@
 package com.backendsys.modules.log.controller;
 
 import cn.hutool.core.convert.Convert;
+import cn.hutool.core.util.StrUtil;
 import com.backendsys.exception.CustomException;
 import com.backendsys.modules.common.config.security.annotations.Anonymous;
 import com.backendsys.modules.sse.emitter.SseEmitterManager;
@@ -39,16 +40,19 @@ public class LogStreamController {
     private String charset;
     @Value("${log-stream.exec}")
     private String exec;
-
+    @Value("${log-stream.sign}")
+    private String signValue;
 
     /**
      * [SSE] 消息监听
      */
     @Anonymous
     @GetMapping(value = "/api/log/stream/watch", produces = "text/event-stream")
-    public SseEmitter stream() {
+    public SseEmitter stream(String sign) {
 
         if (!isEnable) return null;
+        if (StrUtil.isEmpty(sign)) return null;
+        if (!signValue.equals(sign)) return null;
 
         String userId = Convert.toStr(1L);
         SseEmitterUTF8 emitter = new SseEmitterUTF8(Long.MAX_VALUE);

+ 1 - 0
src/main/resources/application-local.yml

@@ -12,6 +12,7 @@ HTTP_BASE_STATIC: http://127.0.0.1:48080
 HTTP_ACTUATOR_URI: https://jsonplaceholder.typicode.com
 
 log-stream:
+  sign: 97e3ef8e-a8b9-46f3-b63a-f0504154efb7
   enable: true
   charset: GBK
   exec: 'ping 127.0.0.1'

+ 1 - 0
src/main/resources/application-prod.yml

@@ -12,6 +12,7 @@ HTTP_BASE_STATIC: http://ai.api.daoguyujia.com
 HTTP_ACTUATOR_URI: http://43.128.1.201:48080
 
 log-stream:
+  sign: 97e3ef8e-a8b9-46f3-b63a-f0504154efb7
   enable: true
   charset: UTF-8
   exec: 'tail -f /logs/backendsys.log'