Fix handling Get() result for async channel storage

initial

1 files changed, 6 insertions, 1 deletions

diff --git a/ydb/library/yql/dq/runtime/dq_output_channel.cpp b/ydb/library/yql/dq/runtime/dq_output_channel.cpp
index 0a8cf3a62af..08864c6556d 100644
--- a/ydb/library/yql/dq/runtime/dq_output_channel.cpp
+++ b/ydb/library/yql/dq/runtime/dq_output_channel.cpp
@@ -176,8 +176,13 @@ public:
         data.ClearRaw();
         if (FirstStoredId < NextStoredId) {
             YQL_ENSURE(Storage);
+            LOG("Loading spilled blob. BlobId: " << FirstStoredId);
             TBuffer blob;
-            YQL_ENSURE(Storage->Get(FirstStoredId++, blob), "Lost block in storage");
+            if (!Storage->Get(FirstStoredId, blob)) {
+                LOG("BlobId " << FirstStoredId << " not ready yet");
+                return false;
+            }
+            ++FirstStoredId;
             ui64 rows;
             YQL_ENSURE(blob.size() >= sizeof(rows));
             std::memcpy((char*)&rows, blob.data(), sizeof(rows));