| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 "components/scheduler/child/task_queue_sets.h" | 5 #include "components/scheduler/child/task_queue_sets.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "components/scheduler/child/task_queue_impl.h" | 8 #include "components/scheduler/child/task_queue_impl.h" |
| 9 | 9 |
| 10 namespace scheduler { | 10 namespace scheduler { |
| 11 namespace internal { | 11 namespace internal { |
| 12 | 12 |
| 13 TaskQueueSets::TaskQueueSets(size_t num_sets) | 13 TaskQueueSets::TaskQueueSets(size_t num_sets) : age_to_queue_maps_(num_sets) {} |
| 14 : enqueue_order_to_queue_maps_(num_sets) {} | |
| 15 | 14 |
| 16 TaskQueueSets::~TaskQueueSets() {} | 15 TaskQueueSets::~TaskQueueSets() {} |
| 17 | 16 |
| 18 void TaskQueueSets::RemoveQueue(internal::TaskQueueImpl* queue) { | 17 void TaskQueueSets::RemoveQueue(internal::TaskQueueImpl* queue) { |
| 19 int enqueue_order; | 18 int age; |
| 20 bool has_enqueue_order = | 19 bool has_age = queue->GetWorkQueueFrontTaskAge(&age); |
| 21 queue->GetWorkQueueFrontTaskEnqueueOrder(&enqueue_order); | 20 if (!has_age) |
| 22 if (!has_enqueue_order) | |
| 23 return; | 21 return; |
| 24 size_t set_index = queue->get_task_queue_set_index(); | 22 size_t set_index = queue->get_task_queue_set_index(); |
| 25 DCHECK_LT(set_index, enqueue_order_to_queue_maps_.size()); | 23 DCHECK_LT(set_index, age_to_queue_maps_.size()); |
| 26 enqueue_order_to_queue_maps_[set_index].erase(enqueue_order); | 24 age_to_queue_maps_[set_index].erase(age); |
| 27 } | 25 } |
| 28 | 26 |
| 29 void TaskQueueSets::AssignQueueToSet(internal::TaskQueueImpl* queue, | 27 void TaskQueueSets::AssignQueueToSet(internal::TaskQueueImpl* queue, |
| 30 size_t set_index) { | 28 size_t set_index) { |
| 31 DCHECK_LT(set_index, enqueue_order_to_queue_maps_.size()); | 29 DCHECK_LT(set_index, age_to_queue_maps_.size()); |
| 32 int enqueue_order; | 30 int age; |
| 33 bool has_enqueue_order = | 31 bool has_age = queue->GetWorkQueueFrontTaskAge(&age); |
| 34 queue->GetWorkQueueFrontTaskEnqueueOrder(&enqueue_order); | |
| 35 size_t old_set = queue->get_task_queue_set_index(); | 32 size_t old_set = queue->get_task_queue_set_index(); |
| 36 DCHECK_LT(old_set, enqueue_order_to_queue_maps_.size()); | 33 DCHECK_LT(old_set, age_to_queue_maps_.size()); |
| 37 queue->set_task_queue_set_index(set_index); | 34 queue->set_task_queue_set_index(set_index); |
| 38 if (!has_enqueue_order) | 35 if (!has_age) |
| 39 return; | 36 return; |
| 40 enqueue_order_to_queue_maps_[old_set].erase(enqueue_order); | 37 age_to_queue_maps_[old_set].erase(age); |
| 41 enqueue_order_to_queue_maps_[set_index].insert( | 38 age_to_queue_maps_[set_index].insert(std::make_pair(age, queue)); |
| 42 std::make_pair(enqueue_order, queue)); | |
| 43 } | 39 } |
| 44 | 40 |
| 45 void TaskQueueSets::OnPushQueue(internal::TaskQueueImpl* queue) { | 41 void TaskQueueSets::OnPushQueue(internal::TaskQueueImpl* queue) { |
| 46 int enqueue_order; | 42 int age; |
| 47 bool has_enqueue_order = | 43 bool has_age = queue->GetWorkQueueFrontTaskAge(&age); |
| 48 queue->GetWorkQueueFrontTaskEnqueueOrder(&enqueue_order); | 44 DCHECK(has_age); |
| 49 DCHECK(has_enqueue_order); | |
| 50 size_t set_index = queue->get_task_queue_set_index(); | 45 size_t set_index = queue->get_task_queue_set_index(); |
| 51 DCHECK_LT(set_index, enqueue_order_to_queue_maps_.size()) << " set_index = " | 46 DCHECK_LT(set_index, age_to_queue_maps_.size()) << " set_index = " |
| 52 << set_index; | 47 << set_index; |
| 53 enqueue_order_to_queue_maps_[set_index].insert( | 48 age_to_queue_maps_[set_index].insert(std::make_pair(age, queue)); |
| 54 std::make_pair(enqueue_order, queue)); | |
| 55 } | 49 } |
| 56 | 50 |
| 57 void TaskQueueSets::OnPopQueue(internal::TaskQueueImpl* queue) { | 51 void TaskQueueSets::OnPopQueue(internal::TaskQueueImpl* queue) { |
| 58 size_t set_index = queue->get_task_queue_set_index(); | 52 size_t set_index = queue->get_task_queue_set_index(); |
| 59 DCHECK_LT(set_index, enqueue_order_to_queue_maps_.size()); | 53 DCHECK_LT(set_index, age_to_queue_maps_.size()); |
| 60 DCHECK(!enqueue_order_to_queue_maps_[set_index].empty()) << " set_index = " | 54 DCHECK(!age_to_queue_maps_[set_index].empty()) << " set_index = " |
| 61 << set_index; | 55 << set_index; |
| 62 DCHECK_EQ(enqueue_order_to_queue_maps_[set_index].begin()->second, queue) | 56 DCHECK_EQ(age_to_queue_maps_[set_index].begin()->second, queue) |
| 63 << " set_index = " << set_index; | 57 << " set_index = " << set_index; |
| 64 // O(1) amortised. | 58 // O(1) amortised. |
| 65 enqueue_order_to_queue_maps_[set_index].erase( | 59 age_to_queue_maps_[set_index].erase(age_to_queue_maps_[set_index].begin()); |
| 66 enqueue_order_to_queue_maps_[set_index].begin()); | 60 int age; |
| 67 int enqueue_order; | 61 bool has_age = queue->GetWorkQueueFrontTaskAge(&age); |
| 68 bool has_enqueue_order = | 62 if (!has_age) |
| 69 queue->GetWorkQueueFrontTaskEnqueueOrder(&enqueue_order); | |
| 70 if (!has_enqueue_order) | |
| 71 return; | 63 return; |
| 72 enqueue_order_to_queue_maps_[set_index].insert( | 64 age_to_queue_maps_[set_index].insert(std::make_pair(age, queue)); |
| 73 std::make_pair(enqueue_order, queue)); | |
| 74 } | 65 } |
| 75 | 66 |
| 76 bool TaskQueueSets::GetOldestQueueInSet( | 67 bool TaskQueueSets::GetOldestQueueInSet( |
| 77 size_t set_index, | 68 size_t set_index, |
| 78 internal::TaskQueueImpl** out_queue) const { | 69 internal::TaskQueueImpl** out_queue) const { |
| 79 DCHECK_LT(set_index, enqueue_order_to_queue_maps_.size()); | 70 DCHECK_LT(set_index, age_to_queue_maps_.size()); |
| 80 if (enqueue_order_to_queue_maps_[set_index].empty()) | 71 if (age_to_queue_maps_[set_index].empty()) |
| 81 return false; | 72 return false; |
| 82 *out_queue = enqueue_order_to_queue_maps_[set_index].begin()->second; | 73 *out_queue = age_to_queue_maps_[set_index].begin()->second; |
| 83 return true; | 74 return true; |
| 84 } | 75 } |
| 85 | 76 |
| 86 } // namespace internal | 77 } // namespace internal |
| 87 } // namespace scheduler | 78 } // namespace scheduler |
| OLD | NEW |