OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 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/gdata/gdata_operation_registry.h" | |
6 | |
7 #include "base/string_number_conversions.h" | |
8 #include "content/public/browser/browser_thread.h" | |
9 | |
10 using content::BrowserThread; | |
11 | |
12 namespace { | |
13 | |
14 const int64 kNotificationFrequencyInMilliseconds = 1000; | |
15 | |
16 } // namespace | |
17 | |
18 namespace gdata { | |
19 | |
20 // static | |
21 std::string GDataOperationRegistry::OperationTypeToString(OperationType type) { | |
22 switch (type) { | |
23 case OPERATION_UPLOAD: return "upload"; | |
24 case OPERATION_DOWNLOAD: return "download"; | |
25 case OPERATION_OTHER: return "other"; | |
26 } | |
27 NOTREACHED(); | |
28 return "unknown_transfer_state"; | |
29 } | |
30 | |
31 // static | |
32 std::string GDataOperationRegistry::OperationTransferStateToString( | |
33 OperationTransferState state) { | |
34 switch (state) { | |
35 case OPERATION_NOT_STARTED: return "not_started"; | |
36 case OPERATION_STARTED: return "started"; | |
37 case OPERATION_IN_PROGRESS: return "in_progress"; | |
38 case OPERATION_COMPLETED: return "completed"; | |
39 case OPERATION_FAILED: return "failed"; | |
40 // Suspended state is opaque to users and looks as same as "in_progress". | |
41 case OPERATION_SUSPENDED: return "in_progress"; | |
42 } | |
43 NOTREACHED(); | |
44 return "unknown_transfer_state"; | |
45 } | |
46 | |
47 GDataOperationRegistry::ProgressStatus::ProgressStatus(OperationType type, | |
48 const FilePath& path) | |
49 : operation_id(-1), | |
50 operation_type(type), | |
51 file_path(path), | |
52 transfer_state(OPERATION_NOT_STARTED), | |
53 progress_current(0), | |
54 progress_total(-1) { | |
55 } | |
56 | |
57 std::string GDataOperationRegistry::ProgressStatus::DebugString() const { | |
58 std::string str; | |
59 str += "id="; | |
60 str += base::IntToString(operation_id); | |
61 str += " type="; | |
62 str += OperationTypeToString(operation_type); | |
63 str += " path="; | |
64 str += file_path.AsUTF8Unsafe(); | |
65 str += " state="; | |
66 str += OperationTransferStateToString(transfer_state); | |
67 str += " progress="; | |
68 str += base::Int64ToString(progress_current); | |
69 str += "/"; | |
70 str += base::Int64ToString(progress_total); | |
71 return str; | |
72 } | |
73 | |
74 GDataOperationRegistry::Operation::Operation(GDataOperationRegistry* registry) | |
75 : registry_(registry), | |
76 progress_status_(GDataOperationRegistry::OPERATION_OTHER, FilePath()) { | |
77 } | |
78 | |
79 GDataOperationRegistry::Operation::Operation(GDataOperationRegistry* registry, | |
80 OperationType type, | |
81 const FilePath& path) | |
82 : registry_(registry), | |
83 progress_status_(type, path) { | |
84 } | |
85 | |
86 GDataOperationRegistry::Operation::~Operation() { | |
87 DCHECK(progress_status_.transfer_state == OPERATION_COMPLETED || | |
88 progress_status_.transfer_state == OPERATION_SUSPENDED || | |
89 progress_status_.transfer_state == OPERATION_FAILED); | |
90 } | |
91 | |
92 void GDataOperationRegistry::Operation::Cancel() { | |
93 DoCancel(); | |
94 NotifyFinish(OPERATION_FAILED); | |
95 } | |
96 | |
97 void GDataOperationRegistry::Operation::NotifyStart() { | |
98 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
99 // Some operations may be restarted. Report only the first "start". | |
100 if (progress_status_.transfer_state == OPERATION_NOT_STARTED) { | |
101 progress_status_.transfer_state = OPERATION_STARTED; | |
102 progress_status_.start_time = base::Time::Now(); | |
103 registry_->OnOperationStart(this, &progress_status_.operation_id); | |
104 } | |
105 } | |
106 | |
107 void GDataOperationRegistry::Operation::NotifyProgress( | |
108 int64 current, int64 total) { | |
109 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
110 DCHECK(progress_status_.transfer_state >= OPERATION_STARTED); | |
111 progress_status_.transfer_state = OPERATION_IN_PROGRESS; | |
112 progress_status_.progress_current = current; | |
113 progress_status_.progress_total = total; | |
114 registry_->OnOperationProgress(progress_status().operation_id); | |
115 } | |
116 | |
117 void GDataOperationRegistry::Operation::NotifyFinish( | |
118 OperationTransferState status) { | |
119 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
120 DCHECK(progress_status_.transfer_state >= OPERATION_STARTED); | |
121 DCHECK(status == OPERATION_COMPLETED || status == OPERATION_FAILED); | |
122 progress_status_.transfer_state = status; | |
123 registry_->OnOperationFinish(progress_status().operation_id); | |
124 } | |
125 | |
126 void GDataOperationRegistry::Operation::NotifySuspend() { | |
127 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
128 DCHECK(progress_status_.transfer_state >= OPERATION_STARTED); | |
129 progress_status_.transfer_state = OPERATION_SUSPENDED; | |
130 registry_->OnOperationSuspend(progress_status().operation_id); | |
131 } | |
132 | |
133 void GDataOperationRegistry::Operation::NotifyResume() { | |
134 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
135 if (progress_status_.transfer_state == OPERATION_NOT_STARTED) { | |
136 progress_status_.transfer_state = OPERATION_IN_PROGRESS; | |
137 registry_->OnOperationResume(this, &progress_status_); | |
138 } | |
139 } | |
140 | |
141 void GDataOperationRegistry::Operation::NotifyAuthFailed() { | |
142 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
143 registry_->OnOperationAuthFailed(); | |
144 } | |
145 | |
146 GDataOperationRegistry::GDataOperationRegistry() | |
147 : do_notification_frequency_control_(true) { | |
148 in_flight_operations_.set_check_on_null_data(true); | |
149 } | |
150 | |
151 GDataOperationRegistry::~GDataOperationRegistry() { | |
152 DCHECK(in_flight_operations_.IsEmpty()); | |
153 } | |
154 | |
155 void GDataOperationRegistry::AddObserver(Observer* observer) { | |
156 observer_list_.AddObserver(observer); | |
157 } | |
158 | |
159 void GDataOperationRegistry::RemoveObserver(Observer* observer) { | |
160 observer_list_.RemoveObserver(observer); | |
161 } | |
162 | |
163 void GDataOperationRegistry::DisableNotificationFrequencyControlForTest() { | |
164 do_notification_frequency_control_ = false; | |
165 } | |
166 | |
167 void GDataOperationRegistry::CancelAll() { | |
168 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
169 | |
170 for (OperationIDMap::iterator iter(&in_flight_operations_); | |
171 !iter.IsAtEnd(); | |
172 iter.Advance()) { | |
173 Operation* operation = iter.GetCurrentValue(); | |
174 operation->Cancel(); | |
175 // Cancel() may immediately trigger OnOperationFinish and remove the | |
176 // operation from the map, but IDMap is designed to be safe on such remove | |
177 // while iteration. | |
178 } | |
179 } | |
180 | |
181 bool GDataOperationRegistry::CancelForFilePath(const FilePath& file_path) { | |
182 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
183 | |
184 for (OperationIDMap::iterator iter(&in_flight_operations_); | |
185 !iter.IsAtEnd(); | |
186 iter.Advance()) { | |
187 Operation* operation = iter.GetCurrentValue(); | |
188 if (operation->progress_status().file_path == file_path) { | |
189 operation->Cancel(); | |
190 return true; | |
191 } | |
192 } | |
193 return false; | |
194 } | |
195 | |
196 void GDataOperationRegistry::OnOperationStart( | |
197 GDataOperationRegistry::Operation* operation, | |
198 OperationID* id) { | |
199 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
200 | |
201 *id = in_flight_operations_.Add(operation); | |
202 DVLOG(1) << "GDataOperation[" << *id << "] started."; | |
203 if (IsFileTransferOperation(operation)) | |
204 NotifyStatusToObservers(); | |
205 } | |
206 | |
207 void GDataOperationRegistry::OnOperationProgress(OperationID id) { | |
208 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
209 | |
210 Operation* operation = in_flight_operations_.Lookup(id); | |
211 DCHECK(operation); | |
212 | |
213 DVLOG(1) << "GDataOperation[" << id << "] " | |
214 << operation->progress_status().DebugString(); | |
215 if (IsFileTransferOperation(operation)) | |
216 NotifyStatusToObservers(); | |
217 } | |
218 | |
219 void GDataOperationRegistry::OnOperationFinish(OperationID id) { | |
220 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
221 | |
222 Operation* operation = in_flight_operations_.Lookup(id); | |
223 DCHECK(operation); | |
224 | |
225 DVLOG(1) << "GDataOperation[" << id << "] finished."; | |
226 if (IsFileTransferOperation(operation)) | |
227 NotifyStatusToObservers(); | |
228 in_flight_operations_.Remove(id); | |
229 } | |
230 | |
231 void GDataOperationRegistry::OnOperationResume( | |
232 GDataOperationRegistry::Operation* operation, | |
233 GDataOperationRegistry::ProgressStatus* new_status) { | |
234 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
235 | |
236 // Find the corresponding suspended task. | |
237 Operation* suspended = NULL; | |
238 for (OperationIDMap::iterator iter(&in_flight_operations_); | |
239 !iter.IsAtEnd(); | |
240 iter.Advance()) { | |
241 Operation* in_flight_operation = iter.GetCurrentValue(); | |
242 const ProgressStatus& status = in_flight_operation->progress_status(); | |
243 if (status.transfer_state == OPERATION_SUSPENDED && | |
244 status.file_path == operation->progress_status().file_path) { | |
245 suspended = in_flight_operation; | |
246 break; | |
247 } | |
248 } | |
249 DCHECK(suspended); | |
250 | |
251 // Copy the progress status. | |
252 const ProgressStatus& old_status = suspended->progress_status(); | |
253 OperationID old_id = old_status.operation_id; | |
254 | |
255 new_status->progress_current = old_status.progress_current; | |
256 new_status->progress_total = old_status.progress_total; | |
257 new_status->start_time = old_status.start_time; | |
258 | |
259 // Remove the old one and initiate the new operation. | |
260 in_flight_operations_.Remove(old_id); | |
261 new_status->operation_id = in_flight_operations_.Add(operation); | |
262 DVLOG(1) << "GDataOperation[" << old_id << " -> " << | |
263 new_status->operation_id << "] resumed."; | |
264 if (IsFileTransferOperation(operation)) | |
265 NotifyStatusToObservers(); | |
266 } | |
267 | |
268 void GDataOperationRegistry::OnOperationSuspend(OperationID id) { | |
269 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
270 | |
271 Operation* operation = in_flight_operations_.Lookup(id); | |
272 DCHECK(operation); | |
273 | |
274 DVLOG(1) << "GDataOperation[" << id << "] suspended."; | |
275 if (IsFileTransferOperation(operation)) | |
276 NotifyStatusToObservers(); | |
277 } | |
278 | |
279 void GDataOperationRegistry::OnOperationAuthFailed() { | |
280 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
281 | |
282 DVLOG(1) << "GDataOperation authentication failed."; | |
283 FOR_EACH_OBSERVER(Observer, observer_list_, OnAuthenticationFailed()); | |
284 } | |
285 | |
286 bool GDataOperationRegistry::IsFileTransferOperation( | |
287 const Operation* operation) const { | |
288 OperationType type = operation->progress_status().operation_type; | |
289 return type == OPERATION_UPLOAD || type == OPERATION_DOWNLOAD; | |
290 } | |
291 | |
292 GDataOperationRegistry::ProgressStatusList | |
293 GDataOperationRegistry::GetProgressStatusList() { | |
294 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
295 | |
296 ProgressStatusList status_list; | |
297 for (OperationIDMap::const_iterator iter(&in_flight_operations_); | |
298 !iter.IsAtEnd(); | |
299 iter.Advance()) { | |
300 const Operation* operation = iter.GetCurrentValue(); | |
301 if (IsFileTransferOperation(operation)) | |
302 status_list.push_back(operation->progress_status()); | |
303 } | |
304 return status_list; | |
305 } | |
306 | |
307 bool GDataOperationRegistry::ShouldNotifyStatusNow( | |
308 const ProgressStatusList& list) { | |
309 if (!do_notification_frequency_control_) | |
310 return true; | |
311 | |
312 base::Time now = base::Time::Now(); | |
313 | |
314 // If it is a first event, or some time abnormality is detected, we should | |
315 // not skip this notification. | |
316 if (last_notification_.is_null() || now < last_notification_) { | |
317 last_notification_ = now; | |
318 return true; | |
319 } | |
320 | |
321 // If sufficiently long time has elapsed since the previous event, we should | |
322 // not skip this notification. | |
323 if ((now - last_notification_).InMilliseconds() >= | |
324 kNotificationFrequencyInMilliseconds) { | |
325 last_notification_ = now; | |
326 return true; | |
327 } | |
328 | |
329 // If important events (OPERATION_STARTED, COMPLETED, or FAILED) are there, | |
330 // we should not skip this notification. | |
331 for (size_t i = 0; i < list.size(); ++i) { | |
332 if (list[i].transfer_state != OPERATION_IN_PROGRESS) { | |
333 last_notification_ = now; | |
334 return true; | |
335 } | |
336 } | |
337 | |
338 // Otherwise we can skip it. | |
339 return false; | |
340 } | |
341 | |
342 void GDataOperationRegistry::NotifyStatusToObservers() { | |
343 ProgressStatusList list(GetProgressStatusList()); | |
344 if (ShouldNotifyStatusNow(list)) | |
345 FOR_EACH_OBSERVER(Observer, observer_list_, OnProgressUpdate(list)); | |
346 } | |
347 | |
348 } // namespace gdata | |
OLD | NEW |