Remove after wakeup logic and replace PumpTask with Fences

third_party/WebKit/Source/platform/scheduler/base/work_queue.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 "platform/scheduler/base/work_queue.h"
 
 #include "platform/scheduler/base/work_queue_sets.h"
 
 namespace blink {
 namespace scheduler {
 namespace internal {
 
 WorkQueue::WorkQueue(TaskQueueImpl* task_queue,
                      const char* name,
                      TaskQueueImpl::Task::ComparatorFn comparator)
     : work_queue_(comparator),
       work_queue_sets_(nullptr),
       task_queue_(task_queue),
       work_queue_set_index_(0),
-      name_(name) {}
+      name_(name),
+      fence_(0),
+      fence_hit_(false) {}
 
 void WorkQueue::AsValueInto(base::trace_event::TracedValue* state) const {
   for (const TaskQueueImpl::Task& task : work_queue_) {
     TaskQueueImpl::TaskAsValueInto(task, state);
   }
 }
 
 WorkQueue::~WorkQueue() {
   DCHECK(!work_queue_sets_) << task_queue_->GetName() << " : "
                             << work_queue_sets_->name() << " : " << name_;
 }
 
 const TaskQueueImpl::Task* WorkQueue::GetFrontTask() const {
   if (work_queue_.empty())
     return nullptr;
   return &*work_queue_.begin();
 }
 
+const TaskQueueImpl::Task* WorkQueue::GetBackTask() const {
+  if (work_queue_.empty())
+    return nullptr;
+  return &*work_queue_.rbegin();
+}
+
 bool WorkQueue::GetFrontTaskEnqueueOrder(EnqueueOrder* enqueue_order) const {
-  if (work_queue_.empty())
+  if (work_queue_.empty() || fence_hit_)
     return false;
   // Quick sanity check.
   DCHECK_LE(work_queue_.begin()->enqueue_order(),
             work_queue_.rbegin()->enqueue_order())
       << task_queue_->GetName() << " : "
       << work_queue_sets_->name() << " : " << name_;
   *enqueue_order = work_queue_.begin()->enqueue_order();
   return true;
 }
 
 void WorkQueue::Push(TaskQueueImpl::Task task) {
   bool was_empty = work_queue_.empty();
 #ifndef NDEBUG
   DCHECK(task.enqueue_order_set());
 #endif
 
   // We expect |task| to be inserted at the end. Amoritized O(1).
   work_queue_.insert(work_queue_.end(), std::move(task));
   DCHECK_EQ(task.enqueue_order(), work_queue_.rbegin()->enqueue_order())
       << task_queue_->GetName() << " : "
       << work_queue_sets_->name() << " : " << name_
       << "task [scheduled_run_time_" << task.delayed_run_time << ", "
       << task.sequence_num <<
       "] rbegin() [" << work_queue_.rbegin()->delayed_run_time << ", "
       << work_queue_.rbegin()->sequence_num << "]";
 
-  if (was_empty && work_queue_sets_) {
+  if (was_empty && work_queue_sets_ && !fence_hit_) {
     work_queue_sets_->OnPushQueue(this);
   }
 }
 
 bool WorkQueue::CancelTask(const TaskQueueImpl::Task& key) {
   TaskQueueImpl::ComparatorQueue::iterator it = work_queue_.find(key);
   if (it == work_queue_.end())
     return false;
 
   if (it == work_queue_.begin()) {
     EnqueueOrder erased_task_enqueue_order = it->enqueue_order();
     work_queue_.erase(it);
     // We can't guarantee this WorkQueue is the lowest value in the WorkQueueSet
     // so we need to use OnQueueHeadChanged instead of OnPopQueue for
     // correctness.
-    work_queue_sets_->OnQueueHeadChanged(this, erased_task_enqueue_order);
+    if (!fence_hit_)
+      work_queue_sets_->OnQueueHeadChanged(this, erased_task_enqueue_order);
   } else {
     work_queue_.erase(it);
   }
   task_queue_->TraceQueueSize(false);
   return true;
 }
 
 bool WorkQueue::IsTaskPending(const TaskQueueImpl::Task& key) const {
   return work_queue_.find(key) != work_queue_.end();
 }
 
 void WorkQueue::PopTaskForTest() {
   if (work_queue_.empty())
     return;
   work_queue_.erase(work_queue_.begin());
 }
 
 void WorkQueue::SwapLocked(TaskQueueImpl::ComparatorQueue& incoming_queue) {
   DCHECK(work_queue_.empty());
   std::swap(work_queue_, incoming_queue);
-  if (!work_queue_.empty() && work_queue_sets_)
+  if (!work_queue_.empty() && work_queue_sets_ && !fence_hit_)
     work_queue_sets_->OnPushQueue(this);
-  task_queue_->TraceQueueSize(true);

[Sami, 2016/08/26 14:37:02]

Did you mean to remove this?

[alex clarke (OOO till 29th), 2016/08/26 16:33:14]

Yes it's not useful to call it here since Swap does not change the total queue
size.

 }
 
 TaskQueueImpl::Task WorkQueue::TakeTaskFromWorkQueue() {
   DCHECK(work_queue_sets_);
   DCHECK(!work_queue_.empty());
   TaskQueueImpl::ComparatorQueue::iterator it = work_queue_.begin();
   TaskQueueImpl::Task pending_task =
       std::move(const_cast<TaskQueueImpl::Task&>(*it));
   work_queue_.erase(it);
+  // If we hit the fence, pretend to WorkQueueSets that we're empty until the
+  // fence is removed.
+  if (fence_ && pending_task.enqueue_order() >= fence_)
+    fence_hit_ = true;
   work_queue_sets_->OnPopQueue(this);
   task_queue_->TraceQueueSize(false);
   return pending_task;
 }
 
 void WorkQueue::AssignToWorkQueueSets(WorkQueueSets* work_queue_sets) {
   work_queue_sets_ = work_queue_sets;
 }
 
 void WorkQueue::AssignSetIndex(size_t work_queue_set_index) {
   work_queue_set_index_ = work_queue_set_index;
 }
 
+bool WorkQueue::InsertFence(EnqueueOrder fence) {
+  bool task_unblocked = false;
+  if (fence) {
+    DCHECK_GE(fence, fence_);
+    if (fence > fence_)
+      task_unblocked = RemoveFence();
+  } else {
+    DCHECK(work_queue_.empty());
+    // Until the fence is removed we should pretend to remain empty even after a
+    // push or a swap.
+    fence_hit_ = true;
+  }
+
+  fence_ = fence;
+  return task_unblocked;
+}
+
+bool WorkQueue::RemoveFence() {
+  fence_ = 0;
+  bool fence_was_hit = fence_hit_;
+  fence_hit_ = false;
+  if (work_queue_sets_ && fence_was_hit && !work_queue_.empty()) {
+    work_queue_sets_->OnPushQueue(this);
+    return true;
+  }
+  return false;
+}
+
 bool WorkQueue::ShouldRunBefore(const WorkQueue* other_queue) const {
   DCHECK(!work_queue_.empty());
   DCHECK(!other_queue->work_queue_.empty());
   EnqueueOrder enqueue_order = 0;
   EnqueueOrder other_enqueue_order = 0;
   bool have_task = GetFrontTaskEnqueueOrder(&enqueue_order);
   bool have_other_task =
       other_queue->GetFrontTaskEnqueueOrder(&other_enqueue_order);
   DCHECK(have_task);
   DCHECK(have_other_task);
   return enqueue_order < other_enqueue_order;
 }
 
 }  // namespace internal
 }  // namespace scheduler
 }  // namespace blink