third_party/blink/renderer/platform/scheduler/worker/worker_scheduler_proxy.cc

// Copyright 2018 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 "third_party/blink/renderer/platform/scheduler/worker/worker_scheduler_proxy.h"

#include "third_party/blink/renderer/platform/scheduler/main_thread/frame_scheduler_impl.h"
#include "third_party/blink/renderer/platform/scheduler/public/worker_scheduler.h"
#include "third_party/blink/renderer/platform/scheduler/worker/worker_thread_scheduler.h"

namespace blink {
namespace scheduler {

WorkerSchedulerProxy::WorkerSchedulerProxy(FrameOrWorkerScheduler* scheduler) {
  DCHECK(scheduler);
  throttling_observer_handle_ = scheduler->AddThrottlingObserver(
      FrameOrWorkerScheduler::ObserverType::kWorkerScheduler, this);
  if (FrameScheduler* frame_scheduler = scheduler->ToFrameScheduler()) {
    parent_frame_type_ = GetFrameOriginType(frame_scheduler);
  }
}

WorkerSchedulerProxy::~WorkerSchedulerProxy() {
  DETACH_FROM_THREAD(parent_thread_checker_);
}

void WorkerSchedulerProxy::OnWorkerSchedulerCreated(
    base::WeakPtr<WorkerScheduler> worker_scheduler) {
  DCHECK(!IsMainThread())
      << "OnWorkerSchedulerCreated should be called from the worker thread";
  DCHECK(!worker_scheduler_) << "OnWorkerSchedulerCreated is called twice";
  DCHECK(worker_scheduler) << "WorkerScheduler is expected to exist";
  worker_scheduler_ = std::move(worker_scheduler);
  worker_thread_task_runner_ =
      worker_scheduler_->GetWorkerThreadScheduler()->ControlTaskQueue();
  initialized_ = true;
}

void WorkerSchedulerProxy::OnThrottlingStateChanged(
    FrameScheduler::ThrottlingState throttling_state) {
  DCHECK_CALLED_ON_VALID_THREAD(parent_thread_checker_);
  if (throttling_state_ == throttling_state)
    return;
  throttling_state_ = throttling_state;

  if (!initialized_)
    return;

  worker_thread_task_runner_->PostTask(
      FROM_HERE, base::BindOnce(&WorkerScheduler::OnThrottlingStateChanged,
                                worker_scheduler_, throttling_state));
}

}  // namespace scheduler
}  // namespace blink