// Copyright 2019 Fuzhou Rockchip Electronics Co., Ltd. All rights reserved.
|
// Use of this source code is governed by a BSD-style license that can be
|
// found in the LICENSE file.
|
|
#include "lock.h"
|
|
#include <stdio.h>
|
|
namespace easymedia
|
{
|
|
LockMutex::LockMutex()
|
#ifndef NDEBUG
|
: lock_times(0)
|
#endif
|
{
|
}
|
LockMutex::~LockMutex()
|
{
|
#ifndef NDEBUG
|
assert(lock_times == 0 && "mutex lock/unlock mismatch");
|
#endif
|
}
|
|
void LockMutex::locktimeinc()
|
{
|
#ifndef NDEBUG
|
lock_times++;
|
#endif
|
}
|
void LockMutex::locktimedec()
|
{
|
#ifndef NDEBUG
|
lock_times--;
|
#endif
|
}
|
|
void ConditionLockMutex::lock()
|
{
|
mtx.lock();
|
locktimeinc();
|
}
|
void ConditionLockMutex::unlock()
|
{
|
locktimedec();
|
mtx.unlock();
|
}
|
void ConditionLockMutex::wait()
|
{
|
cond.wait(mtx);
|
}
|
void ConditionLockMutex::notify()
|
{
|
cond.notify_all();
|
}
|
|
ReadWriteLockMutex::ReadWriteLockMutex() : valid(true)
|
{
|
int ret = pthread_rwlock_init(&rwlock, NULL);
|
if (ret) {
|
fprintf(stderr, "Fail to pthread_rwlock_init\n");
|
valid = false;
|
}
|
}
|
ReadWriteLockMutex::~ReadWriteLockMutex()
|
{
|
if (valid) {
|
pthread_rwlock_destroy(&rwlock);
|
}
|
}
|
void ReadWriteLockMutex::lock()
|
{
|
// write lock
|
if (valid) {
|
pthread_rwlock_wrlock(&rwlock);
|
}
|
locktimeinc();
|
}
|
void ReadWriteLockMutex::unlock()
|
{
|
locktimedec();
|
if (valid) {
|
pthread_rwlock_unlock(&rwlock);
|
}
|
}
|
void ReadWriteLockMutex::read_lock()
|
{
|
if (valid) {
|
pthread_rwlock_rdlock(&rwlock);
|
}
|
locktimeinc();
|
}
|
|
SpinLockMutex::SpinLockMutex() : flag(ATOMIC_FLAG_INIT)
|
{
|
}
|
void SpinLockMutex::lock()
|
{
|
while (flag.test_and_set(std::memory_order_acquire))
|
;
|
locktimeinc();
|
}
|
void SpinLockMutex::unlock()
|
{
|
locktimedec();
|
flag.clear(std::memory_order_release);
|
}
|
|
} // namespace easymedia
|
