third_party/WebKit/Source/core/workers/ParentFrameTaskRunners.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/workers/ParentFrameTaskRunners.h"

#include "core/dom/Document.h"
#include "core/frame/LocalFrame.h"
#include "platform/wtf/Assertions.h"
#include "platform/wtf/ThreadingPrimitives.h"
#include "public/platform/Platform.h"
#include "public/platform/TaskType.h"

namespace blink {

ParentFrameTaskRunners* ParentFrameTaskRunners::Create(LocalFrame& frame) {
  DCHECK(frame.GetDocument());
  DCHECK(frame.GetDocument()->IsContextThread());
  DCHECK(IsMainThread());
  return new ParentFrameTaskRunners(&frame);
}

ParentFrameTaskRunners* ParentFrameTaskRunners::Create() {
  return new ParentFrameTaskRunners(nullptr);
}

ParentFrameTaskRunners::ParentFrameTaskRunners(LocalFrame* frame)
    : ContextLifecycleObserver(frame ? frame->GetDocument() : nullptr) {
  // For now we only support very limited task types.
  for (auto type : {TaskType::kUnspecedTimer, TaskType::kUnspecedLoading,
                    TaskType::kNetworking, TaskType::kPostedMessage,
                    TaskType::kCanvasBlobSerialization, TaskType::kUnthrottled,
                    TaskType::kInternalTest}) {
    auto task_runner =
        frame ? frame->GetTaskRunner(type)
              : Platform::Current()->MainThread()->GetWebTaskRunner();
    task_runners_.insert(type, std::move(task_runner));
  }
}

scoped_refptr<base::SingleThreadTaskRunner> ParentFrameTaskRunners::Get(
    TaskType type) {
  MutexLocker lock(task_runners_mutex_);
  return task_runners_.at(type);
}

void ParentFrameTaskRunners::Trace(blink::Visitor* visitor) {
  ContextLifecycleObserver::Trace(visitor);
}

void ParentFrameTaskRunners::ContextDestroyed(ExecutionContext*) {
  MutexLocker lock(task_runners_mutex_);
  for (auto& entry : task_runners_)
    entry.value = Platform::Current()->CurrentThread()->GetWebTaskRunner();
}

}  // namespace blink