aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/coroutine/engine/mutex.h
diff options
context:
space:
mode:
authorDevtools Arcadia <arcadia-devtools@yandex-team.ru>2022-02-07 18:08:42 +0300
committerDevtools Arcadia <arcadia-devtools@mous.vla.yp-c.yandex.net>2022-02-07 18:08:42 +0300
commit1110808a9d39d4b808aef724c861a2e1a38d2a69 (patch)
treee26c9fed0de5d9873cce7e00bc214573dc2195b7 /library/cpp/coroutine/engine/mutex.h
downloadydb-1110808a9d39d4b808aef724c861a2e1a38d2a69.tar.gz
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'library/cpp/coroutine/engine/mutex.h')
-rw-r--r--library/cpp/coroutine/engine/mutex.h49
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_;
+};