| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/task_scheduler/priority_queue.h" | 5 #include "base/task_scheduler/priority_queue.h" |
| 6 | 6 |
| 7 #include <utility> | 7 #include <utility> |
| 8 | 8 |
| 9 #include "base/logging.h" | 9 #include "base/logging.h" |
| 10 #include "base/memory/ptr_util.h" | 10 #include "base/memory/ptr_util.h" |
| 11 | 11 |
| 12 namespace base { | 12 namespace base { |
| 13 namespace internal { | 13 namespace internal { |
| 14 | 14 |
| 15 PriorityQueue::SequenceAndSortKey::SequenceAndSortKey() | 15 PriorityQueue::SequenceAndSortKey::SequenceAndSortKey() |
| 16 : sort_key(TaskPriority::LOWEST, TimeTicks()) {} | 16 : sort_key(TaskPriority::LOWEST, TimeTicks()) {} |
| 17 | 17 |
| 18 PriorityQueue::SequenceAndSortKey::SequenceAndSortKey( | 18 PriorityQueue::SequenceAndSortKey::SequenceAndSortKey( |
| 19 scoped_refptr<Sequence> sequence, | 19 scoped_refptr<Sequence> sequence, |
| 20 const SequenceSortKey& sort_key) | 20 const SequenceSortKey& sort_key) |
| 21 : sequence(std::move(sequence)), sort_key(sort_key) {} | 21 : sequence(std::move(sequence)), sort_key(sort_key) {} |
| 22 | 22 |
| 23 PriorityQueue::SequenceAndSortKey::~SequenceAndSortKey() = default; | 23 PriorityQueue::SequenceAndSortKey::~SequenceAndSortKey() = default; |
| 24 | 24 |
| 25 PriorityQueue::Transaction::Transaction(PriorityQueue* outer_queue) | 25 PriorityQueue::Transaction::Transaction(PriorityQueue* outer_queue) |
| 26 : auto_lock_(new AutoSchedulerLock(outer_queue->container_lock_)), | 26 : auto_lock_(outer_queue->container_lock_), outer_queue_(outer_queue) { |
| 27 outer_queue_(outer_queue) { | |
| 28 DCHECK(CalledOnValidThread()); | 27 DCHECK(CalledOnValidThread()); |
| 29 } | 28 } |
| 30 | 29 |
| 31 PriorityQueue::Transaction::~Transaction() { | 30 PriorityQueue::Transaction::~Transaction() { |
| 32 DCHECK(CalledOnValidThread()); | 31 DCHECK(CalledOnValidThread()); |
| 33 | |
| 34 // Run the wake up callback once for each call to Push(). Perform this outside | |
| 35 // the scope of PriorityQueue's lock to avoid imposing an unnecessary lock | |
| 36 // dependency on the callback's destination. | |
| 37 auto_lock_.reset(); | |
| 38 for (size_t i = 0; i < num_wake_ups_; ++i) | |
| 39 outer_queue_->wake_up_callback_.Run(); | |
| 40 } | 32 } |
| 41 | 33 |
| 42 void PriorityQueue::Transaction::Push( | 34 void PriorityQueue::Transaction::Push( |
| 43 std::unique_ptr<SequenceAndSortKey> sequence_and_sort_key) { | 35 std::unique_ptr<SequenceAndSortKey> sequence_and_sort_key) { |
| 44 DCHECK(CalledOnValidThread()); | 36 DCHECK(CalledOnValidThread()); |
| 45 DCHECK(!sequence_and_sort_key->is_null()); | 37 DCHECK(!sequence_and_sort_key->is_null()); |
| 46 | 38 |
| 47 PushNoWakeUp(std::move(sequence_and_sort_key)); | |
| 48 | |
| 49 ++num_wake_ups_; | |
| 50 } | |
| 51 | |
| 52 void PriorityQueue::Transaction::PushNoWakeUp( | |
| 53 std::unique_ptr<SequenceAndSortKey> sequence_and_sort_key) { | |
| 54 DCHECK(CalledOnValidThread()); | |
| 55 DCHECK(!sequence_and_sort_key->is_null()); | |
| 56 | |
| 57 outer_queue_->container_.push(std::move(sequence_and_sort_key)); | 39 outer_queue_->container_.push(std::move(sequence_and_sort_key)); |
| 58 } | 40 } |
| 59 | 41 |
| 60 const PriorityQueue::SequenceAndSortKey& PriorityQueue::Transaction::Peek() | 42 const PriorityQueue::SequenceAndSortKey& PriorityQueue::Transaction::Peek() |
| 61 const { | 43 const { |
| 62 DCHECK(CalledOnValidThread()); | 44 DCHECK(CalledOnValidThread()); |
| 63 | 45 |
| 64 // TODO(fdoray): Add an IsEmpty() method to Transaction and require Peek() to | 46 // TODO(fdoray): Add an IsEmpty() method to Transaction and require Peek() to |
| 65 // be called on a non-empty PriorityQueue only. | 47 // be called on a non-empty PriorityQueue only. |
| 66 if (outer_queue_->container_.empty()) | 48 if (outer_queue_->container_.empty()) |
| 67 return outer_queue_->empty_sequence_and_sort_key_; | 49 return outer_queue_->empty_sequence_and_sort_key_; |
| 68 | 50 |
| 69 return *outer_queue_->container_.top(); | 51 return *outer_queue_->container_.top(); |
| 70 } | 52 } |
| 71 | 53 |
| 72 void PriorityQueue::Transaction::Pop() { | 54 void PriorityQueue::Transaction::Pop() { |
| 73 DCHECK(CalledOnValidThread()); | 55 DCHECK(CalledOnValidThread()); |
| 74 DCHECK(!outer_queue_->container_.empty()); | 56 DCHECK(!outer_queue_->container_.empty()); |
| 75 outer_queue_->container_.pop(); | 57 outer_queue_->container_.pop(); |
| 76 } | 58 } |
| 77 | 59 |
| 78 PriorityQueue::PriorityQueue(const Closure& wake_up_callback) | 60 PriorityQueue::PriorityQueue() = default; |
| 79 : wake_up_callback_(wake_up_callback) { | |
| 80 DCHECK(!wake_up_callback_.is_null()); | |
| 81 } | |
| 82 | 61 |
| 83 PriorityQueue::PriorityQueue(const Closure& wake_up_callback, | 62 PriorityQueue::PriorityQueue(const PriorityQueue* predecessor_priority_queue) |
| 84 const PriorityQueue* predecessor_priority_queue) | 63 : container_lock_(&predecessor_priority_queue->container_lock_) { |
| 85 : container_lock_(&predecessor_priority_queue->container_lock_), | |
| 86 wake_up_callback_(wake_up_callback) { | |
| 87 DCHECK(!wake_up_callback_.is_null()); | |
| 88 DCHECK(predecessor_priority_queue); | 64 DCHECK(predecessor_priority_queue); |
| 89 } | 65 } |
| 90 | 66 |
| 91 PriorityQueue::~PriorityQueue() = default; | 67 PriorityQueue::~PriorityQueue() = default; |
| 92 | 68 |
| 93 std::unique_ptr<PriorityQueue::Transaction> PriorityQueue::BeginTransaction() { | 69 std::unique_ptr<PriorityQueue::Transaction> PriorityQueue::BeginTransaction() { |
| 94 return WrapUnique(new Transaction(this)); | 70 return WrapUnique(new Transaction(this)); |
| 95 } | 71 } |
| 96 | 72 |
| 97 } // namespace internal | 73 } // namespace internal |
| 98 } // namespace base | 74 } // namespace base |
| OLD | NEW |