67 lines
1.2 KiB
C++
67 lines
1.2 KiB
C++
|
#include "OysterMutex.h"
|
||
|
|
|
||
|
|
#include <chrono>
|
||
|
|
#include <thread>
|
||
|
|
#include <future>
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
OysterMutex::OysterMutex()
|
||
|
|
{}
|
||
|
|
|
||
|
|
OysterMutex::OysterMutex(bool initialOwnership)
|
||
|
|
{
|
||
|
|
if(initialOwnership)
|
||
|
|
{
|
||
|
|
this->mutex.lock();
|
||
|
|
this->id = std::this_thread::get_id();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
OysterMutex::~OysterMutex()
|
||
|
|
{
|
||
|
|
}
|
||
|
|
void OysterMutex::LockMutex()
|
||
|
|
{
|
||
|
|
if(std::this_thread::get_id() == this->id) return;
|
||
|
|
|
||
|
|
this->mutex.lock();
|
||
|
|
|
||
|
|
this->id = std::this_thread::get_id();
|
||
|
|
}
|
||
|
|
void OysterMutex::LockMutex(unsigned int msec)
|
||
|
|
{
|
||
|
|
if(std::this_thread::get_id() == this->id) return;
|
||
|
|
|
||
|
|
auto start = std::chrono::high_resolution_clock::now();
|
||
|
|
auto end = start + std::chrono::milliseconds(msec);
|
||
|
|
|
||
|
|
do
|
||
|
|
{
|
||
|
|
if(this->mutex.try_lock())
|
||
|
|
{
|
||
|
|
this->mutex.lock();
|
||
|
|
this->id = std::this_thread::get_id();
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
} while (std::chrono::high_resolution_clock::now() < end);
|
||
|
|
|
||
|
|
this->mutex.lock();
|
||
|
|
}
|
||
|
|
void OysterMutex::UnlockMutex()
|
||
|
|
{
|
||
|
|
//Let the owner unlock
|
||
|
|
if(std::this_thread::get_id() != this->id) return;
|
||
|
|
|
||
|
|
this->mutex.unlock();
|
||
|
|
this->id = std::thread::id();
|
||
|
|
|
||
|
|
}
|
||
|
|
bool OysterMutex::IsTaken()
|
||
|
|
{
|
||
|
|
return !this->mutex.try_lock();
|
||
|
|
}
|
||
|
|
void OysterMutex::Reset()
|
||
|
|
{
|
||
|
|
if(!this->mutex.try_lock())
|
||
|
|
this->mutex.unlock();
|
||
|
|
}
|