| OLD | NEW |
| (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/offline_pages/background/change_requests_state_task.h" | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 | |
| 9 namespace offline_pages { | |
| 10 | |
| 11 ChangeRequestsStateTask::ChangeRequestsStateTask( | |
| 12 RequestQueueStore* store, | |
| 13 const std::vector<int64_t>& request_ids, | |
| 14 const SavePageRequest::RequestState new_state, | |
| 15 const RequestQueueStore::UpdateCallback& callback) | |
| 16 : store_(store), | |
| 17 request_ids_(request_ids.begin(), request_ids.end()), | |
| 18 new_state_(new_state), | |
| 19 callback_(callback), | |
| 20 weak_ptr_factory_(this) {} | |
| 21 | |
| 22 ChangeRequestsStateTask::~ChangeRequestsStateTask() {} | |
| 23 | |
| 24 void ChangeRequestsStateTask::Run() { | |
| 25 ReadRequests(); | |
| 26 } | |
| 27 | |
| 28 void ChangeRequestsStateTask::ReadRequests() { | |
| 29 if (request_ids_.empty()) { | |
| 30 CompleteEarly(ItemActionStatus::NOT_FOUND); | |
| 31 return; | |
| 32 } | |
| 33 | |
| 34 store_->GetRequests(base::Bind(&ChangeRequestsStateTask::SelectItemsToUpdate, | |
| 35 weak_ptr_factory_.GetWeakPtr())); | |
| 36 } | |
| 37 | |
| 38 void ChangeRequestsStateTask::SelectItemsToUpdate( | |
| 39 bool success, | |
| 40 std::vector<std::unique_ptr<SavePageRequest>> requests) { | |
| 41 if (!success) { | |
| 42 CompleteEarly(ItemActionStatus::STORE_ERROR); | |
| 43 return; | |
| 44 } | |
| 45 | |
| 46 std::vector<SavePageRequest> items_to_update; | |
| 47 for (const auto& request : requests) { | |
| 48 // If this request is in our list, update it. | |
| 49 if (request_ids_.count(request->request_id()) > 0) { | |
| 50 request->set_request_state(new_state_); | |
| 51 items_to_update.push_back(*request); | |
| 52 // Items that are missing before the update will be marked as not found | |
| 53 // before the callback. | |
| 54 request_ids_.erase(request->request_id()); | |
| 55 } | |
| 56 } | |
| 57 | |
| 58 if (items_to_update.empty()) { | |
| 59 CompleteEarly(ItemActionStatus::NOT_FOUND); | |
| 60 return; | |
| 61 } | |
| 62 | |
| 63 store_->UpdateRequests(items_to_update, | |
| 64 base::Bind(&ChangeRequestsStateTask::UpdateCompleted, | |
| 65 weak_ptr_factory_.GetWeakPtr())); | |
| 66 } | |
| 67 | |
| 68 void ChangeRequestsStateTask::UpdateCompleted( | |
| 69 std::unique_ptr<UpdateRequestsResult> update_result) { | |
| 70 CompleteWithStatus(std::move(update_result), ItemActionStatus::NOT_FOUND); | |
| 71 } | |
| 72 | |
| 73 void ChangeRequestsStateTask::CompleteEarly(ItemActionStatus status) { | |
| 74 // TODO(fgorski): store_->state() once implemented | |
| 75 std::unique_ptr<UpdateRequestsResult> result( | |
| 76 new UpdateRequestsResult(StoreState::LOADED)); | |
| 77 CompleteWithStatus(std::move(result), status); | |
| 78 } | |
| 79 | |
| 80 void ChangeRequestsStateTask::CompleteWithStatus( | |
| 81 std::unique_ptr<UpdateRequestsResult> result, | |
| 82 ItemActionStatus status) { | |
| 83 // Mark items as not found, if they are still in the request IDs set. | |
| 84 for (int64_t request_id : request_ids_) | |
| 85 result->item_statuses.push_back(std::make_pair(request_id, status)); | |
| 86 callback_.Run(std::move(result)); | |
| 87 TaskComplete(); | |
| 88 } | |
| 89 | |
| 90 } // namespace offline_pages | |
| OLD | NEW |