From 196bb14112448857a885e32dc4149e308e00b01a Mon Sep 17 00:00:00 2001
From: feiyu02 <risaku@163.com>
Date: 星期四, 15 八月 2024 11:57:15 +0800
Subject: [PATCH] 2024.8.15 各项修正
---
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..892a5d6 100644
--- a/src/main/kotlin/cn/flightfeather/supervision/SupervisionApplication.kt
+++ b/src/main/kotlin/cn/flightfeather/supervision/SupervisionApplication.kt
@@ -1,16 +1,23 @@
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("\${systemIsApp}") var systemIsApp: Boolean,
+ @Value("\${mode}") var mode: String,
+) {
@Autowired
lateinit var webSocketServer: VMRoomWebSocketServer
@@ -18,10 +25,20 @@
@Autowired
lateinit var taskController: TaskController
+ @Autowired
+ lateinit var wxTokenManager: WxTokenManager
+
@Bean
fun runner() = ApplicationRunner {
- webSocketServer.start()
-// taskController.run()
+ if (systemIsApp) {
+ webSocketServer.start()
+ } else {
+ wxTokenManager.run()
+ }
+ if (mode == "proapp") {
+ taskController.run()
+ }
+ println("mode: $mode")
}
}
--
Gitblit v1.9.3