Program Listing for File recursive_read_write_mutex.h

↰ Return to documentation for file (/home/docs/checkouts/readthedocs.org/user_builds/canonical-mir-documentation/checkouts/2.30/include/common/mir/recursive_read_write_mutex.h)

/*
 * Copyright © Canonical Ltd.
 *
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License version 2 or 3,
 * as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef MIR_RECURSIVE_READ_WRITE_MUTEX_H_
#define MIR_RECURSIVE_READ_WRITE_MUTEX_H_

#include <condition_variable>
#include <thread>
#include <vector>

namespace mir
{
class RecursiveReadWriteMutex
{
public:
    void read_lock();

    void read_unlock();

    void write_lock();

    void write_unlock();

private:
    std::mutex mutex;
    std::condition_variable cv;
    struct ThreadLockCount
    {
        ThreadLockCount() : id(), count(0) {}
        ThreadLockCount(std::thread::id id, unsigned int count) : id(id), count(count) {}
        std::thread::id id;
        unsigned int count;
    };
    std::vector<ThreadLockCount> read_locking_threads;
    ThreadLockCount write_locking_thread;
};

class RecursiveReadLock
{
public:
    explicit RecursiveReadLock(RecursiveReadWriteMutex& mutex) : mutex(mutex) { mutex.read_lock(); }
    ~RecursiveReadLock() { mutex.read_unlock(); }

private:
    RecursiveReadWriteMutex& mutex;
};

class RecursiveWriteLock
{
public:
    explicit RecursiveWriteLock(RecursiveReadWriteMutex& mutex) : mutex(mutex) { mutex.write_lock(); }
    ~RecursiveWriteLock() { mutex.write_unlock(); }

private:
    RecursiveReadWriteMutex& mutex;
};
}

#endif /* MIR_RECURSIVE_READ_WRITE_MUTEX_H_ */