| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "net/base/prioritized_dispatcher.h" | 5 #include "net/base/prioritized_dispatcher.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 | 8 |
| 9 namespace net { | 9 namespace net { |
| 10 | 10 |
| 11 PrioritizedDispatcher::Limits::Limits(Priority num_priorities, | 11 PrioritizedDispatcher::Limits::Limits(Priority num_priorities, |
| 12 size_t total_jobs) | 12 size_t total_jobs) |
| 13 : total_jobs(total_jobs), reserved_slots(num_priorities) {} | 13 : total_jobs(total_jobs), reserved_slots(num_priorities) {} |
| 14 | 14 |
| 15 PrioritizedDispatcher::Limits::~Limits() {} | 15 PrioritizedDispatcher::Limits::~Limits() {} |
| 16 | 16 |
| 17 PrioritizedDispatcher::PrioritizedDispatcher(const Limits& limits) | 17 PrioritizedDispatcher::PrioritizedDispatcher(const Limits& limits) |
| 18 : queue_(limits.reserved_slots.size()), | 18 : queue_(limits.reserved_slots.size()), |
| 19 max_running_jobs_(limits.reserved_slots.size()), | 19 max_running_jobs_(limits.reserved_slots.size()), |
| 20 num_running_jobs_(0) { | 20 num_running_jobs_(0) { |
| 21 size_t total = 0; | 21 size_t total = 0; |
| 22 for (size_t i = limits.reserved_slots.size(); i > 0; --i) { | 22 for (size_t i = 0; i < limits.reserved_slots.size(); ++i) { |
| 23 total += limits.reserved_slots[i - 1]; | 23 total += limits.reserved_slots[i]; |
| 24 max_running_jobs_[i - 1] = total; | 24 max_running_jobs_[i] = total; |
| 25 } | 25 } |
| 26 // Unreserved slots are available for all priorities. | 26 // Unreserved slots are available for all priorities. |
| 27 DCHECK_LE(total, limits.total_jobs) << "sum(reserved_slots) <= total_jobs"; | 27 DCHECK_LE(total, limits.total_jobs) << "sum(reserved_slots) <= total_jobs"; |
| 28 size_t spare = limits.total_jobs - total; | 28 size_t spare = limits.total_jobs - total; |
| 29 for (size_t i = 0; i < max_running_jobs_.size(); ++i) { | 29 for (size_t i = limits.reserved_slots.size(); i > 0; --i) { |
| 30 max_running_jobs_[i] += spare; | 30 max_running_jobs_[i - 1] += spare; |
| 31 } | 31 } |
| 32 } | 32 } |
| 33 | 33 |
| 34 PrioritizedDispatcher::~PrioritizedDispatcher() {} | 34 PrioritizedDispatcher::~PrioritizedDispatcher() {} |
| 35 | 35 |
| 36 PrioritizedDispatcher::Handle PrioritizedDispatcher::Add( | 36 PrioritizedDispatcher::Handle PrioritizedDispatcher::Add( |
| 37 Job* job, Priority priority) { | 37 Job* job, Priority priority) { |
| 38 DCHECK(job); | 38 DCHECK(job); |
| 39 DCHECK_LT(priority, num_priorities()); | 39 DCHECK_LT(priority, num_priorities()); |
| 40 if (num_running_jobs_ < max_running_jobs_[priority]) { | 40 if (num_running_jobs_ < max_running_jobs_[priority]) { |
| 41 ++num_running_jobs_; | 41 ++num_running_jobs_; |
| 42 job->Start(); | 42 job->Start(); |
| 43 return Handle(); | 43 return Handle(); |
| 44 } | 44 } |
| 45 return queue_.Insert(job, priority); | 45 return queue_.Insert(job, priority); |
| 46 } | 46 } |
| 47 | 47 |
| 48 void PrioritizedDispatcher::Cancel(const Handle& handle) { | 48 void PrioritizedDispatcher::Cancel(const Handle& handle) { |
| 49 queue_.Erase(handle); | 49 queue_.Erase(handle); |
| 50 } | 50 } |
| 51 | 51 |
| 52 PrioritizedDispatcher::Job* PrioritizedDispatcher::EvictOldestLowest() { | 52 PrioritizedDispatcher::Job* PrioritizedDispatcher::EvictOldestLowest() { |
| 53 Handle handle = queue_.FirstMax(); | 53 Handle handle = queue_.FirstMin(); |
| 54 if (handle.is_null()) | 54 if (handle.is_null()) |
| 55 return NULL; | 55 return NULL; |
| 56 Job* job = handle.value(); | 56 Job* job = handle.value(); |
| 57 Cancel(handle); | 57 Cancel(handle); |
| 58 return job; | 58 return job; |
| 59 } | 59 } |
| 60 | 60 |
| 61 PrioritizedDispatcher::Handle PrioritizedDispatcher::ChangePriority( | 61 PrioritizedDispatcher::Handle PrioritizedDispatcher::ChangePriority( |
| 62 const Handle& handle, Priority priority) { | 62 const Handle& handle, Priority priority) { |
| 63 DCHECK(!handle.is_null()); | 63 DCHECK(!handle.is_null()); |
| 64 DCHECK_LT(priority, num_priorities()); | 64 DCHECK_LT(priority, num_priorities()); |
| 65 DCHECK_GE(num_running_jobs_, max_running_jobs_[handle.priority()]) << | 65 DCHECK_GE(num_running_jobs_, max_running_jobs_[handle.priority()]) << |
| 66 "Job should not be in queue when limits permit it to start."; | 66 "Job should not be in queue when limits permit it to start."; |
| 67 | 67 |
| 68 if (handle.priority() == priority) | 68 if (handle.priority() == priority) |
| 69 return handle; | 69 return handle; |
| 70 | 70 |
| 71 if (MaybeDispatchJob(handle, priority)) | 71 if (MaybeDispatchJob(handle, priority)) |
| 72 return Handle(); | 72 return Handle(); |
| 73 Job* job = handle.value(); | 73 Job* job = handle.value(); |
| 74 queue_.Erase(handle); | 74 queue_.Erase(handle); |
| 75 return queue_.Insert(job, priority); | 75 return queue_.Insert(job, priority); |
| 76 } | 76 } |
| 77 | 77 |
| 78 void PrioritizedDispatcher::OnJobFinished() { | 78 void PrioritizedDispatcher::OnJobFinished() { |
| 79 DCHECK_GT(num_running_jobs_, 0u); | 79 DCHECK_GT(num_running_jobs_, 0u); |
| 80 --num_running_jobs_; | 80 --num_running_jobs_; |
| 81 Handle handle = queue_.FirstMin(); | 81 Handle handle = queue_.FirstMax(); |
| 82 if (handle.is_null()) { | 82 if (handle.is_null()) { |
| 83 DCHECK_EQ(0u, queue_.size()); | 83 DCHECK_EQ(0u, queue_.size()); |
| 84 return; | 84 return; |
| 85 } | 85 } |
| 86 MaybeDispatchJob(handle, handle.priority()); | 86 MaybeDispatchJob(handle, handle.priority()); |
| 87 } | 87 } |
| 88 | 88 |
| 89 bool PrioritizedDispatcher::MaybeDispatchJob(const Handle& handle, | 89 bool PrioritizedDispatcher::MaybeDispatchJob(const Handle& handle, |
| 90 Priority job_priority) { | 90 Priority job_priority) { |
| 91 DCHECK_LT(job_priority, num_priorities()); | 91 DCHECK_LT(job_priority, num_priorities()); |
| 92 if (num_running_jobs_ >= max_running_jobs_[job_priority]) | 92 if (num_running_jobs_ >= max_running_jobs_[job_priority]) |
| 93 return false; | 93 return false; |
| 94 Job* job = handle.value(); | 94 Job* job = handle.value(); |
| 95 queue_.Erase(handle); | 95 queue_.Erase(handle); |
| 96 ++num_running_jobs_; | 96 ++num_running_jobs_; |
| 97 job->Start(); | 97 job->Start(); |
| 98 return true; | 98 return true; |
| 99 } | 99 } |
| 100 | 100 |
| 101 } // namespace net | 101 } // namespace net |
| 102 | 102 |
| 103 | 103 |
| OLD | NEW |