OLD | NEW |
---|---|
(Empty) | |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome/browser/chromeos/customization_wallpaper_downloader.h" | |
6 | |
7 #include "base/file_util.h" | |
8 #include "chrome/browser/chromeos/customization_document.h" | |
9 #include "content/public/browser/browser_thread.h" | |
10 #include "net/base/load_flags.h" | |
11 #include "net/url_request/url_request_context_getter.h" | |
12 #include "url/gurl.h" | |
13 | |
14 namespace { | |
15 // This is temporary file suffix (for downloading or resizing). | |
16 const char kTemporarySuffix[] = ".temp"; | |
17 | |
18 // Sleep between wallpaper retries (used multiplied by retry number). | |
19 const unsigned kRetrySleepSeconds = 10; | |
20 | |
21 // Do not increase retry sleep beyond this number of attempts. | |
22 const unsigned kRetriesInfinite = 30; | |
23 } | |
24 | |
25 namespace chromeos { | |
26 | |
27 CustomizationWallpaperDownloader::CustomizationWallpaperDownloader( | |
28 net::URLRequestContextGetter* url_context_getter, | |
29 const GURL& wallpaper_url, | |
30 const base::FilePath& wallpaper_dir, | |
31 const base::FilePath& wallpaper_downloaded_file) | |
32 : url_context_getter_(url_context_getter), | |
33 wallpaper_url_(wallpaper_url), | |
34 wallpaper_dir_(wallpaper_dir), | |
35 wallpaper_downloaded_file_(wallpaper_downloaded_file), | |
36 wallpaper_temporary_file_(wallpaper_downloaded_file.value() + | |
37 kTemporarySuffix), | |
38 retries_(0), | |
39 weak_factory_(this) { | |
40 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); | |
41 } | |
42 | |
43 CustomizationWallpaperDownloader::~CustomizationWallpaperDownloader() { | |
44 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); | |
45 } | |
46 | |
47 void CustomizationWallpaperDownloader::StartRequest() { | |
48 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); | |
49 DCHECK(wallpaper_url_.is_valid()); | |
50 ++retries_; | |
51 | |
52 url_fetcher_.reset( | |
53 net::URLFetcher::Create(wallpaper_url_, net::URLFetcher::GET, this)); | |
54 url_fetcher_->SetRequestContext(url_context_getter_); | |
55 url_fetcher_->SetLoadFlags(net::LOAD_BYPASS_CACHE | net::LOAD_DISABLE_CACHE | | |
56 net::LOAD_DO_NOT_SAVE_COOKIES | | |
57 net::LOAD_DO_NOT_SEND_COOKIES | | |
58 net::LOAD_DO_NOT_SEND_AUTH_DATA); | |
59 base::SequencedWorkerPool* blocking_pool = | |
60 content::BrowserThread::GetBlockingPool(); | |
61 url_fetcher_->SaveResponseToFileAtPath( | |
62 wallpaper_temporary_file_, | |
63 blocking_pool->GetSequencedTaskRunner(blocking_pool->GetSequenceToken())); | |
64 url_fetcher_->Start(); | |
Dmitry Polukhin
2014/04/11 23:17:17
Perhaps you need to check network condition to don
Alexander Alekseev
2014/04/12 01:42:30
I can use DelayNetworkCall here, but I think it ad
| |
65 } | |
66 | |
67 void CustomizationWallpaperDownloader::Retry() { | |
Dmitry Polukhin
2014/04/11 23:17:17
We need to give up after several retries to don't
Alexander Alekseev
2014/04/12 01:42:30
If file doesn't exist, it usually means "temporary
Dmitry Polukhin
2014/04/14 20:26:37
Because you don't remember throttling info between
Alexander Alekseev
2014/04/15 01:57:16
The idea was to increase delay fast enough to make
| |
68 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); | |
69 const unsigned delay_multiplier = | |
70 (retries_ > kRetriesInfinite ? kRetriesInfinite : retries_); | |
71 const base::TimeDelta delay = | |
72 base::TimeDelta::FromSeconds(kRetrySleepSeconds) * delay_multiplier; | |
73 VLOG(1) << "Schedule Customized Wallpaper download in " << delay.InSecondsF() | |
74 << " seconds."; | |
75 request_scheduled_.Start( | |
76 FROM_HERE, delay, this, &CustomizationWallpaperDownloader::StartRequest); | |
77 } | |
78 | |
79 void CustomizationWallpaperDownloader::Start() { | |
80 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); | |
81 scoped_ptr<bool> success(new bool(false)); | |
82 | |
83 base::Closure mkdir_closure = | |
84 base::Bind(&CustomizationWallpaperDownloader::CreateWallpaperDirectory, | |
85 wallpaper_dir_, | |
86 base::Unretained(success.get())); | |
87 base::Closure on_created_closure = | |
88 base::Bind(&CustomizationWallpaperDownloader::OnWallpaperDirectoryCreated, | |
89 weak_factory_.GetWeakPtr(), | |
90 base::Passed(success.Pass())); | |
91 if (!content::BrowserThread::PostBlockingPoolTaskAndReply( | |
92 FROM_HERE, mkdir_closure, on_created_closure)) { | |
93 LOG(WARNING) << "Failed to start Customized Wallpaper download."; | |
94 } | |
95 } | |
96 | |
97 // static | |
98 void CustomizationWallpaperDownloader::CreateWallpaperDirectory( | |
99 const base::FilePath& wallpaper_dir, | |
100 bool* success) { | |
101 DCHECK(content::BrowserThread::GetBlockingPool()->RunsTasksOnCurrentThread()); | |
102 DCHECK(success); | |
103 | |
104 *success = CreateDirectoryAndGetError(wallpaper_dir, NULL); | |
105 } | |
106 | |
107 void CustomizationWallpaperDownloader::OnWallpaperDirectoryCreated( | |
108 scoped_ptr<bool> success) { | |
109 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); | |
110 if (*success) | |
111 StartRequest(); | |
112 } | |
113 | |
114 void CustomizationWallpaperDownloader::OnURLFetchComplete( | |
115 const net::URLFetcher* source) { | |
116 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); | |
117 DCHECK_EQ(url_fetcher_.get(), source); | |
118 | |
119 const net::URLRequestStatus status = source->GetStatus(); | |
120 const int response_code = source->GetResponseCode(); | |
121 | |
122 const bool server_error = | |
123 !status.is_success() || (response_code >= 500 && response_code < 600); | |
124 | |
125 VLOG(1) << "CustomizationWallpaperDownloader::OnURLFetchComplete(): status=" | |
126 << status.status(); | |
127 | |
128 if (server_error) { | |
129 url_fetcher_.reset(); | |
130 Retry(); | |
131 return; | |
132 } | |
133 | |
134 base::FilePath response_path; | |
135 url_fetcher_->GetResponseAsFilePath(true, &response_path); | |
136 url_fetcher_.reset(); | |
137 | |
138 scoped_ptr<bool> success(new bool(false)); | |
139 | |
140 base::Closure rename_closure = | |
141 base::Bind(&CustomizationWallpaperDownloader::RenameTemporaryFile, | |
142 response_path, | |
143 wallpaper_downloaded_file_, | |
144 base::Unretained(success.get())); | |
145 base::Closure on_rename_closure = | |
146 base::Bind(&CustomizationWallpaperDownloader::OnTemporaryFileRenamed, | |
147 weak_factory_.GetWeakPtr(), | |
148 base::Passed(success.Pass())); | |
149 if (!content::BrowserThread::PostBlockingPoolTaskAndReply( | |
150 FROM_HERE, rename_closure, on_rename_closure)) { | |
151 LOG(WARNING) | |
152 << "Failed to start Customized Wallpaper Rename DownloadedFile."; | |
153 // Destroy self. | |
154 ServicesCustomizationDocument::GetInstance()->DestroyWallpaperDownloader(); | |
Dmitry Polukhin
2014/04/11 23:17:17
I think you can just delete self if ServicesCustom
Alexander Alekseev
2014/04/12 01:42:30
I think this would lead to problems in tests. Our
Dmitry Polukhin
2014/04/14 20:26:37
Singletons shouldn't be a problem in test if you u
Alexander Alekseev
2014/04/15 01:57:16
I've dropped DestroyWallpaperDownloader method.
| |
155 } | |
156 } | |
157 | |
158 void CustomizationWallpaperDownloader::OnTemporaryFileRenamed( | |
159 scoped_ptr<bool> success) { | |
160 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); | |
161 if (!success) { | |
162 // Destroy self. | |
163 ServicesCustomizationDocument::GetInstance()->DestroyWallpaperDownloader(); | |
164 return; | |
165 } | |
166 ServicesCustomizationDocument::OnCustomizedWallpaperDownloaded( | |
167 wallpaper_url_); | |
168 } | |
169 | |
170 // static | |
171 void CustomizationWallpaperDownloader::RenameTemporaryFile( | |
172 const base::FilePath& from, | |
173 const base::FilePath& to, | |
174 bool* success) { | |
175 DCHECK(content::BrowserThread::GetBlockingPool()->RunsTasksOnCurrentThread()); | |
176 DCHECK(success); | |
177 | |
178 base::File::Error error; | |
179 if (ReplaceFile(from, to, &error)) { | |
180 *success = true; | |
181 } else { | |
182 LOG(WARNING) | |
183 << "Failed to rename temporary file of Customized Wallpaper. error=" | |
184 << error; | |
185 *success = false; | |
186 } | |
187 } | |
188 | |
189 } // namespace chromeos | |
OLD | NEW |