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/coroutine/engine/mutex.h | |
download | ydb-1110808a9d39d4b808aef724c861a2e1a38d2a69.tar.gz |
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'library/cpp/coroutine/engine/mutex.h')
-rw-r--r-- | library/cpp/coroutine/engine/mutex.h | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/library/cpp/coroutine/engine/mutex.h b/library/cpp/coroutine/engine/mutex.h new file mode 100644 index 0000000000..93e9119503 --- /dev/null +++ b/library/cpp/coroutine/engine/mutex.h @@ -0,0 +1,49 @@ +#pragma once + +#include "impl.h" +#include "events.h" + +class TContMutex { +public: + TContMutex() noexcept + : Token_(true) + { + } + + ~TContMutex() { + Y_ASSERT(Token_); + } + + int LockD(TCont* current, TInstant deadline) { + while (!Token_) { + const int ret = WaitQueue_.WaitD(current, deadline); + + if (ret != EWAKEDUP) { + return ret; + } + } + + Token_ = false; + + return 0; + } + + int LockT(TCont* current, TDuration timeout) { + return LockD(current, timeout.ToDeadLine()); + } + + int LockI(TCont* current) { + return LockD(current, TInstant::Max()); + } + + void UnLock() noexcept { + Y_ASSERT(!Token_); + + Token_ = true; + WaitQueue_.Signal(); + } + +private: + TContWaitQueue WaitQueue_; + bool Token_; +}; |