fix reading logic in case memory control (#11768)

1 files changed, 10 insertions, 1 deletions

diff --git a/ydb/core/tx/columnshard/engines/reader/plain_reader/iterator/source.h b/ydb/core/tx/columnshard/engines/reader/plain_reader/iterator/source.h
index 3647bb52a3..563dab4477 100644
--- a/ydb/core/tx/columnshard/engines/reader/plain_reader/iterator/source.h
+++ b/ydb/core/tx/columnshard/engines/reader/plain_reader/iterator/source.h
@@ -75,6 +75,7 @@ protected:
     virtual bool DoStartFetchingAccessor(const std::shared_ptr<IDataSource>& sourcePtr, const TFetchingScriptCursor& step) = 0;
 
 public:
+    virtual bool NeedAccessorsForRead() const = 0;
     virtual bool NeedAccessorsFetching() const = 0;
 
     bool StartFetchingAccessor(const std::shared_ptr<IDataSource>& sourcePtr, const TFetchingScriptCursor& step) {
@@ -110,7 +111,7 @@ public:
     void SetSourceInMemory(const bool value) {
         AFL_VERIFY(!IsSourceInMemoryFlag);
         IsSourceInMemoryFlag = value;
-        if (NeedAccessorsFetching()) {
+        if (NeedAccessorsForRead()) {
             AFL_VERIFY(StageData);
             if (!value) {
                 StageData->SetUseFilter(value);
@@ -317,6 +318,10 @@ private:
     virtual bool DoStartFetchingAccessor(const std::shared_ptr<IDataSource>& sourcePtr, const TFetchingScriptCursor& step) override;
 
 public:
+    virtual bool NeedAccessorsForRead() const override {
+        return true;
+    }
+
     virtual bool NeedAccessorsFetching() const override {
         return !StageData  || !StageData->HasPortionAccessor();
     }
@@ -425,6 +430,10 @@ private:
     }
 
 public:
+    virtual bool NeedAccessorsForRead() const override {
+        return false;
+    }
+
     virtual bool NeedAccessorsFetching() const override {
         return false;
     }