Move has_incoming_immediate_work to the TaskQueueManager

third_party/WebKit/Source/platform/scheduler/base/task_queue_manager.h

 // Copyright 2014 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_BASE_TASK_QUEUE_MANAGER_H_
 #define THIRD_PARTY_WEBKIT_SOURCE_PLATFORM_SCHEDULER_BASE_TASK_QUEUE_MANAGER_H_
 
 #include <map>
 
 #include "base/atomic_sequence_num.h"
@@ skipping 162 matching lines @@
   void OnBeginNestedMessageLoop() override;
 
   // Called by the task queue to register a new pending task.
   void DidQueueTask(const internal::TaskQueueImpl::Task& pending_task);
 
   // Use the selector to choose a pending task and run it.
   void DoWork(base::TimeTicks run_time, bool from_main_thread);
 
   // Delayed Tasks with run_times <= Now() are enqueued onto the work queue and
   // reloads any empty work queues.
-  void UpdateWorkQueues(LazyNow* lazy_now);
+  void WakeupReadyDelayedQueues(LazyNow* lazy_now);
 
   // Chooses the next work queue to service. Returns true if |out_queue|
   // indicates the queue from which the next task should be run, false to
   // avoid running any tasks.
   bool SelectWorkQueueToService(internal::WorkQueue** out_work_queue);
 
   enum class ProcessTaskResult {
     DEFERRED,
     EXECUTED,
     TASK_QUEUE_MANAGER_DELETED
@@ skipping 21 matching lines @@
   base::Optional<base::TimeDelta> ComputeDelayTillNextTask(LazyNow* lazy_now);
 
   void MaybeRecordTaskDelayHistograms(
       const internal::TaskQueueImpl::Task& pending_task,
       const internal::TaskQueueImpl* queue);
 
   std::unique_ptr<base::trace_event::ConvertableToTraceFormat>
   AsValueWithSelectorResult(bool should_run,
                             internal::WorkQueue* selected_work_queue) const;
 
+  // Adds |queue| to |any_thread().has_incoming_immediate_work_| and if
+  // |queue_is_blocked| is false it makes sure a DoWork is posted.
+  // Can be called from any thread.
+  void OnQueueHasIncomingImmediateWork(internal::TaskQueueImpl* queue,
+                                       bool queue_is_blocked);
+
+  // Calls |ReloadImmediateWorkQueueIfEmpty| on all queues in
+  // |queues_to_reload|.
+  void ReloadEmptyWorkQueues(const std::unordered_set<internal::TaskQueueImpl*>&
+                                 queues_to_reload) const;
+
   std::set<TimeDomain*> time_domains_;
   std::unique_ptr<RealTimeDomain> real_time_domain_;
 
   std::set<scoped_refptr<internal::TaskQueueImpl>> queues_;
 
   // We have to be careful when deleting a queue because some of the code uses
   // raw pointers and doesn't expect the rug to be pulled out from underneath.
   std::set<scoped_refptr<internal::TaskQueueImpl>> queues_to_delete_;
 
   internal::EnqueueOrderGenerator enqueue_order_generator_;
   base::debug::TaskAnnotator task_annotator_;
 
   base::ThreadChecker main_thread_checker_;
   scoped_refptr<TaskQueueManagerDelegate> delegate_;
   internal::TaskQueueSelector selector_;
 
   base::Closure from_main_thread_immediate_do_work_closure_;
   base::Closure from_other_thread_immediate_do_work_closure_;
 
   bool task_was_run_on_quiescence_monitored_queue_;
 
   // To reduce locking overhead we track pending calls to DoWork separately for
   // the main thread and other threads.
   std::set<base::TimeTicks> main_thread_pending_wakeups_;
 
   struct AnyThread {
     AnyThread();
 
+    // Set of task queues with newly available work on the incoming queue.
+    std::unordered_set<internal::TaskQueueImpl*> has_incoming_immediate_work;
+
     bool other_thread_pending_wakeup;
   };
 
   // TODO(alexclarke): Add a MainThreadOnly struct too.
 
   mutable base::Lock any_thread_lock_;
   AnyThread any_thread_;
 
   struct AnyThread& any_thread() {
     any_thread_lock_.AssertAcquired();
@@ skipping 23 matching lines @@
   scoped_refptr<DeletionSentinel> deletion_sentinel_;
   base::WeakPtrFactory<TaskQueueManager> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(TaskQueueManager);
 };
 
 }  // namespace scheduler
 }  // namespace blink
 
 #endif  // THIRD_PARTY_WEBKIT_SOURCE_PLATFORM_SCHEDULER_BASE_TASK_QUEUE_MANAGER_H_