SharedWriteLock.hpp

#pragma once

#include <atomic>
#include <samchon/library/RWMutex.hpp>

namespace samchon
{
namespace library
{
    class SharedWriteLock
    {
    private:
        RWMutex *rw_mutex;

        std::atomic<size_t> *reference;

        std::atomic<bool> *locked;

    public:
        /* -----------------------------------------------------------
            CONSTRUCTORS
        ----------------------------------------------------------- */
        SharedWriteLock(RWMutex &rw_mutex, bool lock = true)
        {
            this->rw_mutex = &rw_mutex;
            this->reference = new std::atomic<size_t>(0);
            this->locked = new std::atomic<bool>(false);

            if (lock == true)
                this->lock();
        };

        SharedWriteLock(const SharedWriteLock &obj)
        {
            obj.reference->operator++();

            this->rw_mutex = obj.rw_mutex;
            this->reference = obj.reference;
            this->locked = obj.locked;
        };

        SharedWriteLock(SharedWriteLock &&obj)
        {
            //MOVE
            this->rw_mutex = obj.rw_mutex;
            this->reference = obj.reference;
            this->locked = obj.locked;

            //TRUNCATE
            obj.rw_mutex = nullptr;
            obj.reference = nullptr;
            obj.locked = nullptr;
        };

        ~SharedWriteLock()
        {
            if (reference == nullptr || reference->operator--() > 0)
                return;

            if (locked->load() == true)
                rw_mutex->writeUnlock();

            delete reference;
            delete locked;
        };

        /* -----------------------------------------------------------
            LOCKERS
        ----------------------------------------------------------- */
        void lock()
        {
            if (locked->load() == true)
                return;

            rw_mutex->writeLock();
            locked->store(true);
        };

        void unlock()
        {
            if (locked->load() == false)
                return;

            rw_mutex->writeUnlock();
            locked->store(false);
        };

        //auto tryLock() -> bool;
    };
};
};