From 95521e23b4a3614de9033169b0d0ad9b3d482e43 Mon Sep 17 00:00:00 2001
From: Kashif Faraz <kashif.faraz@gmail.com>
Date: Thu, 5 Dec 2024 07:50:33 +0530
Subject: [PATCH] Fix NPE in segment allocation when reduceMetadataIO is true

---
 .../druid/metadata/IndexerSQLMetadataStorageCoordinator.java    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/server/src/main/java/org/apache/druid/metadata/IndexerSQLMetadataStorageCoordinator.java b/server/src/main/java/org/apache/druid/metadata/IndexerSQLMetadataStorageCoordinator.java
index e3a3ee047088..69d86ab80cf8 100644
--- a/server/src/main/java/org/apache/druid/metadata/IndexerSQLMetadataStorageCoordinator.java
+++ b/server/src/main/java/org/apache/druid/metadata/IndexerSQLMetadataStorageCoordinator.java
@@ -2943,7 +2943,7 @@ Set<DataSegment> retrieveUsedSegmentsForAllocation(
           = versionIntervalToSmallestSegmentId.computeIfAbsent(segmentId.getVersion(), v -> new HashMap<>());
       final SegmentId value = map.get(segmentId.getInterval());
       if (value == null || value.getPartitionNum() > segmentId.getPartitionNum()) {
-        map.put(interval, segmentId);
+        map.put(segmentId.getInterval(), segmentId);
       }
     }