Fix overflow case in stream index lookup join (ydb-platform#11818)

GrigoriyPA · Nov 20, 2024 · 41baadd · 41baadd
1 parent 779684a
commit 41baadd
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/ydb/core/kqp/runtime/kqp_stream_lookup_worker.cpp b/ydb/core/kqp/runtime/kqp_stream_lookup_worker.cpp
@@ -231,7 +231,7 @@ class TKqpLookupRows : public TKqpStreamLookupWorker {
         }
     }
 
-    std::vector<THolder<TEvDataShard::TEvRead>> RebuildRequest(const ui64& prevReadId, ui32 firstUnprocessedQuery, 
+    std::vector<THolder<TEvDataShard::TEvRead>> RebuildRequest(const ui64& prevReadId, ui32 firstUnprocessedQuery,
         TMaybe<TOwnedCellVec> lastProcessedKey, ui64& newReadId) final {
 
         auto it = PendingKeysByReadId.find(prevReadId);
@@ -280,7 +280,7 @@ class TKqpLookupRows : public TKqpStreamLookupWorker {
             requests.emplace_back(std::move(request));
             PendingKeysByReadId.insert({newReadId, std::move(unprocessedRanges)});
         }
-        
+
         return requests;
     }
 
@@ -495,7 +495,7 @@ class TKqpJoinRows : public TKqpStreamLookupWorker {
         UnprocessedRows.emplace_back(std::make_pair(TOwnedCellVec(joinKeyCells), std::move(inputRow.GetElement(1))));
     }
 
-    std::vector<THolder<TEvDataShard::TEvRead>> RebuildRequest(const ui64& prevReadId, ui32 firstUnprocessedQuery, 
+    std::vector<THolder<TEvDataShard::TEvRead>> RebuildRequest(const ui64& prevReadId, ui32 firstUnprocessedQuery,
         TMaybe<TOwnedCellVec> lastProcessedKey, ui64& newReadId) final {
 
         auto readIt = PendingKeysByReadId.find(prevReadId);
@@ -934,6 +934,7 @@ class TKqpJoinRows : public TKqpStreamLookupWorker {
         if (rowStats.ResultBytesCount > (ui64)freeSpace) {
             resultRow.DeleteUnreferenced();
             rowStats.Clear();
+            return NUdf::TUnboxedValuePod();
         }
 
         return resultRow;