/
githubmirror
/
nbs
Обзор
Документация
Войти
/
githubmirror
/
nbs
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
library/cpp/messagebus/latch.h
55 строк
954 B
qkrorlqr
Initial NBS OpenSource export
28 апр 2023, 21:54
28 апр 2023, 21:54
783a858
Код
Авторство
О чём код?
#pragma once #include <library/cpp/deprecated/atomic/atomic.h> #include <util/system/condvar.h> #include <util/system/mutex.h> class TLatch { private: // 0 for unlocked, 1 for locked TAtomic Locked; TMutex Mutex; TCondVar CondVar; public: TLatch() : Locked(0) { } void Wait() { // optimistic path if (AtomicGet(Locked) == 0) { return; } TGuard<TMutex> guard(Mutex); while (AtomicGet(Locked) == 1) { CondVar.WaitI(Mutex); } } bool TryWait() { return AtomicGet(Locked) == 0; } void Unlock() { // optimistic path if (AtomicGet(Locked) == 0) { return; } TGuard<TMutex> guard(Mutex); AtomicSet(Locked, 0); CondVar.BroadCast(); } void Lock() { AtomicSet(Locked, 1); } bool IsLocked() { return AtomicGet(Locked); } };