DeferredWorkTimer.cpp [plain text]
#include "config.h"
#include "DeferredWorkTimer.h"
#include "JSPromise.h"
#include "StrongInlines.h"
#include "VM.h"
#include <wtf/RunLoop.h>
namespace JSC {
namespace DeferredWorkTimerInternal {
static const bool verbose = false;
}
DeferredWorkTimer::DeferredWorkTimer(VM& vm)
: Base(vm)
{
}
void DeferredWorkTimer::doWork(VM& vm)
{
ASSERT(vm.currentThreadIsHoldingAPILock());
m_taskLock.lock();
cancelTimer();
if (!m_runTasks) {
m_taskLock.unlock();
return;
}
while (!m_tasks.isEmpty()) {
auto [ticket, task] = m_tasks.takeFirst();
dataLogLnIf(DeferredWorkTimerInternal::verbose, "Doing work on: ", RawPointer(ticket));
if (m_pendingTickets.contains(ticket)) {
m_currentlyRunningTask = true;
m_taskLock.unlock();
vm.finalizeSynchronousJSExecution();
auto scope = DECLARE_CATCH_SCOPE(vm);
task();
if (Exception* exception = scope.exception()) {
auto* globalObject = ticket->globalObject();
scope.clearException();
globalObject->globalObjectMethodTable()->reportUncaughtExceptionAtEventLoop(globalObject, exception);
}
vm.drainMicrotasks();
ASSERT(!vm.exceptionForInspection());
m_taskLock.lock();
m_currentlyRunningTask = false;
}
}
if (m_pendingTickets.isEmpty() && m_shouldStopRunLoopWhenAllTicketsFinish)
RunLoop::current().stop();
m_taskLock.unlock();
}
void DeferredWorkTimer::runRunLoop()
{
ASSERT(!m_apiLock->vm()->currentThreadIsHoldingAPILock());
ASSERT(&RunLoop::current() == &m_apiLock->vm()->runLoop());
m_shouldStopRunLoopWhenAllTicketsFinish = true;
if (m_pendingTickets.size())
RunLoop::run();
}
void DeferredWorkTimer::addPendingWork(VM& vm, Ticket ticket, Vector<Strong<JSCell>>&& dependencies)
{
ASSERT(vm.currentThreadIsHoldingAPILock() || (Thread::mayBeGCThread() && vm.heap.worldIsStopped()));
for (unsigned i = 0; i < dependencies.size(); ++i)
ASSERT(dependencies[i].get() != ticket);
auto result = m_pendingTickets.add(ticket, Vector<Strong<JSCell>>());
if (result.isNewEntry) {
dataLogLnIf(DeferredWorkTimerInternal::verbose, "Adding new pending ticket: ", RawPointer(ticket));
dependencies.append(Strong<JSCell>(vm, ticket));
result.iterator->value = WTFMove(dependencies);
} else {
dataLogLnIf(DeferredWorkTimerInternal::verbose, "Adding new dependencies for ticket: ", RawPointer(ticket));
result.iterator->value.appendVector(dependencies);
}
}
bool DeferredWorkTimer::hasPendingWork(Ticket ticket)
{
ASSERT(ticket->vm().currentThreadIsHoldingAPILock() || (Thread::mayBeGCThread() && ticket->vm().heap.worldIsStopped()));
return m_pendingTickets.contains(ticket);
}
bool DeferredWorkTimer::hasDependancyInPendingWork(Ticket ticket, JSCell* dependency)
{
ASSERT(ticket->vm().currentThreadIsHoldingAPILock() || (Thread::mayBeGCThread() && ticket->vm().heap.worldIsStopped()));
ASSERT(m_pendingTickets.contains(ticket));
auto result = m_pendingTickets.get(ticket);
return result.contains(dependency);
}
bool DeferredWorkTimer::cancelPendingWork(Ticket ticket)
{
ASSERT(ticket->vm().currentThreadIsHoldingAPILock() || (Thread::mayBeGCThread() && ticket->vm().heap.worldIsStopped()));
bool result = m_pendingTickets.remove(ticket);
if (result)
dataLogLnIf(DeferredWorkTimerInternal::verbose, "Canceling ticket: ", RawPointer(ticket));
return result;
}
void DeferredWorkTimer::scheduleWorkSoon(Ticket ticket, Task&& task)
{
LockHolder locker(m_taskLock);
m_tasks.append(std::make_tuple(ticket, WTFMove(task)));
if (!isScheduled() && !m_currentlyRunningTask)
setTimeUntilFire(0_s);
}
}