123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262 |
- #ifndef BOOST_LOG_SINKS_BOUNDED_ORDERING_QUEUE_HPP_INCLUDED_
- #define BOOST_LOG_SINKS_BOUNDED_ORDERING_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 <vector>
- #include <boost/cstdint.hpp>
- #include <boost/thread/locks.hpp>
- #include <boost/thread/mutex.hpp>
- #include <boost/thread/condition_variable.hpp>
- #include <boost/thread/thread_time.hpp>
- #include <boost/date_time/posix_time/posix_time_types.hpp>
- #include <boost/log/detail/timestamp.hpp>
- #include <boost/log/detail/enqueued_record.hpp>
- #include <boost/log/keywords/order.hpp>
- #include <boost/log/keywords/ordering_window.hpp>
- #include <boost/log/core/record_view.hpp>
- #include <boost/log/detail/header.hpp>
- namespace boost {
- BOOST_LOG_OPEN_NAMESPACE
- namespace sinks {
- template< typename OrderT, std::size_t MaxQueueSizeV, typename OverflowStrategyT >
- class bounded_ordering_queue :
- private OverflowStrategyT
- {
- private:
- typedef OverflowStrategyT overflow_strategy;
- typedef boost::mutex mutex_type;
- typedef sinks::aux::enqueued_record enqueued_record;
- typedef std::priority_queue<
- enqueued_record,
- std::vector< enqueued_record >,
- enqueued_record::order< OrderT >
- > queue_type;
- private:
-
- const uint64_t m_ordering_window;
-
- mutex_type m_mutex;
-
- condition_variable m_cond;
-
- queue_type m_queue;
-
- bool m_interruption_requested;
- public:
-
- posix_time::time_duration get_ordering_window() const
- {
- return posix_time::milliseconds(m_ordering_window);
- }
-
- static posix_time::time_duration get_default_ordering_window()
- {
-
-
-
-
-
-
-
- return posix_time::milliseconds(30);
- }
- protected:
-
- template< typename ArgsT >
- explicit bounded_ordering_queue(ArgsT const& args) :
- m_ordering_window(args[keywords::ordering_window || &bounded_ordering_queue::get_default_ordering_window].total_milliseconds()),
- m_queue(args[keywords::order]),
- 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(enqueued_record(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(enqueued_record(rec));
- if (size == 0)
- m_cond.notify_one();
- return true;
- }
- }
- return false;
- }
- //! Attempts to dequeue a log record ready for processing from the queue, does not block if the queue is empty
- bool try_dequeue_ready(record_view& rec)
- {
- lock_guard< mutex_type > lock(m_mutex);
- const std::size_t size = m_queue.size();
- if (size > 0)
- {
- const boost::log::aux::timestamp now = boost::log::aux::get_timestamp();
- enqueued_record const& elem = m_queue.top();
- if (static_cast< uint64_t >((now - elem.m_timestamp).milliseconds()) >= m_ordering_window)
- {
-
- rec = elem.m_record;
- m_queue.pop();
- overflow_strategy::on_queue_space_available();
- return true;
- }
- }
- return false;
- }
-
- bool try_dequeue(record_view& rec)
- {
- lock_guard< mutex_type > lock(m_mutex);
- const std::size_t size = m_queue.size();
- if (size > 0)
- {
- enqueued_record const& elem = m_queue.top();
- rec = elem.m_record;
- 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)
- {
- const boost::log::aux::timestamp now = boost::log::aux::get_timestamp();
- enqueued_record const& elem = m_queue.top();
- const uint64_t difference = (now - elem.m_timestamp).milliseconds();
- if (difference >= m_ordering_window)
- {
- rec = elem.m_record;
- m_queue.pop();
- overflow_strategy::on_queue_space_available();
- return true;
- }
- else
- {
-
- m_cond.timed_wait(lock, posix_time::milliseconds(m_ordering_window - difference));
- }
- }
- 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
|