From 7cae7ed7a51d7303cac40a3573a3f614cdf5c64a Mon Sep 17 00:00:00 2001
From: Oleg Doronin <dorooleg@yandex.ru>
Date: Fri, 24 Jan 2025 00:04:13 +0300
Subject: [PATCH] TotalCpuLimitPercentPerNode has been added to FQ WM (#13768)
 (#13782)

---
 .../synchronization_service.cpp                     | 13 +++++++++----
 ydb/core/fq/libs/config/protos/compute.proto        |  1 +
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/ydb/core/fq/libs/compute/ydb/synchronization_service/synchronization_service.cpp b/ydb/core/fq/libs/compute/ydb/synchronization_service/synchronization_service.cpp
index fe8df83ebd6c..fef5777b69fb 100644
--- a/ydb/core/fq/libs/compute/ydb/synchronization_service/synchronization_service.cpp
+++ b/ydb/core/fq/libs/compute/ydb/synchronization_service/synchronization_service.cpp
@@ -627,13 +627,15 @@ class TSynchronizeScopeActor : public NActors::TActorBootstrapped<TSynchronizeSc
                         CREATE RESOURCE POOL `{resource_pool_name}` WITH (
                             CONCURRENT_QUERY_LIMIT="{concurrent_query_limit}",
                             QUEUE_SIZE="{queue_size}",
-                            DATABASE_LOAD_CPU_THRESHOLD="{database_load_cpu_threshold}"
+                            DATABASE_LOAD_CPU_THRESHOLD="{database_load_cpu_threshold}",
+                            TOTAL_CPU_LIMIT_PERCENT_PER_NODE="{total_cpu_limit_percent_per_node}"
                         );
                     )",
                     "resource_pool_name"_a = resourcePool.GetName(),
                     "concurrent_query_limit"_a = resourcePool.GetConcurrentQueryLimit(),
                     "queue_size"_a = resourcePool.GetQueueSize(),
-                    "database_load_cpu_threshold"_a = resourcePool.GetDatabaseLoadCpuThreshold()));
+                    "database_load_cpu_threshold"_a = resourcePool.GetDatabaseLoadCpuThreshold(),
+                    "total_cpu_limit_percent_per_node"_a = resourcePool.GetTotalCpuLimitPercentPerNode()));
                 })
                 .Subscribe([actorSystem = TActivationContext::ActorSystem(), self = SelfId(), i](const NYdb::TAsyncStatus& future) {
                     actorSystem->Send(self, new TEvYdbCompute::TEvCreateResourcePoolResponse(ExtractStatus(future)), 0, i);
@@ -663,13 +665,16 @@ class TSynchronizeScopeActor : public NActors::TActorBootstrapped<TSynchronizeSc
                     ALTER RESOURCE POOL `{resource_pool_name}` SET (
                         CONCURRENT_QUERY_LIMIT="{concurrent_query_limit}",
                         QUEUE_SIZE="{queue_size}",
-                        DATABASE_LOAD_CPU_THRESHOLD="{database_load_cpu_threshold}"
+                        DATABASE_LOAD_CPU_THRESHOLD="{database_load_cpu_threshold}",
+                        TOTAL_CPU_LIMIT_PERCENT_PER_NODE="{total_cpu_limit_percent_per_node}"
                     );
                 )",
                 "resource_pool_name"_a = resourcePool.GetName(),
                 "concurrent_query_limit"_a = resourcePool.GetConcurrentQueryLimit(),
                 "queue_size"_a = resourcePool.GetQueueSize(),
-                "database_load_cpu_threshold"_a = resourcePool.GetDatabaseLoadCpuThreshold()));
+                "database_load_cpu_threshold"_a = resourcePool.GetDatabaseLoadCpuThreshold(),
+                "total_cpu_limit_percent_per_node"_a = resourcePool.GetTotalCpuLimitPercentPerNode()
+                ));
             })
             .Subscribe([actorSystem = TActivationContext::ActorSystem(), self = SelfId(), index](const NYdb::TAsyncStatus& future) {
                 actorSystem->Send(self, new TEvYdbCompute::TEvCreateResourcePoolResponse(ExtractStatus(future)), 0, index);
diff --git a/ydb/core/fq/libs/config/protos/compute.proto b/ydb/core/fq/libs/config/protos/compute.proto
index 72b6826b692f..172a6092b1cc 100644
--- a/ydb/core/fq/libs/config/protos/compute.proto
+++ b/ydb/core/fq/libs/config/protos/compute.proto
@@ -35,6 +35,7 @@ message TWorkloadManagerConfig {
         int32 ConcurrentQueryLimit = 2;
         int32 QueueSize = 3;
         double DatabaseLoadCpuThreshold = 4;
+        double TotalCpuLimitPercentPerNode = 5;
     }
 
     bool Enable = 1;