diff options
author | Devtools Arcadia <arcadia-devtools@yandex-team.ru> | 2022-02-07 18:08:42 +0300 |
---|---|---|
committer | Devtools Arcadia <arcadia-devtools@mous.vla.yp-c.yandex.net> | 2022-02-07 18:08:42 +0300 |
commit | 1110808a9d39d4b808aef724c861a2e1a38d2a69 (patch) | |
tree | e26c9fed0de5d9873cce7e00bc214573dc2195b7 /library/cpp/messagebus/scheduler/scheduler.h | |
download | ydb-1110808a9d39d4b808aef724c861a2e1a38d2a69.tar.gz |
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'library/cpp/messagebus/scheduler/scheduler.h')
-rw-r--r-- | library/cpp/messagebus/scheduler/scheduler.h | 68 |
1 files changed, 68 insertions, 0 deletions
diff --git a/library/cpp/messagebus/scheduler/scheduler.h b/library/cpp/messagebus/scheduler/scheduler.h new file mode 100644 index 0000000000..afcc0de55d --- /dev/null +++ b/library/cpp/messagebus/scheduler/scheduler.h @@ -0,0 +1,68 @@ +#pragma once + +#include <library/cpp/threading/future/legacy_future.h> + +#include <util/datetime/base.h> +#include <util/generic/object_counter.h> +#include <util/generic/ptr.h> +#include <util/generic/vector.h> +#include <util/system/atomic.h> +#include <util/system/condvar.h> +#include <util/system/mutex.h> +#include <util/system/thread.h> + +namespace NBus { + namespace NPrivate { + class IScheduleItem { + public: + inline IScheduleItem(TInstant scheduleTime) noexcept; + virtual ~IScheduleItem() { + } + + virtual void Do() = 0; + inline TInstant GetScheduleTime() const noexcept; + + private: + TInstant ScheduleTime; + }; + + using IScheduleItemAutoPtr = TAutoPtr<IScheduleItem>; + + class TScheduler { + public: + TScheduler(); + ~TScheduler(); + void Stop(); + void Schedule(TAutoPtr<IScheduleItem> i); + + size_t Size() const; + + private: + void SchedulerThread(); + + void FillNextItem(); + + private: + TVector<IScheduleItemAutoPtr> Items; + IScheduleItemAutoPtr NextItem; + typedef TMutex TLock; + TLock Lock; + TCondVar CondVar; + + TObjectCounter<TScheduler> ObjectCounter; + + bool StopThread; + NThreading::TLegacyFuture<> Thread; + }; + + inline IScheduleItem::IScheduleItem(TInstant scheduleTime) noexcept + : ScheduleTime(scheduleTime) + { + } + + inline TInstant IScheduleItem::GetScheduleTime() const noexcept { + return ScheduleTime; + } + + } +} |