Skip to content

Commit e968445

Browse files
authored
Merge b827e7d into ae1af5f
2 parents ae1af5f + b827e7d commit e968445

File tree

7 files changed

+275
-36
lines changed

7 files changed

+275
-36
lines changed

ydb/core/kqp/executer_actor/kqp_planner.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -260,6 +260,9 @@ std::unique_ptr<TEvKqpNode::TEvStartKqpTasksRequest> TKqpPlanner::SerializeReque
260260
if (UserRequestContext->PoolConfig->QueryCpuLimitPercentPerNode >= 0) {
261261
request.SetQueryCpuShare(UserRequestContext->PoolConfig->QueryCpuLimitPercentPerNode / 100.0);
262262
}
263+
if (UserRequestContext->PoolConfig->ResourceWeight >= 0) {
264+
request.SetResourceWeight(UserRequestContext->PoolConfig->ResourceWeight);
265+
}
263266
}
264267

265268
return result;

ydb/core/kqp/node_service/kqp_node_service.cpp

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -203,12 +203,17 @@ class TKqpNodeService : public TActorBootstrapped<TKqpNodeService> {
203203

204204
if (SchedulerOptions.Scheduler->Disabled(schedulerGroup)) {
205205
auto share = msg.GetPoolMaxCpuShare();
206-
if (share <= 0 && msg.HasQueryCpuShare()) {
206+
if (share <= 0 && (msg.HasQueryCpuShare() || msg.HasResourceWeight())) {
207207
share = 1.0;
208208
}
209+
std::optional<double> resourceWeight;
210+
if (msg.GetResourceWeight() >= 0) {
211+
resourceWeight = msg.GetResourceWeight();
212+
}
213+
209214
if (share > 0) {
210-
Scheduler->UpdateGroupShare(schedulerGroup, share, schedulerNow);
211-
Send(SchedulerActorId, new TEvSchedulerNewPool(msg.GetDatabaseId(), schedulerGroup));
215+
Scheduler->UpdateGroupShare(schedulerGroup, share, schedulerNow, resourceWeight);
216+
Send(SchedulerActorId, new TEvSchedulerNewPool(msg.GetDatabase(), schedulerGroup));
212217
} else {
213218
schedulerGroup = "";
214219
}

0 commit comments

Comments
 (0)