KAFKA-17402: DefaultStateUpdated should transite task atomically (#18607

) Reviewers: Bruno Cadonna <[email protected]>, Lucas Brutschy <[email protected]>
apache · Jan 21, 2025 · 9f64ef2 · 9f64ef2
1 parent 379049f
commit 9f64ef2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/streams/src/main/java/org/apache/kafka/streams/processor/internals/DefaultStateUpdater.java b/streams/src/main/java/org/apache/kafka/streams/processor/internals/DefaultStateUpdater.java
@@ -675,7 +675,6 @@ private void maybeCompleteRestoration(final StreamTask task,
                 measureCheckpointLatency(() -> task.maybeCheckpoint(true));
                 changelogReader.unregister(changelogPartitions);
                 addToRestoredTasks(task);
-                updatingTasks.remove(task.id());
                 log.info("Stateful active task " + task.id() + " completed restoration");
                 transitToUpdateStandbysIfOnlyStandbysLeft();
             }
@@ -691,6 +690,7 @@ private void addToRestoredTasks(final StreamTask task) {
             restoredActiveTasksLock.lock();
             try {
                 restoredActiveTasks.add(task);
+                updatingTasks.remove(task.id());
                 log.debug("Active task " + task.id() + " was added to the restored tasks");
                 restoredActiveTasksCondition.signalAll();
             } finally {