Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code

Unified Diff: src/plugin/ActiveQueue.h

Issue 29323611: Issue #1234, #2058 - Rewrite log facility, improving thread implementation
Patch Set: rebase only Created July 27, 2016, 9:11 p.m.
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « adblockplus.gyp ('k') | src/plugin/Config.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: src/plugin/ActiveQueue.h
===================================================================
new file mode 100644
--- /dev/null
+++ b/src/plugin/ActiveQueue.h
@@ -0,0 +1,281 @@
+/*
+ * This file is part of Adblock Plus <https://adblockplus.org/>,
+ * Copyright (C) 2006-2016 Eyeo GmbH
+ *
+ * Adblock Plus is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 3 as
+ * published by the Free Software Foundation.
+ *
+ * Adblock Plus is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include <thread>
+#include <mutex>
+#include <condition_variable>
+#include <deque>
+#include "Placeholder.h"
+
+/**
+ * A synchronized FIFO queue with thread notice on receipt.
+ *
+ * \tparam T
+ * Class of the elements of the queue.
+ * It must have a move constructor.
+ */
+template<class T>
+class MessageQueue
+{
+ std::deque<T> queue;
+ typedef std::lock_guard<std::mutex> SentryType;
+ typedef std::unique_lock<std::mutex> UniqueLockType;
+ std::mutex mutex;
+ std::condition_variable cv;
+public:
+ MessageQueue() {} // = default
+
+ /*
+ * This class does not have any responsibility with regard to the elements of its queue.
+ * An instance may be destroyed with elements still in the queue.
+ */
+ ~MessageQueue() {} // = default;
+
+ /**
+ * Insert an element, copy semantics version.
+ */
+ void Insert(const T& t)
+ {
+ SentryType sentry(mutex);
+ queue.push_back(t);
+ cv.notify_one();
+ }
+
+ /**
+ * Insert an element, move semantics version.
+ */
+ void Insert(T&& t)
+ {
+ SentryType sentry(mutex);
+ queue.push_back(std::move(t));
+ cv.notify_one();
+ }
+
+ /**
+ * Wake up anything waiting on the condition variable.
+ *
+ * \tparam Function
+ * Functor type for argument
+ * \param f
+ * Functor to execute _inside_ the protection of the queue's mutex
+ */
+ template<typename Function>
+ void Rouse(Function& f)
+ {
+ SentryType sentry(mutex);
+ f();
+ cv.notify_one();
+ }
+
+ /**
+ * Test, Remove, and Execute
+ *
+ * If the queue is not empty, remove the next element from the consumer end
+ * of the queue and execute the argument functor on it.
+ *
+ * \param f
+ * A functor to which to pass the element at the consumer end of the queue.
+ * This functor is executed _outside_ the protection of the queue's mutex.
+ * \return
+ * True if and only if an element was removed from the front of the queue.
+ */
+ template<typename Function>
+ bool Remove(Function& f)
+ {
+ /*
+ * We need to remove an element atomically from the queue,
+ * so we need to construct that element _inside_ the protection of the mutex,
+ * On the other hand, we need to execute the function _outside_ the protection
+ * of the queue, because otherwise performance may suffer.
+ * Hence we use 'Placeholder' so that we can declare a variable to hold the
+ * removed element without requiring it to have a default constructor or
+ * assignment operators.
+ */
+ Placeholder<T> x;
+ {
+ SentryType sentry(mutex);
+ if (queue.empty())
+ return false;
+ x.Construct(std::move(queue.front())); // only require move-constructibility
+ queue.pop_front();
+ }
+ f(std::move(x.Object()));
+ return true;
+ }
+
+ /**
+ * If a condition is satisfied, wait indefinitely. If not, don't wait.
+ *
+ * \return False, immediately without waiting, if the argument evaluates false.
+ * \return True, only after waiting, if the argument evaluates true.
+ */
+ template<typename Predicate>
+ bool WaitIf(Predicate& p)
+ {
+ UniqueLockType ul(mutex);
+ if (!p())
+ {
+ return false;
+ }
+ cv.wait(ul);
+ return true;
+ }
+
+ /*
+ * A defect in the compiler for VS2012 requires these definitions.
+ */
+#if defined(_MSC_VER) && _MSC_VER <= 1700
+#define WAIT_FOR_RETURN_TYPE std::cv_status::cv_status
+#else
+#define WAIT_FOR_RETURN_TYPE std::cv_status
+#endif
+
+ /**
+ * Wait for a limited duration.
+ *
+ * Return type was going to be "decltype(cv.wait_for)" for clarity.
+ * A defect in the toolset for VS 2012 causes that declaration not to compile.
+ */
+ template <class R, class P>
+ WAIT_FOR_RETURN_TYPE WaitFor(const std::chrono::duration<R,P>& relativeDuration)
+ {
+ return cv.wait_for(UniqueLockType(mutex), relativeDuration);
+ }
+};
+
+/**
+ * An active queue with a single, internal consumer.
+ *
+ * This class presents the front of a message queue for arbitrary producers.
+ * The back of the message queue is managed by an internal consumer,
+ * which calls the processor for each element passed through the queue.
+ *
+ * The internal thread has the same lifetime as the object.
+ * There's no external interface to pause or kill the thread.
+ * Destroy the object to terminate the thread.
+ * The consumer drains the queue before terminating the thread.
+ *
+ * \tparam T Class of elements in the queue
+ * \tparam F Type of functor to process each element consumed from the queue
+ */
+template<class T, class F>
+class ActiveQueue
+{
+ /**
+ * Signal flag indicating to the consumer thread that it should continue running.
+ *
+ * This flag is true for the entire duration of the object lifetime;
+ * it's set to false only in the destructor.
+ * To avoid race conditions, it's only accessed under mutex protection with calls to notify and wait.
+ * We accomplish this with lambdas passed as the functor arguments of the message queue functions 'Rouse' and 'WaitIf'.
+ * These lambdas are the only references to the variable after construction.
+ */
+ bool running;
+
+ /**
+ * Functor to process each element as it is removed from the queue.
+ */
+ F& processor;
+
+ /**
+ * The queue that the active thread behavior is wrapped around.
+ */
+ MessageQueue<T> queue;
+
+ /**
+ * Thread running the consumer process.
+ *
+ * This thread runs the entire lifetime of this object.
+ * It's started in the constructor and joined in the destructor.
+ *
+ * This member variable is declared last so that it is constructed last,
+ * after all the other member necessary for proper functioning of the consumer.
+ */
+ std::thread thread;
+
+ /**
+ * Main function for the consumer thread.
+ *
+ * Strictly speaking, it's the effective main function for the thread.
+ * The actual main function is a lambda that only calls this function and nothing else.
+ */
+ void Consumer()
+ {
+ while (queue.WaitIf([this]() { return running; }))
+ {
+ // Drain the queue, processing all of its elements
+ while (
+ queue.Remove(
+ [&](T&& t) {
+ try
+ {
+ processor(std::move(t));
+ }
+ catch (...)
+ {
+ // Ignore any exception 'processor' may throw.
+ }
+ }
+ )
+ )
+ {
+ }
+ // Loop termination: queue.Remove() removed nothing from the queue, so it's now empty
+ }
+ }
+
+public:
+ ActiveQueue(F& f)
+ : running(true), processor(f), queue(), thread([this]() { Consumer(); })
+ {}
+
+ ~ActiveQueue()
+ {
+ /*
+ * The consumer thread is waiting on the condition variable.
+ * If we don't wake it up, our thread won't terminate.
+ *
+ * Note that we don't have a race condition here with new elements arriving
+ * while we are draining the queue for the last time.
+ * We are executing in the destructor, so our member function Insert() is unavailable,
+ * and it's the only way that new element enter the message queue.
+ */
+ queue.Rouse([this](){ running = false; });
+ thread.join();
+ }
+
+ ActiveQueue(const ActiveQueue&); // = delete
+ ActiveQueue(ActiveQueue&&); // = delete
+ ActiveQueue& operator=(const ActiveQueue&); // = delete
+ ActiveQueue& operator=(ActiveQueue&&); // = delete
+
+ /**
+ * Insert
+ */
+ void Insert(const T& t)
+ {
+ queue.Insert(t);
+ }
+
+ /**
+ * Insert
+ */
+ void Insert(T&& t)
+ {
+ queue.Insert(std::move(t));
+ }
+};
« no previous file with comments | « adblockplus.gyp ('k') | src/plugin/Config.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld