AuroraRuntime/Source/Threading/Primitives/AuCriticalSection.cpp

165 lines
3.1 KiB
C++

/***
Copyright (C) 2021 J Reece Wilson (a/k/a "Reece"). All rights reserved.
File: AuCriticalSection.cpp
Date: 2021-6-12
Author: Reece
***/
#include <Source/RuntimeInternal.hpp>
#include "AuCriticalSection.hpp"
namespace Aurora::Threading::Primitives
{
CriticalSectionImpl::CriticalSectionImpl()
{
this->owner_ = {};
this->count_ = 0;
}
CriticalSectionImpl::~CriticalSectionImpl()
{
}
bool CriticalSectionImpl::HasOSHandle(AuMach &mach)
{
return false;
}
bool CriticalSectionImpl::TryLock()
{
auto cur = GetThreadCookie();
if (this->owner_ == cur)
{
this->count_++;
return true;
}
if (!this->mutex_.TryLock())
{
return false;
}
this->owner_ = cur;
this->count_ = 1;
return true;
}
void CriticalSectionImpl::Lock()
{
auto status = LockMS(0);
SysAssert(status, "Spurious critical section wakeup");
}
bool CriticalSectionImpl::LockMS(AuUInt64 timeout)
{
auto cur = GetThreadCookie();
if (this->owner_ == cur)
{
this->count_++;
return true;
}
if (!this->mutex_.LockMS(timeout))
{
return false;
}
this->owner_ = cur;
this->count_ = 1;
return true;
}
bool CriticalSectionImpl::LockNS(AuUInt64 timeout)
{
auto cur = GetThreadCookie();
if (this->owner_ == cur)
{
this->count_++;
return true;
}
if (!this->mutex_.LockNS(timeout))
{
return false;
}
this->owner_ = cur;
this->count_ = 1;
return true;
}
bool CriticalSectionImpl::LockAbsNS(AuUInt64 timeout)
{
auto cur = GetThreadCookie();
if (this->owner_ == cur)
{
this->count_++;
return true;
}
if (!this->mutex_.LockAbsNS(timeout))
{
return false;
}
this->owner_ = cur;
this->count_ = 1;
return true;
}
bool CriticalSectionImpl::LockAbsMS(AuUInt64 timeout)
{
auto cur = GetThreadCookie();
if (this->owner_ == cur)
{
this->count_++;
return true;
}
if (!this->mutex_.LockAbsMS(timeout))
{
return false;
}
this->owner_ = cur;
this->count_ = 1;
return true;
}
void CriticalSectionImpl::Unlock()
{
if (--this->count_ == 0)
{
this->owner_ = {};
this->mutex_.Unlock();
}
}
bool CriticalSectionImpl::HasLockImplementation()
{
return true;
}
AUKN_SYM IWaitable *CriticalSectionNew()
{
return _new CriticalSectionImpl();
}
AUKN_SYM void CriticalSectionRelease(IWaitable *pCriticalSection)
{
AuSafeDelete<CriticalSectionImpl *>(pCriticalSection);
}
AUROXTL_INTERFACE_SOO_SRC_EX(AURORA_SYMBOL_EXPORT, CriticalSection, CriticalSectionImpl)
}