123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192 |
- #ifndef BOOST_LOG_SINKS_BOUNDED_FIFO_QUEUE_HPP_INCLUDED_
- #define BOOST_LOG_SINKS_BOUNDED_FIFO_QUEUE_HPP_INCLUDED_
- #include <boost/log/detail/config.hpp>
- #ifdef BOOST_HAS_PRAGMA_ONCE
- #pragma once
- #endif
- #if defined(BOOST_LOG_NO_THREADS)
- #error Boost.Log: This header content is only supported in multithreaded environment
- #endif
- #include <cstddef>
- #include <queue>
- #include <boost/thread/locks.hpp>
- #include <boost/thread/mutex.hpp>
- #include <boost/thread/condition_variable.hpp>
- #include <boost/log/core/record_view.hpp>
- #include <boost/log/detail/header.hpp>
- namespace boost {
- BOOST_LOG_OPEN_NAMESPACE
- namespace sinks {
- template< std::size_t MaxQueueSizeV, typename OverflowStrategyT >
- class bounded_fifo_queue :
- private OverflowStrategyT
- {
- private:
- typedef OverflowStrategyT overflow_strategy;
- typedef std::queue< record_view > queue_type;
- typedef boost::mutex mutex_type;
- private:
-
- mutex_type m_mutex;
-
- condition_variable m_cond;
-
- queue_type m_queue;
-
- bool m_interruption_requested;
- protected:
-
- bounded_fifo_queue() : m_interruption_requested(false)
- {
- }
-
- template< typename ArgsT >
- explicit bounded_fifo_queue(ArgsT const&) : m_interruption_requested(false)
- {
- }
-
- void enqueue(record_view const& rec)
- {
- unique_lock< mutex_type > lock(m_mutex);
- std::size_t size = m_queue.size();
- for (; size >= MaxQueueSizeV; size = m_queue.size())
- {
- if (!overflow_strategy::on_overflow(rec, lock))
- return;
- }
- m_queue.push(rec);
- if (size == 0)
- m_cond.notify_one();
- }
-
- bool try_enqueue(record_view const& rec)
- {
- unique_lock< mutex_type > lock(m_mutex, try_to_lock);
- if (lock.owns_lock())
- {
- const std::size_t size = m_queue.size();
-
- if (size < MaxQueueSizeV)
- {
- m_queue.push(rec);
- if (size == 0)
- m_cond.notify_one();
- return true;
- }
- }
- return false;
- }
-
- bool try_dequeue_ready(record_view& rec)
- {
- return try_dequeue(rec);
- }
-
- bool try_dequeue(record_view& rec)
- {
- lock_guard< mutex_type > lock(m_mutex);
- const std::size_t size = m_queue.size();
- if (size > 0)
- {
- rec.swap(m_queue.front());
- m_queue.pop();
- overflow_strategy::on_queue_space_available();
- return true;
- }
- return false;
- }
-
- bool dequeue_ready(record_view& rec)
- {
- unique_lock< mutex_type > lock(m_mutex);
- while (!m_interruption_requested)
- {
- const std::size_t size = m_queue.size();
- if (size > 0)
- {
- rec.swap(m_queue.front());
- m_queue.pop();
- overflow_strategy::on_queue_space_available();
- return true;
- }
- else
- {
- m_cond.wait(lock);
- }
- }
- m_interruption_requested = false;
- return false;
- }
-
- void interrupt_dequeue()
- {
- lock_guard< mutex_type > lock(m_mutex);
- m_interruption_requested = true;
- overflow_strategy::interrupt();
- m_cond.notify_one();
- }
- };
- }
- BOOST_LOG_CLOSE_NAMESPACE
- }
- #include <boost/log/detail/footer.hpp>
- #endif
|