| 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 |