jprdonnelly/kestra
1
0
Fork 0
mirror of https://github.com/kestra-io/kestra.git synced 2025-12-19 18:05:41 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat(core): add onResume inputs on Pause tasks (#3640)

Browse Source 
close kestra-io/kestra#1581
This commit is contained in:
Ludovic DEHON
2024-04-30 09:54:09 +02:00
committed by GitHub
parent c60f78ffb4
commit 3145e1c71e
22 changed files with 880 additions and 281 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
runner-memory/src/main/java/io/kestra/runner/memory/MemoryExecutor.java
View File

@@ -222,6 +222,7 @@ public class MemoryExecutor implements ExecutorInterface {
if (workerTaskResultDelay.getDelayType().equals(ExecutionDelay.DelayType.RESUME_FLOW)) {
Execution markAsExecution = executionService.markAs(
executionState.execution,
flow,
workerTaskResultDelay.getTaskRunId(),
workerTaskResultDelay.getState()
);
@@ -518,17 +519,18 @@ public class MemoryExecutor implements ExecutorInterface {
executorService.log(log, true, message);
}
final Flow flowFromRepository = this.flowRepository.findByExecution(EXECUTIONS.get(message.getExecutionId()).execution);
// save WorkerTaskResult on current QueuedExecution
EXECUTIONS.compute(message.getExecutionId(), (s, executionState) -> {
if (executionState == null) {
throw new IllegalStateException("Execution state don't exist for " + s + ", receive " + message);
}
return executionState.from(executionService.kill(executionState.execution));
return executionState.from(executionService.kill(executionState.execution, flowFromRepository));
});
Flow flow = this.flowRepository.findByExecution(EXECUTIONS.get(message.getExecutionId()).execution);
flow = transform(flow, EXECUTIONS.get(message.getExecutionId()).execution);
Flow flow = transform(flowFromRepository, EXECUTIONS.get(message.getExecutionId()).execution);
this.toExecution(new Executor(EXECUTIONS.get(message.getExecutionId()).execution, null).withFlow(flow));
}