StreamAppenderator: Add sink prior to announcing it. (#17649)

Fixes #17633.
apache · Jan 22, 2025 · a964220 · a964220
1 parent a805f4b
commit a964220
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/server/src/main/java/org/apache/druid/segment/realtime/appenderator/StreamAppenderator.java b/server/src/main/java/org/apache/druid/segment/realtime/appenderator/StreamAppenderator.java
@@ -528,6 +528,9 @@ private Sink getOrCreateSink(final SegmentIdWithShardSpec identifier)
       );
       bytesCurrentlyInMemory.addAndGet(calculateSinkMemoryInUsed(retVal));
 
+      // Add sink prior to announcing it, to ensure it is immediately queryable.
+      addSink(identifier, retVal);
+
       try {
         segmentAnnouncer.announceSegment(retVal.getSegment());
       }
@@ -536,8 +539,6 @@ private Sink getOrCreateSink(final SegmentIdWithShardSpec identifier)
            .addData("interval", retVal.getInterval())
            .emit();
       }
-
-      addSink(identifier, retVal);
     }
 
     return retVal;