Handle another init race

apache · Sep 10, 2024 · 8790626 · 8790626
1 parent 4181969
commit 8790626
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metadata/src/main/java/org/apache/kafka/metadata/migration/KRaftMigrationDriver.java b/metadata/src/main/java/org/apache/kafka/metadata/migration/KRaftMigrationDriver.java
@@ -522,7 +522,7 @@ public void run() throws Exception {
             KRaftMigrationDriver.this.image = image;
             String metadataType = isSnapshot ? "snapshot" : "delta";
 
-            if (migrationState.equals(MigrationDriverState.INACTIVE)) {
+            if (EnumSet.of(MigrationDriverState.UNINITIALIZED, MigrationDriverState.INACTIVE).contains(migrationState)) {
                 // No need to log anything if this node is not the active controller
                 completionHandler.accept(null);
                 return;