Introduce TaskRunnerHelper::get()

third_party/WebKit/Source/core/dom/TaskRunnerHelper.cpp

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "core/dom/TaskRunnerHelper.h"
 
 #include "core/dom/Document.h"
 #include "core/frame/LocalFrame.h"
 #include "public/platform/Platform.h"
 #include "public/platform/WebFrameScheduler.h"
 #include "public/platform/WebTaskRunner.h"
 #include "public/platform/WebThread.h"
 
 namespace blink {
 
+WebTaskRunner* TaskRunnerHelper::get(TaskRunnerType type, LocalFrame* frame)
+{
+    // TODO(haraken): Optimize the mapping from TaskRunnerTypes to task runners.
+    switch (type) {
+    case DOMManipulationTask:
+    case UserInteractionTask:
+    case HistoryTraversalTask:
+    case EmbedTask:
+    case MediaElementEventTask:
+    case CanvasBlobSerializationTask:
+    case RemoteEventTask:
+    case WebSocketTask:
+    case MicrotaskTask:
+    case PostedMessageTask:
+    case UnshippedPortMessageTask:
+    case TimerTask:
+    case InternalTask:
+        return frame ? frame->frameScheduler()->timerTaskRunner() : Platform::current()->currentThread()->getWebTaskRunner();
+    case NetworkingTask:
+        return frame ? frame->frameScheduler()->loadingTaskRunner() : Platform::current()->currentThread()->getWebTaskRunner();
+    case IdleTask:
+        // TODO(haraken): Implement this. We need to expose frameScheduler()->idleTaskRunner().
+        NOTREACHED();
+    case UnthrottledTask:
+        return frame ? frame->frameScheduler()->unthrottledTaskRunner() : Platform::current()->currentThread()->getWebTaskRunner();
+    default:
+        NOTREACHED();
+    }
+}
+
+WebTaskRunner* TaskRunnerHelper::get(TaskRunnerType type, Document* document)
+{
+    return get(type, document ? document->frame() : nullptr);
+}
+
+WebTaskRunner* TaskRunnerHelper::get(TaskRunnerType type, ExecutionContext* executionContext)
+{
+    return get(type, executionContext && executionContext->isDocument() ? static_cast<Document*>(executionContext) : nullptr);
+}
+
+WebTaskRunner* TaskRunnerHelper::get(TaskRunnerType type, ScriptState* scriptState)
+{
+    return get(type, scriptState ? scriptState->getExecutionContext() : nullptr);
+}
+
 WebTaskRunner* TaskRunnerHelper::getUnthrottledTaskRunner(LocalFrame* frame)
 {
     return frame ? frame->frameScheduler()->unthrottledTaskRunner() : Platform::current()->currentThread()->getWebTaskRunner();
 }
 
 WebTaskRunner* TaskRunnerHelper::getTimerTaskRunner(LocalFrame* frame)
 {
     return frame ? frame->frameScheduler()->timerTaskRunner() : Platform::current()->currentThread()->getWebTaskRunner();
 }
 
@@ skipping 21 matching lines @@
 {
     return getUnthrottledTaskRunner(executionContext && executionContext->isDocument() ? static_cast<Document*>(executionContext) : nullptr);
 }
 
 WebTaskRunner* TaskRunnerHelper::getUnthrottledTaskRunner(ScriptState* scriptState)
 {
     return getUnthrottledTaskRunner(scriptState ? scriptState->getExecutionContext() : nullptr);
 }
 
 } // namespace blink