SharedAcquire.hpp

#pragma once

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

namespace samchon
{
namespace library
{
    class SharedAcquire
    {
    private:
        Semaphore *semaphore;

        std::atomic<size_t> *reference;

        std::atomic<bool> *locked;

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

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

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

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

        SharedAcquire(SharedAcquire &&obj)
        {
            //MOVE
            this->semaphore = obj.semaphore;
            this->reference = obj.reference;
            this->locked = obj.locked;

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

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

            if (locked->load() == true)
                semaphore->release();

            delete reference;
            delete locked;
        };

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

            semaphore->acquire();
            locked->store(true);
        };

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

            semaphore->release();
            locked->store(false);
        };

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