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

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