| 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/request_queue_in_memory_store.h" |
| 6 |
| 7 #include "base/bind.h" |
| 8 #include "base/location.h" |
| 9 #include "base/thread_task_runner_handle.h" |
| 10 #include "components/offline_pages/background/save_page_request.h" |
| 11 |
| 12 namespace offline_pages { |
| 13 |
| 14 RequestQueueInMemoryStore::RequestQueueInMemoryStore() {} |
| 15 |
| 16 RequestQueueInMemoryStore::~RequestQueueInMemoryStore() {} |
| 17 |
| 18 void RequestQueueInMemoryStore::GetRequests( |
| 19 const GetRequestsCallback& callback) { |
| 20 std::vector<SavePageRequest> result_requests; |
| 21 for (const auto& id_request_pair : requests_) |
| 22 result_requests.push_back(id_request_pair.second); |
| 23 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 24 FROM_HERE, base::Bind(callback, true, result_requests)); |
| 25 } |
| 26 |
| 27 void RequestQueueInMemoryStore::AddOrUpdateRequest( |
| 28 const SavePageRequest& request, |
| 29 const UpdateCallback& callback) { |
| 30 RequestsMap::iterator iter = requests_.find(request.request_id()); |
| 31 UpdateStatus status = UpdateStatus::kAdded; |
| 32 if (iter != requests_.end()) { |
| 33 requests_.erase(iter); |
| 34 status = UpdateStatus::kUpdated; |
| 35 } |
| 36 requests_.insert(std::make_pair(request.request_id(), request)); |
| 37 base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
| 38 base::Bind(callback, status)); |
| 39 } |
| 40 |
| 41 void RequestQueueInMemoryStore::RemoveRequests( |
| 42 const std::vector<int64_t>& request_ids, |
| 43 const RemoveCallback& callback) { |
| 44 // In case the |request_ids| is empty, the result will be true, but the count |
| 45 // of deleted pages will be empty. |
| 46 bool result = true; |
| 47 int count = 0; |
| 48 RequestsMap::iterator iter; |
| 49 for (auto request_id : request_ids) { |
| 50 iter = requests_.find(request_id); |
| 51 if (iter != requests_.end()) { |
| 52 requests_.erase(iter); |
| 53 ++count; |
| 54 } else { |
| 55 result = false; |
| 56 } |
| 57 } |
| 58 |
| 59 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 60 FROM_HERE, base::Bind(callback, result, count)); |
| 61 } |
| 62 |
| 63 void RequestQueueInMemoryStore::Reset(const ResetCallback& callback) { |
| 64 requests_.clear(); |
| 65 base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
| 66 base::Bind(callback, true)); |
| 67 } |
| 68 |
| 69 } // namespace offline_pages |
| OLD | NEW |