| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "components/offline_pages/background/request_queue_in_memory_store.h" | 5 #include "components/offline_pages/background/request_queue_in_memory_store.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/location.h" | 8 #include "base/location.h" |
| 9 #include "base/threading/thread_task_runner_handle.h" | 9 #include "base/threading/thread_task_runner_handle.h" |
| 10 #include "components/offline_pages/background/save_page_request.h" | 10 #include "components/offline_pages/background/save_page_request.h" |
| (...skipping 22 matching lines...) Expand all Loading... |
| 33 requests_.insert(std::make_pair(request.request_id(), request)); | 33 requests_.insert(std::make_pair(request.request_id(), request)); |
| 34 base::ThreadTaskRunnerHandle::Get()->PostTask( | 34 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 35 FROM_HERE, base::Bind(callback, UpdateStatus::UPDATED)); | 35 FROM_HERE, base::Bind(callback, UpdateStatus::UPDATED)); |
| 36 } | 36 } |
| 37 | 37 |
| 38 void RequestQueueInMemoryStore::RemoveRequests( | 38 void RequestQueueInMemoryStore::RemoveRequests( |
| 39 const std::vector<int64_t>& request_ids, | 39 const std::vector<int64_t>& request_ids, |
| 40 const RemoveCallback& callback) { | 40 const RemoveCallback& callback) { |
| 41 RequestQueue::UpdateMultipleRequestResults results; | 41 RequestQueue::UpdateMultipleRequestResults results; |
| 42 RequestQueue::UpdateRequestResult result; | 42 RequestQueue::UpdateRequestResult result; |
| 43 std::vector<SavePageRequest> requests; |
| 43 RequestsMap::iterator iter; | 44 RequestsMap::iterator iter; |
| 45 |
| 46 // If we find a request, mark it as succeeded, and put it in the request list. |
| 47 // Otherwise mark it as failed. |
| 44 for (auto request_id : request_ids) { | 48 for (auto request_id : request_ids) { |
| 45 iter = requests_.find(request_id); | 49 iter = requests_.find(request_id); |
| 46 if (iter != requests_.end()) { | 50 if (iter != requests_.end()) { |
| 51 SavePageRequest request = iter->second; |
| 47 requests_.erase(iter); | 52 requests_.erase(iter); |
| 48 result = RequestQueue::UpdateRequestResult::SUCCESS; | 53 result = RequestQueue::UpdateRequestResult::SUCCESS; |
| 54 requests.push_back(request); |
| 49 } else { | 55 } else { |
| 50 result = RequestQueue::UpdateRequestResult::REQUEST_DOES_NOT_EXIST; | 56 result = RequestQueue::UpdateRequestResult::REQUEST_DOES_NOT_EXIST; |
| 51 } | 57 } |
| 52 results.push_back(std::make_pair(request_id, result)); | 58 results.push_back(std::make_pair(request_id, result)); |
| 53 } | 59 } |
| 54 | 60 |
| 55 base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, | 61 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 56 base::Bind(callback, results)); | 62 FROM_HERE, base::Bind(callback, results, requests)); |
| 57 } | 63 } |
| 58 | 64 |
| 59 void RequestQueueInMemoryStore::ChangeRequestsState( | 65 void RequestQueueInMemoryStore::ChangeRequestsState( |
| 60 const std::vector<int64_t>& request_ids, | 66 const std::vector<int64_t>& request_ids, |
| 61 const SavePageRequest::RequestState new_state, | 67 const SavePageRequest::RequestState new_state, |
| 62 const UpdateCallback& callback) { | 68 const UpdateMultipleRequestsCallback& callback) { |
| 63 int count = 0; | 69 RequestQueue::UpdateMultipleRequestResults results; |
| 64 | 70 std::vector<SavePageRequest> requests; |
| 71 RequestQueue::UpdateRequestResult result; |
| 65 for (int64_t request_id : request_ids) { | 72 for (int64_t request_id : request_ids) { |
| 66 auto pair = requests_.find(request_id); | 73 auto pair = requests_.find(request_id); |
| 74 // If we find this request id, modify it, and return the modified request in |
| 75 // the request list. |
| 67 if (pair != requests_.end()) { | 76 if (pair != requests_.end()) { |
| 68 ++count; | |
| 69 pair->second.set_request_state(new_state); | 77 pair->second.set_request_state(new_state); |
| 78 requests.push_back(pair->second); |
| 79 result = RequestQueue::UpdateRequestResult::SUCCESS; |
| 80 } else { |
| 81 result = RequestQueue::UpdateRequestResult::REQUEST_DOES_NOT_EXIST;; |
| 70 } | 82 } |
| 83 |
| 84 results.push_back(std::make_pair(request_id, result)); |
| 71 } | 85 } |
| 72 | 86 |
| 73 RequestQueueStore::UpdateStatus status = | 87 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 74 count > 0 ? UpdateStatus::UPDATED : UpdateStatus::FAILED; | 88 FROM_HERE, base::Bind(callback, results, requests)); |
| 75 | |
| 76 base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, | |
| 77 base::Bind(callback, status)); | |
| 78 } | 89 } |
| 79 | 90 |
| 80 void RequestQueueInMemoryStore::Reset(const ResetCallback& callback) { | 91 void RequestQueueInMemoryStore::Reset(const ResetCallback& callback) { |
| 81 requests_.clear(); | 92 requests_.clear(); |
| 82 base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, | 93 base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
| 83 base::Bind(callback, true)); | 94 base::Bind(callback, true)); |
| 84 } | 95 } |
| 85 | 96 |
| 86 } // namespace offline_pages | 97 } // namespace offline_pages |
| OLD | NEW |