[scheduler] Add WakeupBudgetPool.

third_party/WebKit/Source/platform/scheduler/renderer/cpu_time_budget_pool.cc

 // Copyright 2017 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 "platform/scheduler/renderer/budget_pool.h"
+#include "platform/scheduler/renderer/cpu_time_budget_pool.h"
 
 #include <cstdint>
 
 #include "base/format_macros.h"
 #include "base/logging.h"
 #include "base/memory/ptr_util.h"
 #include "base/optional.h"
 #include "base/strings/stringprintf.h"
-#include "platform/WebFrameScheduler.h"
-#include "platform/scheduler/base/real_time_domain.h"
-#include "platform/scheduler/child/scheduler_tqm_delegate.h"
-#include "platform/scheduler/renderer/renderer_scheduler_impl.h"
 #include "platform/scheduler/renderer/task_queue_throttler.h"
-#include "platform/scheduler/renderer/throttled_time_domain.h"
-#include "platform/scheduler/renderer/web_frame_scheduler_impl.h"
 
 namespace blink {
 namespace scheduler {
 
 namespace {
 
 std::string PointerToId(void* pointer) {
   return base::StringPrintf(
       "0x%" PRIx64,
       static_cast<uint64_t>(reinterpret_cast<uintptr_t>(pointer)));
 }
 
 }  // namespace
 
-BudgetPool::BudgetPool(const char* name,
-                       BudgetPoolController* budget_pool_controller)
-    : name_(name),
-      budget_pool_controller_(budget_pool_controller),
-      is_enabled_(true) {}
-
-BudgetPool::~BudgetPool() {}
-
-const char* BudgetPool::Name() const {
-  return name_;
-}
-
-void BudgetPool::AddQueue(base::TimeTicks now, TaskQueue* queue) {
-  budget_pool_controller_->AddQueueToBudgetPool(queue, this);
-  associated_task_queues_.insert(queue);
-
-  if (!is_enabled_ || !budget_pool_controller_->IsThrottled(queue))
-    return;
-
-  budget_pool_controller_->BlockQueue(now, queue);
-}
-
-void BudgetPool::RemoveQueue(base::TimeTicks now, TaskQueue* queue) {
-  budget_pool_controller_->RemoveQueueFromBudgetPool(queue, this);
-  associated_task_queues_.erase(queue);
-
-  if (!is_enabled_ || !budget_pool_controller_->IsThrottled(queue))
-    return;
-
-  budget_pool_controller_->UnblockQueue(now, queue);
-}
-
-void BudgetPool::EnableThrottling(LazyNow* lazy_now) {
-  if (is_enabled_)
-    return;
-  is_enabled_ = true;
-
-  TRACE_EVENT0("renderer.scheduler", "BudgetPool_EnableThrottling");
-
-  BlockThrottledQueues(lazy_now->Now());
-}
-
-void BudgetPool::DisableThrottling(LazyNow* lazy_now) {
-  if (!is_enabled_)
-    return;
-  is_enabled_ = false;
-
-  TRACE_EVENT0("renderer.scheduler", "BudgetPool_DisableThrottling");
-
-  for (TaskQueue* queue : associated_task_queues_) {
-    if (!budget_pool_controller_->IsThrottled(queue))
-      continue;
-
-    budget_pool_controller_->UnblockQueue(lazy_now->Now(), queue);
-  }
-
-  // TODO(altimin): We need to disable TimeBudgetQueues here or they will
-  // regenerate extra time budget when they are disabled.
-}
-
-bool BudgetPool::IsThrottlingEnabled() const {
-  return is_enabled_;
-}
-
-void BudgetPool::Close() {
-  DCHECK_EQ(0u, associated_task_queues_.size());
-
-  budget_pool_controller_->UnregisterBudgetPool(this);
-}
-
-void BudgetPool::BlockThrottledQueues(base::TimeTicks now) {
-  for (TaskQueue* queue : associated_task_queues_)
-    budget_pool_controller_->BlockQueue(now, queue);
-}
-
 CPUTimeBudgetPool::CPUTimeBudgetPool(
     const char* name,
     BudgetPoolController* budget_pool_controller,
     base::TimeTicks now)
     : BudgetPool(name, budget_pool_controller),
       last_checkpoint_(now),
       cpu_percentage_(1) {}
 
 CPUTimeBudgetPool::~CPUTimeBudgetPool() {}
 
+QueueBlockType CPUTimeBudgetPool::GetBlockType() const {
+  return QueueBlockType::kAllTasks;
+}
+
 void CPUTimeBudgetPool::SetMaxBudgetLevel(
     base::TimeTicks now,
     base::Optional<base::TimeDelta> max_budget_level) {
   Advance(now);
   max_budget_level_ = max_budget_level;
   EnforceBudgetLevelRestrictions();
 }
 
 void CPUTimeBudgetPool::SetMaxThrottlingDelay(
     base::TimeTicks now,
@@ skipping 22 matching lines @@
   Advance(now);
   current_budget_level_ += budget_level;
   EnforceBudgetLevelRestrictions();
 }
 
 void CPUTimeBudgetPool::SetReportingCallback(
     base::Callback<void(base::TimeDelta)> reporting_callback) {
   reporting_callback_ = reporting_callback;
 }
 
-bool CPUTimeBudgetPool::HasEnoughBudgetToRun(base::TimeTicks now) {
-  return now >= GetNextAllowedRunTime();
+bool CPUTimeBudgetPool::CanRunTasksAt(base::TimeTicks now,
+                                      bool is_wake_up) const {
+  return now >= GetNextAllowedRunTime(now);
 }
 
-base::TimeTicks CPUTimeBudgetPool::GetNextAllowedRunTime() {
+bool CPUTimeBudgetPool::CanRunTasksUntil(base::TimeTicks now,
+                                         base::TimeTicks moment) const {
+  return CanRunTasksAt(now, false);

[Sami, 2017/04/29 17:43:02]

I think this is slightly misleading: I would expect CanRanTasksUntil to return
true until the point at which the budget becomes negative. To do that we would
need to look at |moment| which we're not doing here. Should we do that?

[altimin, 2017/05/02 18:16:58]

Scrapped CanRunTasksUntil completely.

+}
+
+base::TimeTicks CPUTimeBudgetPool::GetNextAllowedRunTime(
+    base::TimeTicks desired_run_time) const {
   if (!is_enabled_ || current_budget_level_.InMicroseconds() >= 0) {
     return last_checkpoint_;
   } else {
     // Subtract because current_budget is negative.
     return last_checkpoint_ +
            (-current_budget_level_ + min_budget_level_to_run_) /
                cpu_percentage_;
   }
 }
 
-void CPUTimeBudgetPool::RecordTaskRunTime(base::TimeTicks start_time,
+void CPUTimeBudgetPool::RecordTaskRunTime(TaskQueue* queue,
+                                          base::TimeTicks start_time,
                                           base::TimeTicks end_time) {
   DCHECK_LE(start_time, end_time);
   Advance(end_time);
   if (is_enabled_) {
     base::TimeDelta old_budget_level = current_budget_level_;
     current_budget_level_ -= (end_time - start_time);
     EnforceBudgetLevelRestrictions();
 
     if (!reporting_callback_.is_null() && old_budget_level.InSecondsF() > 0 &&
         current_budget_level_.InSecondsF() < 0) {
       reporting_callback_.Run(-current_budget_level_ / cpu_percentage_);
     }
   }
+
+  if (current_budget_level_.InSecondsF() < 0)
+    BlockThrottledQueues(end_time);
 }
 
+void CPUTimeBudgetPool::OnQueueNextWakeUpChanged(
+    TaskQueue* queue,
+    base::TimeTicks now,
+    base::TimeTicks desired_run_time) {
+  budget_pool_controller_->UpdateQueueThrottlingState(now, queue);
+}
+
+void CPUTimeBudgetPool::OnWakeUp(base::TimeTicks now) {}
+
 void CPUTimeBudgetPool::AsValueInto(base::trace_event::TracedValue* state,
                                     base::TimeTicks now) const {
   state->BeginDictionary(name_);
 
   state->SetString("name", name_);
   state->SetDouble("time_budget", cpu_percentage_);
   state->SetDouble("time_budget_level_in_seconds",
                    current_budget_level_.InSecondsF());
   state->SetDouble("last_checkpoint_seconds_ago",
                    (now - last_checkpoint_).InSecondsF());
@@ skipping 37 matching lines @@
   if (max_throttling_delay_) {
     // Current budget level may be negative.
     current_budget_level_ =
         std::max(current_budget_level_,
                  -max_throttling_delay_.value() * cpu_percentage_);
   }
 }
 
 }  // namespace scheduler
 }  // namespace blink