third_party/WebKit/Source/platform/scheduler/public/non_main_thread_scheduler.h

// Copyright 2015 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.

#ifndef THIRD_PARTY_WEBKIT_SOURCE_PLATFORM_SCHEDULER_PUBLIC_NON_MAIN_THREAD_SCHEDULER_H_
#define THIRD_PARTY_WEBKIT_SOURCE_PLATFORM_SCHEDULER_PUBLIC_NON_MAIN_THREAD_SCHEDULER_H_

#include <memory>

#include "base/macros.h"
#include "base/message_loop/message_loop.h"
#include "platform/PlatformExport.h"
#include "platform/scheduler/base/task_queue.h"
#include "platform/scheduler/child/worker_scheduler_helper.h"
#include "platform/scheduler/child/worker_task_queue.h"
#include "public/platform/WebThreadType.h"
#include "public/platform/scheduler/single_thread_idle_task_runner.h"
#include "public/platform/scheduler/web_thread_scheduler.h"

namespace blink {
namespace scheduler {
class WorkerSchedulerProxy;

// TODO(yutak): Rename this class to NonMainThreadScheduler.
class PLATFORM_EXPORT WorkerScheduler : public ChildScheduler {
 public:
  ~WorkerScheduler() override;

  static std::unique_ptr<WorkerScheduler> Create(WebThreadType thread_type,
                                                 WorkerSchedulerProxy* proxy);

  // Blink should use WorkerScheduler::DefaultTaskQueue instead of
  // ChildScheduler::DefaultTaskRunner.
  virtual scoped_refptr<WorkerTaskQueue> DefaultTaskQueue() = 0;

  // Must be called before the scheduler can be used. Does any post construction
  // initialization needed such as initializing idle period detection.
  virtual void Init() = 0;

  virtual void OnTaskCompleted(WorkerTaskQueue* worker_task_queue,
                               const TaskQueue::Task& task,
                               base::TimeTicks start,
                               base::TimeTicks end,
                               base::Optional<base::TimeDelta> thread_time) = 0;

  scoped_refptr<WorkerTaskQueue> CreateTaskRunner();

 protected:
  explicit WorkerScheduler(std::unique_ptr<WorkerSchedulerHelper> helper);

  std::unique_ptr<WorkerSchedulerHelper> helper_;

  DISALLOW_COPY_AND_ASSIGN(WorkerScheduler);
};

}  // namespace scheduler
}  // namespace blink

#endif  // THIRD_PARTY_WEBKIT_SOURCE_PLATFORM_SCHEDULER_PUBLIC_NON_MAIN_THREAD_SCHEDULER_H_