compaction priorities weight (#21675)

Co-authored-by: ivanmorozov333 <[email protected]>

Author: ivanmorozov333

ydb/core/protos/flat_scheme_op.proto

@@ -577,6 +577,7 @@ message TCompactionSelectorConstructorContainer {
 message TCompactionPlannerConstructorContainer {
     optional string ClassName = 1;
     optional uint64 NodePortionsCountLimit = 2;
+    optional double WeightKff = 3 [default = 1.0];
 
     message TLOptimizer {
 

ydb/core/tx/columnshard/engines/storage/optimizer/abstract/optimizer.h

@@ -36,6 +36,10 @@ class TOptimizationPriority {
     }
 
 public:
+    void Mul(const ui32 kff) {
+        InternalLevelWeight *= kff;
+    }
+
     ui64 GetGeneralPriority() const {
         return ((ui64)Level << 56) + InternalLevelWeight;
     }
@@ -88,13 +92,15 @@ using TPortionInfoForCompaction = NPortion::TPortionInfoForCompaction;
 
 class IOptimizerPlanner {
 private:
+    friend class IOptimizerPlannerConstructor;
     const TInternalPathId PathId;
     YDB_READONLY(TInstant, ActualizationInstant, TInstant::Zero());
 
     virtual bool DoIsOverloaded() const {
         return false;
     }
     const ui32 NodePortionsCountLimit = 0;
+    double WeightKff = 1;
     static inline TAtomicCounter NodePortionsCounter = 0;
     TPositiveControlInteger LocalPortionsCount;
     std::shared_ptr<TCounters> Counters = std::make_shared<TCounters>();
@@ -191,7 +197,9 @@ class IOptimizerPlanner {
     std::shared_ptr<TColumnEngineChanges> GetOptimizationTask(
         std::shared_ptr<TGranuleMeta> granule, const std::shared_ptr<NDataLocks::TManager>& dataLocksManager) const;
     TOptimizationPriority GetUsefulMetric() const {
-        return DoGetUsefulMetric();
+        auto result = DoGetUsefulMetric();
+        result.Mul(WeightKff);
+        return result;
     }
     void Actualize(const TInstant currentInstant) {
         ActualizationInstant = currentInstant;
@@ -228,6 +236,7 @@ class IOptimizerPlannerConstructor {
 
 private:
     ui32 NodePortionsCountLimit = 1000000;
+    double WeightKff = 1.0;
 
     virtual TConclusion<std::shared_ptr<IOptimizerPlanner>> DoBuildPlanner(const TBuildContext& context) const = 0;
     virtual void DoSerializeToProto(TProto& proto) const = 0;
@@ -263,16 +272,29 @@ class IOptimizerPlannerConstructor {
             }
             NodePortionsCountLimit = jsonValue.GetUInteger();
         }
+        if (jsonInfo.Has("weight_kff")) {
+            const auto& jsonValue = jsonInfo["weight_kff"];
+            if (!jsonValue.IsDouble()) {
+                return TConclusionStatus::Fail("incorrect weight_kff value have to be double");
+            }
+            WeightKff = jsonValue.GetDouble();
+        }
         return DoDeserializeFromJson(jsonInfo);
     }
 
     TConclusion<std::shared_ptr<IOptimizerPlanner>> BuildPlanner(const TBuildContext& context) const {
-        return DoBuildPlanner(context);
+        auto result = DoBuildPlanner(context);
+        if (result.IsFail()) {
+            return result;
+        }
+        (*result)->WeightKff = WeightKff;
+        return result;
     }
 
     virtual TString GetClassName() const = 0;
     void SerializeToProto(TProto& proto) const {
         proto.SetNodePortionsCountLimit(NodePortionsCountLimit);
+        proto.SetWeightKff(WeightKff);
         DoSerializeToProto(proto);
     }
 
@@ -292,6 +314,9 @@ class IOptimizerPlannerConstructor {
         if (proto.HasNodePortionsCountLimit()) {
             NodePortionsCountLimit = proto.GetNodePortionsCountLimit();
         }
+        if (proto.HasWeightKff()) {
+            WeightKff = proto.GetWeightKff();
+        }
         return DoDeserializeFromProto(proto);
     }
 };