ladybird/Userland/Libraries/LibThreading/BackgroundAction.cpp

/*
 * Copyright (c) 2019-2020, Sergey Bugaev <bugaevc@serenityos.org>
 * Copyright (c) 2021, Andreas Kling <andreas@ladybird.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <AK/Queue.h>
#include <LibThreading/BackgroundAction.h>
#include <LibThreading/Mutex.h>
#include <LibThreading/Thread.h>
#include <unistd.h>

static pthread_mutex_t s_mutex = PTHREAD_MUTEX_INITIALIZER;
static pthread_cond_t s_condition = PTHREAD_COND_INITIALIZER;
static Queue<Function<void()>>* s_all_actions;
static Threading::Thread* s_background_thread;
static Atomic<bool> s_background_thread_should_run = true;

static intptr_t background_thread_func()
{
    Vector<Function<void()>> actions;
    while (s_background_thread_should_run.load(AK::MemoryOrder::memory_order_acquire)) {
        pthread_mutex_lock(&s_mutex);

        while (s_all_actions->is_empty() && s_background_thread_should_run.load(AK::MemoryOrder::memory_order_acquire))
            pthread_cond_wait(&s_condition, &s_mutex);

        while (!s_all_actions->is_empty())
            actions.append(s_all_actions->dequeue());

        pthread_mutex_unlock(&s_mutex);

        for (auto& action : actions) {
            if (s_background_thread_should_run.load(AK::MemoryOrder::memory_order_acquire))
                action();
        }
        actions.clear();
    }
    return 0;
}

static void init()
{
    s_all_actions = new Queue<Function<void()>>;
    s_background_thread = &Threading::Thread::construct(background_thread_func, "Background Thread"sv).leak_ref();
    s_background_thread->start();
}

void Threading::quit_background_thread()
{
    if (!s_background_thread)
        return;

    s_background_thread_should_run.store(false, AK::MemoryOrder::memory_order_release);

    pthread_mutex_lock(&s_mutex);
    pthread_cond_broadcast(&s_condition);
    pthread_mutex_unlock(&s_mutex);

    MUST(s_background_thread->join());

    delete s_all_actions;
    s_background_thread->unref();
    s_all_actions = nullptr;
    s_background_thread = nullptr;

    s_background_thread_should_run.store(true, AK::MemoryOrder::memory_order_release);
}

Threading::Thread& Threading::BackgroundActionBase::background_thread()
{
    if (s_background_thread == nullptr)
        init();
    return *s_background_thread;
}

void Threading::BackgroundActionBase::enqueue_work(Function<void()> work)
{
    if (s_all_actions == nullptr)
        init();

    pthread_mutex_lock(&s_mutex);
    s_all_actions->enqueue(move(work));
    pthread_cond_broadcast(&s_condition);
    pthread_mutex_unlock(&s_mutex);
}