Optimize perfomance of creating topic write sessions (#10085)

nshestakov · web-flow · commit 605e422002e7 · 2024-10-07T15:52:39.000+03:00
diff --git a/ydb/core/persqueue/partition_scale_manager.cpp b/ydb/core/persqueue/partition_scale_manager.cpp
@@ -11,11 +11,13 @@ TPartitionScaleManager::TPartitionScaleManager(
     const TString& databasePath,
     ui64 pathId,
     int version,
-    const NKikimrPQ::TPQTabletConfig& config
+    const NKikimrPQ::TPQTabletConfig& config,
+    const TPartitionGraph& partitionGraph
 )
     : TopicName(topicName)
     , DatabasePath(databasePath)
-    , BalancerConfig(pathId, version, config) {
+    , BalancerConfig(pathId, version, config)
+    , PartitionGraph(partitionGraph) {
     }
 
 void TPartitionScaleManager::HandleScaleStatusChange(const ui32 partitionId, NKikimrPQ::EScaleStatus scaleStatus, const TActorContext& ctx) {
@@ -69,7 +71,7 @@ std::pair<std::vector<TPartitionSplit>, std::vector<TPartitionMerge>> TPartition
     size_t allowedSplitsCount = BalancerConfig.MaxActivePartitions > BalancerConfig.CurPartitions ? BalancerConfig.MaxActivePartitions - BalancerConfig.CurPartitions : 0;
     auto partitionId = PartitionsToSplit.begin();
     while (allowedSplitsCount > 0 && partitionId != PartitionsToSplit.end()) {
-        auto* node = BalancerConfig.PartitionGraph.GetPartition(*partitionId);
+        auto* node = PartitionGraph.GetPartition(*partitionId);
         if (node->Children.empty()) {
             auto from = node->From;
             auto to = node->To;
diff --git a/ydb/core/persqueue/partition_scale_manager.h b/ydb/core/persqueue/partition_scale_manager.h
@@ -30,7 +30,6 @@ class TPartitionScaleManager {
         )
             : PathId(pathId)
             , PathVersion(version)
-            , PartitionGraph(MakePartitionGraph(config))
             , MaxActivePartitions(config.GetPartitionStrategy().GetMaxPartitionCount())
             , MinActivePartitions(config.GetPartitionStrategy().GetMinPartitionCount())
             , CurPartitions(std::count_if(config.GetAllPartitions().begin(), config.GetAllPartitions().end(), [](auto& p) {
@@ -40,14 +39,13 @@ class TPartitionScaleManager {
 
         ui64 PathId;
         int PathVersion;
-        TPartitionGraph PartitionGraph;
         ui64 MaxActivePartitions;
         ui64 MinActivePartitions;
         ui64 CurPartitions;
     };
 
 public:
-    TPartitionScaleManager(const TString& topicPath, const TString& databasePath, ui64 pathId, int version, const NKikimrPQ::TPQTabletConfig& config);
+    TPartitionScaleManager(const TString& topicPath, const TString& databasePath, ui64 pathId, int version, const NKikimrPQ::TPQTabletConfig& config, const TPartitionGraph& partitionGraph);
 
 public:
     void HandleScaleStatusChange(const ui32 partition, NKikimrPQ::EScaleStatus scaleStatus, const TActorContext& ctx);
@@ -79,6 +77,8 @@ class TPartitionScaleManager {
     std::unordered_set<ui32> PartitionsToSplit;
 
     TBalancerConfig BalancerConfig;
+    const TPartitionGraph& PartitionGraph;
+
     bool RequestInflight = false;
 };
 
diff --git a/ydb/core/persqueue/read_balancer.cpp b/ydb/core/persqueue/read_balancer.cpp
@@ -537,7 +537,7 @@ void TPersQueueReadBalancer::Handle(TEvPersQueue::TEvUpdateBalancerConfig::TPtr
 
     if (SplitMergeEnabled(TabletConfig)) {
         if (!PartitionsScaleManager) {
-            PartitionsScaleManager = std::make_unique<TPartitionScaleManager>(Topic, DatabasePath, PathId, Version, TabletConfig);
+            PartitionsScaleManager = std::make_unique<TPartitionScaleManager>(Topic, DatabasePath, PathId, Version, TabletConfig, PartitionGraph);
         } else {
             PartitionsScaleManager->UpdateBalancerConfig(PathId, Version, TabletConfig);
         }
diff --git a/ydb/core/persqueue/read_balancer__txinit.h b/ydb/core/persqueue/read_balancer__txinit.h
@@ -60,7 +60,7 @@ struct TPersQueueReadBalancer::TTxInit : public ITransaction {
                     Self->PartitionGraph = MakePartitionGraph(Self->TabletConfig);
 
                     if (SplitMergeEnabled(Self->TabletConfig)) {
-                        Self->PartitionsScaleManager = std::make_unique<TPartitionScaleManager>(Self->Topic, Self->DatabasePath, Self->PathId, Self->Version, Self->TabletConfig);
+                        Self->PartitionsScaleManager = std::make_unique<TPartitionScaleManager>(Self->Topic, Self->DatabasePath, Self->PathId, Self->Version, Self->TabletConfig, Self->PartitionGraph);
                     }
                     Self->UpdateConfigCounters();
                 }
diff --git a/ydb/core/persqueue/ut/partition_chooser_ut.cpp b/ydb/core/persqueue/ut/partition_chooser_ut.cpp
@@ -162,7 +162,6 @@ Y_UNIT_TEST(THashChooser_GetTabletIdTest) {
     UNIT_ASSERT_VALUES_EQUAL(chooser.GetPartition(2)->PartitionId, 2);
 
     // Not found
-    UNIT_ASSERT(!chooser.GetPartition(3));
     UNIT_ASSERT(!chooser.GetPartition(666));
 }
 
@@ -209,13 +208,17 @@ TWriteSessionMock* ChoosePartition(NPersQueue::TTestServer& server,
     NPersQueue::TTopicConverterPtr fullConverter = CreateTopicConverter();
     TWriteSessionMock* mock = new TWriteSessionMock();
 
+    auto chooser = NPQ::CreatePartitionChooser(config, true);
+    auto graph = std::make_shared<NPQ::TPartitionGraph>(NPQ::MakePartitionGraph(config));
+
     NActors::TActorId parentId = server.GetRuntime()->Register(mock);
-    server.GetRuntime()->Register(NKikimr::NPQ::CreatePartitionChooserActorM(parentId,
+    server.GetRuntime()->Register(NKikimr::NPQ::CreatePartitionChooserActor<NTabletPipe::NTest::TPipeMock>(parentId,
                                                                                    config,
+                                                                                   chooser,
+                                                                                   graph,
                                                                                    fullConverter,
                                                                                    sourceId,
-                                                                                   preferedPartition,
-                                                                                   true));
+                                                                                   preferedPartition));
 
     mock->Promise.GetFuture().GetValueSync();
 
@@ -679,20 +682,6 @@ Y_UNIT_TEST(TPartitionChooserActor_SplitMergeDisabled_RegisteredSourceId_Test) {
     UNIT_ASSERT_VALUES_EQUAL(r->Result->Get()->PartitionId, 1);
 }
 
-Y_UNIT_TEST(TPartitionChooserActor_SplitMergeDisabled_Inactive_Test) {
-    NPersQueue::TTestServer server = CreateServer();
-
-    auto config = CreateConfig0(false);
-    AddPartition(config, 0, {}, {}, {1});
-    AddPartition(config, 1);
-
-    WriteToTable(server, "A_Source", 0);
-    auto r = ChoosePartition(server, config, "A_Source");
-
-    UNIT_ASSERT(r->Result);
-    UNIT_ASSERT_VALUES_EQUAL(r->Result->Get()->PartitionId, 1);
-}
-
 Y_UNIT_TEST(TPartitionChooserActor_SplitMergeDisabled_PreferedPartition_Test) {
     NPersQueue::TTestServer server = CreateServer();
 
@@ -706,23 +695,11 @@ Y_UNIT_TEST(TPartitionChooserActor_SplitMergeDisabled_PreferedPartition_Test) {
     UNIT_ASSERT_VALUES_EQUAL(r->Result->Get()->PartitionId, 0);
 }
 
-Y_UNIT_TEST(TPartitionChooserActor_SplitMergeDisabled_PreferedPartition_Inactive_Test) {
-    NPersQueue::TTestServer server = CreateServer();
-
-    auto config = CreateConfig0(false);
-    AddPartition(config, 0, {}, {}, {1});
-    AddPartition(config, 1);
-
-    auto r = ChoosePartition(server, config, "A_Source", 0);
-
-    UNIT_ASSERT(r->Error);
-}
-
 Y_UNIT_TEST(TPartitionChooserActor_SplitMergeDisabled_BadSourceId_Test) {
     NPersQueue::TTestServer server = CreateServer();
 
     auto config = CreateConfig0(false);
-    AddPartition(config, 0, {}, {});
+    AddPartition(config, 0);
 
     auto r = ChoosePartition(server, config, "base64:a***");
 
diff --git a/ydb/core/persqueue/writer/partition_chooser.h b/ydb/core/persqueue/writer/partition_chooser.h
@@ -2,6 +2,8 @@
 
 #include <ydb/library/actors/core/actor.h>
 #include <ydb/core/base/events.h>
+#include <ydb/core/persqueue/utils.h>
+#include <ydb/core/persqueue/writer/pipe_utils.h>
 #include <ydb/core/persqueue/writer/source_id_encoding.h>
 #include <ydb/core/protos/flat_scheme_op.pb.h>
 #include <ydb/library/persqueue/topic_parser/topic_parser.h>
@@ -53,7 +55,7 @@ struct TEvPartitionChooser {
 class IPartitionChooser {
 public:
     struct TPartitionInfo {
-        TPartitionInfo(ui32 partitionId, ui64 tabletId)
+        explicit TPartitionInfo(ui32 partitionId = 0, ui64 tabletId = 0)
             : PartitionId(partitionId)
             , TabletId(tabletId) {}
 
@@ -70,11 +72,13 @@ class IPartitionChooser {
 
 std::shared_ptr<IPartitionChooser> CreatePartitionChooser(const NKikimrSchemeOp::TPersQueueGroupDescription& config, bool withoutHash = false);
 
+template<typename TPipeHelper = NTabletPipe::TPipeHelper>
 NActors::IActor* CreatePartitionChooserActor(TActorId parentId,
                                              const NKikimrSchemeOp::TPersQueueGroupDescription& config,
+                                             const std::shared_ptr<NPQ::IPartitionChooser>& chooser,
+                                             const std::shared_ptr<NPQ::TPartitionGraph>& graph,
                                              NPersQueue::TTopicConverterPtr& fullConverter,
                                              const TString& sourceId,
-                                             std::optional<ui32> preferedPartition,
-                                             bool withoutHash = false);
+                                             std::optional<ui32> preferedPartition);
 
 } // namespace NKikimr::NPQ
diff --git a/ydb/core/persqueue/writer/partition_chooser_impl.cpp b/ydb/core/persqueue/writer/partition_chooser_impl.cpp
@@ -47,20 +47,39 @@ std::shared_ptr<IPartitionChooser> CreatePartitionChooser(const NKikimrSchemeOp:
     }
 }
 
+template<typename TPipeHelper>
 IActor* CreatePartitionChooserActor(TActorId parentId,
                                     const NKikimrSchemeOp::TPersQueueGroupDescription& config,
+                                    const std::shared_ptr<NPQ::IPartitionChooser>& chooser,
+                                    const std::shared_ptr<NPQ::TPartitionGraph>& graph,
                                     NPersQueue::TTopicConverterPtr& fullConverter,
                                     const TString& sourceId,
-                                    std::optional<ui32> preferedPartition,
-                                    bool withoutHash) {
-    auto chooser = CreatePartitionChooser(config, withoutHash);
+                                    std::optional<ui32> preferedPartition) {
     if (SplitMergeEnabled(config.GetPQTabletConfig())) {
-        return new NPartitionChooser::TSMPartitionChooserActor<NTabletPipe::TPipeHelper>(parentId, config, chooser, fullConverter, sourceId, preferedPartition);
+        return new NPartitionChooser::TSMPartitionChooserActor<TPipeHelper>(parentId, chooser, graph, fullConverter, sourceId, preferedPartition);
     } else {
-        return new NPartitionChooser::TPartitionChooserActor<NTabletPipe::TPipeHelper>(parentId, config, chooser, fullConverter, sourceId, preferedPartition);
+        return new NPartitionChooser::TPartitionChooserActor<TPipeHelper>(parentId, config, chooser, fullConverter, sourceId, preferedPartition);
     }
 }
 
+template
+IActor* CreatePartitionChooserActor<NTabletPipe::TPipeHelper>(TActorId parentId,
+                                    const NKikimrSchemeOp::TPersQueueGroupDescription& config,
+                                    const std::shared_ptr<NPQ::IPartitionChooser>& chooser,
+                                    const std::shared_ptr<NPQ::TPartitionGraph>& graph,
+                                    NPersQueue::TTopicConverterPtr& fullConverter,
+                                    const TString& sourceId,
+                                    std::optional<ui32> preferedPartition);
+
+template
+IActor* CreatePartitionChooserActor<NTabletPipe::NTest::TPipeMock>(TActorId parentId,
+                                    const NKikimrSchemeOp::TPersQueueGroupDescription& config,
+                                    const std::shared_ptr<NPQ::IPartitionChooser>& chooser,
+                                    const std::shared_ptr<NPQ::TPartitionGraph>& graph,
+                                    NPersQueue::TTopicConverterPtr& fullConverter,
+                                    const TString& sourceId,
+                                    std::optional<ui32> preferedPartition);
+
 } // namespace NKikimr::NPQ
 
 std::unordered_map<ui64, NActors::TActorId> NKikimr::NTabletPipe::NTest::TPipeMock::Tablets;
diff --git a/ydb/core/persqueue/writer/partition_chooser_impl.h b/ydb/core/persqueue/writer/partition_chooser_impl.h
@@ -121,15 +121,10 @@ const typename TBoundaryChooser<THasher>::TPartitionInfo* TBoundaryChooser<THash
 //
 template<class THasher>
 THashChooser<THasher>::THashChooser(const NKikimrSchemeOp::TPersQueueGroupDescription& config) {
+    Partitions.resize(config.GetPartitions().size());
     for(const auto& p : config.GetPartitions()) {
-        if (NKikimrPQ::ETopicPartitionStatus::Active == p.GetStatus()) {
-            Partitions.emplace_back(TPartitionInfo{p.GetPartitionId(),
-                                    p.GetTabletId()});
-        }
+        Partitions[p.GetPartitionId()] = TPartitionInfo{p.GetPartitionId(), p.GetTabletId()};
     }
-
-    std::sort(Partitions.begin(), Partitions.end(),
-        [](const TPartitionInfo& a, const TPartitionInfo& b) { return a.PartitionId < b.PartitionId; });
 }
 
 template<class THasher>
@@ -142,12 +137,10 @@ const typename THashChooser<THasher>::TPartitionInfo* THashChooser<THasher>::Get
 
 template<class THasher>
 const typename THashChooser<THasher>::TPartitionInfo* THashChooser<THasher>::GetPartition(ui32 partitionId) const {
-    auto it = std::lower_bound(Partitions.begin(), Partitions.end(), partitionId,
-                    [](const TPartitionInfo& partition, const ui32 value) { return value > partition.PartitionId; });
-    if (it == Partitions.end()) {
+    if (partitionId >= Partitions.size()) {
         return nullptr;
     }
-    return it->PartitionId == partitionId ? it : nullptr;
+    return &Partitions[partitionId];
 }
 
 template<class THasher>
@@ -161,21 +154,4 @@ const typename THashChooser<THasher>::TPartitionInfo* THashChooser<THasher>::Get
 
 
 } // namespace NPartitionChooser
-
-
-inline IActor* CreatePartitionChooserActorM(TActorId parentId,
-                                    const NKikimrSchemeOp::TPersQueueGroupDescription& config,
-                                    NPersQueue::TTopicConverterPtr& fullConverter,
-                                    const TString& sourceId,
-                                    std::optional<ui32> preferedPartition,
-                                    bool withoutHash) {
-    auto chooser = CreatePartitionChooser(config, withoutHash);
-    if (SplitMergeEnabled(config.GetPQTabletConfig())) {
-        return new NPartitionChooser::TSMPartitionChooserActor<NTabletPipe::NTest::TPipeMock>(parentId, config, chooser, fullConverter, sourceId, preferedPartition);
-    } else {
-        return new NPartitionChooser::TPartitionChooserActor<NTabletPipe::NTest::TPipeMock>(parentId, config, chooser, fullConverter, sourceId, preferedPartition);
-    }
-}
-
-
 } // namespace NKikimr::NPQ
diff --git a/ydb/core/persqueue/writer/partition_chooser_impl__abstract_chooser_actor.h b/ydb/core/persqueue/writer/partition_chooser_impl__abstract_chooser_actor.h
@@ -38,7 +38,7 @@ class TAbstractPartitionChooserActor: public TActorBootstrapped<TDerived> {
 
 
     TAbstractPartitionChooserActor(TActorId parentId,
-                                   std::shared_ptr<IPartitionChooser>& chooser,
+                                   const std::shared_ptr<IPartitionChooser>& chooser,
                                    NPersQueue::TTopicConverterPtr& fullConverter,
                                    const TString& sourceId,
                                    std::optional<ui32> preferedPartition)
diff --git a/ydb/core/persqueue/writer/partition_chooser_impl__old_chooser_actor.h b/ydb/core/persqueue/writer/partition_chooser_impl__old_chooser_actor.h
@@ -28,7 +28,7 @@ class TPartitionChooserActor: public TAbstractPartitionChooserActor<TPartitionCh
 
     TPartitionChooserActor(TActorId parentId,
                            const NKikimrSchemeOp::TPersQueueGroupDescription& config,
-                           std::shared_ptr<IPartitionChooser>& chooser,
+                           const std::shared_ptr<IPartitionChooser>& chooser,
                            NPersQueue::TTopicConverterPtr& fullConverter,
                            const TString& sourceId,
                            std::optional<ui32> preferedPartition)
diff --git a/ydb/core/persqueue/writer/partition_chooser_impl__sm_chooser_actor.h b/ydb/core/persqueue/writer/partition_chooser_impl__sm_chooser_actor.h
@@ -28,13 +28,13 @@ class TSMPartitionChooserActor: public TAbstractPartitionChooserActor<TSMPartiti
     using TParentActor = TAbstractPartitionChooserActor<TSMPartitionChooserActor<TPipeCreator>, TPipeCreator>;
 
     TSMPartitionChooserActor(TActorId parentId,
-                           const NKikimrSchemeOp::TPersQueueGroupDescription& config,
-                           std::shared_ptr<IPartitionChooser>& chooser,
+                           const std::shared_ptr<IPartitionChooser>& chooser,
+                           const std::shared_ptr<NPQ::TPartitionGraph>& graph,
                            NPersQueue::TTopicConverterPtr& fullConverter,
                            const TString& sourceId,
                            std::optional<ui32> preferedPartition)
         : TAbstractPartitionChooserActor<TSMPartitionChooserActor<TPipeCreator>, TPipeCreator>(parentId, chooser, fullConverter, sourceId, preferedPartition)
-        , Graph(MakePartitionGraph(config)) {
+        , Graph(graph) {
     }
 
     void Bootstrap(const TActorContext& ctx) {
@@ -69,7 +69,7 @@ class TSMPartitionChooserActor: public TAbstractPartitionChooserActor<TSMPartiti
             return OnPartitionChosen(ctx);
         }
 
-        const auto* node = Graph.GetPartition(TThis::TableHelper.PartitionId().value());
+        const auto* node = Graph->GetPartition(TThis::TableHelper.PartitionId().value());
         if (!node) {
             // The partition where the writting was performed earlier has already been deleted.
             // We can write without taking into account the hierarchy of the partition.
@@ -78,7 +78,7 @@ class TSMPartitionChooserActor: public TAbstractPartitionChooserActor<TSMPartiti
         }
 
         // Choosing a partition based on the split and merge hierarchy.
-        auto activeChildren = Graph.GetActiveChildren(TThis::TableHelper.PartitionId().value());
+        auto activeChildren = Graph->GetActiveChildren(TThis::TableHelper.PartitionId().value());
         if (activeChildren.empty()) {
             return TThis::ReplyError(ErrorCode::ERROR, TStringBuilder() << "has't active partition Marker# PC01", ctx);
         }
@@ -150,7 +150,7 @@ class TSMPartitionChooserActor: public TAbstractPartitionChooserActor<TSMPartiti
         DEBUG("GetOldSeqNo");
         TThis::Become(&TThis::StateGetMaxSeqNo);
 
-        const auto* oldNode = Graph.GetPartition(TThis::TableHelper.PartitionId().value());
+        const auto* oldNode = Graph->GetPartition(TThis::TableHelper.PartitionId().value());
 
         if (!oldNode) {
             return TThis::ReplyError(ErrorCode::ERROR, TStringBuilder() << "Inconsistent status Marker# PC03", ctx);
@@ -243,7 +243,7 @@ class TSMPartitionChooserActor: public TAbstractPartitionChooserActor<TSMPartiti
 
 private:
     const TPartitionInfo* BoundaryPartition = nullptr;
-    const TPartitionGraph Graph;
+    const std::shared_ptr<TPartitionGraph> Graph;
 };
 
 #undef LOG_PREFIX
diff --git a/ydb/core/tx/datashard/change_sender_cdc_stream.cpp b/ydb/core/tx/datashard/change_sender_cdc_stream.cpp
@@ -329,8 +329,8 @@ class TMd5PartitionResolver final: public NChangeExchange::TBasePartitionResolve
 
 class TBoundaryPartitionResolver final: public NChangeExchange::TBasePartitionResolver {
 public:
-    TBoundaryPartitionResolver(const NKikimrSchemeOp::TPersQueueGroupDescription& config) {
-        Chooser = NPQ::CreatePartitionChooser(config);
+    TBoundaryPartitionResolver(const std::shared_ptr<NPQ::IPartitionChooser>& chooser) {
+        Chooser = chooser;
     }
 
     void Visit(const TChangeRecord& record) override {
@@ -585,7 +585,8 @@ class TCdcChangeSenderMain
         KeyDesc->Partitioning = std::make_shared<TVector<NKikimr::TKeyDesc::TPartitionInfo>>(entry.PQGroupInfo->Partitioning);
 
         if (topicAutoPartitioning) {
-            SetPartitionResolver(new TBoundaryPartitionResolver(pqDesc));
+            Y_ABORT_UNLESS(entry.PQGroupInfo->PartitionChooser);
+            SetPartitionResolver(new TBoundaryPartitionResolver(entry.PQGroupInfo->PartitionChooser));
         } else if (NKikimrSchemeOp::ECdcStreamFormatProto == Stream.Format) {
             SetPartitionResolver(CreateDefaultPartitionResolver(*KeyDesc.Get()));
         } else {
diff --git a/ydb/core/tx/scheme_board/cache.cpp b/ydb/core/tx/scheme_board/cache.cpp
diff --git a/ydb/core/tx/scheme_cache/scheme_cache.h b/ydb/core/tx/scheme_cache/scheme_cache.h
diff --git a/ydb/services/deprecated/persqueue_v0/grpc_pq_write_actor.cpp b/ydb/services/deprecated/persqueue_v0/grpc_pq_write_actor.cpp
diff --git a/ydb/services/persqueue_v1/actors/read_init_auth_actor.cpp b/ydb/services/persqueue_v1/actors/read_init_auth_actor.cpp
diff --git a/ydb/services/persqueue_v1/actors/write_session_actor.cpp b/ydb/services/persqueue_v1/actors/write_session_actor.cpp
diff --git a/ydb/services/persqueue_v1/actors/write_session_actor.h b/ydb/services/persqueue_v1/actors/write_session_actor.h

Original file line number	Diff line number	Diff line change
`@@ -537,7 +537,7 @@ void TPersQueueReadBalancer::Handle(TEvPersQueue::TEvUpdateBalancerConfig::TPtr`
`537`	`537`
`538`	`538`	`if (SplitMergeEnabled(TabletConfig)) {`
`539`	`539`	`if (!PartitionsScaleManager) {`
`540`		`- PartitionsScaleManager = std::make_unique<TPartitionScaleManager>(Topic, DatabasePath, PathId, Version, TabletConfig);`
	`540`	`+ PartitionsScaleManager = std::make_unique<TPartitionScaleManager>(Topic, DatabasePath, PathId, Version, TabletConfig, PartitionGraph);`
`541`	`541`	`} else {`
`542`	`542`	`PartitionsScaleManager->UpdateBalancerConfig(PathId, Version, TabletConfig);`
`543`	`543`	`}`
Original file line number	Diff line number	Diff line change
`@@ -60,7 +60,7 @@ struct TPersQueueReadBalancer::TTxInit : public ITransaction {`
`60`	`60`	`Self->PartitionGraph = MakePartitionGraph(Self->TabletConfig);`
`61`	`61`
`62`	`62`	`if (SplitMergeEnabled(Self->TabletConfig)) {`
`63`		`- Self->PartitionsScaleManager = std::make_unique<TPartitionScaleManager>(Self->Topic, Self->DatabasePath, Self->PathId, Self->Version, Self->TabletConfig);`
	`63`	`+ Self->PartitionsScaleManager = std::make_unique<TPartitionScaleManager>(Self->Topic, Self->DatabasePath, Self->PathId, Self->Version, Self->TabletConfig, Self->PartitionGraph);`
`64`	`64`	`}`
`65`	`65`	`Self->UpdateConfigCounters();`
`66`	`66`	`}`