From c95f66f4b81e81df9432c0c6d306ad22e3f5e587 Mon Sep 17 00:00:00 2001
From: hcong <1050828145@qq.com>
Date: 星期一, 02 十二月 2024 11:41:09 +0800
Subject: [PATCH] socket以及后台任务状态实时刷新

---
 src/main/kotlin/cn/flightfeather/supervision/common/executor/BgTask.kt |   14 +++++++++-----
 1 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/src/main/kotlin/cn/flightfeather/supervision/common/executor/BgTask.kt b/src/main/kotlin/cn/flightfeather/supervision/common/executor/BgTask.kt
index 953eac3..66a9315 100644
--- a/src/main/kotlin/cn/flightfeather/supervision/common/executor/BgTask.kt
+++ b/src/main/kotlin/cn/flightfeather/supervision/common/executor/BgTask.kt
@@ -1,6 +1,5 @@
 package cn.flightfeather.supervision.common.executor
 
-import org.springframework.scheduling.annotation.Async
 import java.time.LocalDateTime
 import java.util.concurrent.Future
 
@@ -22,8 +21,13 @@
     }
 
     fun execute() {
-        val res = task()
-        if (res) success() else fail()
+        try {
+            val res = task()
+            if (res) success() else fail()
+        } catch (e: Exception) {
+            fail()
+            taskStatus.extra = e.message
+        }
     }
 
     fun success() {
@@ -39,9 +43,9 @@
     fun shutdown() {
         if (future?.isCancelled == false && !future!!.isDone) {
             future!!.cancel(true)
-            taskStatus.status = TaskStatus.SHUTDOWN
-            complete()
         }
+        taskStatus.status = TaskStatus.SHUTDOWN
+        complete()
     }
 
     fun complete() {

--
Gitblit v1.9.3