correction

Author: ivanmorozov333

ydb/core/tx/columnshard/data_accessor/abstract/collector.cpp

@@ -0,0 +1,5 @@
+#include "collector.h"
+
+namespace NKikimr::NOlap::NDataAccessorControl {
+
+}

ydb/core/tx/columnshard/data_accessor/abstract/collector.h

@@ -0,0 +1,32 @@
+#pragma once
+
+namespace NKikimr::NOlap::NDataAccessorControl {
+class IGranuleDataAccessor {
+private:
+    const ui64 PathId;
+
+    virtual void DoAskData(const std::shared_ptr<TDataAccessorsRequest>& request) = 0;
+    virtual void DoModifyPortions(const std::vector<TPortionDataAccessor>& add, const std::vector<ui64>& remove) = 0;
+
+public:
+    virtual ~IGranuleDataAccessor() = default;
+
+    ui64 GetPathId() const {
+        return PathId;
+    }
+
+    IGranuleDataAccessor(const ui64 pathId)
+        : PathId(pathId) {
+    }
+
+    void AskData(const std::shared_ptr<TDataAccessorsRequest>& request) {
+        AFL_VERIFY(request);
+        AFL_VERIFY(request->HasSubscriber());
+        return DoAskData(request);
+    }
+    void ModifyPortions(const std::vector<TPortionDataAccessor>& add, const std::vector<ui64>& remove) {
+        return DoModifyPortions(add, remove);
+    }
+};
+
+}   // namespace NKikimr::NOlap::NDataAccessorControl

ydb/core/tx/columnshard/data_accessor/abstract/constructor.cpp

@@ -0,0 +1,5 @@
+#include "constructor.h"
+
+namespace NKikimr::NOlap::NDataAccessorControl {
+
+}

ydb/core/tx/columnshard/data_accessor/abstract/constructor.h

@@ -0,0 +1,8 @@
+#pragma once
+
+namespace NKikimr::NOlap::NDataAccessorControl {
+class IConstructor {
+private:
+public:
+};
+}

ydb/core/tx/columnshard/data_accessor/abstract/manager.cpp

@@ -0,0 +1,5 @@
+#include "manager.h"
+
+namespace NKikimr::NOlap::NDataAccessorControl {
+
+}

ydb/core/tx/columnshard/data_accessor/abstract/manager.h

@@ -0,0 +1,24 @@
+#pragma once
+
+namespace NKikimr::NOlap {
+class TGranuleMeta;
+}
+
+namespace NKikimr::NOlap::NDataAccessorControl {
+class IMetadataMemoryManager {
+private:
+    virtual std::unique_ptr<IGranuleDataAccessor> DoBuildCollector() = 0;
+    virtual std::shared_ptr<ITxReader> DoBuildGranuleLoader(
+        const TVersionedIndex& versionedIndex, TGranuleMeta* granule, const std::shared_ptr<IBlobGroupSelector>& dsGroupSelector) = 0;
+
+public:
+    virtual std::unique_ptr<IGranuleDataAccessor> BuildCollector() override {
+        return DoBuildCollector();
+    }
+
+    std::shared_ptr<ITxReader> BuildGranuleLoader(
+        const TVersionedIndex& versionedIndex, TGranuleMeta* granule, const std::shared_ptr<IBlobGroupSelector>& dsGroupSelector) {
+        return DoBuildGranuleLoader(versionedIndex, granule, dsGroupSelector);
+    }
+};
+}

ydb/core/tx/columnshard/data_accessor/abstract/ya.make

@@ -0,0 +1,12 @@
+LIBRARY()
+
+SRCS(
+    manager.cpp
+    collector.cpp
+    constructor.cpp
+)
+
+PEERDIR(
+)
+
+END()

ydb/core/tx/columnshard/data_accessor/controller.h

@@ -0,0 +1,5 @@
+#include "collector.h"
+
+namespace NKikimr::NOlap::NDataAccessorControl::NInMem {
+
+}

ydb/core/tx/columnshard/data_accessor/in_mem/collector.cpp

@@ -0,0 +1,33 @@
+#pragma once
+
+namespace NKikimr::NOlap::NDataAccessorControl::NInMem {
+class TCollector: public IGranuleDataAccessor {
+private:
+    using TBase = IGranuleDataAccessor;
+    THashMap<ui64, TPortionDataAccessor> Accessors;
+    virtual void DoAskData(const std::shared_ptr<TDataAccessorsRequest>& request) override {
+        std::vector<TPortionDataAccessor> accessors;
+        auto& portions = request->StartFetching(GetPathId());
+        for (auto&& i : portions) {
+            auto it = Accessors.find(i->GetPortionId());
+            AFL_VERIFY(it != Accessors.end());
+            accessors.emplace_back(it->second);
+        }
+        request->AddData(GetPathId(), std::move(accessors));
+    }
+    virtual void DoModifyPortions(const std::vector<TPortionDataAccessor>& add, const std::vector<ui64>& remove) override {
+        for (auto&& i : remove) {
+            AFL_VERIFY(Accessors.erase(i));
+        }
+        for (auto&& i : add) {
+            AFL_VERIFY(Accessors.emplace(i.GetPortionInfo().GetPortionId(), i).second);
+        }
+    }
+
+public:
+    TCollector(const ui64 pathId)
+        : TBase(pathId) {
+    }
+};
+
+}   // namespace NKikimr::NOlap::NDataAccessorControl