Skip to content

Commit 99fd95e

Browse files
authored
Merge 25febb0 into 207f846
2 parents 207f846 + 25febb0 commit 99fd95e

File tree

3 files changed

+6
-1
lines changed

3 files changed

+6
-1
lines changed

ydb/core/kqp/workload_service/kqp_workload_service.cpp

+4-1
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,7 @@ class TKqpWorkloadService : public TActorBootstrapped<TKqpWorkloadService> {
7474

7575
EnabledResourcePools = AppData()->FeatureFlags.GetEnableResourcePools();
7676
EnabledResourcePoolsOnServerless = AppData()->FeatureFlags.GetEnableResourcePoolsOnServerless();
77+
EnableResourcePoolsCounters = AppData()->FeatureFlags.GetEnableResourcePoolsCounters();
7778
if (EnabledResourcePools) {
7879
InitializeWorkloadService();
7980
}
@@ -101,6 +102,7 @@ class TKqpWorkloadService : public TActorBootstrapped<TKqpWorkloadService> {
101102

102103
EnabledResourcePools = event.GetConfig().GetFeatureFlags().GetEnableResourcePools();
103104
EnabledResourcePoolsOnServerless = event.GetConfig().GetFeatureFlags().GetEnableResourcePoolsOnServerless();
105+
EnableResourcePoolsCounters = event.GetConfig().GetFeatureFlags().GetEnableResourcePoolsCounters();
104106
if (EnabledResourcePools) {
105107
LOG_I("Resource pools was enanbled");
106108
InitializeWorkloadService();
@@ -526,7 +528,7 @@ class TKqpWorkloadService : public TActorBootstrapped<TKqpWorkloadService> {
526528

527529
LOG_I("Creating new handler for pool " << poolKey);
528530

529-
const auto poolHandler = Register(CreatePoolHandlerActor(database, poolId, poolConfig, Counters.Counters));
531+
const auto poolHandler = Register(CreatePoolHandlerActor(database, poolId, poolConfig, EnableResourcePoolsCounters ? Counters.Counters : MakeIntrusive<NMonitoring::TDynamicCounters>()));
530532
const auto poolState = &PoolIdToState.insert({poolKey, TPoolState{.PoolHandler = poolHandler, .ActorContext = ActorContext()}}).first->second;
531533

532534
Counters.ActivePools->Inc();
@@ -560,6 +562,7 @@ class TKqpWorkloadService : public TActorBootstrapped<TKqpWorkloadService> {
560562

561563
bool EnabledResourcePools = false;
562564
bool EnabledResourcePoolsOnServerless = false;
565+
bool EnableResourcePoolsCounters = false;
563566
bool ServiceInitialized = false;
564567
bool IdleChecksStarted = false;
565568
ETablesCreationStatus TablesCreationStatus = ETablesCreationStatus::Cleanup;

ydb/core/kqp/workload_service/ut/common/kqp_workload_service_ut_common.cpp

+1
Original file line numberDiff line numberDiff line change
@@ -230,6 +230,7 @@ class TWorkloadServiceYdbSetup : public IYdbSetup {
230230
TAppConfig GetAppConfig() const {
231231
TAppConfig appConfig;
232232
appConfig.MutableFeatureFlags()->SetEnableResourcePools(Settings_.EnableResourcePools_);
233+
appConfig.MutableFeatureFlags()->SetEnableResourcePoolsCounters(true);
233234

234235
return appConfig;
235236
}

ydb/core/protos/feature_flags.proto

+1
Original file line numberDiff line numberDiff line change
@@ -151,4 +151,5 @@ message TFeatureFlags {
151151
optional bool EnableResourcePoolsOnServerless = 132 [default = false];
152152
optional bool EnableVectorIndex = 133 [default = false];
153153
optional bool EnableChangefeedsOnIndexTables = 134 [default = false];
154+
optional bool EnableResourcePoolsCounters = 135 [default = false];
154155
}

0 commit comments

Comments
 (0)