#pragma once
#include <wtf/Lock.h>
#include <wtf/Threading.h>
namespace WTF {
#if !defined(ENABLE_DATA_MUTEX_CHECKS)
#if defined(NDEBUG)
#define ENABLE_DATA_MUTEX_CHECKS 0
#else
#define ENABLE_DATA_MUTEX_CHECKS 1
#endif
#endif
#if ENABLE_DATA_MUTEX_CHECKS
#define DATA_MUTEX_CHECK(expr) RELEASE_ASSERT(expr)
#else
#define DATA_MUTEX_CHECK(expr)
#endif
template<typename LockType>
class OwnerAwareLockAdapter {
public:
void lock()
{
DATA_MUTEX_CHECK(m_owner != &Thread::current()); m_lock.lock();
#if ENABLE_DATA_MUTEX_CHECKS
ASSERT(!m_owner);
m_owner = &Thread::current();
#endif
}
void unlock()
{
#if ENABLE_DATA_MUTEX_CHECKS
m_owner = nullptr;
#endif
m_lock.unlock();
}
bool tryLock()
{
DATA_MUTEX_CHECK(m_owner != &Thread::current()); if (!m_lock.tryLock())
return false;
#if ENABLE_DATA_MUTEX_CHECKS
ASSERT(!m_owner);
m_owner = &Thread::current();
#endif
return true;
}
bool isLocked() const
{
return m_lock.isLocked();
}
private:
#if ENABLE_DATA_MUTEX_CHECKS
Thread* m_owner { nullptr }; #endif
LockType m_lock;
};
using OwnerAwareLock = OwnerAwareLockAdapter<Lock>;
template<typename T, typename LockType = OwnerAwareLock>
class DataMutex {
WTF_MAKE_FAST_ALLOCATED;
WTF_MAKE_NONCOPYABLE(DataMutex);
public:
template<typename ...Args>
explicit DataMutex(Args&&... args)
: m_data(std::forward<Args>(args)...)
{ }
class LockedWrapper {
public:
explicit LockedWrapper(DataMutex& dataMutex)
: m_mutex(dataMutex.m_mutex)
, m_lockHolder(dataMutex.m_mutex)
, m_data(dataMutex.m_data)
{ }
T* operator->()
{
DATA_MUTEX_CHECK(m_mutex.isLocked());
return &m_data;
}
T& operator*()
{
DATA_MUTEX_CHECK(m_mutex.isLocked());
return m_data;
}
LockType& mutex()
{
return m_mutex;
}
Locker<LockType>& lockHolder()
{
return m_lockHolder;
}
void runUnlocked(WTF::Function<void()> callback)
{
m_mutex.unlock();
callback();
m_mutex.lock();
}
private:
LockType& m_mutex;
Locker<LockType> m_lockHolder;
T& m_data;
};
private:
LockType m_mutex;
T m_data;
};
}