#include "config.h"
#include "Microtasks.h"
#include <wtf/MainThread.h>
#include <wtf/TemporaryChange.h>
namespace WebCore {
void Microtask::removeSelfFromQueue(MicrotaskQueue& queue)
{
queue.remove(*this);
}
MicrotaskQueue::MicrotaskQueue()
: m_timer(*this, &MicrotaskQueue::timerFired)
{
}
MicrotaskQueue::~MicrotaskQueue()
{
}
MicrotaskQueue& MicrotaskQueue::mainThreadQueue()
{
ASSERT(isMainThread());
static NeverDestroyed<MicrotaskQueue> queue;
return queue;
}
void MicrotaskQueue::append(std::unique_ptr<Microtask>&& task)
{
if (m_performingMicrotaskCheckpoint)
m_tasksAppendedDuringMicrotaskCheckpoint.append(WTFMove(task));
else
m_microtaskQueue.append(WTFMove(task));
m_timer.startOneShot(0);
}
void MicrotaskQueue::remove(const Microtask& task)
{
for (size_t i = 0; i < m_microtaskQueue.size(); ++i) {
if (m_microtaskQueue[i].get() == &task) {
m_microtaskQueue.remove(i);
return;
}
}
for (size_t i = 0; i < m_tasksAppendedDuringMicrotaskCheckpoint.size(); ++i) {
if (m_tasksAppendedDuringMicrotaskCheckpoint[i].get() == &task) {
m_tasksAppendedDuringMicrotaskCheckpoint.remove(i);
return;
}
}
}
void MicrotaskQueue::timerFired()
{
performMicrotaskCheckpoint();
}
void MicrotaskQueue::performMicrotaskCheckpoint()
{
if (m_performingMicrotaskCheckpoint)
return;
TemporaryChange<bool> change(m_performingMicrotaskCheckpoint, true);
Vector<std::unique_ptr<Microtask>> queue = WTFMove(m_microtaskQueue);
for (auto& task : queue) {
auto result = task->run();
switch (result) {
case Microtask::Result::Done:
break;
case Microtask::Result::KeepInQueue:
m_microtaskQueue.append(WTFMove(task));
break;
}
}
for (auto& task : m_tasksAppendedDuringMicrotaskCheckpoint)
m_microtaskQueue.append(WTFMove(task));
m_tasksAppendedDuringMicrotaskCheckpoint.clear();
}
}