1 #ifndef OSMIUM_THREAD_QUEUE_HPP
2 #define OSMIUM_THREAD_QUEUE_HPP
38 #include <condition_variable>
76 #ifdef OSMIUM_DEBUG_QUEUE_SIZE
77 size_t m_largest_size;
82 std::atomic<int> m_full_counter;
94 Queue(
size_t max_size = 0,
const std::string& name =
"") :
101 #ifdef OSMIUM_DEBUG_QUEUE_SIZE
111 #ifdef OSMIUM_DEBUG_QUEUE_SIZE
112 std::cerr <<
"queue '" << m_name <<
"' with max_size=" << m_max_size <<
" had largest size " << m_largest_size <<
" and was full " << m_full_counter <<
" times\n";
122 while (
size() >= m_max_size) {
124 #ifdef OSMIUM_DEBUG_QUEUE_SIZE
129 std::lock_guard<std::mutex> lock(m_mutex);
130 m_queue.push(std::move(value));
131 #ifdef OSMIUM_DEBUG_QUEUE_SIZE
132 if (m_largest_size < m_queue.size()) {
133 m_largest_size = m_queue.size();
136 m_data_available.notify_one();
141 m_data_available.notify_all();
145 std::unique_lock<std::mutex> lock(m_mutex);
146 m_data_available.wait(lock, [
this] {
147 return !m_queue.empty() ||
m_done;
149 if (!m_queue.empty()) {
150 value = std::move(m_queue.front());
156 std::unique_lock<std::mutex> lock(m_mutex);
157 if (!m_data_available.wait_for(lock, std::chrono::seconds(1), [
this] {
158 return !m_queue.empty() || m_done;
162 if (!m_queue.empty()) {
163 value = std::move(m_queue.front());
169 std::lock_guard<std::mutex> lock(m_mutex);
170 if (m_queue.empty()) {
173 value = std::move(m_queue.front());
179 std::lock_guard<std::mutex> lock(m_mutex);
180 return m_queue.empty();
184 std::lock_guard<std::mutex> lock(m_mutex);
185 return m_queue.size();
194 #endif // OSMIUM_THREAD_QUEUE_HPP
size_t size() const
Definition: queue.hpp:183
bool empty() const
Definition: queue.hpp:178
std::mutex m_mutex
Definition: queue.hpp:67
std::atomic< bool > m_done
Definition: queue.hpp:74
void shutdown()
Definition: queue.hpp:139
~Queue()
Definition: queue.hpp:109
std::queue< T > m_queue
Definition: queue.hpp:69
const std::string m_name
Name of this queue (for debugging only).
Definition: queue.hpp:65
const size_t m_max_size
Definition: queue.hpp:62
Namespace for everything in the Osmium library.
Definition: assembler.hpp:55
void wait_and_pop_with_timeout(T &value)
Definition: queue.hpp:155
bool try_pop(T &value)
Definition: queue.hpp:168
static const std::chrono::milliseconds full_queue_sleep_duration
Definition: queue.hpp:52
void wait_and_pop(T &value)
Definition: queue.hpp:144
void push(T value)
Definition: queue.hpp:120
Queue(size_t max_size=0, const std::string &name="")
Definition: queue.hpp:94
std::condition_variable m_data_available
Used to signal readers when data is available in the queue.
Definition: queue.hpp:72