LockGuard.h

#pragma once
#include "AudioConfig.h"
#include "AudioTools/AudioLogger.h"
 
#ifdef USE_STD_CONCURRENCY
#  include <mutex>
#endif
 
#if defined(USE_CONCURRENCY)
#ifdef ESP32
#  include "freertos/FreeRTOS.h"
#  include "freertos/semphr.h"
#else
#  include "FreeRTOS.h"
#  include "semphr.h"
#endif
 
namespace audio_tools {
 
class MutexBase {
public:
  virtual void lock() {}
  virtual void unlock() {}
};
 
#if defined(USE_STD_CONCURRENCY) 
 
class StdMutex : public MutexBase {
public:
  void lock() override { std_mutex.lock(); }
  void unlock() override { std_mutex.unlock(); }
 
protected:
  std::mutex std_mutex;
};
 
#endif
 
 
class Mutex : public MutexBase {
public:
  Mutex() {
    TRACED();
    xSemaphore = xSemaphoreCreateBinary();
    xSemaphoreGive(xSemaphore);
  }
  ~Mutex() {
    TRACED();
    vSemaphoreDelete(xSemaphore);
  }
  void lock() override {
    TRACED();
    xSemaphoreTake(xSemaphore, portMAX_DELAY);
  }
  void unlock() override {
    TRACED();
    xSemaphoreGive(xSemaphore);
  }
 
protected:
  SemaphoreHandle_t xSemaphore = NULL;
};
 
 
class LockGuard {
public:
  LockGuard(Mutex &mutex) {
    TRACED();
    p_mutex = &mutex;
    p_mutex->lock();
  }
  LockGuard(Mutex *mutex) {
    TRACED();
    p_mutex = mutex;
    p_mutex->lock();
  }
  ~LockGuard() {
    TRACED();
    p_mutex->unlock();
  }
 
protected:
  Mutex *p_mutex = nullptr;
};
 
}
 
#endif