C++ 读写锁实现

源码来自Godot

#ifndef RW_LOCK_H
#define RW_LOCK_H

#include "core/error/error_list.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 {
        mutex.lock_shared();
    }

    // Unlock the rwlock, let other threads continue
    void read_unlock() const {
        mutex.unlock_shared();
    }

    // Attempt to lock the rwlock, OK on success, ERR_BUSY means it can't lock.
    Error read_try_lock() const {
        return mutex.try_lock_shared() ? OK : ERR_BUSY;
    }

    // Lock the rwlock, block if locked by someone else
    void write_lock() {
        mutex.lock();
    }

    // Unlock the rwlock, let other thwrites continue
    void write_unlock() {
        mutex.unlock();
    }

    // Attempt to lock the rwlock, OK on success, ERR_BUSY means it can't lock.
    Error write_try_lock() {
        return mutex.try_lock() ? OK : ERR_BUSY;
    }
};

class RWLockRead {
    const RWLock &lock;

public:
    RWLockRead(const RWLock &p_lock) :
            lock(p_lock) {
        lock.read_lock();
    }
    ~RWLockRead() {
        lock.read_unlock();
    }
};

class RWLockWrite {
    RWLock &lock;

public:
    RWLockWrite(RWLock &p_lock) :
            lock(p_lock) {
        lock.write_lock();
    }
    ~RWLockWrite() {
        lock.write_unlock();
    }
};

#endif // RW_LOCK_H