SimpleSpinLock.h

/*
 * Copyright (c) 2018 https://www.thecoderscorner.com (Dave Cherry)..
 * This product is licensed under an Apache license, see the LICENSE file in the top-level directory.
 */
 
#ifndef TASKMANAGERIO_RENTRANTYIELDINGLOCK_H
#define TASKMANAGERIO_RENTRANTYIELDINGLOCK_H
 
#include "TaskManagerIO.h"

class SimpleSpinLock {
private:
    tm_internal::TimerTaskAtomicPtr initiatingTask;
#if defined(IOA_MULTITHREADED)
    volatile void* currentThread = nullptr;
#endif
    tm_internal::TmAtomicBool locked;
    volatile uint8_t count;
public:
    SimpleSpinLock() {
        initiatingTask = nullptr;
        locked = false;
        count = 0;
    }

    void lock() {
        spinLock(0xFFFFFFFFUL);
    }

    bool tryLock();

    bool spinLock(unsigned long iterations);

    void unlock();
 
    uint8_t getLockCount() const { return count; }
 
    bool isLocked() const { return locked; }
};

class TaskMgrLock {
private:
    SimpleSpinLock& lock;
 
public:
    TaskMgrLock(SimpleSpinLock& theLock) : lock(theLock) {
        lock.lock();
    }
 
    ~TaskMgrLock() {
        lock.unlock();
    }
};
 
#endif //TASKMANAGERIO_RENTRANTYIELDINGLOCK_H