fix tx volume calculation for inserted chunks (#10813)

ydb-platform · Oct 24, 2024 · c71ca85 · c71ca85
1 parent 71df826
commit c71ca85
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/ydb/core/tx/columnshard/engines/insert_table/meta.h b/ydb/core/tx/columnshard/engines/insert_table/meta.h
@@ -25,7 +25,8 @@ class TInsertedDataMeta {
 
 public:
     ui64 GetTxVolume() const {
-        return 2 * sizeof(ui64) + sizeof(ui32) + sizeof(OriginalProto) + (SpecialKeysParsed ? SpecialKeysParsed->GetMemoryBytes() : 0);
+        return 512 + 2 * sizeof(ui64) + sizeof(ui32) + sizeof(OriginalProto) + (SpecialKeysParsed ? SpecialKeysParsed->GetMemoryBytes() : 0) +
+               SchemaSubset.GetTxVolume();
     }
 
     TInsertedDataMeta(const NKikimrTxColumnShard::TLogicalMetadata& proto)

diff --git a/ydb/library/formats/arrow/modifier/subset.h b/ydb/library/formats/arrow/modifier/subset.h
@@ -14,6 +14,10 @@ class TSchemaSubset {
     TSchemaSubset() = default;
     TSchemaSubset(const std::set<ui32>& fieldsIdx, const ui32 fieldsCount);
 
+    ui64 GetTxVolume() const {
+        return FieldBits.size() + FieldIdx.size() * sizeof(ui32) + 1;
+    }
+
     static TSchemaSubset AllFieldsAccepted() {
         TSchemaSubset result;
         result.Exclude = true;