45 lines
1.2 KiB
C++
45 lines
1.2 KiB
C++
#pragma once
|
|
#ifndef MESSMER_CPPUTILS_LOCK_MUTEXPOOLLOCK_H
|
|
#define MESSMER_CPPUTILS_LOCK_MUTEXPOOLLOCK_H
|
|
|
|
#include "LockPool.h"
|
|
|
|
namespace cpputils {
|
|
template<class LockName>
|
|
class MutexPoolLock final {
|
|
public:
|
|
MutexPoolLock(LockPool<LockName> *pool, const LockName &lockName): _pool(pool), _lockName(lockName) {
|
|
_pool->lock(_lockName);
|
|
}
|
|
|
|
MutexPoolLock(LockPool<LockName> *pool, const LockName &lockName, std::unique_lock<std::mutex> *lockToFreeWhileWaiting)
|
|
: _pool(pool), _lockName(lockName) {
|
|
_pool->lock(_lockName, lockToFreeWhileWaiting);
|
|
}
|
|
|
|
MutexPoolLock(MutexPoolLock &&rhs) noexcept: _pool(rhs._pool), _lockName(std::move(rhs._lockName)) {
|
|
rhs._pool = nullptr;
|
|
}
|
|
|
|
~MutexPoolLock() {
|
|
if (_pool != nullptr) {
|
|
unlock();
|
|
}
|
|
}
|
|
|
|
void unlock() {
|
|
ASSERT(_pool != nullptr, "MutexPoolLock is not locked");
|
|
_pool->release(_lockName);
|
|
_pool = nullptr;
|
|
}
|
|
|
|
private:
|
|
LockPool<LockName> *_pool;
|
|
LockName _lockName;
|
|
|
|
DISALLOW_COPY_AND_ASSIGN(MutexPoolLock);
|
|
};
|
|
}
|
|
|
|
#endif
|