fix stream lookup bytes calculation (ydb-platform#12026) (ydb-platfor…

…m#12101)
zinal · Dec 3, 2024 · 5734702 · 5734702
1 parent c861790
commit 5734702
Showing 1 changed file with 3 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
@@ -377,9 +377,7 @@ class TKqpLookupRows : public TKqpStreamLookupWorker {
                     }
                 }
 
-                storageRowSize = std::max(storageRowSize, (i64)8);
-
-                if (rowsInBatch && storageRowSize > freeSpace - (i64)resultStats.ResultBytesCount) {
+                if (rowsInBatch && rowSize > freeSpace - (i64)resultStats.ResultBytesCount) {
                     row.DeleteUnreferenced();
                     sizeLimitExceeded = true;
                     break;
@@ -388,6 +386,8 @@ class TKqpLookupRows : public TKqpStreamLookupWorker {
                 batch.push_back(std::move(row));
                 ++rowsInBatch;
 
+                storageRowSize = std::max(storageRowSize, (i64)8);
+
                 resultStats.ReadRowsCount += 1;
                 resultStats.ReadBytesCount += storageRowSize;
                 resultStats.ResultRowsCount += 1;