lock.h

Go to the documentation of this file.

// Evo C++ Library
/* Copyright 2019 Justin Crowell
Distributed under the BSD 2-Clause License -- see included file LICENSE.txt for details.
*/

#pragma once
#ifndef INCL_evo_lock_h
#define INCL_evo_lock_h

namespace evo {


template<class T>
struct SmartLock {
    SmartLock(T& object, bool lock=true) : object_(object) {
        locked_ = false;
        if (lock) {
            object_.lock();
            locked_ = true;
        }
    }

    ~SmartLock() {
        if (locked_)
            object_.unlock();
    }

    SmartLock& lock() {
        if (!locked_) {
            object_.lock();
            locked_ = true;
        }
        return *this;
    }

    SmartLock& unlock() {
        if (locked_) {
            object_.unlock();
            locked_ = false;
        }
        return *this;
    }

    T& object_;     

protected:
    bool locked_;   
};


template<class T>
struct SmartLockRead {
    SmartLockRead(T& object, bool lock=true) : object_(object) {
        locked_ = false;
        if (lock) {
            object_.lock_read();
            locked_ = true;
        }
    }

    ~SmartLockRead() {
        if (locked_)
            object_.unlock_read();
    }

    SmartLockRead& lock() {
        if (!locked_) {
            object_.lock_read();
            locked_ = true;
        }
        return *this;
    }

    SmartLockRead& unlock() {
        if (locked_) {
            object_.unlock_read();
            locked_ = false;
        }
        return *this;
    }

    T& object_;     

protected:
    bool locked_;   
};


template<class T>
struct SmartSleepLock {
    SmartSleepLock(T& object, ulong sleep_ms=1) : object_(object) {
        locked_ = false;
        if (sleep_ms > 0) {
            object_.sleeplock(sleep_ms);
            locked_ = true;
        }
    }

    ~SmartSleepLock() {
        if (locked_)
            object_.unlock();
    }

    SmartSleepLock& lock(ulong sleep_ms=1) {
        if (!locked_ && sleep_ms > 0) {
            object_.sleeplock(sleep_ms);
            locked_ = true;
        }
        return *this;
    }

    SmartSleepLock& unlock() {
        if (locked_) {
            object_.unlock();
            locked_ = false;
        }
        return *this;
    }

    T& object_;     

protected:
    bool locked_;   
};

}
#endif