#ifndef WorkQueue_h
#define WorkQueue_h
#include <wtf/Forward.h>
#include <wtf/FunctionDispatcher.h>
#include <wtf/Seconds.h>
#include <wtf/Threading.h>
#if USE(COCOA_EVENT_LOOP)
#include <dispatch/dispatch.h>
#endif
#if USE(WINDOWS_EVENT_LOOP)
#include <wtf/HashMap.h>
#include <wtf/ThreadingPrimitives.h>
#include <wtf/Vector.h>
#endif
#if USE(GLIB_EVENT_LOOP) || USE(GENERIC_EVENT_LOOP)
#include <wtf/Condition.h>
#include <wtf/RunLoop.h>
#endif
namespace WTF {
#if USE(WINDOWS_EVENT_LOOP)
class WorkItemContext;
#endif
class WorkQueue final : public FunctionDispatcher {
public:
enum class Type {
Serial,
Concurrent
};
enum class QOS {
UserInteractive,
UserInitiated,
Default,
Utility,
Background
};
WTF_EXPORT_PRIVATE static Ref<WorkQueue> create(const char* name, Type = Type::Serial, QOS = QOS::Default);
virtual ~WorkQueue();
WTF_EXPORT_PRIVATE void dispatch(Function<void()>&&) override;
WTF_EXPORT_PRIVATE void dispatchAfter(Seconds, Function<void()>&&);
WTF_EXPORT_PRIVATE static void concurrentApply(size_t iterations, WTF::Function<void(size_t index)>&&);
#if USE(COCOA_EVENT_LOOP)
dispatch_queue_t dispatchQueue() const { return m_dispatchQueue; }
#elif USE(GLIB_EVENT_LOOP) || USE(GENERIC_EVENT_LOOP)
RunLoop& runLoop() const { return *m_runLoop; }
#elif USE(WINDOWS_EVENT_LOOP)
WTF_EXPORT_PRIVATE void registerHandle(HANDLE, Function<void()>&&);
WTF_EXPORT_PRIVATE void unregisterAndCloseHandle(HANDLE);
#endif
private:
explicit WorkQueue(const char* name, Type, QOS);
void platformInitialize(const char* name, Type, QOS);
void platformInvalidate();
#if USE(WINDOWS_EVENT_LOOP)
static void CALLBACK handleCallback(void* context, BOOLEAN timerOrWaitFired);
static void CALLBACK timerCallback(void* context, BOOLEAN timerOrWaitFired);
static DWORD WINAPI workThreadCallback(void* context);
bool tryRegisterAsWorkThread();
void unregisterAsWorkThread();
void performWorkOnRegisteredWorkThread();
static void unregisterWaitAndDestroyItemSoon(Ref<WorkItemContext>&&);
static DWORD WINAPI unregisterWaitAndDestroyItemCallback(void* context);
#endif
#if USE(COCOA_EVENT_LOOP)
static void executeFunction(void*);
dispatch_queue_t m_dispatchQueue;
#elif USE(WINDOWS_EVENT_LOOP)
volatile LONG m_isWorkThreadRegistered;
Lock m_functionQueueLock;
Lock m_itemsMapLock;
Vector<Function<void()>> m_functionQueue;
HashMap<HANDLE, Ref<WorkItemContext>> m_itemsMap;
HANDLE m_timerQueue;
#elif USE(GLIB_EVENT_LOOP) || USE(GENERIC_EVENT_LOOP)
RunLoop* m_runLoop;
#endif
};
}
using WTF::WorkQueue;
#endif