56 lines
1.1 KiB
C++
56 lines
1.1 KiB
C++
#pragma once
|
|
|
|
#include <queue>
|
|
#include <mutex>
|
|
#include <condition_variable>
|
|
#include <robin_hood.h>
|
|
|
|
namespace data {
|
|
/// Thread safe queue
|
|
template <class T>
|
|
class safe_queue {
|
|
private:
|
|
std::queue<T> queue;
|
|
std::mutex mutex;
|
|
std::condition_variable cv;
|
|
|
|
public:
|
|
void push(const T& in) {
|
|
std::unique_lock<std::mutex> lock(mutex);
|
|
queue.push(in);
|
|
cv.notify_one();
|
|
}
|
|
|
|
bool pop(T& out) {
|
|
std::unique_lock<std::mutex> lock(mutex);
|
|
if (queue.empty())
|
|
return false;
|
|
|
|
out = queue.front();
|
|
queue.pop();
|
|
return true;
|
|
}
|
|
|
|
bool empty() {
|
|
std::unique_lock<std::mutex> lock(mutex);
|
|
return queue.empty();
|
|
}
|
|
|
|
size_t size() {
|
|
std::unique_lock<std::mutex> lock(mutex);
|
|
return queue.size();
|
|
}
|
|
|
|
void notify() {
|
|
cv.notify_all();
|
|
}
|
|
|
|
void wait() {
|
|
std::unique_lock<std::mutex> lock(mutex);
|
|
if(queue.empty())
|
|
cv.wait(lock);
|
|
}
|
|
|
|
};
|
|
}
|