Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(42)

Side by Side Diff: multi_http_fetcher.h

Issue 5205002: AU: Manual proxy support (Closed) Base URL: http://git.chromium.org/git/update_engine.git@master
Patch Set: missed one fix for review Created 10 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
« no previous file with comments | « mock_http_fetcher.h ('k') | omaha_request_action_unittest.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright (c) 2010 The Chromium OS Authors. All rights reserved. 1 // Copyright (c) 2010 The Chromium OS 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 #ifndef CHROMEOS_PLATFORM_UPDATE_ENGINE_MULTI_HTTP_FETCHER_H__ 5 #ifndef CHROMEOS_PLATFORM_UPDATE_ENGINE_MULTI_HTTP_FETCHER_H__
6 #define CHROMEOS_PLATFORM_UPDATE_ENGINE_MULTI_HTTP_FETCHER_H__ 6 #define CHROMEOS_PLATFORM_UPDATE_ENGINE_MULTI_HTTP_FETCHER_H__
7 7
8 #include <deque>
8 #include <tr1/memory> 9 #include <tr1/memory>
9 #include <utility> 10 #include <utility>
10 #include <vector> 11 #include <vector>
11 12
12 #include "update_engine/http_fetcher.h" 13 #include "update_engine/http_fetcher.h"
13 #include "update_engine/utils.h" 14 #include "update_engine/utils.h"
14 15
15 // This class is a simple wrapper around an HttpFetcher. The client 16 // This class is a simple wrapper around an HttpFetcher. The client
16 // specifies a vector of byte ranges. MultiHttpFetcher will fetch bytes 17 // specifies a vector of byte ranges. MultiHttpFetcher will fetch bytes
17 // from those offsets. Pass -1 as a length to specify unlimited length. 18 // from those offsets. Pass -1 as a length to specify unlimited length.
18 // It really only would make sense for the last range specified to have 19 // It really only would make sense for the last range specified to have
19 // unlimited length. 20 // unlimited length.
20 21
21 namespace chromeos_update_engine { 22 namespace chromeos_update_engine {
22 23
23 template<typename BaseHttpFetcher> 24 template<typename BaseHttpFetcher>
24 class MultiHttpFetcher : public HttpFetcher, public HttpFetcherDelegate { 25 class MultiHttpFetcher : public HttpFetcher, public HttpFetcherDelegate {
25 public: 26 public:
26 typedef std::vector<std::pair<off_t, off_t> > RangesVect; 27 typedef std::vector<std::pair<off_t, off_t> > RangesVect;
28 typedef std::vector<std::tr1::shared_ptr<BaseHttpFetcher> > FetchersVect;
27 29
28 MultiHttpFetcher() 30 MultiHttpFetcher(ProxyResolver* proxy_resolver)
29 : sent_transfer_complete_(false), 31 : HttpFetcher(proxy_resolver),
32 sent_transfer_complete_(false),
30 pending_next_fetcher_(false), 33 pending_next_fetcher_(false),
31 current_index_(0), 34 current_index_(0),
32 bytes_received_this_fetcher_(0) {} 35 bytes_received_this_fetcher_(0) {}
33 ~MultiHttpFetcher() {} 36 ~MultiHttpFetcher() {}
34 37
35 void set_ranges(const RangesVect& ranges) { 38 void set_ranges(const RangesVect& ranges) {
36 ranges_ = ranges; 39 ranges_ = ranges;
37 fetchers_.resize(ranges_.size()); // Allocate the fetchers 40 fetchers_.resize(ranges_.size()); // Allocate the fetchers
38 for (typename std::vector<std::tr1::shared_ptr<BaseHttpFetcher> 41 for (typename std::vector<std::tr1::shared_ptr<BaseHttpFetcher>
39 >::iterator it = fetchers_.begin(), e = fetchers_.end(); 42 >::iterator it = fetchers_.begin(), e = fetchers_.end();
40 it != e; ++it) { 43 it != e; ++it) {
41 (*it) = std::tr1::shared_ptr<BaseHttpFetcher>(new BaseHttpFetcher); 44 (*it) = std::tr1::shared_ptr<BaseHttpFetcher>(
45 new BaseHttpFetcher(proxy_resolver_));
42 (*it)->set_delegate(this); 46 (*it)->set_delegate(this);
43 } 47 }
48 LOG(INFO) << "done w/ list";
44 } 49 }
45 50
46 void SetOffset(off_t offset) {} // for now, doesn't support this 51 void SetOffset(off_t offset) {} // for now, doesn't support this
47 52
48 // Begins the transfer to the specified URL. 53 // Begins the transfer to the specified URL.
49 void BeginTransfer(const std::string& url) { 54 void BeginTransfer(const std::string& url) {
50 url_ = url; 55 url_ = url;
51 if (ranges_.empty()) { 56 if (ranges_.empty()) {
52 if (delegate_) 57 if (delegate_)
53 delegate_->TransferComplete(this, true); 58 delegate_->TransferComplete(this, true);
(...skipping 73 matching lines...) Expand 10 before | Expand all | Expand 10 after
127 (*it)->SetConnectionAsExpensive(is_expensive); 132 (*it)->SetConnectionAsExpensive(is_expensive);
128 } 133 }
129 } 134 }
130 void SetBuildType(bool is_official) { 135 void SetBuildType(bool is_official) {
131 for (typename std::vector<std::tr1::shared_ptr<BaseHttpFetcher> >::iterator 136 for (typename std::vector<std::tr1::shared_ptr<BaseHttpFetcher> >::iterator
132 it = fetchers_.begin(), 137 it = fetchers_.begin(),
133 e = fetchers_.end(); it != e; ++it) { 138 e = fetchers_.end(); it != e; ++it) {
134 (*it)->SetBuildType(is_official); 139 (*it)->SetBuildType(is_official);
135 } 140 }
136 } 141 }
142
143 virtual void SetProxies(const std::deque<std::string>& proxies) {
144 for (typename FetchersVect::iterator it = fetchers_.begin(),
145 e = fetchers_.end(); it != e; ++it) {
146 (*it)->SetProxies(proxies);
147 }
148 }
137 149
138 private: 150 private:
139 void SendTransferComplete(HttpFetcher* fetcher, bool successful) { 151 void SendTransferComplete(HttpFetcher* fetcher, bool successful) {
140 if (sent_transfer_complete_) 152 if (sent_transfer_complete_)
141 return; 153 return;
142 LOG(INFO) << "Sending transfer complete"; 154 LOG(INFO) << "Sending transfer complete";
143 sent_transfer_complete_ = true; 155 sent_transfer_complete_ = true;
144 http_response_code_ = fetcher->http_response_code(); 156 http_response_code_ = fetcher->http_response_code();
145 if (delegate_) 157 if (delegate_)
146 delegate_->TransferComplete(this, successful); 158 delegate_->TransferComplete(this, successful);
(...skipping 66 matching lines...) Expand 10 before | Expand all | Expand 10 after
213 } 225 }
214 226
215 // If true, do not send any more data or TransferComplete to the delegate. 227 // If true, do not send any more data or TransferComplete to the delegate.
216 bool sent_transfer_complete_; 228 bool sent_transfer_complete_;
217 229
218 // If true, the next fetcher needs to be started when TransferTerminated is 230 // If true, the next fetcher needs to be started when TransferTerminated is
219 // received from the current fetcher. 231 // received from the current fetcher.
220 bool pending_next_fetcher_; 232 bool pending_next_fetcher_;
221 233
222 RangesVect ranges_; 234 RangesVect ranges_;
223 std::vector<std::tr1::shared_ptr<BaseHttpFetcher> > fetchers_; 235 FetchersVect fetchers_;
224 236
225 RangesVect::size_type current_index_; // index into ranges_, fetchers_ 237 RangesVect::size_type current_index_; // index into ranges_, fetchers_
226 off_t bytes_received_this_fetcher_; 238 off_t bytes_received_this_fetcher_;
227 239
228 DISALLOW_COPY_AND_ASSIGN(MultiHttpFetcher); 240 DISALLOW_COPY_AND_ASSIGN(MultiHttpFetcher);
229 }; 241 };
230 242
231 } // namespace chromeos_update_engine 243 } // namespace chromeos_update_engine
232 244
233 #endif // CHROMEOS_PLATFORM_UPDATE_ENGINE_MULTI_HTTP_FETCHER_H__ 245 #endif // CHROMEOS_PLATFORM_UPDATE_ENGINE_MULTI_HTTP_FETCHER_H__
OLDNEW
« no previous file with comments | « mock_http_fetcher.h ('k') | omaha_request_action_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698