#ifndef LLARP_CODEL_QUEUE_HPP #define LLARP_CODEL_QUEUE_HPP #ifdef _MSC_VER #define NOMINMAX #ifdef min #undef min #endif #endif #include #include #include #include #include #include #include #include #include namespace llarp { namespace util { struct DummyMutex { }; struct DummyLock { DummyLock(const DummyMutex& mtx){}; ~DummyLock() { } }; template < typename T, typename GetTime, typename PutTime, typename Compare, typename Mutex_t = util::Mutex, typename Lock_t = util::Lock, llarp_time_t dropMs = 5, llarp_time_t initialIntervalMs = 100 > struct CoDelQueue { CoDelQueue(const std::string& name) : m_name(name) { } size_t Size() { Lock_t lock(m_QueueMutex); return m_Queue.size(); } template < typename... Args > bool EmplaceIf(std::function< bool(T&) > pred, Args&&... args) { T t(std::forward< Args >(args)...); if(!pred(t)) { return false; } PutTime()(t); { Lock_t lock(m_QueueMutex); if(firstPut == 0) firstPut = GetTime()(t); m_Queue.push(t); } return true; } template < typename... Args > void Emplace(Args&&... args) { T t(std::forward< Args >(args)...); PutTime()(t); { Lock_t lock(m_QueueMutex); if(firstPut == 0) firstPut = GetTime()(t); m_Queue.push(t); } } void Put(T& t) { PutTime()(t); { Lock_t lock(m_QueueMutex); if(firstPut == 0) firstPut = GetTime()(t); m_Queue.push(t); } } template < typename Visit > void Process(Visit visitor) { llarp_time_t lowest = 0xFFFFFFFFFFFFFFFFUL; // auto start = llarp_time_now_ms(); // llarp::LogInfo("CoDelQueue::Process - start at ", start); Lock_t lock(m_QueueMutex); auto start = firstPut; while(m_Queue.size()) { llarp::LogDebug("CoDelQueue::Process - queue has ", m_Queue.size()); auto& item = m_Queue.front(); auto dlt = start - GetTime()(item); // llarp::LogInfo("CoDelQueue::Process - dlt ", dlt); lowest = std::min(dlt, lowest); if(m_Queue.size() == 1) { // llarp::LogInfo("CoDelQueue::Process - single item: lowest ", // lowest, " dropMs: ", dropMs); if(lowest > dropMs) { nextTickInterval += initialIntervalMs / std::sqrt(++dropNum); m_Queue.pop(); firstPut = 0; return; } else { nextTickInterval = initialIntervalMs; dropNum = 0; } } // llarp::LogInfo("CoDelQueue::Process - passing"); visitor(item); m_Queue.pop(); } firstPut = 0; } llarp_time_t firstPut = 0; size_t dropNum = 0; llarp_time_t nextTickInterval = initialIntervalMs; Mutex_t m_QueueMutex; typedef std::queue< T > Queue_t; Queue_t m_Queue; std::string m_name; }; } // namespace util } // namespace llarp #endif