OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "chrome/browser/predictors/resource_prefetcher.h" | 5 #include "chrome/browser/predictors/resource_prefetcher.h" |
6 | 6 |
7 #include <iterator> | 7 #include <iterator> |
8 #include <utility> | 8 #include <utility> |
9 | 9 |
10 #include "base/stl_util.h" | |
11 #include "content/public/browser/browser_thread.h" | 10 #include "content/public/browser/browser_thread.h" |
12 #include "net/base/io_buffer.h" | 11 #include "net/base/io_buffer.h" |
13 #include "net/base/load_flags.h" | 12 #include "net/base/load_flags.h" |
14 #include "net/base/request_priority.h" | 13 #include "net/base/request_priority.h" |
15 #include "net/url_request/url_request_context.h" | 14 #include "net/url_request/url_request_context.h" |
16 #include "url/origin.h" | 15 #include "url/origin.h" |
17 | 16 |
18 namespace { | 17 namespace { |
19 | 18 |
20 // The size of the buffer used to read the resource. | 19 // The size of the buffer used to read the resource. |
(...skipping 27 matching lines...) Expand all Loading... |
48 navigation_id_(navigation_id), | 47 navigation_id_(navigation_id), |
49 key_type_(key_type), | 48 key_type_(key_type), |
50 request_vector_(std::move(requests)) { | 49 request_vector_(std::move(requests)) { |
51 DCHECK_CURRENTLY_ON(content::BrowserThread::IO); | 50 DCHECK_CURRENTLY_ON(content::BrowserThread::IO); |
52 DCHECK(request_vector_.get()); | 51 DCHECK(request_vector_.get()); |
53 | 52 |
54 std::copy(request_vector_->begin(), request_vector_->end(), | 53 std::copy(request_vector_->begin(), request_vector_->end(), |
55 std::back_inserter(request_queue_)); | 54 std::back_inserter(request_queue_)); |
56 } | 55 } |
57 | 56 |
58 ResourcePrefetcher::~ResourcePrefetcher() { | 57 ResourcePrefetcher::~ResourcePrefetcher() {} |
59 // Delete any pending net::URLRequests. | |
60 base::STLDeleteContainerPairFirstPointers(inflight_requests_.begin(), | |
61 inflight_requests_.end()); | |
62 } | |
63 | 58 |
64 void ResourcePrefetcher::Start() { | 59 void ResourcePrefetcher::Start() { |
65 DCHECK(thread_checker_.CalledOnValidThread()); | 60 DCHECK(thread_checker_.CalledOnValidThread()); |
66 | 61 |
67 CHECK_EQ(state_, INITIALIZED); | 62 CHECK_EQ(state_, INITIALIZED); |
68 state_ = RUNNING; | 63 state_ = RUNNING; |
69 | 64 |
70 TryToLaunchPrefetchRequests(); | 65 TryToLaunchPrefetchRequests(); |
71 } | 66 } |
72 | 67 |
(...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
106 } | 101 } |
107 request_available = request_it != request_queue_.end(); | 102 request_available = request_it != request_queue_.end(); |
108 | 103 |
109 if (request_available) { | 104 if (request_available) { |
110 SendRequest(*request_it); | 105 SendRequest(*request_it); |
111 request_queue_.erase(request_it); | 106 request_queue_.erase(request_it); |
112 } | 107 } |
113 } | 108 } |
114 } | 109 } |
115 | 110 |
116 // If the inflight_requests_ is empty, we cant launch any more. Finish. | 111 // If the inflight_requests_ is empty, we can't launch any more. Finish. |
117 if (inflight_requests_.empty()) { | 112 if (inflight_requests_.empty()) { |
118 CHECK(host_inflight_counts_.empty()); | 113 CHECK(host_inflight_counts_.empty()); |
119 CHECK(request_queue_.empty() || state_ == STOPPED); | 114 CHECK(request_queue_.empty() || state_ == STOPPED); |
120 | 115 |
121 state_ = FINISHED; | 116 state_ = FINISHED; |
122 delegate_->ResourcePrefetcherFinished(this, request_vector_.release()); | 117 delegate_->ResourcePrefetcherFinished(this, request_vector_.release()); |
123 } | 118 } |
124 } | 119 } |
125 | 120 |
126 void ResourcePrefetcher::SendRequest(Request* request) { | 121 void ResourcePrefetcher::SendRequest(Request* request) { |
127 request->prefetch_status = Request::PREFETCH_STATUS_STARTED; | 122 request->prefetch_status = Request::PREFETCH_STATUS_STARTED; |
128 | 123 |
129 net::URLRequest* url_request = | 124 std::unique_ptr<net::URLRequest> url_request_ptr = |
130 delegate_->GetURLRequestContext()->CreateRequest( | 125 delegate_->GetURLRequestContext()->CreateRequest(request->resource_url, |
131 request->resource_url, net::LOW, this).release(); | 126 net::LOW, this); |
| 127 net::URLRequest* url_request = url_request_ptr.get(); |
132 | 128 |
133 inflight_requests_[url_request] = request; | 129 inflight_requests_[url_request] = {std::move(url_request_ptr), request}; |
134 host_inflight_counts_[url_request->original_url().host()] += 1; | 130 host_inflight_counts_[url_request->original_url().host()] += 1; |
135 | 131 |
136 url_request->set_method("GET"); | 132 url_request->set_method("GET"); |
137 url_request->set_first_party_for_cookies(navigation_id_.main_frame_url); | 133 url_request->set_first_party_for_cookies(navigation_id_.main_frame_url); |
138 url_request->set_initiator(url::Origin(navigation_id_.main_frame_url)); | 134 url_request->set_initiator(url::Origin(navigation_id_.main_frame_url)); |
139 url_request->SetReferrer(navigation_id_.main_frame_url.spec()); | 135 url_request->SetReferrer(navigation_id_.main_frame_url.spec()); |
140 url_request->SetLoadFlags(url_request->load_flags() | net::LOAD_PREFETCH); | 136 url_request->SetLoadFlags(url_request->load_flags() | net::LOAD_PREFETCH); |
141 StartURLRequest(url_request); | 137 StartURLRequest(url_request); |
142 } | 138 } |
143 | 139 |
144 void ResourcePrefetcher::StartURLRequest(net::URLRequest* request) { | 140 void ResourcePrefetcher::StartURLRequest(net::URLRequest* request) { |
145 request->Start(); | 141 request->Start(); |
146 } | 142 } |
147 | 143 |
148 void ResourcePrefetcher::FinishRequest(net::URLRequest* request, | 144 void ResourcePrefetcher::FinishRequest(net::URLRequest* request, |
149 Request::PrefetchStatus status) { | 145 Request::PrefetchStatus status) { |
150 std::map<net::URLRequest*, Request*>::iterator request_it = | 146 auto request_it = inflight_requests_.find(request); |
151 inflight_requests_.find(request); | |
152 CHECK(request_it != inflight_requests_.end()); | 147 CHECK(request_it != inflight_requests_.end()); |
153 | 148 |
154 const std::string host = request->original_url().host(); | 149 const std::string host = request->original_url().host(); |
155 std::map<std::string, size_t>::iterator host_it = host_inflight_counts_.find( | 150 std::map<std::string, size_t>::iterator host_it = host_inflight_counts_.find( |
156 host); | 151 host); |
157 CHECK_GT(host_it->second, 0U); | 152 CHECK_GT(host_it->second, 0U); |
158 host_it->second -= 1; | 153 host_it->second -= 1; |
159 if (host_it->second == 0) | 154 if (host_it->second == 0) |
160 host_inflight_counts_.erase(host); | 155 host_inflight_counts_.erase(host); |
161 | 156 |
162 request_it->second->prefetch_status = status; | 157 request_it->second.second->prefetch_status = status; |
163 inflight_requests_.erase(request_it); | 158 inflight_requests_.erase(request_it); |
164 | 159 |
165 delete request; | |
166 | |
167 TryToLaunchPrefetchRequests(); | 160 TryToLaunchPrefetchRequests(); |
168 } | 161 } |
169 | 162 |
170 void ResourcePrefetcher::ReadFullResponse(net::URLRequest* request) { | 163 void ResourcePrefetcher::ReadFullResponse(net::URLRequest* request) { |
171 bool status = true; | 164 bool status = true; |
172 while (status) { | 165 while (status) { |
173 int bytes_read = 0; | 166 int bytes_read = 0; |
174 scoped_refptr<net::IOBuffer> buffer(new net::IOBuffer( | 167 scoped_refptr<net::IOBuffer> buffer(new net::IOBuffer( |
175 kResourceBufferSizeBytes)); | 168 kResourceBufferSizeBytes)); |
176 status = request->Read(buffer.get(), kResourceBufferSizeBytes, &bytes_read); | 169 status = request->Read(buffer.get(), kResourceBufferSizeBytes, &bytes_read); |
(...skipping 59 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
236 if (!request->status().is_success()) { | 229 if (!request->status().is_success()) { |
237 FinishRequest(request, Request::PREFETCH_STATUS_FAILED); | 230 FinishRequest(request, Request::PREFETCH_STATUS_FAILED); |
238 return; | 231 return; |
239 } | 232 } |
240 | 233 |
241 if (ShouldContinueReadingRequest(request, bytes_read)) | 234 if (ShouldContinueReadingRequest(request, bytes_read)) |
242 ReadFullResponse(request); | 235 ReadFullResponse(request); |
243 } | 236 } |
244 | 237 |
245 } // namespace predictors | 238 } // namespace predictors |
OLD | NEW |