UniqueWriteLock.hpp

#pragma once

#include <samchon/library/RWMutex.hpp>

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

        bool locked;

    public:
        /* -----------------------------------------------------------
            CONSTRUCTORS
        ----------------------------------------------------------- */
        UniqueWriteLock(RWMutex &rw_mutex, bool lock = true)
        {
            this->rw_mutex = &rw_mutex;
            this->locked = false;

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

        UniqueWriteLock(const UniqueWriteLock &) = delete;

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

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

        ~UniqueWriteLock()
        {
            if (locked == true)
                rw_mutex->writeUnlock();
        };

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

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

        void unlock()
        {
            if (locked == false)
                return;

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

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