slhaf/Partner
1
0
Fork 0
mirror of https://github.com/slhaf/Partner.git synced 2026-05-12 16:53:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(ActionScheduler): make TimeWheel load scheduled actions dynamically instead of using init snapshot

Browse Source
This commit is contained in:
slhafzjw
2026-02-09 00:13:36 +08:00
parent b05b665960
commit 227c735667
1 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
Partner-Main/src/main/java/work/slhaf/partner/module/modules/action/dispatcher/scheduler/ActionScheduler.kt
View File

@@ -48,15 +48,18 @@ class ActionScheduler : AgentRunningSubModule<Set<ScheduledActionData>, Void>()
@Init @Init
fun init() { fun init() {
val actions = actionCapability.listActions(null, null) val listScheduledActions: () -> Set<ScheduledActionData> = {
actionCapability.listActions(null, null)
.stream() .stream()
.filter { actionData -> actionData is ScheduledActionData } .filter { it is ScheduledActionData }
.map { actionData -> actionData as ScheduledActionData } .map { it as ScheduledActionData }
.collect(Collectors.toSet()) .collect(Collectors.toSet())
timeWheel = TimeWheel(actions) { actionDataSet ->
actionExecutor.execute(ActionExecutorInput(actionDataSet))
} }
val onTrigger: (Set<ScheduledActionData>) -> Unit = { actionExecutor.execute(ActionExecutorInput(it)) }
timeWheel = TimeWheel(listScheduledActions, onTrigger)
setupShutdownHook() setupShutdownHook()
} }
@@ -80,8 +83,8 @@ class ActionScheduler : AgentRunningSubModule<Set<ScheduledActionData>, Void>()
} }
private class TimeWheel( private class TimeWheel(
val primaryActions: Set<ScheduledActionData>, val listScheduledActions: () -> Set<ScheduledActionData>,
val onTrigger: (Set<ScheduledActionData>) -> Unit val onTrigger: (toTrigger: Set<ScheduledActionData>) -> Unit
) : Closeable { ) : Closeable {
private val actionsGroupByHour = Array<MutableSet<ScheduledActionData>>(24) { mutableSetOf() } private val actionsGroupByHour = Array<MutableSet<ScheduledActionData>>(24) { mutableSetOf() }
@@ -232,7 +235,7 @@ class ActionScheduler : AgentRunningSubModule<Set<ScheduledActionData>, Void>()
) { ) {
val runLoading = { val runLoading = {
val now = ZonedDateTime.now() val now = ZonedDateTime.now()
for (actionData in primaryActions) { for (actionData in listScheduledActions()) {
val latestExecutingTime = val latestExecutingTime =
parseToZonedDateTime( parseToZonedDateTime(
actionData.scheduleType, actionData.scheduleType,