fix

Author: ssmike

ydb/core/kqp/runtime/kqp_compute_scheduler.cpp

@@ -480,7 +480,6 @@ class TSchedulerActor : public TActorBootstrapped<TSchedulerActor> {
         }
         Opts.Scheduler->SetForgetInterval(Opts.ForgetOverflowTimeout);
         Opts.Scheduler->ReportCounters(Opts.Counters);
-        SetCapacity();
     }
 
     void Bootstrap() {
@@ -491,13 +490,14 @@ class TSchedulerActor : public TActorBootstrapped<TSchedulerActor> {
              new NConsole::TEvConfigsDispatcher::TEvSetConfigSubscriptionRequest({tableServiceConfigKind}),
              IEventHandle::FlagTrackDelivery);
 
+        SetCapacity(SelfId().PoolID());
         Become(&TSchedulerActor::State);
     }
 
-    void SetCapacity() {
+    void SetCapacity(ui32 poolId) {
         NActors::TExecutorPoolStats poolStats;
         TVector<NActors::TExecutorThreadStats> threadsStats;
-        TlsActivationContext->ActorSystem()->GetPoolStats(SelfId().PoolID(), poolStats, threadsStats);
+        TlsActivationContext->ActorSystem()->GetPoolStats(poolId, poolStats, threadsStats);
         Y_ENSURE(poolStats.MaxThreadCount > 0);
         Opts.Counters->SchedulerCapacity->Set(poolStats.MaxThreadCount);
         Opts.Scheduler->SetCapacity(poolStats.MaxThreadCount);
@@ -550,7 +550,7 @@ class TSchedulerActor : public TActorBootstrapped<TSchedulerActor> {
     }
 
     void Handle(TEvents::TEvWakeup::TPtr&) {
-        SetCapacity();
+        SetCapacity(SelfId().PoolID());
         Opts.Scheduler->AdvanceTime(TlsActivationContext->Monotonic());
         Schedule(Opts.AdvanceTimeInterval, new TEvents::TEvWakeup());
     }