Move throttling of background timers into the renderer scheduler

components/scheduler/base/time_domain.cc

 // 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.
 
 #include "components/scheduler/base/time_domain.h"
 
 #include <set>
 
 #include "components/scheduler/base/task_queue_impl.h"
 #include "components/scheduler/base/task_queue_manager_delegate.h"
 #include "components/scheduler/scheduler_export.h"
 
 namespace scheduler {
 
-TimeDomain::TimeDomain() {}
+TimeDomain::TimeDomain(Observer* observer) : observer_(observer) {}
 
-TimeDomain::~TimeDomain() {}
+TimeDomain::~TimeDomain() {
+  for (internal::TaskQueueImpl* queue : registered_task_queues_) {
+    queue->SetTimeDomain(nullptr);
+  }
+}
+
+void TimeDomain::RegisterQueue(internal::TaskQueueImpl* queue) {
+  registered_task_queues_.insert(queue);
+}
 
 void TimeDomain::UnregisterQueue(internal::TaskQueueImpl* queue) {
+  registered_task_queues_.erase(queue);
+
   // We need to remove |task_queue| from delayed_wakeup_multimap_ which is a
   // little awkward since it's keyed by time. O(n) running time.
   for (DelayedWakeupMultimap::iterator iter = delayed_wakeup_multimap_.begin();
        iter != delayed_wakeup_multimap_.end();) {
     if (iter->second == queue) {
       DelayedWakeupMultimap::iterator temp = iter;
       iter++;
       // O(1) amortized.
       delayed_wakeup_multimap_.erase(temp);
     } else {
       iter++;
     }
   }
 
   // |newly_updatable_| might contain |queue|, we use
   // MoveNewlyUpdatableQueuesIntoUpdatableQueueSet to flush it out.
   MoveNewlyUpdatableQueuesIntoUpdatableQueueSet();
   updatable_queue_set_.erase(queue);
 }
 
 void TimeDomain::MigrateQueue(internal::TaskQueueImpl* queue,
                               TimeDomain* destination_time_domain) {
+  DCHECK(destination_time_domain);
+  registered_task_queues_.erase(queue);
+
   LazyNow destination_lazy_now = destination_time_domain->CreateLazyNow();
   // We need to remove |task_queue| from delayed_wakeup_multimap_ which is a
   // little awkward since it's keyed by time. O(n) running time.
   for (DelayedWakeupMultimap::iterator iter = delayed_wakeup_multimap_.begin();
        iter != delayed_wakeup_multimap_.end();) {
     if (iter->second == queue) {
       destination_time_domain->ScheduleDelayedWork(queue, iter->first,
                                                    &destination_lazy_now);
       DelayedWakeupMultimap::iterator temp = iter;
       iter++;
       // O(1) amortized.
       delayed_wakeup_multimap_.erase(temp);
     } else {
       iter++;
     }
   }
 
   // |newly_updatable_| might contain |queue|, we use
   // MoveNewlyUpdatableQueuesIntoUpdatableQueueSet to flush it out.
   MoveNewlyUpdatableQueuesIntoUpdatableQueueSet();
   updatable_queue_set_.erase(queue);
+
+  destination_time_domain->RegisterQueue(queue);
 }
 
 void TimeDomain::ScheduleDelayedWork(internal::TaskQueueImpl* queue,
                                      base::TimeTicks delayed_run_time,
                                      LazyNow* lazy_now) {
   DCHECK(main_thread_checker_.CalledOnValidThread());
 
-  if (delayed_wakeup_multimap_.empty() ||
+  bool delayed_wakeup_multimap_was_empty = delayed_wakeup_multimap_.empty();
+  if (delayed_wakeup_multimap_was_empty ||
       delayed_run_time < delayed_wakeup_multimap_.begin()->first) {
     base::TimeDelta delay =
         std::max(base::TimeDelta(), delayed_run_time - lazy_now->Now());
     RequestWakeup(lazy_now, delay);
   }
+
+  if (observer_ && delayed_wakeup_multimap_was_empty)
+    observer_->OnTimeDomainHasDelayedWork();
   delayed_wakeup_multimap_.insert(std::make_pair(delayed_run_time, queue));
 }
 
 void TimeDomain::RegisterAsUpdatableTaskQueue(internal::TaskQueueImpl* queue) {
-  base::AutoLock lock(newly_updatable_lock_);
-  newly_updatable_.push_back(queue);
+  {
+    base::AutoLock lock(newly_updatable_lock_);
+    newly_updatable_.push_back(queue);
+  }
+  if (observer_)
+    observer_->OnTimeDomainHasImmediateWork();
 }
 
 void TimeDomain::UnregisterAsUpdatableTaskQueue(
     internal::TaskQueueImpl* queue) {
   DCHECK(main_thread_checker_.CalledOnValidThread());
   MoveNewlyUpdatableQueuesIntoUpdatableQueueSet();
 #ifndef NDEBUG
   {
     base::AutoLock lock(newly_updatable_lock_);
     DCHECK(!(updatable_queue_set_.find(queue) == updatable_queue_set_.end() &&
@@ skipping 77 matching lines @@
   state->SetString("name", GetName());
   state->BeginArray("updatable_queue_set");
   for (auto& queue : updatable_queue_set_)
     state->AppendString(queue->GetName());
   state->EndArray();
   AsValueIntoInternal(state);
   state->EndDictionary();
 }
 
 }  // namespace scheduler