1 #ifndef OSMIUM_THREAD_QUEUE_HPP
2 #define OSMIUM_THREAD_QUEUE_HPP
38 #include <condition_variable>
74 #ifdef OSMIUM_DEBUG_QUEUE_SIZE
75 size_t m_largest_size;
80 std::atomic<int> m_full_counter;
92 Queue(
size_t max_size = 0,
const std::string& name =
"") :
99 #ifdef OSMIUM_DEBUG_QUEUE_SIZE
109 #ifdef OSMIUM_DEBUG_QUEUE_SIZE
110 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";
120 while (
size() >= m_max_size) {
122 #ifdef OSMIUM_DEBUG_QUEUE_SIZE
127 std::lock_guard<std::mutex> lock(m_mutex);
128 m_queue.push(std::move(value));
129 #ifdef OSMIUM_DEBUG_QUEUE_SIZE
130 if (m_largest_size < m_queue.size()) {
131 m_largest_size = m_queue.size();
134 m_data_available.notify_one();
139 m_data_available.notify_all();
143 std::unique_lock<std::mutex> lock(m_mutex);
144 m_data_available.wait(lock, [
this] {
145 return !m_queue.empty() ||
m_done;
147 if (!m_queue.empty()) {
148 value = std::move(m_queue.front());
154 std::unique_lock<std::mutex> lock(m_mutex);
155 if (!m_data_available.wait_for(lock, std::chrono::seconds(1), [
this] {
156 return !m_queue.empty() || m_done;
160 if (!m_queue.empty()) {
161 value = std::move(m_queue.front());
167 std::lock_guard<std::mutex> lock(m_mutex);
168 if (m_queue.empty()) {
171 value = std::move(m_queue.front());
177 std::lock_guard<std::mutex> lock(m_mutex);
178 return m_queue.empty();
182 std::lock_guard<std::mutex> lock(m_mutex);
183 return m_queue.size();
192 #endif // OSMIUM_THREAD_QUEUE_HPP
size_t size() const
Definition: queue.hpp:181
bool empty() const
Definition: queue.hpp:176
std::mutex m_mutex
Definition: queue.hpp:65
std::atomic< bool > m_done
Definition: queue.hpp:72
void shutdown()
Definition: queue.hpp:137
~Queue()
Definition: queue.hpp:107
std::queue< T > m_queue
Definition: queue.hpp:67
const std::string m_name
Name of this queue (for debugging only).
Definition: queue.hpp:63
const size_t m_max_size
Definition: queue.hpp:60
Namespace for everything in the Osmium library.
Definition: assembler.hpp:55
void wait_and_pop_with_timeout(T &value)
Definition: queue.hpp:153
bool try_pop(T &value)
Definition: queue.hpp:166
static const std::chrono::milliseconds full_queue_sleep_duration
Definition: queue.hpp:50
void wait_and_pop(T &value)
Definition: queue.hpp:142
void push(T value)
Definition: queue.hpp:118
Queue(size_t max_size=0, const std::string &name="")
Definition: queue.hpp:92
std::condition_variable m_data_available
Used to signal readers when data is available in the queue.
Definition: queue.hpp:70