mirror of
https://github.com/BeamMP/BeamMP-Launcher.git
synced 2026-04-05 07:16:29 +00:00
Custom thread safe wrapper for std::queue
This commit is contained in:
56
include/atomic_queue.h
Normal file
56
include/atomic_queue.h
Normal file
@@ -0,0 +1,56 @@
|
||||
//
|
||||
// Created by Anonymous275 on 24/07/22.
|
||||
//
|
||||
|
||||
#pragma once
|
||||
#include <semaphore>
|
||||
#include <queue>
|
||||
|
||||
template <class T, size_t Size>
|
||||
class atomic_queue {
|
||||
public:
|
||||
bool try_pop(T& val) {
|
||||
lock_guard guard(semaphore);
|
||||
if(queue.empty())return false;
|
||||
val = queue.front();
|
||||
queue.pop();
|
||||
full.release();
|
||||
return true;
|
||||
}
|
||||
|
||||
void push(const T& val) {
|
||||
check_full();
|
||||
lock_guard guard(semaphore);
|
||||
queue.push(val);
|
||||
}
|
||||
|
||||
size_t size() {
|
||||
lock_guard guard(semaphore);
|
||||
return queue.size();
|
||||
}
|
||||
|
||||
bool empty() {
|
||||
lock_guard guard(semaphore);
|
||||
return queue.empty();
|
||||
}
|
||||
private:
|
||||
void check_full() {
|
||||
if(size() >= Size) {
|
||||
full.acquire();
|
||||
}
|
||||
}
|
||||
private:
|
||||
struct lock_guard {
|
||||
explicit lock_guard(std::binary_semaphore& lock) : lock(lock){
|
||||
lock.acquire();
|
||||
}
|
||||
~lock_guard() {
|
||||
lock.release();
|
||||
}
|
||||
private:
|
||||
std::binary_semaphore& lock;
|
||||
};
|
||||
std::binary_semaphore semaphore{1}, full{0};
|
||||
std::queue<T> queue{};
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user