Skip to content

Commit 308d9c9

Browse files
committed
[Fix #1414] Gonzalos comments
1 parent df68b6f commit 308d9c9

2 files changed

Lines changed: 15 additions & 13 deletions

File tree

impl/core/src/main/java/io/serverlessworkflow/impl/executors/TryExecutor.java

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,16 +71,15 @@ protected TryExecutorBuilder(
7171
this.errorFilter = buildErrorFilter(catchInfo.getErrors());
7272
this.whenFilter = WorkflowUtils.optionalPredicate(application, catchInfo.getWhen());
7373
this.exceptFilter = WorkflowUtils.optionalPredicate(application, catchInfo.getExceptWhen());
74-
this.taskExecutor =
75-
TaskExecutorHelper.createExecutorList(position, task.getTry(), definition, "try");
7674
TryTaskCatch catchTask = task.getCatch();
7775
if (catchTask != null) {
7876
this.errorVariable = catchTask.getAs();
7977
List<TaskItem> catchTaskDo = catchTask.getDo();
8078
this.catchTaskExecutor =
8179
catchTaskDo != null && !catchTaskDo.isEmpty()
8280
? Optional.of(
83-
TaskExecutorHelper.createExecutorList(position.copy(), catchTaskDo, definition))
81+
TaskExecutorHelper.createExecutorList(
82+
position.copy().addProperty("catch"), catchTaskDo, definition))
8483
: Optional.empty();
8584

8685
Retry retry = catchTask.getRetry();
@@ -89,6 +88,8 @@ protected TryExecutorBuilder(
8988
this.catchTaskExecutor = Optional.empty();
9089
this.retryIntervalExecutor = Optional.empty();
9190
}
91+
this.taskExecutor =
92+
TaskExecutorHelper.createExecutorList(position, task.getTry(), definition, "try");
9293
}
9394

9495
private Optional<RetryExecutor> buildRetryInterval(Retry retry) {

impl/test/src/test/java/io/serverlessworkflow/impl/test/RetryTimeoutTest.java

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
import static org.assertj.core.api.Assertions.assertThatThrownBy;
2121

2222
import com.fasterxml.jackson.databind.JsonNode;
23-
import io.serverlessworkflow.api.types.TryTask;
23+
import io.serverlessworkflow.impl.TaskContextData;
2424
import io.serverlessworkflow.impl.WorkflowApplication;
2525
import io.serverlessworkflow.impl.WorkflowException;
2626
import io.serverlessworkflow.impl.WorkflowModel;
@@ -65,17 +65,17 @@ void tearDown() throws IOException {
6565
private class RetryListener implements WorkflowExecutionListener {
6666

6767
private Map<String, Short> taskRetried = new ConcurrentHashMap<>();
68-
private Map<String, Short> tryTaskCompleted = new ConcurrentHashMap<>();
68+
private Map<String, Short> taskCompleted = new ConcurrentHashMap<>();
6969

70+
@Override
7071
public void onTaskRetried(TaskRetriedEvent ev) {
7172
taskRetried.put(ev.taskContext().position().jsonPointer(), ev.taskContext().retryAttempt());
7273
}
7374

75+
@Override
7476
public void onTaskCompleted(TaskCompletedEvent ev) {
75-
if (ev.taskContext().task() instanceof TryTask) {
76-
tryTaskCompleted.put(
77-
ev.taskContext().position().jsonPointer(), ev.taskContext().retryAttempt());
78-
}
77+
TaskContextData taskContext = ev.taskContext();
78+
taskCompleted.put(taskContext.position().jsonPointer(), taskContext.retryAttempt());
7979
}
8080
}
8181

@@ -103,7 +103,6 @@ void testRetry(String path) throws IOException {
103103
.until(() -> future.join().as(JsonNode.class).orElseThrow().equals(result));
104104
assertThat(retryListener.taskRetried).hasSize(1);
105105
assertThat(retryListener.taskRetried.get("do/0/tryGetPet/try/0/getPet")).isEqualTo((short) 2);
106-
assertThat(retryListener.tryTaskCompleted.values()).containsOnly((short) 0);
107106
}
108107

109108
@Test
@@ -136,9 +135,9 @@ void testNestedRetry() throws IOException {
136135
retryListener.taskRetried.get(
137136
"do/0/tryServerError/try/0/tryCommunication/try/0/getPet"))
138137
.isEqualTo((short) 5);
139-
assertThat(retryListener.tryTaskCompleted.get("do/0/tryServerError/try/0/tryCommunication/try"))
138+
assertThat(retryListener.taskCompleted.get("do/0/tryServerError/try/0/tryCommunication/try"))
140139
.isEqualTo((short) 2);
141-
assertThat(retryListener.tryTaskCompleted.get("do/0/tryServerError/try")).isEqualTo((short) 0);
140+
assertThat(retryListener.taskCompleted.get("do/0/tryServerError/try")).isEqualTo((short) 0);
142141
}
143142

144143
@Test
@@ -158,7 +157,9 @@ void testRetryDo() throws IOException {
158157
.orElseThrow()
159158
.equals(Map.of("setAfterFailingTask", "No problem")));
160159

161-
assertThat(retryListener.tryTaskCompleted.get("do/0/attemptTask/try")).isEqualTo((short) 0);
160+
assertThat(retryListener.taskCompleted.get("do/0/attemptTask/try")).isEqualTo((short) 0);
161+
assertThat(retryListener.taskCompleted)
162+
.containsKey("do/0/attemptTask/catch/do/0/executeAfterFailingTask");
162163
}
163164

164165
@Test

0 commit comments

Comments
 (0)