package cn.flightfeather.supervision.common.executor
|
|
import java.time.LocalDateTime
|
import java.util.concurrent.Future
|
|
/**
|
* 后台任务
|
*/
|
class BgTask(
|
val type: BgTaskType,
|
val id: String,
|
val name: String,
|
private val task: () -> Boolean,
|
private val onStatusChange: (status: BgTaskStatus) -> Unit
|
) {
|
var taskStatus = BgTaskStatus(type, id, name)
|
var future: Future<*>? = null
|
|
fun ready() {
|
taskStatus.startTime = LocalDateTime.now()
|
setStatus(TaskStatus.RUNNING)
|
}
|
|
fun execute() {
|
try {
|
val res = task()
|
if (res) success() else fail()
|
} catch (e: Exception) {
|
fail()
|
taskStatus.extra = e.message
|
}
|
}
|
|
fun success() {
|
complete()
|
setStatus(TaskStatus.SUCCESS)
|
}
|
|
fun fail() {
|
complete()
|
setStatus(TaskStatus.FAIL)
|
}
|
|
fun shutdown() {
|
if (future?.isCancelled == false && !future!!.isDone) {
|
future!!.cancel(true)
|
}
|
complete()
|
setStatus(TaskStatus.SHUTDOWN)
|
}
|
|
fun complete() {
|
taskStatus.endTime = LocalDateTime.now()
|
}
|
|
fun setStatus(status: TaskStatus) {
|
taskStatus.status = status
|
onStatusChange(taskStatus)
|
}
|
}
|
