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 | 22 +++++++++++++++++----- 1 files changed, 17 insertions(+), 5 deletions(-) diff --git a/src/main/kotlin/cn/flightfeather/supervision/SupervisionApplication.kt b/src/main/kotlin/cn/flightfeather/supervision/SupervisionApplication.kt index 63fe9e1..892a5d6 100644 --- a/src/main/kotlin/cn/flightfeather/supervision/SupervisionApplication.kt +++ b/src/main/kotlin/cn/flightfeather/supervision/SupervisionApplication.kt @@ -1,17 +1,23 @@ package cn.flightfeather.supervision import cn.flightfeather.supervision.common.wx.WxTokenManager -import cn.flightfeather.supervision.timingtask.TaskController +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 @@ -24,9 +30,15 @@ @Bean fun runner() = ApplicationRunner { - webSocketServer.start() - taskController.run() - wxTokenManager.run() + if (systemIsApp) { + webSocketServer.start() + } else { + wxTokenManager.run() + } + if (mode == "proapp") { + taskController.run() + } + println("mode: $mode") } } -- Gitblit v1.9.3