content/child/scheduler/worker_scheduler_impl.h

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CONTENT_CHILD_SCHEDULER_WORKER_SCHEDULER_IMPL_H_
   6 #define CONTENT_CHILD_SCHEDULER_WORKER_SCHEDULER_IMPL_H_
   7
   8 #include "content/child/scheduler/worker_scheduler.h"
   9 #include "content/child/scheduler/scheduler_helper.h"
  10
  11 namespace base {
  12 namespace trace_event {
  13 class ConvertableToTraceFormat;
  14 }
  15 }
  16
  17 namespace content {
  18
  19 class NestableSingleThreadTaskRunner;
  20
  21 class CONTENT_EXPORT WorkerSchedulerImpl
  22     : public WorkerScheduler,
  23       public SchedulerHelper::SchedulerHelperDelegate {
  24  public:
  25   explicit WorkerSchedulerImpl(
  26       scoped_refptr<NestableSingleThreadTaskRunner> main_task_runner);
  27   ~WorkerSchedulerImpl() override;
  28
  29   // WorkerScheduler implementation:
  30   scoped_refptr<base::SingleThreadTaskRunner> DefaultTaskRunner() override;
  31   scoped_refptr<SingleThreadIdleTaskRunner> IdleTaskRunner() override;
  32   bool CanExceedIdleDeadlineIfRequired() const override;
  33   bool ShouldYieldForHighPriorityWork() override;
  34   void AddTaskObserver(base::MessageLoop::TaskObserver* task_observer) override;
  35   void RemoveTaskObserver(
  36       base::MessageLoop::TaskObserver* task_observer) override;
  37   void Init() override;
  38   void Shutdown() override;
  39
  40   void SetTimeSourceForTesting(scoped_refptr<cc::TestNowSource> time_source);
  41   void SetWorkBatchSizeForTesting(size_t work_batch_size);
  42   base::TimeTicks CurrentIdleTaskDeadlineForTesting() const;
  43
  44  protected:
  45   // SchedulerHelperDelegate implementation:
  46   bool CanEnterLongIdlePeriod(
  47       base::TimeTicks now,
  48       base::TimeDelta* next_long_idle_period_delay_out) override;
  49   void IsNotQuiescent() override {}
  50
  51  private:
  52   void MaybeStartLongIdlePeriod();
  53
  54   SchedulerHelper helper_;
  55   bool initialized_;
  56
  57   DISALLOW_COPY_AND_ASSIGN(WorkerSchedulerImpl);
  58 };
  59
  60 }  // namespace content
  61
  62 #endif  // CONTENT_CHILD_SCHEDULER_WORKER_SCHEDULER_IMPL_H_