mirrors/logiops
1
0
Fork 0
mirror of https://github.com/PixlOne/logiops.git synced 2025-07-13 21:02:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Exit gracefully upon return

Browse Source 
Stop all worker threads when main thread exits.
This commit is contained in:
pixl 2023-05-12 17:25:28 -04:00 committed by pixl
parent a361f206ff
commit be5ee9f793
No known key found for this signature in database
GPG Key ID: 1866C148CD593B6E
1 changed files with 29 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
src/logid/util/task.cpp
View File

@ -18,6 +18,7 @@
#include <util/task.h> #include <util/task.h>
#include <queue> #include <queue>
#include <optional> #include <optional>
#include <cassert>
using namespace logid; using namespace logid;
using namespace std::chrono; using namespace std::chrono;
@ -35,18 +36,38 @@ static std::priority_queue<task, std::vector<task>, task_less> tasks {};
static std::mutex task_mutex {}; static std::mutex task_mutex {};
static std::condition_variable task_cv {}; static std::condition_variable task_cv {};
static std::atomic_bool workers_init = false; static std::atomic_bool workers_init = false;
static std::atomic_bool workers_run = false;
[[noreturn]] static void worker() { void stop_workers() {
std::unique_lock lock(task_mutex); std::unique_lock lock(task_mutex);
while (true) { if (workers_init) {
task_cv.wait(lock, []() { return !tasks.empty(); }); workers_run = false;
lock.unlock();
task_cv.notify_all();
/* Wait for all workers to end */
lock.lock();
}
}
void worker() {
std::unique_lock lock(task_mutex);
while (workers_run) {
task_cv.wait(lock, []() { return !tasks.empty() || !workers_run; });
if (!workers_run)
break;
/* top task is in the future, wait */ /* top task is in the future, wait */
if (tasks.top().time >= system_clock::now()) { if (tasks.top().time >= system_clock::now()) {
auto wait = tasks.top().time - system_clock::now(); auto wait = tasks.top().time - system_clock::now();
task_cv.wait_for(lock, wait, []() { task_cv.wait_for(lock, wait, []() {
return !tasks.empty() && (tasks.top().time < system_clock::now()); return (!tasks.empty() && (tasks.top().time < system_clock::now())) ||
!workers_run;
}); });
if (!workers_run)
break;
} }
if (!tasks.empty()) { if (!tasks.empty()) {
@ -67,11 +88,15 @@ static std::atomic_bool workers_init = false;
void logid::init_workers(int worker_count) { void logid::init_workers(int worker_count) {
std::lock_guard lock(task_mutex); std::lock_guard lock(task_mutex);
assert(!workers_init);
for (int i = 0; i < worker_count; ++i) for (int i = 0; i < worker_count; ++i)
std::thread(&worker).detach(); std::thread(&worker).detach();
workers_init = true; workers_init = true;
workers_run = true;
atexit(&stop_workers);
} }
void logid::run_task(std::function<void()> function) { void logid::run_task(std::function<void()> function) {