#include "config.h"
#include "EventLoop.h"
#include "Microtasks.h"
namespace WebCore {
void EventLoop::queueTask(std::unique_ptr<EventLoopTask>&& task)
{
ASSERT(task->taskSource() != TaskSource::Microtask);
ASSERT(task->group());
ASSERT(isContextThread());
scheduleToRunIfNeeded();
m_tasks.append(WTFMove(task));
}
void EventLoop::queueMicrotask(std::unique_ptr<EventLoopTask>&& microtask)
{
ASSERT(microtask->taskSource() == TaskSource::Microtask);
microtaskQueue().append(WTFMove(microtask));
scheduleToRunIfNeeded(); }
void EventLoop::performMicrotaskCheckpoint()
{
microtaskQueue().performMicrotaskCheckpoint();
}
void EventLoop::resumeGroup(EventLoopTaskGroup& group)
{
ASSERT(isContextThread());
if (!m_groupsWithSuspenedTasks.contains(group))
return;
scheduleToRunIfNeeded();
}
void EventLoop::stopGroup(EventLoopTaskGroup& group)
{
ASSERT(isContextThread());
m_tasks.removeAllMatching([&group] (auto& task) {
return group.matchesTask(*task);
});
}
void EventLoop::scheduleToRunIfNeeded()
{
if (m_isScheduledToRun)
return;
m_isScheduledToRun = true;
scheduleToRun();
}
void EventLoop::run()
{
m_isScheduledToRun = false;
bool didPerformMicrotaskCheckpoint = false;
if (!m_tasks.isEmpty()) {
auto tasks = std::exchange(m_tasks, { });
m_groupsWithSuspenedTasks.clear();
Vector<std::unique_ptr<EventLoopTask>> remainingTasks;
for (auto& task : tasks) {
auto* group = task->group();
if (!group || group->isStoppedPermanently())
continue;
if (group->isSuspended()) {
m_groupsWithSuspenedTasks.add(group);
remainingTasks.append(WTFMove(task));
continue;
}
task->execute();
didPerformMicrotaskCheckpoint = true;
microtaskQueue().performMicrotaskCheckpoint();
}
for (auto& task : m_tasks)
remainingTasks.append(WTFMove(task));
m_tasks = WTFMove(remainingTasks);
}
if (!didPerformMicrotaskCheckpoint)
microtaskQueue().performMicrotaskCheckpoint();
}
void EventLoop::clearAllTasks()
{
m_tasks.clear();
m_groupsWithSuspenedTasks.clear();
}
void EventLoopTaskGroup::queueTask(std::unique_ptr<EventLoopTask>&& task)
{
if (m_state == State::Stopped || !m_eventLoop)
return;
ASSERT(task->group() == this);
m_eventLoop->queueTask(WTFMove(task));
}
class EventLoopFunctionDispatchTask : public EventLoopTask {
public:
EventLoopFunctionDispatchTask(TaskSource source, EventLoopTaskGroup& group, EventLoop::TaskFunction&& function)
: EventLoopTask(source, group)
, m_function(WTFMove(function))
{
}
void execute() final { m_function(); }
private:
EventLoop::TaskFunction m_function;
};
void EventLoopTaskGroup::queueTask(TaskSource source, EventLoop::TaskFunction&& function)
{
return queueTask(makeUnique<EventLoopFunctionDispatchTask>(source, *this, WTFMove(function)));
}
void EventLoopTaskGroup::queueMicrotask(EventLoop::TaskFunction&& function)
{
if (m_state == State::Stopped || !m_eventLoop)
return;
m_eventLoop->queueMicrotask(makeUnique<EventLoopFunctionDispatchTask>(TaskSource::Microtask, *this, WTFMove(function)));
}
void EventLoopTaskGroup::performMicrotaskCheckpoint()
{
if (m_eventLoop)
m_eventLoop->performMicrotaskCheckpoint();
}
}