Merge pull request #72172 from RandomShaper/semaphore_codestyle
Fix code style and consistency of RWLock and Semaphore
This commit is contained in:
commit
8365530cb0
|
@ -31,39 +31,41 @@
|
|||
#ifndef RW_LOCK_H
|
||||
#define RW_LOCK_H
|
||||
|
||||
#include "core/typedefs.h"
|
||||
|
||||
#include <shared_mutex>
|
||||
|
||||
class RWLock {
|
||||
mutable std::shared_timed_mutex mutex;
|
||||
|
||||
public:
|
||||
// Lock the rwlock, block if locked by someone else
|
||||
void read_lock() const {
|
||||
// Lock the RWLock, block if locked by someone else.
|
||||
_ALWAYS_INLINE_ void read_lock() const {
|
||||
mutex.lock_shared();
|
||||
}
|
||||
|
||||
// Unlock the rwlock, let other threads continue
|
||||
void read_unlock() const {
|
||||
// Unlock the RWLock, let other threads continue.
|
||||
_ALWAYS_INLINE_ void read_unlock() const {
|
||||
mutex.unlock_shared();
|
||||
}
|
||||
|
||||
// Attempt to lock the RWLock for reading. True on success, false means it can't lock.
|
||||
bool read_try_lock() const {
|
||||
_ALWAYS_INLINE_ bool read_try_lock() const {
|
||||
return mutex.try_lock_shared();
|
||||
}
|
||||
|
||||
// Lock the rwlock, block if locked by someone else
|
||||
void write_lock() {
|
||||
// Lock the RWLock, block if locked by someone else.
|
||||
_ALWAYS_INLINE_ void write_lock() {
|
||||
mutex.lock();
|
||||
}
|
||||
|
||||
// Unlock the rwlock, let other thwrites continue
|
||||
void write_unlock() {
|
||||
// Unlock the RWLock, let other threads continue.
|
||||
_ALWAYS_INLINE_ void write_unlock() {
|
||||
mutex.unlock();
|
||||
}
|
||||
|
||||
// Attempt to lock the RWLock for writing. True on success, false means it can't lock.
|
||||
bool write_try_lock() {
|
||||
_ALWAYS_INLINE_ bool write_try_lock() {
|
||||
return mutex.try_lock();
|
||||
}
|
||||
};
|
||||
|
@ -72,11 +74,11 @@ class RWLockRead {
|
|||
const RWLock &lock;
|
||||
|
||||
public:
|
||||
RWLockRead(const RWLock &p_lock) :
|
||||
_ALWAYS_INLINE_ RWLockRead(const RWLock &p_lock) :
|
||||
lock(p_lock) {
|
||||
lock.read_lock();
|
||||
}
|
||||
~RWLockRead() {
|
||||
_ALWAYS_INLINE_ ~RWLockRead() {
|
||||
lock.read_unlock();
|
||||
}
|
||||
};
|
||||
|
@ -85,11 +87,11 @@ class RWLockWrite {
|
|||
RWLock &lock;
|
||||
|
||||
public:
|
||||
RWLockWrite(RWLock &p_lock) :
|
||||
_ALWAYS_INLINE_ RWLockWrite(RWLock &p_lock) :
|
||||
lock(p_lock) {
|
||||
lock.write_lock();
|
||||
}
|
||||
~RWLockWrite() {
|
||||
_ALWAYS_INLINE_ ~RWLockWrite() {
|
||||
lock.write_unlock();
|
||||
}
|
||||
};
|
||||
|
|
|
@ -39,33 +39,34 @@
|
|||
|
||||
class Semaphore {
|
||||
private:
|
||||
mutable std::mutex mutex_;
|
||||
mutable std::condition_variable condition_;
|
||||
mutable unsigned long count_ = 0; // Initialized as locked.
|
||||
mutable std::mutex mutex;
|
||||
mutable std::condition_variable condition;
|
||||
mutable uint32_t count = 0; // Initialized as locked.
|
||||
|
||||
public:
|
||||
_ALWAYS_INLINE_ void post() const {
|
||||
std::lock_guard<decltype(mutex_)> lock(mutex_);
|
||||
++count_;
|
||||
condition_.notify_one();
|
||||
std::lock_guard lock(mutex);
|
||||
count++;
|
||||
condition.notify_one();
|
||||
}
|
||||
|
||||
_ALWAYS_INLINE_ void wait() const {
|
||||
std::unique_lock<decltype(mutex_)> lock(mutex_);
|
||||
while (!count_) { // Handle spurious wake-ups.
|
||||
condition_.wait(lock);
|
||||
std::unique_lock lock(mutex);
|
||||
while (!count) { // Handle spurious wake-ups.
|
||||
condition.wait(lock);
|
||||
}
|
||||
--count_;
|
||||
count--;
|
||||
}
|
||||
|
||||
_ALWAYS_INLINE_ bool try_wait() const {
|
||||
std::lock_guard<decltype(mutex_)> lock(mutex_);
|
||||
if (count_) {
|
||||
--count_;
|
||||
std::lock_guard lock(mutex);
|
||||
if (count) {
|
||||
count--;
|
||||
return true;
|
||||
}
|
||||
} else {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
#endif // SEMAPHORE_H
|
||||
|
|
Loading…
Reference in New Issue