Fix SegmentTransactionalReplaceAction

kfaraz · Oct 9, 2023 · 46a1b1e · 46a1b1e
1 parent b7a7e6f
commit 46a1b1e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/...main/java/org/apache/druid/indexing/common/actions/SegmentTransactionalReplaceAction.java b/...main/java/org/apache/druid/indexing/common/actions/SegmentTransactionalReplaceAction.java
@@ -33,7 +33,6 @@
 import org.apache.druid.segment.realtime.appenderator.SegmentIdWithShardSpec;
 import org.apache.druid.timeline.DataSegment;
 
-import java.util.List;
 import java.util.Set;
 import java.util.stream.Collectors;
 
@@ -123,7 +122,7 @@ public SegmentPublishResult perform(Task task, TaskActionToolbox toolbox)
       // transaction as the commit of replace segments and failure to upgrade
       // pending segments should not affect success of replace commit.
       try {
-        List<SegmentIdWithShardSpec> upgradedPendingSegments =
+        Set<SegmentIdWithShardSpec> upgradedPendingSegments =
             toolbox.getIndexerMetadataStorageCoordinator().upgradePendingSegments(segments);
         log.info(
             "Upgraded [%d] pending segments for REPLACE task[%s]: [%s]",