| OLD | NEW |
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #ifndef CHROME_BROWSER_EXTENSIONS_UPDATER_REQUEST_QUEUE_IMPL_H_ | |
| 6 #define CHROME_BROWSER_EXTENSIONS_UPDATER_REQUEST_QUEUE_IMPL_H_ | |
| 7 | |
| 8 #include <algorithm> | |
| 9 | |
| 10 #include "base/bind.h" | |
| 11 #include "base/compiler_specific.h" | |
| 12 #include "base/message_loop/message_loop.h" | |
| 13 #include "base/stl_util.h" | |
| 14 #include "chrome/browser/extensions/updater/request_queue.h" | |
| 15 | |
| 16 namespace extensions { | |
| 17 | |
| 18 template<typename T> | |
| 19 RequestQueue<T>::RequestQueue( | |
| 20 const net::BackoffEntry::Policy* const backoff_policy, | |
| 21 const base::Closure& start_request_callback) | |
| 22 : backoff_policy_(backoff_policy), | |
| 23 start_request_callback_(start_request_callback), | |
| 24 timer_(false, false) { | |
| 25 } | |
| 26 | |
| 27 template<typename T> | |
| 28 RequestQueue<T>::~RequestQueue() {} | |
| 29 | |
| 30 template<typename T> | |
| 31 T* RequestQueue<T>::active_request() { | |
| 32 return active_request_.get(); | |
| 33 } | |
| 34 | |
| 35 template<typename T> | |
| 36 int RequestQueue<T>::active_request_failure_count() { | |
| 37 return active_backoff_entry_->failure_count(); | |
| 38 } | |
| 39 | |
| 40 template<typename T> | |
| 41 scoped_ptr<T> RequestQueue<T>::reset_active_request() { | |
| 42 active_backoff_entry_.reset(); | |
| 43 return active_request_.Pass(); | |
| 44 } | |
| 45 | |
| 46 template<typename T> | |
| 47 void RequestQueue<T>::ScheduleRequest(scoped_ptr<T> request) { | |
| 48 PushImpl(request.Pass(), scoped_ptr<net::BackoffEntry>( | |
| 49 new net::BackoffEntry(backoff_policy_))); | |
| 50 StartNextRequest(); | |
| 51 } | |
| 52 | |
| 53 template<typename T> | |
| 54 void RequestQueue<T>::PushImpl(scoped_ptr<T> request, | |
| 55 scoped_ptr<net::BackoffEntry> backoff_entry) { | |
| 56 pending_requests_.push_back(Request( | |
| 57 backoff_entry.release(), request.release())); | |
| 58 std::push_heap(pending_requests_.begin(), pending_requests_.end(), | |
| 59 CompareRequests); | |
| 60 } | |
| 61 | |
| 62 template<typename T> | |
| 63 bool RequestQueue<T>::empty() const { | |
| 64 return pending_requests_.empty(); | |
| 65 } | |
| 66 | |
| 67 template<typename T> | |
| 68 size_t RequestQueue<T>::size() const { | |
| 69 return pending_requests_.size(); | |
| 70 } | |
| 71 | |
| 72 template<typename T> | |
| 73 base::TimeTicks RequestQueue<T>::NextReleaseTime() const { | |
| 74 return pending_requests_.front().backoff_entry->GetReleaseTime(); | |
| 75 } | |
| 76 | |
| 77 template<typename T> | |
| 78 void RequestQueue<T>::StartNextRequest() { | |
| 79 if (active_request_) | |
| 80 // Already running a request, assume this method will be called again when | |
| 81 // the request is done. | |
| 82 return; | |
| 83 | |
| 84 if (empty()) | |
| 85 // No requests in the queue, so we're done. | |
| 86 return; | |
| 87 | |
| 88 base::TimeTicks next_release = NextReleaseTime(); | |
| 89 base::TimeTicks now = base::TimeTicks::Now(); | |
| 90 if (next_release > now) { | |
| 91 // Not ready for the next update check yet, call this method when it is | |
| 92 // time. | |
| 93 timer_.Start(FROM_HERE, next_release - now, | |
| 94 base::Bind(&RequestQueue<T>::StartNextRequest, | |
| 95 base::Unretained(this))); | |
| 96 return; | |
| 97 } | |
| 98 | |
| 99 // pop_heap swaps the first and last elements of pending_requests_, and after | |
| 100 // that assures that the rest of pending_requests_ (excluding the | |
| 101 // now last/formerly first element) forms a proper heap. After pop_heap | |
| 102 // [begin, end-1) is a valid heap, and *(end - 1) contains the element that | |
| 103 // used to be at the top of the heap. Since no elements are actually | |
| 104 // removed from the container it is safe to read the entry being removed after | |
| 105 // pop_heap is called (but before pop_back is called). | |
| 106 std::pop_heap(pending_requests_.begin(), pending_requests_.end(), | |
| 107 CompareRequests); | |
| 108 | |
| 109 active_backoff_entry_.reset(pending_requests_.back().backoff_entry.release()); | |
| 110 active_request_.reset(pending_requests_.back().request.release()); | |
| 111 | |
| 112 pending_requests_.pop_back(); | |
| 113 | |
| 114 start_request_callback_.Run(); | |
| 115 } | |
| 116 | |
| 117 template<typename T> | |
| 118 void RequestQueue<T>::RetryRequest(const base::TimeDelta& min_backoff_delay) { | |
| 119 active_backoff_entry_->InformOfRequest(false); | |
| 120 if (active_backoff_entry_->GetTimeUntilRelease() < min_backoff_delay) { | |
| 121 active_backoff_entry_->SetCustomReleaseTime( | |
| 122 base::TimeTicks::Now() + min_backoff_delay); | |
| 123 } | |
| 124 PushImpl(active_request_.Pass(), active_backoff_entry_.Pass()); | |
| 125 } | |
| 126 | |
| 127 template<typename T> | |
| 128 typename RequestQueue<T>::iterator RequestQueue<T>::begin() { | |
| 129 return iterator(pending_requests_.begin()); | |
| 130 } | |
| 131 | |
| 132 template<typename T> | |
| 133 typename RequestQueue<T>::iterator RequestQueue<T>::end() { | |
| 134 return iterator(pending_requests_.end()); | |
| 135 } | |
| 136 | |
| 137 template<typename T> | |
| 138 void RequestQueue<T>::set_backoff_policy( | |
| 139 const net::BackoffEntry::Policy* backoff_policy) { | |
| 140 backoff_policy_ = backoff_policy; | |
| 141 } | |
| 142 | |
| 143 // static | |
| 144 template<typename T> | |
| 145 bool RequestQueue<T>::CompareRequests( | |
| 146 const Request& a, | |
| 147 const Request& b) { | |
| 148 return a.backoff_entry->GetReleaseTime() > | |
| 149 b.backoff_entry->GetReleaseTime(); | |
| 150 } | |
| 151 | |
| 152 } // namespace extensions | |
| 153 | |
| 154 #endif // CHROME_BROWSER_EXTENSIONS_UPDATER_REQUEST_QUEUE_IMPL_H_ | |
| OLD | NEW |