Merge #19090: refactor: Misc scheduler cleanups

fa8337fcdb clang-format scheduler (MarcoFalke)
fa3d41b5ab doc: Switch scheduler to doxygen comments (MarcoFalke)
fac43f9889 scheduler: Replace stop(true) with StopWhenDrained() (MarcoFalke)
fa9cca0550 doc: Remove unused documentation about unimplemented features (MarcoFalke)
fab2950d70 doc: Switch boost::thread to std::thread in scheduler (MarcoFalke)
fa9819695a test: Remove unused scheduler.h include from the common setup (MarcoFalke)
fa609c4f76 scheduler: Remove unused REVERSE_LOCK (MarcoFalke)

Pull request description:

  This accumulates a bunch of cleanup that was long overdue, but I haven't yet gotten around to address. Specifically, but not limited to:

  * Remove unused code, documentation and includes
  * Upgrade to doxygen documentation

  Please refer to the individual commits for more details.

ACKs for top commit:
  jnewbery:
    utACK fa8337fcdb

Tree-SHA512: 0c825ad9767e2697a3ef1ec1be13fdc2b18eeb7493ad0be5b65cc9f209391e78b17ee66e35e094c5e171c12b0f1624f287a110f6bddaf3024b708877afa8552e
This commit is contained in:
MarcoFalke committed 2020-06-25 09:24:58 -04:00
commit 3a4a3729d9
5 files changed
+72 -73

No files matched your search

+17 -25
View File
@@ -30,9 +30,6 @@ void CScheduler::serviceQueue()
// is called. // is called.
while (!shouldStop()) { while (!shouldStop()) {
try { try {
if (!shouldStop() && taskQueue.empty()) {
REVERSE_LOCK(lock);
}
while (!shouldStop() && taskQueue.empty()) { while (!shouldStop() && taskQueue.empty()) {
// Wait until there is something to do. // Wait until there is something to do.
newTaskScheduled.wait(lock); newTaskScheduled.wait(lock);
@@ -71,18 +68,6 @@ void CScheduler::serviceQueue()
newTaskScheduled.notify_one(); newTaskScheduled.notify_one();
} }
void CScheduler::stop(bool drain)
{
{
LOCK(newTaskMutex);
if (drain)
stopWhenEmpty = true;
else
stopRequested = true;
}
newTaskScheduled.notify_all();
}
void CScheduler::schedule(CScheduler::Function f, std::chrono::system_clock::time_point t) void CScheduler::schedule(CScheduler::Function f, std::chrono::system_clock::time_point t)
{ {
{ {
@@ -125,8 +110,8 @@ void CScheduler::scheduleEvery(CScheduler::Function f, std::chrono::milliseconds
scheduleFromNow([=] { Repeat(*this, f, delta); }, delta); scheduleFromNow([=] { Repeat(*this, f, delta); }, delta);
} }
size_t CScheduler::getQueueInfo(std::chrono::system_clock::time_point &first, size_t CScheduler::getQueueInfo(std::chrono::system_clock::time_point& first,
std::chrono::system_clock::time_point &last) const std::chrono::system_clock::time_point& last) const
{ {
LOCK(newTaskMutex); LOCK(newTaskMutex);
size_t result = taskQueue.size(); size_t result = taskQueue.size();
@@ -137,13 +122,15 @@ size_t CScheduler::getQueueInfo(std::chrono::system_clock::time_point &first,
return result; return result;
} }
bool CScheduler::AreThreadsServicingQueue() const { bool CScheduler::AreThreadsServicingQueue() const
{
LOCK(newTaskMutex); LOCK(newTaskMutex);
return nThreadsServicingQueue; return nThreadsServicingQueue;
} }
void SingleThreadedSchedulerClient::MaybeScheduleProcessQueue() { void SingleThreadedSchedulerClient::MaybeScheduleProcessQueue()
{
{ {
LOCK(m_cs_callbacks_pending); LOCK(m_cs_callbacks_pending);
// Try to avoid scheduling too many copies here, but if we // Try to avoid scheduling too many copies here, but if we
@@ -155,8 +142,9 @@ void SingleThreadedSchedulerClient::MaybeScheduleProcessQueue() {
m_pscheduler->schedule(std::bind(&SingleThreadedSchedulerClient::ProcessQueue, this), std::chrono::system_clock::now()); m_pscheduler->schedule(std::bind(&SingleThreadedSchedulerClient::ProcessQueue, this), std::chrono::system_clock::now());
} }
void SingleThreadedSchedulerClient::ProcessQueue() { void SingleThreadedSchedulerClient::ProcessQueue()
std::function<void ()> callback; {
std::function<void()> callback;
{ {
LOCK(m_cs_callbacks_pending); LOCK(m_cs_callbacks_pending);
if (m_are_callbacks_running) return; if (m_are_callbacks_running) return;
@@ -172,7 +160,8 @@ void SingleThreadedSchedulerClient::ProcessQueue() {
struct RAIICallbacksRunning { struct RAIICallbacksRunning {
SingleThreadedSchedulerClient* instance; SingleThreadedSchedulerClient* instance;
explicit RAIICallbacksRunning(SingleThreadedSchedulerClient* _instance) : instance(_instance) {} explicit RAIICallbacksRunning(SingleThreadedSchedulerClient* _instance) : instance(_instance) {}
~RAIICallbacksRunning() { ~RAIICallbacksRunning()
{
{ {
LOCK(instance->m_cs_callbacks_pending); LOCK(instance->m_cs_callbacks_pending);
instance->m_are_callbacks_running = false; instance->m_are_callbacks_running = false;
@@ -184,7 +173,8 @@ void SingleThreadedSchedulerClient::ProcessQueue() {
callback(); callback();
} }
void SingleThreadedSchedulerClient::AddToProcessQueue(std::function<void ()> func) { void SingleThreadedSchedulerClient::AddToProcessQueue(std::function<void()> func)
{
assert(m_pscheduler); assert(m_pscheduler);
{ {
@@ -194,7 +184,8 @@ void SingleThreadedSchedulerClient::AddToProcessQueue(std::function<void ()> fun
MaybeScheduleProcessQueue(); MaybeScheduleProcessQueue();
} }
void SingleThreadedSchedulerClient::EmptyQueue() { void SingleThreadedSchedulerClient::EmptyQueue()
{
assert(!m_pscheduler->AreThreadsServicingQueue()); assert(!m_pscheduler->AreThreadsServicingQueue());
bool should_continue = true; bool should_continue = true;
while (should_continue) { while (should_continue) {
@@ -204,7 +195,8 @@ void SingleThreadedSchedulerClient::EmptyQueue() {
} }
} }
size_t SingleThreadedSchedulerClient::CallbacksPending() { size_t SingleThreadedSchedulerClient::CallbacksPending()
{
LOCK(m_cs_callbacks_pending); LOCK(m_cs_callbacks_pending);
return m_callbacks_pending.size(); return m_callbacks_pending.size();
} }
+51 -44
View File
@@ -5,11 +5,6 @@
#ifndef BITCOIN_SCHEDULER_H #ifndef BITCOIN_SCHEDULER_H
#define BITCOIN_SCHEDULER_H #define BITCOIN_SCHEDULER_H
//
// NOTE:
// boost::thread should be ported to std::thread
// when we support C++11.
//
#include <condition_variable> #include <condition_variable>
#include <functional> #include <functional>
#include <list> #include <list>
@@ -17,24 +12,23 @@
#include <sync.h> #include <sync.h>
// /**
// Simple class for background tasks that should be run * Simple class for background tasks that should be run
// periodically or once "after a while" * periodically or once "after a while"
// *
// Usage: * Usage:
// *
// CScheduler* s = new CScheduler(); * CScheduler* s = new CScheduler();
// s->scheduleFromNow(doSomething, std::chrono::milliseconds{11}); // Assuming a: void doSomething() { } * s->scheduleFromNow(doSomething, std::chrono::milliseconds{11}); // Assuming a: void doSomething() { }
// s->scheduleFromNow([=] { this->func(argument); }, std::chrono::milliseconds{3}); * s->scheduleFromNow([=] { this->func(argument); }, std::chrono::milliseconds{3});
// boost::thread* t = new boost::thread(std::bind(CScheduler::serviceQueue, s)); * std::thread* t = new std::thread([&] { s->serviceQueue(); });
// *
// ... then at program shutdown, make sure to call stop() to clean up the thread(s) running serviceQueue: * ... then at program shutdown, make sure to call stop() to clean up the thread(s) running serviceQueue:
// s->stop(); * s->stop();
// t->join(); * t->join();
// delete t; * delete t;
// delete s; // Must be done after thread is interrupted/joined. * delete s; // Must be done after thread is interrupted/joined.
// */
class CScheduler class CScheduler
{ {
public: public:
@@ -43,7 +37,7 @@ public:
typedef std::function<void()> Function; typedef std::function<void()> Function;
// Call func at/after time t /** Call func at/after time t */
void schedule(Function f, std::chrono::system_clock::time_point t); void schedule(Function f, std::chrono::system_clock::time_point t);
/** Call f once after the delta has passed */ /** Call f once after the delta has passed */
@@ -67,23 +61,33 @@ public:
*/ */
void MockForward(std::chrono::seconds delta_seconds); void MockForward(std::chrono::seconds delta_seconds);
// To keep things as simple as possible, there is no unschedule. /**
* Services the queue 'forever'. Should be run in a thread,
// Services the queue 'forever'. Should be run in a thread, * and interrupted using boost::interrupt_thread
// and interrupted using boost::interrupt_thread */
void serviceQueue(); void serviceQueue();
// Tell any threads running serviceQueue to stop as soon as they're /** Tell any threads running serviceQueue to stop as soon as the current task is done */
// done servicing whatever task they're currently servicing (drain=false) void stop()
// or when there is no work left to be done (drain=true) {
void stop(bool drain=false); WITH_LOCK(newTaskMutex, stopRequested = true);
newTaskScheduled.notify_all();
}
/** Tell any threads running serviceQueue to stop when there is no work left to be done */
void StopWhenDrained()
{
WITH_LOCK(newTaskMutex, stopWhenEmpty = true);
newTaskScheduled.notify_all();
}
// Returns number of tasks waiting to be serviced, /**
// and first and last task times * Returns number of tasks waiting to be serviced,
size_t getQueueInfo(std::chrono::system_clock::time_point &first, * and first and last task times
std::chrono::system_clock::time_point &last) const; */
size_t getQueueInfo(std::chrono::system_clock::time_point& first,
std::chrono::system_clock::time_point& last) const;
// Returns true if there are threads actively running in serviceQueue() /** Returns true if there are threads actively running in serviceQueue() */
bool AreThreadsServicingQueue() const; bool AreThreadsServicingQueue() const;
private: private:
@@ -106,19 +110,20 @@ private:
* B() will be able to observe all of the effects of callback A() which executed * B() will be able to observe all of the effects of callback A() which executed
* before it. * before it.
*/ */
class SingleThreadedSchedulerClient { class SingleThreadedSchedulerClient
{
private: private:
CScheduler *m_pscheduler; CScheduler* m_pscheduler;
RecursiveMutex m_cs_callbacks_pending; RecursiveMutex m_cs_callbacks_pending;
std::list<std::function<void ()>> m_callbacks_pending GUARDED_BY(m_cs_callbacks_pending); std::list<std::function<void()>> m_callbacks_pending GUARDED_BY(m_cs_callbacks_pending);
bool m_are_callbacks_running GUARDED_BY(m_cs_callbacks_pending) = false; bool m_are_callbacks_running GUARDED_BY(m_cs_callbacks_pending) = false;
void MaybeScheduleProcessQueue(); void MaybeScheduleProcessQueue();
void ProcessQueue(); void ProcessQueue();
public: public:
explicit SingleThreadedSchedulerClient(CScheduler *pschedulerIn) : m_pscheduler(pschedulerIn) {} explicit SingleThreadedSchedulerClient(CScheduler* pschedulerIn) : m_pscheduler(pschedulerIn) {}
/** /**
* Add a callback to be executed. Callbacks are executed serially * Add a callback to be executed. Callbacks are executed serially
@@ -126,10 +131,12 @@ public:
* Practically, this means that callbacks can behave as if they are executed * Practically, this means that callbacks can behave as if they are executed
* in order by a single thread. * in order by a single thread.
*/ */
void AddToProcessQueue(std::function<void ()> func); void AddToProcessQueue(std::function<void()> func);
// Processes all remaining queue members on the calling thread, blocking until queue is empty /**
// Must be called after the CScheduler has no remaining processing threads! * Processes all remaining queue members on the calling thread, blocking until queue is empty
* Must be called after the CScheduler has no remaining processing threads!
*/
void EmptyQueue(); void EmptyQueue();
size_t CallbacksPending(); size_t CallbacksPending();
+3 -3
View File
@@ -89,7 +89,7 @@ BOOST_AUTO_TEST_CASE(manythreads)
} }
// Drain the task queue then exit threads // Drain the task queue then exit threads
microTasks.stop(true); microTasks.StopWhenDrained();
microThreads.join_all(); // ... wait until all the threads are done microThreads.join_all(); // ... wait until all the threads are done
int counterSum = 0; int counterSum = 0;
@@ -155,7 +155,7 @@ BOOST_AUTO_TEST_CASE(singlethreadedscheduler_ordered)
} }
// finish up // finish up
scheduler.stop(true); scheduler.StopWhenDrained();
threads.join_all(); threads.join_all();
BOOST_CHECK_EQUAL(counter1, 100); BOOST_CHECK_EQUAL(counter1, 100);
@@ -186,7 +186,7 @@ BOOST_AUTO_TEST_CASE(mockforward)
scheduler.MockForward(std::chrono::minutes{5}); scheduler.MockForward(std::chrono::minutes{5});
// ensure scheduler has chance to process all tasks queued for before 1 ms from now. // ensure scheduler has chance to process all tasks queued for before 1 ms from now.
scheduler.scheduleFromNow([&scheduler] { scheduler.stop(false); }, std::chrono::milliseconds{1}); scheduler.scheduleFromNow([&scheduler] { scheduler.stop(); }, std::chrono::milliseconds{1});
scheduler_thread.join(); scheduler_thread.join();
// check that the queue only has one job remaining // check that the queue only has one job remaining
+1
View File
@@ -19,6 +19,7 @@
#include <rpc/blockchain.h> #include <rpc/blockchain.h>
#include <rpc/register.h> #include <rpc/register.h>
#include <rpc/server.h> #include <rpc/server.h>
#include <scheduler.h>
#include <script/sigcache.h> #include <script/sigcache.h>
#include <streams.h> #include <streams.h>
#include <txdb.h> #include <txdb.h>
-1
View File
@@ -11,7 +11,6 @@
#include <node/context.h> #include <node/context.h>
#include <pubkey.h> #include <pubkey.h>
#include <random.h> #include <random.h>
#include <scheduler.h>
#include <txmempool.h> #include <txmempool.h>
#include <util/string.h> #include <util/string.h>