| 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.h" | 5 #include "components/offline_pages/background/request_queue.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/request_queue_store.h" | 10 #include "components/offline_pages/background/request_queue_store.h" |
| 11 #include "components/offline_pages/background/save_page_request.h" | 11 #include "components/offline_pages/background/save_page_request.h" |
| 12 | 12 |
| 13 namespace offline_pages { | 13 namespace offline_pages { |
| 14 | 14 |
| 15 namespace { | 15 namespace { |
| 16 // Completes the get requests call. | 16 // Completes the get requests call. |
| 17 void GetRequestsDone(const RequestQueue::GetRequestsCallback& callback, | 17 void GetRequestsDone(const RequestQueue::GetRequestsCallback& callback, |
| 18 bool success, | 18 bool success, |
| 19 const std::vector<SavePageRequest>& requests) { | 19 std::vector<std::unique_ptr<SavePageRequest>> requests) { |
| 20 RequestQueue::GetRequestsResult result = | 20 RequestQueue::GetRequestsResult result = |
| 21 success ? RequestQueue::GetRequestsResult::SUCCESS | 21 success ? RequestQueue::GetRequestsResult::SUCCESS |
| 22 : RequestQueue::GetRequestsResult::STORE_FAILURE; | 22 : RequestQueue::GetRequestsResult::STORE_FAILURE; |
| 23 // TODO(fgorski): Filter out expired requests based on policy. | 23 // TODO(fgorski): Filter out expired requests based on policy. |
| 24 // This may trigger the purging if necessary. | 24 // This may trigger the purging if necessary. |
| 25 // Also this may be turned into a method on the request queue or add a policy | 25 // Also this may be turned into a method on the request queue or add a policy |
| 26 // parameter in the process. | 26 // parameter in the process. |
| 27 callback.Run(result, requests); | 27 callback.Run(result, std::move(requests)); |
| 28 } | 28 } |
| 29 | 29 |
| 30 // Completes the add request call. | 30 // Completes the add request call. |
| 31 void AddRequestDone(const RequestQueue::AddRequestCallback& callback, | 31 void AddRequestDone(const RequestQueue::AddRequestCallback& callback, |
| 32 const SavePageRequest& request, | 32 const SavePageRequest& request, |
| 33 RequestQueueStore::UpdateStatus status) { | 33 RequestQueueStore::UpdateStatus status) { |
| 34 RequestQueue::AddRequestResult result = | 34 RequestQueue::AddRequestResult result = |
| 35 (status == RequestQueueStore::UpdateStatus::UPDATED) | 35 (status == RequestQueueStore::UpdateStatus::UPDATED) |
| 36 ? RequestQueue::AddRequestResult::SUCCESS | 36 ? RequestQueue::AddRequestResult::SUCCESS |
| 37 : RequestQueue::AddRequestResult::STORE_FAILURE; | 37 : RequestQueue::AddRequestResult::STORE_FAILURE; |
| 38 callback.Run(result, request); | 38 callback.Run(result, request); |
| 39 } | 39 } |
| 40 | 40 |
| 41 // Completes the update request call. | 41 // Completes the update request call. |
| 42 // TODO(petewil): Move callers to the UpdateMultipleRequestDone callback | 42 // TODO(petewil): Move callers to the UpdateMultipleRequestDone callback |
| 43 void UpdateRequestDone(const RequestQueue::UpdateRequestCallback& callback, | 43 void UpdateRequestDone(const RequestQueue::UpdateRequestCallback& callback, |
| 44 RequestQueueStore::UpdateStatus status) { | 44 RequestQueueStore::UpdateStatus status) { |
| 45 RequestQueue::UpdateRequestResult result = | 45 RequestQueue::UpdateRequestResult result = |
| 46 (status == RequestQueueStore::UpdateStatus::UPDATED) | 46 (status == RequestQueueStore::UpdateStatus::UPDATED) |
| 47 ? RequestQueue::UpdateRequestResult::SUCCESS | 47 ? RequestQueue::UpdateRequestResult::SUCCESS |
| 48 : RequestQueue::UpdateRequestResult::STORE_FAILURE; | 48 : RequestQueue::UpdateRequestResult::STORE_FAILURE; |
| 49 callback.Run(result); | 49 callback.Run(result); |
| 50 } | 50 } |
| 51 | 51 |
| 52 // Handles updating multiple requests at the same time. | 52 // Handles updating multiple requests at the same time. |
| 53 void UpdateMultipleRequestsDone( | 53 void UpdateMultipleRequestsDone( |
| 54 const RequestQueue::UpdateMultipleRequestsCallback& callback, | 54 const RequestQueue::UpdateMultipleRequestsCallback& callback, |
| 55 const RequestQueue::UpdateMultipleRequestResults& results, | 55 const RequestQueue::UpdateMultipleRequestResults& results, |
| 56 const std::vector<SavePageRequest>& requests) { | 56 std::vector<std::unique_ptr<SavePageRequest>> requests) { |
| 57 callback.Run(results, requests); | 57 callback.Run(results, std::move(requests)); |
| 58 } | 58 } |
| 59 | 59 |
| 60 // Completes the remove request call. | 60 // Completes the remove request call. |
| 61 void RemoveRequestsDone( | 61 void RemoveRequestsDone( |
| 62 const RequestQueue::RemoveRequestsCallback& callback, | 62 const RequestQueue::RemoveRequestsCallback& callback, |
| 63 const RequestQueue::UpdateMultipleRequestResults& results, | 63 const RequestQueue::UpdateMultipleRequestResults& results, |
| 64 const std::vector<SavePageRequest>& requests) { | 64 std::vector<std::unique_ptr<SavePageRequest>> requests) { |
| 65 callback.Run(results, requests); | 65 callback.Run(results, std::move(requests)); |
| 66 } | 66 } |
| 67 | 67 |
| 68 } // namespace | 68 } // namespace |
| 69 | 69 |
| 70 RequestQueue::RequestQueue(std::unique_ptr<RequestQueueStore> store) | 70 RequestQueue::RequestQueue(std::unique_ptr<RequestQueueStore> store) |
| 71 : store_(std::move(store)), weak_ptr_factory_(this) {} | 71 : store_(std::move(store)), weak_ptr_factory_(this) {} |
| 72 | 72 |
| 73 RequestQueue::~RequestQueue() {} | 73 RequestQueue::~RequestQueue() {} |
| 74 | 74 |
| 75 void RequestQueue::GetRequests(const GetRequestsCallback& callback) { | 75 void RequestQueue::GetRequests(const GetRequestsCallback& callback) { |
| (...skipping 25 matching lines...) Expand all Loading... |
| 101 } | 101 } |
| 102 | 102 |
| 103 // We need a different version of the GetCallback that can take the curried | 103 // We need a different version of the GetCallback that can take the curried |
| 104 // update_callback as a parameter, and call back into the request queue store | 104 // update_callback as a parameter, and call back into the request queue store |
| 105 // implementation. This must be a member function because we need access to | 105 // implementation. This must be a member function because we need access to |
| 106 // the store pointer to call AddOrUpdateRequest. | 106 // the store pointer to call AddOrUpdateRequest. |
| 107 void RequestQueue::GetForUpdateDone( | 107 void RequestQueue::GetForUpdateDone( |
| 108 const UpdateRequestCallback& update_callback, | 108 const UpdateRequestCallback& update_callback, |
| 109 const SavePageRequest& update_request, | 109 const SavePageRequest& update_request, |
| 110 bool success, | 110 bool success, |
| 111 const std::vector<SavePageRequest>& found_requests) { | 111 std::vector<std::unique_ptr<SavePageRequest>> found_requests) { |
| 112 // If the result was not found, return now. | 112 // If the result was not found, return now. |
| 113 if (!success) { | 113 if (!success) { |
| 114 update_callback.Run( | 114 update_callback.Run( |
| 115 RequestQueue::UpdateRequestResult::REQUEST_DOES_NOT_EXIST); | 115 RequestQueue::UpdateRequestResult::REQUEST_DOES_NOT_EXIST); |
| 116 return; | 116 return; |
| 117 } | 117 } |
| 118 // If the found result does not contain the request we are looking for, return | 118 // If the found result does not contain the request we are looking for, return |
| 119 // now. | 119 // now. |
| 120 bool found = false; | 120 bool found = false; |
| 121 std::vector<SavePageRequest>::const_iterator iter; | 121 std::vector<std::unique_ptr<SavePageRequest>>::const_iterator iter; |
| 122 for (iter = found_requests.begin(); iter != found_requests.end(); ++iter) { | 122 for (iter = found_requests.begin(); iter != found_requests.end(); ++iter) { |
| 123 if (iter->request_id() == update_request.request_id()) | 123 if ((*iter)->request_id() == update_request.request_id()) |
| 124 found = true; | 124 found = true; |
| 125 } | 125 } |
| 126 if (!found) { | 126 if (!found) { |
| 127 update_callback.Run( | 127 update_callback.Run( |
| 128 RequestQueue::UpdateRequestResult::REQUEST_DOES_NOT_EXIST); | 128 RequestQueue::UpdateRequestResult::REQUEST_DOES_NOT_EXIST); |
| 129 return; | 129 return; |
| 130 } | 130 } |
| 131 | 131 |
| 132 // Since the request exists, update it. | 132 // Since the request exists, update it. |
| 133 store_->AddOrUpdateRequest(update_request, | 133 store_->AddOrUpdateRequest(update_request, |
| 134 base::Bind(&UpdateRequestDone, update_callback)); | 134 base::Bind(&UpdateRequestDone, update_callback)); |
| 135 } | 135 } |
| 136 | 136 |
| 137 void RequestQueue::RemoveRequests(const std::vector<int64_t>& request_ids, | 137 void RequestQueue::RemoveRequests(const std::vector<int64_t>& request_ids, |
| 138 const RemoveRequestsCallback& callback) { | 138 const RemoveRequestsCallback& callback) { |
| 139 store_->RemoveRequests(request_ids, base::Bind(RemoveRequestsDone, callback)); | 139 store_->RemoveRequests(request_ids, base::Bind(RemoveRequestsDone, callback)); |
| 140 } | 140 } |
| 141 | 141 |
| 142 void RequestQueue::ChangeRequestsState( | 142 void RequestQueue::ChangeRequestsState( |
| 143 const std::vector<int64_t>& request_ids, | 143 const std::vector<int64_t>& request_ids, |
| 144 const SavePageRequest::RequestState new_state, | 144 const SavePageRequest::RequestState new_state, |
| 145 const UpdateMultipleRequestsCallback& callback) { | 145 const UpdateMultipleRequestsCallback& callback) { |
| 146 store_->ChangeRequestsState(request_ids, new_state, | 146 store_->ChangeRequestsState(request_ids, new_state, |
| 147 base::Bind(UpdateMultipleRequestsDone, callback)); | 147 base::Bind(UpdateMultipleRequestsDone, callback)); |
| 148 } | 148 } |
| 149 | 149 |
| 150 void RequestQueue::PurgeRequests(const PurgeRequestsCallback& callback) {} | 150 void RequestQueue::PurgeRequests(const PurgeRequestsCallback& callback) {} |
| 151 | 151 |
| 152 } // namespace offline_pages | 152 } // namespace offline_pages |
| OLD | NEW |