100 lines
2.5 KiB
C++
100 lines
2.5 KiB
C++
/***
|
|
Copyright (C) 2021 J Reece Wilson (a/k/a "Reece"). All rights reserved.
|
|
|
|
File: AuRWLock.hpp
|
|
Date: 2021-6-12
|
|
Author: Reece
|
|
***/
|
|
#pragma once
|
|
|
|
#include "AuConditionVariable.Generic.hpp"
|
|
#include "AuConditionMutex.Generic.hpp"
|
|
#include "ThreadCookie.hpp"
|
|
|
|
namespace Aurora::Threading::Primitives
|
|
{
|
|
template<bool bIsWriteRecursionAllowed>
|
|
struct RWLockImpl;
|
|
|
|
template<bool bIsReadView, typename T>
|
|
struct RWLockAccessView : IWaitable
|
|
{
|
|
RWLockAccessView(T &impl) : parent_(impl)
|
|
{
|
|
|
|
}
|
|
|
|
bool LockMS(AuUInt64 timeout) override;
|
|
bool LockNS(AuUInt64 timeout) override;
|
|
|
|
bool TryLock() override;
|
|
|
|
bool HasOSHandle(AuMach &mach) override
|
|
{
|
|
return false;
|
|
}
|
|
|
|
bool HasLockImplementation() override
|
|
{
|
|
return true;
|
|
}
|
|
|
|
void Lock() override
|
|
{
|
|
SysAssert(LockNS(0));
|
|
}
|
|
|
|
void Unlock() override;
|
|
|
|
private:
|
|
T &parent_;
|
|
};
|
|
|
|
template<bool bIsWriteRecursionAllowed>
|
|
struct RWLockImpl final : IRWLock
|
|
{
|
|
RWLockImpl();
|
|
~RWLockImpl();
|
|
|
|
// i dont think i'll expose a high performance interface yet
|
|
auline bool LockReadNS(AuUInt64 timeout);// override;
|
|
auline bool LockWriteNS(AuUInt64 timeout);// override;
|
|
auline bool TryLockRead();// override;
|
|
auline bool TryLockWrite();// override;
|
|
auline void UnlockRead();// override;
|
|
auline void UnlockWrite();// override;
|
|
|
|
bool UpgradeReadToWrite(AuUInt64 timeout) override;
|
|
bool DowngradeWriteToRead() override;
|
|
|
|
IWaitable *AsReadable() override;
|
|
IWaitable *AsWritable() override;
|
|
|
|
bool Init();
|
|
|
|
auline ConditionVariableImpl &GetCondition();
|
|
auline ConditionVariableImpl &GetConditionWriter();
|
|
|
|
private:
|
|
|
|
//bool reentrantWriteLock_ {true};
|
|
ThreadCookie_t reentrantWriteLockHandle_ {};
|
|
|
|
RWLockAccessView<true, RWLockImpl> read_;
|
|
RWLockAccessView<false, RWLockImpl> write_;
|
|
|
|
ConditionMutexImpl mutex_;
|
|
#if defined(AURWLOCK_NO_SIZE_OPTIMIZED_CONDVAR)
|
|
ConditionVariableImpl condition_;
|
|
ConditionVariableImpl conditionWriter_;
|
|
#else
|
|
char conditionVariable_[kSizeOfDummyCondVar] {};
|
|
char conditionVariableWriter_[kSizeOfDummyCondVar] {};
|
|
#endif
|
|
volatile AuInt32 state_ {};
|
|
AuInt32 writersPending_ : 31 {};
|
|
AuInt32 bElevaterPending_ : 1 {};
|
|
//bool bElevaterPending_{};
|
|
};
|
|
}
|