ddnet/src/base/tl/threading.h

#ifndef BASE_TL_THREADING_H
#define BASE_TL_THREADING_H

#include "../system.h"
#include <atomic>

class CSemaphore
{
	SEMAPHORE m_Sem;
	// implement the counter seperatly, because the `sem_getvalue`-API is
	// deprecated on macOS: https://stackoverflow.com/a/16655541
	std::atomic_int count;

public:
	CSemaphore() { sphore_init(&m_Sem); }
	~CSemaphore() { sphore_destroy(&m_Sem); }
	CSemaphore(const CSemaphore &) = delete;
	int GetApproximateValue() { return count.load(); }
	void Wait()
	{
		sphore_wait(&m_Sem);
		count.fetch_sub(1);
	}
	void Signal()
	{
		count.fetch_add(1);
		sphore_signal(&m_Sem);
	}
};

class SCOPED_CAPABILITY CLock
{
	LOCK m_Lock;

public:
	CLock() ACQUIRE(m_Lock)
	{
		m_Lock = lock_create();
	}

	~CLock() RELEASE()
	{
		lock_destroy(m_Lock);
	}

	CLock(const CLock &) = delete;

	void Take() ACQUIRE(m_Lock) { lock_wait(m_Lock); }
	void Release() RELEASE() { lock_unlock(m_Lock); }
};

class CScopeLock
{
	CLock *m_pLock;

public:
	CScopeLock(CLock *pLock)
	{
		m_pLock = pLock;
		m_pLock->Take();
	}

	~CScopeLock()
	{
		m_pLock->Release();
	}

	CScopeLock(const CScopeLock &) = delete;
};

#endif // BASE_TL_THREADING_H