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

Side by Side Diff: components/image_fetcher/image_data_fetcher.cc

Issue 2073753002: Simplify the ImageDataFetcher code. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: '"' Created 4 years, 6 months 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
« no previous file with comments | « components/image_fetcher/image_data_fetcher.h ('k') | no next file » | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright 2016 The Chromium Authors. All rights reserved. 1 // Copyright 2016 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/image_fetcher/image_data_fetcher.h" 5 #include "components/image_fetcher/image_data_fetcher.h"
6 6
7 #include "net/base/load_flags.h" 7 #include "net/base/load_flags.h"
8 #include "net/url_request/url_fetcher.h" 8 #include "net/url_request/url_fetcher.h"
9 #include "net/url_request/url_fetcher_delegate.h"
10 #include "net/url_request/url_request.h" 9 #include "net/url_request/url_request.h"
11 #include "net/url_request/url_request_context_getter.h" 10 #include "net/url_request/url_request_context_getter.h"
12 #include "net/url_request/url_request_status.h" 11 #include "net/url_request/url_request_status.h"
13 #include "url/gurl.h" 12 #include "url/gurl.h"
14 13
15 namespace image_fetcher { 14 namespace image_fetcher {
16 15
17 // An active image URL fetcher request. The class contains any related request 16 // An active image URL fetcher request. The struct contains the related requests
18 // state and logic for handling a single image url request. 17 // state.
19 class ImageDataFetcher::ImageDataFetcherRequest 18 struct ImageDataFetcher::ImageDataFetcherRequest {
20 : public net::URLFetcherDelegate { 19 ImageDataFetcherRequest(const ImageDataFetcherCallback& callback,
21 public: 20 std::unique_ptr<net::URLFetcher> url_fetcher);
22 ImageDataFetcherRequest(const GURL& url, 21 ~ImageDataFetcherRequest() {}
23 const ImageDataFetcherCallback& callback,
24 ImageDataFetcher* image_url_fetcher);
25 ~ImageDataFetcherRequest() override {}
26
27 // Sends the URL requests.
28 void Start(net::URLRequestContextGetter* request_context);
29
30 private:
31 // Method inherited from URLFetcherDelegate
32 void OnURLFetchComplete(const net::URLFetcher* source) override;
33
34 // The URL of the image to fetch.
35 const GURL url_;
36 22
37 // The callback to run after the image data was fetched. The callback will 23 // The callback to run after the image data was fetched. The callback will
38 // be run even if the image data could not be fetched successfully. 24 // be run even if the image data could not be fetched successfully.
39 ImageDataFetcherCallback callback_; 25 ImageDataFetcherCallback callback;
40 26
41 // The ImageDataFetcher that owns the ImageDataFetcherRequest. 27 std::unique_ptr<net::URLFetcher> url_fetcher;
42 ImageDataFetcher* image_url_fetcher_;
43
44 std::unique_ptr<net::URLFetcher> url_fetcher_;
45
46 DISALLOW_COPY_AND_ASSIGN(ImageDataFetcherRequest);
47 }; 28 };
48 29
49 ImageDataFetcher::ImageDataFetcherRequest::ImageDataFetcherRequest( 30 ImageDataFetcher::ImageDataFetcherRequest::ImageDataFetcherRequest(
50 const GURL& url,
51 const ImageDataFetcherCallback& callback, 31 const ImageDataFetcherCallback& callback,
52 ImageDataFetcher* image_url_fetcher) 32 std::unique_ptr<net::URLFetcher> url_fetcher)
53 : url_(url), 33 : callback(callback),
54 callback_(callback), 34 url_fetcher(std::move(url_fetcher)) {
Marc Treib 2016/06/16 11:52:09 optional: You could just inline this above.
55 image_url_fetcher_(image_url_fetcher) {
56 }
57
58 void ImageDataFetcher::ImageDataFetcherRequest::Start(
59 net::URLRequestContextGetter* request_context) {
60 DCHECK(!url_fetcher_);
61
62 url_fetcher_ = net::URLFetcher::Create(url_, net::URLFetcher::GET, this);
63 url_fetcher_->SetRequestContext(request_context);
64 url_fetcher_->Start();
65 }
66
67 void ImageDataFetcher::ImageDataFetcherRequest::OnURLFetchComplete(
68 const net::URLFetcher* source) {
69 // An empty string is passed to the callback in case on an unsuccessfull URL
70 // request.
71 std::string image_data;
72 if (source->GetStatus().status() == net::URLRequestStatus::SUCCESS) {
73 source->GetResponseAsString(&image_data);
74 }
75 callback_.Run(image_data);
76
77 // Remove the completed ImageDataFetcherRequest from the internal request
78 // queue. This must be last in the method.
79 image_url_fetcher_->RemoveImageDataFetcherRequest(url_);
80 } 35 }
81 36
82 ImageDataFetcher::ImageDataFetcher( 37 ImageDataFetcher::ImageDataFetcher(
83 net::URLRequestContextGetter* url_request_context_getter) 38 net::URLRequestContextGetter* url_request_context_getter)
84 : url_request_context_getter_(url_request_context_getter) {} 39 : url_request_context_getter_(url_request_context_getter) {}
85 40
86 ImageDataFetcher::~ImageDataFetcher() {} 41 ImageDataFetcher::~ImageDataFetcher() {}
87 42
88 void ImageDataFetcher::FetchImageData( 43 void ImageDataFetcher::FetchImageData(
89 const GURL& url, const ImageDataFetcherCallback& callback) { 44 const GURL& url, const ImageDataFetcherCallback& callback) {
45
46 std::unique_ptr<net::URLFetcher> url_fetcher =
47 net::URLFetcher::Create(url, net::URLFetcher::GET, this);
48
90 std::unique_ptr<ImageDataFetcherRequest> request( 49 std::unique_ptr<ImageDataFetcherRequest> request(
91 new ImageDataFetcherRequest(url, callback, this)); 50 new ImageDataFetcherRequest(callback, std::move(url_fetcher)));
92 request->Start(url_request_context_getter_.get()); 51 request->url_fetcher->SetRequestContext(url_request_context_getter_.get());
93 pending_requests_[url] = std::move(request); 52 request->url_fetcher->Start();
53
54 pending_requests_[request->url_fetcher.get()] = std::move(request);
94 } 55 }
95 56
96 void ImageDataFetcher::RemoveImageDataFetcherRequest(const GURL& image_url) { 57 void ImageDataFetcher::OnURLFetchComplete(const net::URLFetcher* source) {
97 size_t count = pending_requests_.erase(image_url); 58 auto request_iter = pending_requests_.find(source);
98 DCHECK(count); 59 DCHECK(request_iter != pending_requests_.end());
60
61 std::string image_data;
62 if (source->GetStatus().status() == net::URLRequestStatus::SUCCESS) {
63 source->GetResponseAsString(&image_data);
64 }
65 request_iter->second->callback.Run(image_data);
66
67 // Remove the finished request.
68 pending_requests_.erase(request_iter);
99 } 69 }
100 70
101 } // namespace image_fetcher 71 } // namespace image_fetcher
OLDNEW
« no previous file with comments | « components/image_fetcher/image_data_fetcher.h ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698