From 752e00503f672ddfe2066afb6c235721a3a912b5 Mon Sep 17 00:00:00 2001
From: feiyu02 <risaku@163.com>
Date: 星期二, 19 十一月 2024 10:25:55 +0800
Subject: [PATCH] 2024.11.19 各项修正

---
 src/main/kotlin/cn/flightfeather/supervision/SupervisionApplication.kt |   25 +++++++++++++++++++++----
 1 files changed, 21 insertions(+), 4 deletions(-)

diff --git a/src/main/kotlin/cn/flightfeather/supervision/SupervisionApplication.kt b/src/main/kotlin/cn/flightfeather/supervision/SupervisionApplication.kt
index c339a5a..f70f3d8 100644
--- a/src/main/kotlin/cn/flightfeather/supervision/SupervisionApplication.kt
+++ b/src/main/kotlin/cn/flightfeather/supervision/SupervisionApplication.kt
@@ -1,16 +1,22 @@
 package cn.flightfeather.supervision
 
-import cn.flightfeather.supervision.timingtask.TaskController
+import cn.flightfeather.supervision.common.wx.WxTokenManager
+import cn.flightfeather.supervision.bgtask.TaskController
 import cn.flightfeather.supervision.websocket.VMRoomWebSocketServer
 import org.springframework.beans.factory.annotation.Autowired
+import org.springframework.beans.factory.annotation.Value
 import org.springframework.boot.ApplicationRunner
 import org.springframework.boot.autoconfigure.SpringBootApplication
 import org.springframework.boot.runApplication
 import org.springframework.context.annotation.Bean
+import org.springframework.scheduling.annotation.EnableScheduling
 
 
 @SpringBootApplication
-class SupervisionApplication {
+@EnableScheduling
+class SupervisionApplication(
+    @Value("\${mode}") var mode: String,
+) {
 
     @Autowired
     lateinit var webSocketServer: VMRoomWebSocketServer
@@ -18,10 +24,21 @@
     @Autowired
     lateinit var taskController: TaskController
 
+    @Autowired
+    lateinit var wxTokenManager: WxTokenManager
+
     @Bean
     fun runner() = ApplicationRunner {
-        webSocketServer.start()
-//        taskController.run()
+        taskController.run()
+        // 2023.8.20 搴旂敤浜庡畨鍗揳pp鍚庡彴锛屽悗缁簲璇ュ悎骞朵簬灏忕▼搴忓悗鍙�
+        if (mode == "proapp") {
+            webSocketServer.start()
+        }
+        // 搴旂敤浜庡井淇″皬绋嬪簭鍚庡彴
+        else if (mode == "pro") {
+            wxTokenManager.run()
+        }
+        println("mode: $mode")
     }
 }
 

--
Gitblit v1.9.3