CrossThreadTaskHandler.cpp [plain text]
#include "config.h"
#include <wtf/CrossThreadTaskHandler.h>
#include <wtf/AutodrainedPool.h>
namespace WTF {
CrossThreadTaskHandler::CrossThreadTaskHandler(const char* threadName, AutodrainedPoolForRunLoop useAutodrainedPool)
: m_useAutodrainedPool(useAutodrainedPool)
{
ASSERT(isMainThread());
Locker<Lock> locker(m_taskThreadCreationLock);
Thread::create(threadName, [this] {
taskRunLoop();
if (m_completionCallback)
m_completionCallback();
})->detach();
}
CrossThreadTaskHandler::~CrossThreadTaskHandler()
{
ASSERT(isMainThread());
}
void CrossThreadTaskHandler::postTask(CrossThreadTask&& task)
{
m_taskQueue.append(WTFMove(task));
}
void CrossThreadTaskHandler::postTaskReply(CrossThreadTask&& task)
{
m_taskReplyQueue.append(WTFMove(task));
Locker<Lock> locker(m_mainThreadReplyLock);
if (m_mainThreadReplyScheduled)
return;
m_mainThreadReplyScheduled = true;
callOnMainThread([this] {
handleTaskRepliesOnMainThread();
});
}
void CrossThreadTaskHandler::taskRunLoop()
{
ASSERT(!isMainThread());
{
Locker<Lock> locker(m_taskThreadCreationLock);
}
while (!m_taskQueue.isKilled()) {
std::unique_ptr<AutodrainedPool> autodrainedPool = (m_useAutodrainedPool == AutodrainedPoolForRunLoop::Use) ? makeUnique<AutodrainedPool>() : nullptr;
m_taskQueue.waitForMessage().performTask();
}
}
void CrossThreadTaskHandler::handleTaskRepliesOnMainThread()
{
{
Locker<Lock> locker(m_mainThreadReplyLock);
m_mainThreadReplyScheduled = false;
}
while (auto task = m_taskReplyQueue.tryGetMessage())
task->performTask();
}
void CrossThreadTaskHandler::setCompletionCallback(Function<void ()>&& completionCallback)
{
m_completionCallback = WTFMove(completionCallback);
}
void CrossThreadTaskHandler::kill()
{
m_taskQueue.kill();
m_taskReplyQueue.kill();
}
}