aboutsummaryrefslogtreecommitdiffstats
path: root/yt/yt/core/misc/fair_scheduler.h
blob: 1399dcd0cdb8a940990535118cb77634cdc90de1 (plain) (blame)
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
28
29
30
#pragma once

#include "public.h"

namespace NYT {

////////////////////////////////////////////////////////////////////////////////

template <class TTask>
struct IFairScheduler
    : public TRefCounted
{
    virtual void Enqueue(TTask task, const std::string& user) = 0;
    virtual std::optional<TTask> TryDequeue() = 0;

    virtual void ChargeUser(const std::string& user, TDuration time) = 0;
};

////////////////////////////////////////////////////////////////////////////////

template <class TTask>
IFairSchedulerPtr<TTask> CreateFairScheduler();

////////////////////////////////////////////////////////////////////////////////

} // namespace NYT

#define FAIR_SCHEDULER_INL_H_
#include "fair_scheduler-inl.h"
#undef FAIR_SCHEDULER_INL_H_