dn724/dncurrency
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions

Small processing_queue improvements (#4159)

Browse source 
* Add batch type alias

* Add `joinable` function

* Use universal reference
This commit is contained in:
Piotr Wójcik 2023-03-01 12:28:12 +01:00 • committed by GitHub
commit 9f682b49a4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

18
nano/lib/processing_queue.hpp
View file

@ -6,6 +6,7 @@
#include <nano/lib/threading.hpp>
#include <nano/lib/utility.hpp>
#include <algorithm>
#include <condition_variable>
#include <deque>
#include <functional>
@ -23,6 +24,7 @@ class processing_queue final
{
public:
using value_t = T;
using batch_t = std::deque<value_t>;
/**
* @param thread_role Spawned processing threads will use this name
@ -71,15 +73,23 @@ public:
threads.clear ();
}
bool joinable () const
{
return std::any_of (threads.cbegin (), threads.cend (), [] (auto const & thread) {
return thread.joinable ();
});
}
/**
* Queues item for batch processing
*/
void add (T && item)
template <class Item>
void add (Item && item)
{
nano::unique_lock<nano::mutex> lock{ mutex };
if (queue.size () < max_queue_size)
{
queue.emplace_back (std::forward<T> (item));
queue.push_back (std::forward<T> (item));
lock.unlock ();
condition.notify_one ();
stats.inc (stat_type, nano::stat::detail::queue);
@ -136,7 +146,7 @@ private:
for (int n = 0; n < max_batch_size; ++n)
{
debug_assert (!queue.empty ());
queue_l.emplace_back (std::move (queue.front ()));
queue_l.push_back (std::move (queue.front ()));
queue.pop_front ();
}
return queue_l;
@ -160,7 +170,7 @@ private:
}
public:
std::function<void (std::deque<value_t> &)> process_batch{ [] (auto &) { debug_assert (false, "processing queue callback empty"); } };
std::function<void (batch_t &)> process_batch{ [] (auto &) { debug_assert (false, "processing queue callback empty"); } };
private:
nano::stats & stats;