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

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

Issue 2761303002: Move common ImageFetcher component files to core/ (Closed)
Patch Set: Rebased. Created 3 years, 9 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
OLDNEW
(Empty)
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
3 // found in the LICENSE file.
4
5 #include "components/image_fetcher/image_data_fetcher.h"
6
7 #include "net/base/load_flags.h"
8 #include "net/http/http_response_headers.h"
9 #include "net/http/http_status_code.h"
10 #include "net/url_request/url_fetcher.h"
11 #include "net/url_request/url_request_context_getter.h"
12 #include "net/url_request/url_request_status.h"
13 #include "url/gurl.h"
14
15 using data_use_measurement::DataUseUserData;
16
17 namespace image_fetcher {
18
19 // An active image URL fetcher request. The struct contains the related requests
20 // state.
21 struct ImageDataFetcher::ImageDataFetcherRequest {
22 ImageDataFetcherRequest(const ImageDataFetcherCallback& callback,
23 std::unique_ptr<net::URLFetcher> url_fetcher)
24 : callback(callback),
25 url_fetcher(std::move(url_fetcher)) {}
26
27 ~ImageDataFetcherRequest() {}
28
29 // The callback to run after the image data was fetched. The callback will
30 // be run even if the image data could not be fetched successfully.
31 ImageDataFetcherCallback callback;
32
33 std::unique_ptr<net::URLFetcher> url_fetcher;
34 };
35
36 ImageDataFetcher::ImageDataFetcher(
37 net::URLRequestContextGetter* url_request_context_getter)
38 : url_request_context_getter_(url_request_context_getter),
39 data_use_service_name_(DataUseUserData::IMAGE_FETCHER_UNTAGGED),
40 next_url_fetcher_id_(0) {}
41
42 ImageDataFetcher::~ImageDataFetcher() {}
43
44 void ImageDataFetcher::SetDataUseServiceName(
45 DataUseServiceName data_use_service_name) {
46 data_use_service_name_ = data_use_service_name;
47 }
48
49 void ImageDataFetcher::FetchImageData(
50 const GURL& image_url,
51 const ImageDataFetcherCallback& callback) {
52 FetchImageData(
53 image_url, callback, /*referrer=*/std::string(),
54 net::URLRequest::CLEAR_REFERRER_ON_TRANSITION_FROM_SECURE_TO_INSECURE);
55 }
56
57 void ImageDataFetcher::FetchImageData(
58 const GURL& image_url,
59 const ImageDataFetcherCallback& callback,
60 const std::string& referrer,
61 net::URLRequest::ReferrerPolicy referrer_policy) {
62 std::unique_ptr<net::URLFetcher> url_fetcher = net::URLFetcher::Create(
63 next_url_fetcher_id_++, image_url, net::URLFetcher::GET, this);
64
65 DataUseUserData::AttachToFetcher(url_fetcher.get(), data_use_service_name_);
66
67 std::unique_ptr<ImageDataFetcherRequest> request(
68 new ImageDataFetcherRequest(callback, std::move(url_fetcher)));
69 request->url_fetcher->SetRequestContext(url_request_context_getter_.get());
70 request->url_fetcher->SetReferrer(referrer);
71 request->url_fetcher->SetReferrerPolicy(referrer_policy);
72 request->url_fetcher->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES |
73 net::LOAD_DO_NOT_SAVE_COOKIES |
74 net::LOAD_DO_NOT_SEND_AUTH_DATA);
75 request->url_fetcher->Start();
76
77 pending_requests_[request->url_fetcher.get()] = std::move(request);
78 }
79
80 void ImageDataFetcher::OnURLFetchComplete(const net::URLFetcher* source) {
81 auto request_iter = pending_requests_.find(source);
82 DCHECK(request_iter != pending_requests_.end());
83
84 bool success = source->GetStatus().status() == net::URLRequestStatus::SUCCESS;
85
86 RequestMetadata metadata;
87 if (success && source->GetResponseHeaders()) {
88 source->GetResponseHeaders()->GetMimeType(&metadata.mime_type);
89 metadata.http_response_code = source->GetResponseHeaders()->response_code();
90 success &= (metadata.http_response_code == net::HTTP_OK);
91 }
92 metadata.from_http_cache = source->WasCached();
93
94 std::string image_data;
95 if (success) {
96 source->GetResponseAsString(&image_data);
97 }
98 request_iter->second->callback.Run(image_data, metadata);
99
100 // Remove the finished request.
101 pending_requests_.erase(request_iter);
102 }
103
104 } // namespace image_fetcher
OLDNEW
« no previous file with comments | « components/image_fetcher/image_data_fetcher.h ('k') | components/image_fetcher/image_data_fetcher_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698