reuse code for portion meta (#13065)

ydb-platform · Dec 28, 2024 · 03f9a13 · 03f9a13
1 parent 7155bd1
commit 03f9a13
Show file tree

Hide file tree

Showing 4 changed files with 55 additions and 66 deletions.
diff --git a/ydb/core/tx/columnshard/engines/portions/constructor_meta.cpp b/ydb/core/tx/columnshard/engines/portions/constructor_meta.cpp
@@ -49,7 +49,7 @@ TPortionMeta TPortionMetaConstructor::Build() {
     if (TierName) {
         result.TierName = *TierName;
     }
-    AFL_VERIFY(BlobIds.size());
+    TBase::FullValidation();
     result.BlobIds = BlobIds;
     result.BlobIds.shrink_to_fit();
     result.CompactionLevel = *TValidator::CheckNotNull(CompactionLevel);

diff --git a/ydb/core/tx/columnshard/engines/portions/constructor_meta.h b/ydb/core/tx/columnshard/engines/portions/constructor_meta.h
@@ -10,8 +10,9 @@ namespace NKikimr::NOlap {
 class TPortionInfoConstructor;
 struct TIndexInfo;
 
-class TPortionMetaConstructor {
+class TPortionMetaConstructor: public TPortionMetaBase {
 private:
+    using TBase = TPortionMetaBase;
     std::optional<NArrow::TFirstLastSpecialKeys> FirstAndLastPK;
     std::optional<TString> TierName;
     std::optional<TSnapshot> RecordSnapshotMin;
@@ -27,8 +28,6 @@ class TPortionMetaConstructor {
 
     std::optional<ui32> DeletionsCount;
 
-    std::vector<TUnifiedBlobId> BlobIds;
-
     friend class TPortionInfoConstructor;
     friend class TPortionAccessorConstructor;
     void FillMetaInfo(const NArrow::TFirstLastSpecialKeys& primaryKeys, const ui32 deletionsCount,
@@ -42,15 +41,6 @@ class TPortionMetaConstructor {
         return linkRange.RestoreRange(GetBlobId(linkRange.GetBlobIdxVerified()));
     }
 
-    ui32 GetBlobIdsCount() const {
-        return BlobIds.size();
-    }
-
-    const TUnifiedBlobId& GetBlobId(const TBlobRangeLink16::TLinkId linkId) const {
-        AFL_VERIFY(linkId < BlobIds.size());
-        return BlobIds[linkId];
-    }
-
     TBlobRangeLink16::TLinkId RegisterBlobId(const TUnifiedBlobId& blobId) {
         AFL_VERIFY(blobId.IsValid());
         TBlobRangeLink16::TLinkId idx = 0;
@@ -64,24 +54,6 @@ class TPortionMetaConstructor {
         return idx;
     }
 
-    std::optional<TBlobRangeLink16::TLinkId> GetBlobIdxOptional(const TUnifiedBlobId& blobId) const {
-        AFL_VERIFY(blobId.IsValid());
-        TBlobRangeLink16::TLinkId idx = 0;
-        for (auto&& i : BlobIds) {
-            if (i == blobId) {
-                return idx;
-            }
-            ++idx;
-        }
-        return std::nullopt;
-    }
-
-    TBlobRangeLink16::TLinkId GetBlobIdxVerified(const TUnifiedBlobId& blobId) const {
-        auto result = GetBlobIdxOptional(blobId);
-        AFL_VERIFY(result);
-        return *result;
-    }
-
     void SetCompactionLevel(const ui64 level) {
         CompactionLevel = level;
     }

diff --git a/ydb/core/tx/columnshard/engines/portions/meta.cpp b/ydb/core/tx/columnshard/engines/portions/meta.cpp
@@ -44,7 +44,7 @@ NKikimrTxColumnShard::TIndexPortionMeta TPortionMeta::SerializeToProto() const {
 
     RecordSnapshotMin.SerializeToProto(*portionMeta.MutableRecordSnapshotMin());
     RecordSnapshotMax.SerializeToProto(*portionMeta.MutableRecordSnapshotMax());
-    for (auto&& i : BlobIds) {
+    for (auto&& i : GetBlobIds()) {
         *portionMeta.AddBlobIds() = i.GetLogoBlobId().AsBinaryString();
     }
     return portionMeta;

diff --git a/ydb/core/tx/columnshard/engines/portions/meta.h b/ydb/core/tx/columnshard/engines/portions/meta.h
@@ -14,8 +14,56 @@ namespace NKikimr::NOlap {
 
 struct TIndexInfo;
 
-struct TPortionMeta {
+class TPortionMetaBase {
+protected:
+    std::vector<TUnifiedBlobId> BlobIds;
+public:
+    const std::vector<TUnifiedBlobId>& GetBlobIds() const {
+        return BlobIds;
+    }
+
+    const TUnifiedBlobId& GetBlobId(const TBlobRangeLink16::TLinkId linkId) const {
+        AFL_VERIFY(linkId < GetBlobIds().size());
+        return BlobIds[linkId];
+    }
+
+    ui32 GetBlobIdsCount() const {
+        return BlobIds.size();
+    }
+
+    void FullValidation() const {
+        for (auto&& i : BlobIds) {
+            AFL_VERIFY(i.BlobSize());
+        }
+        AFL_VERIFY(BlobIds.size());
+    }
+
+    std::optional<TBlobRangeLink16::TLinkId> GetBlobIdxOptional(const TUnifiedBlobId& blobId) const {
+        AFL_VERIFY(blobId.IsValid());
+        TBlobRangeLink16::TLinkId idx = 0;
+        for (auto&& i : BlobIds) {
+            if (i == blobId) {
+                return idx;
+            }
+            ++idx;
+        }
+        return std::nullopt;
+    }
+
+    ui64 GetMetadataMemorySize() const {
+        return GetBlobIds().size() * sizeof(TUnifiedBlobId);
+    }
+
+    TBlobRangeLink16::TLinkId GetBlobIdxVerified(const TUnifiedBlobId& blobId) const {
+        auto result = GetBlobIdxOptional(blobId);
+        AFL_VERIFY(result);
+        return *result;
+    }
+};
+
+class TPortionMeta: public TPortionMetaBase {
 private:
+    using TBase = TPortionMetaBase;
     NArrow::TFirstLastSpecialKeys ReplaceKeyEdges;   // first and last PK rows
     YDB_READONLY_DEF(TString, TierName);
     YDB_READONLY(ui32, DeletionsCount, 0);
@@ -25,7 +73,6 @@ struct TPortionMeta {
     YDB_READONLY(ui32, ColumnBlobBytes, 0);
     YDB_READONLY(ui32, IndexRawBytes, 0);
     YDB_READONLY(ui32, IndexBlobBytes, 0);
-    YDB_READONLY_DEF(std::vector<TUnifiedBlobId>, BlobIds);
 
     friend class TPortionMetaConstructor;
     friend class TPortionInfo;
@@ -41,10 +88,7 @@ struct TPortionMeta {
     TSnapshot RecordSnapshotMax;
 
     void FullValidation() const {
-        for (auto&& i : BlobIds) {
-            AFL_VERIFY(i.BlobSize());
-        }
-        AFL_VERIFY(BlobIds.size());
+        TBase::FullValidation();
         AFL_VERIFY(RecordsCount);
         AFL_VERIFY(ColumnRawBytes);
         AFL_VERIFY(ColumnBlobBytes);
@@ -55,37 +99,10 @@ struct TPortionMeta {
         return ReplaceKeyEdges;
     }
 
-    const TUnifiedBlobId& GetBlobId(const TBlobRangeLink16::TLinkId linkId) const {
-        AFL_VERIFY(linkId < GetBlobIds().size());
-        return BlobIds[linkId];
-    }
-
-    ui32 GetBlobIdsCount() const {
-        return BlobIds.size();
-    }
-
     void ResetCompactionLevel(const ui32 level) {
         CompactionLevel = level;
     }
 
-    std::optional<TBlobRangeLink16::TLinkId> GetBlobIdxOptional(const TUnifiedBlobId& blobId) const {
-        AFL_VERIFY(blobId.IsValid());
-        TBlobRangeLink16::TLinkId idx = 0;
-        for (auto&& i : BlobIds) {
-            if (i == blobId) {
-                return idx;
-            }
-            ++idx;
-        }
-        return std::nullopt;
-    }
-
-    TBlobRangeLink16::TLinkId GetBlobIdxVerified(const TUnifiedBlobId& blobId) const {
-        auto result = GetBlobIdxOptional(blobId);
-        AFL_VERIFY(result);
-        return *result;
-    }
-
     using EProduced = NPortion::EProduced;
 
     NArrow::TReplaceKey IndexKeyStart;
@@ -96,7 +113,7 @@ struct TPortionMeta {
     std::optional<TString> GetTierNameOptional() const;
 
     ui64 GetMetadataMemorySize() const {
-        return sizeof(TPortionMeta) + ReplaceKeyEdges.GetMemorySize() + BlobIds.size() * sizeof(TUnifiedBlobId);
+        return sizeof(TPortionMeta) + ReplaceKeyEdges.GetMemorySize() + TBase::GetMetadataMemorySize();
     }
 
     NKikimrTxColumnShard::TIndexPortionMeta SerializeToProto() const;