[scheduler] Add WakeupBudgetPool.

third_party/WebKit/Source/platform/scheduler/renderer/task_queue_throttler.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_RENDERER_TASK_QUEUE_THROTTLER_H_
 #define THIRD_PARTY_WEBKIT_SOURCE_PLATFORM_SCHEDULER_RENDERER_TASK_QUEUE_THROTTLER_H_
 
 #include <set>
 #include <unordered_map>
 
@@ skipping 12 matching lines @@
 }
 }
 
 namespace blink {
 namespace scheduler {
 
 class BudgetPool;
 class RendererSchedulerImpl;
 class ThrottledTimeDomain;
 class CPUTimeBudgetPool;
+class WakeupBudgetPool;
+
+enum class QueueBlockType { FULL, NEW_TASKS_ONLY };

[alex clarke (OOO till 29th), 2017/04/21 09:14:19]

These could do with some more documentation.   I wonder if ALL_TASKS might make
the call sites easier to understand.

[alex clarke (OOO till 29th), 2017/04/21 09:14:19]

Are you thinking of adding more blocking types?  Currently there is a 1:1
mapping to fence type, do we need a new enum? (I'm not against adding this btw)

[altimin, 2017/04/25 13:22:36]

ALL_TASKS is a good idea, done.

I hoped that it would be self-explanatory. 

 
 // Interface for BudgetPool to interact with TaskQueueThrottler.
 class BLINK_PLATFORM_EXPORT BudgetPoolController {
  public:
   virtual ~BudgetPoolController() {}
 
   // To be used by BudgetPool only, use BudgetPool::{Add,Remove}Queue
   // methods instead.
   virtual void AddQueueToBudgetPool(TaskQueue* queue,
                                     BudgetPool* budget_pool) = 0;
   virtual void RemoveQueueFromBudgetPool(TaskQueue* queue,
                                          BudgetPool* budget_pool) = 0;
 
   // Deletes the budget pool.
   virtual void UnregisterBudgetPool(BudgetPool* budget_pool) = 0;
 
   // Insert a fence to prevent tasks from running and schedule a wake-up at
   // an appropriate time.
-  virtual void BlockQueue(base::TimeTicks now, TaskQueue* queue) = 0;
+  virtual void BlockQueue(QueueBlockType block_type,
+                          base::TimeTicks now,
+                          TaskQueue* queue) = 0;
 
   // Schedule a call to unblock queue at an appropriate moment.
   virtual void UnblockQueue(base::TimeTicks now, TaskQueue* queue) = 0;
 
   // Returns true if the |queue| is throttled (i.e. added to TaskQueueThrottler
   // and throttling is not disabled).
   virtual bool IsThrottled(TaskQueue* queue) const = 0;
 };
 
 // The job of the TaskQueueThrottler is to control when tasks posted on
@@ skipping 30 matching lines @@
 
   // TaskQueue::Observer implementation:
   void OnQueueNextWakeUpChanged(TaskQueue* queue,
                                 base::TimeTicks wake_up) override;
 
   // BudgetPoolController implementation:
   void AddQueueToBudgetPool(TaskQueue* queue, BudgetPool* budget_pool) override;
   void RemoveQueueFromBudgetPool(TaskQueue* queue,
                                  BudgetPool* budget_pool) override;
   void UnregisterBudgetPool(BudgetPool* budget_pool) override;
-  void BlockQueue(base::TimeTicks now, TaskQueue* queue) override;
+  void BlockQueue(QueueBlockType block_type,
+                  base::TimeTicks now,
+                  TaskQueue* queue) override;
   void UnblockQueue(base::TimeTicks now, TaskQueue* queue) override;
   bool IsThrottled(TaskQueue* queue) const override;
 
   // Increments the throttled refcount and causes |task_queue| to be throttled
   // if its not already throttled.
   void IncreaseThrottleRefCount(TaskQueue* task_queue);
 
   // If the refcouint is non-zero it's decremented.  If the throttled refcount
   // becomes zero then |task_queue| is unthrottled.  If the refcount was already
   // zero this function does nothing.
@@ skipping 12 matching lines @@
 
   const ThrottledTimeDomain* time_domain() const { return time_domain_.get(); }
 
   static base::TimeTicks AlignedThrottledRunTime(
       base::TimeTicks unthrottled_runtime);
 
   const scoped_refptr<TaskQueue>& task_runner() const { return task_runner_; }
 
   // Returned object is owned by |TaskQueueThrottler|.
   CPUTimeBudgetPool* CreateCPUTimeBudgetPool(const char* name);
+  WakeupBudgetPool* CreateWakeupBudgetPool(const char* name);
 
   // Accounts for given task for cpu-based throttling needs.
   void OnTaskRunTimeReported(TaskQueue* task_queue,
                              base::TimeTicks start_time,
                              base::TimeTicks end_time);
 
   void AsValueInto(base::trace_event::TracedValue* state,
                    base::TimeTicks now) const;
  private:
   struct Metadata {
@@ skipping 10 matching lines @@
   // Note |unthrottled_runtime| might be in the past. When this happens we
   // compute the delay to the next runtime based on now rather than
   // unthrottled_runtime.
   void MaybeSchedulePumpThrottledTasks(
       const tracked_objects::Location& from_here,
       base::TimeTicks now,
       base::TimeTicks runtime);
 
   // Return next possible time when queue is allowed to run in accordance
   // with throttling policy.
-  base::TimeTicks GetNextAllowedRunTime(base::TimeTicks now, TaskQueue* queue);
+  base::TimeTicks GetNextAllowedRunTime(TaskQueue* queue,
+                                        base::TimeTicks now,
+                                        base::TimeTicks desired_run_time);
+
+  bool CanRunTasksAt(TaskQueue* queue, base::TimeTicks moment);
+
+  bool CanRunTasksUntil(TaskQueue* queue,
+                        base::TimeTicks now,
+                        base::TimeTicks moment);
 
   void MaybeDeleteQueueMetadata(TaskQueueMap::iterator it);
 
   // Schedule a call PumpThrottledTasks at an appropriate moment for this queue.
   void SchedulePumpQueue(const tracked_objects::Location& from_here,
                          base::TimeTicks now,
                          TaskQueue* queue);
 
+  void OnTaskQueueHasWork(TaskQueue* queue,
+                          base::TimeTicks now,
+                          base::TimeTicks next_task_run_time);
+
   TaskQueueMap queue_details_;
   base::Callback<void(TaskQueue*, base::TimeTicks)>
       forward_immediate_work_callback_;
   scoped_refptr<TaskQueue> task_runner_;
   RendererSchedulerImpl* renderer_scheduler_;  // NOT OWNED
   base::TickClock* tick_clock_;                // NOT OWNED
   const char* tracing_category_;               // NOT OWNED
   std::unique_ptr<ThrottledTimeDomain> time_domain_;
 
   CancelableClosureHolder pump_throttled_tasks_closure_;
   base::Optional<base::TimeTicks> pending_pump_throttled_tasks_runtime_;
   bool allow_throttling_;
 
   std::unordered_map<BudgetPool*, std::unique_ptr<BudgetPool>> budget_pools_;
 
   base::WeakPtrFactory<TaskQueueThrottler> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(TaskQueueThrottler);
 };
 
 }  // namespace scheduler
 }  // namespace blink
 
 #endif  // THIRD_PARTY_WEBKIT_SOURCE_PLATFORM_SCHEDULER_RENDERER_TASK_QUEUE_THROTTLER_H_