ydb/services/bg_tasks/ds_table/assign_tasks.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

#pragma once
#include "executor_controller.h"

#include <ydb/services/metadata/request/request_actor.h>

namespace NKikimr::NBackgroundTasks {

class TAssignTasksActor: public NInternal::NRequest::TSessionedActor {
private:
    using TBase = NInternal::NRequest::TSessionedActor;
    TExecutorController::TPtr Controller;
    const ui32 TasksCount;
    const TString ExecutorId;

    virtual void OnResult(const NInternal::NRequest::TDialogYQLRequest::TResponse& result) override;
    virtual std::optional<NInternal::NRequest::TDialogYQLRequest::TRequest> OnSessionId(const TString& sessionId) override;
public:
    TAssignTasksActor(const ui32 tasksCount, TExecutorController::TPtr controller, const TString& executorId)
        : TBase(controller->GetRequestConfig(), controller->GetUserToken())
        , Controller(controller)
        , TasksCount(tasksCount)
        , ExecutorId(executorId)
    {

    }
};
}