#pragma once
#include "Timer.h"
#include <wtf/Forward.h>
#include <wtf/Vector.h>
namespace WebCore {
class MicrotaskQueue;
class ScriptExecutionContext;
class Microtask {
public:
virtual ~Microtask()
{
}
enum class Result {
Done,
KeepInQueue
};
virtual Result run() = 0;
protected:
void removeSelfFromQueue(MicrotaskQueue&);
};
class VoidMicrotask final : public Microtask {
WTF_MAKE_FAST_ALLOCATED;
public:
explicit VoidMicrotask(Function<void()>&& function)
: m_function(WTFMove(function))
{
}
private:
Result run() final
{
m_function();
return Result::Done;
}
Function<void()> m_function;
};
class MicrotaskQueue {
friend NeverDestroyed<MicrotaskQueue>;
friend class Microtask;
public:
WEBCORE_EXPORT static MicrotaskQueue& mainThreadQueue();
WEBCORE_EXPORT static MicrotaskQueue& contextQueue(ScriptExecutionContext&);
WEBCORE_EXPORT MicrotaskQueue();
WEBCORE_EXPORT ~MicrotaskQueue();
WEBCORE_EXPORT void append(std::unique_ptr<Microtask>&&);
WEBCORE_EXPORT void performMicrotaskCheckpoint();
private:
WEBCORE_EXPORT void remove(const Microtask&);
void timerFired();
bool m_performingMicrotaskCheckpoint = false;
Vector<std::unique_ptr<Microtask>> m_microtaskQueue;
Timer m_timer;
};
}