| 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 <set> | |
| 8 | |
| 9 #include "base/bind.h" | 7 #include "base/bind.h" |
| 10 #include "base/location.h" | 8 #include "base/location.h" |
| 11 #include "base/threading/thread_task_runner_handle.h" | 9 #include "base/threading/thread_task_runner_handle.h" |
| 12 #include "components/offline_pages/background/save_page_request.h" | 10 #include "components/offline_pages/background/save_page_request.h" |
| 13 | 11 |
| 14 namespace offline_pages { | 12 namespace offline_pages { |
| 15 | 13 |
| 16 RequestQueueInMemoryStore::RequestQueueInMemoryStore() {} | 14 RequestQueueInMemoryStore::RequestQueueInMemoryStore() {} |
| 17 | 15 |
| 18 RequestQueueInMemoryStore::~RequestQueueInMemoryStore() {} | 16 RequestQueueInMemoryStore::~RequestQueueInMemoryStore() {} |
| (...skipping 14 matching lines...) Expand all Loading... |
| 33 if (iter != requests_.end()) | 31 if (iter != requests_.end()) |
| 34 requests_.erase(iter); | 32 requests_.erase(iter); |
| 35 requests_.insert(std::make_pair(request.request_id(), request)); | 33 requests_.insert(std::make_pair(request.request_id(), request)); |
| 36 base::ThreadTaskRunnerHandle::Get()->PostTask( | 34 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 37 FROM_HERE, base::Bind(callback, UpdateStatus::UPDATED)); | 35 FROM_HERE, base::Bind(callback, UpdateStatus::UPDATED)); |
| 38 } | 36 } |
| 39 | 37 |
| 40 void RequestQueueInMemoryStore::RemoveRequests( | 38 void RequestQueueInMemoryStore::RemoveRequests( |
| 41 const std::vector<int64_t>& request_ids, | 39 const std::vector<int64_t>& request_ids, |
| 42 const RemoveCallback& callback) { | 40 const RemoveCallback& callback) { |
| 43 int count = 0; | 41 RequestQueue::UpdateMultipleRequestResults results; |
| 42 RequestQueue::UpdateRequestResult result; |
| 44 RequestsMap::iterator iter; | 43 RequestsMap::iterator iter; |
| 45 for (auto request_id : request_ids) { | 44 for (auto request_id : request_ids) { |
| 46 iter = requests_.find(request_id); | 45 iter = requests_.find(request_id); |
| 47 if (iter != requests_.end()) { | 46 if (iter != requests_.end()) { |
| 48 requests_.erase(iter); | 47 requests_.erase(iter); |
| 49 ++count; | 48 result = RequestQueue::UpdateRequestResult::SUCCESS; |
| 49 } else { |
| 50 result = RequestQueue::UpdateRequestResult::REQUEST_DOES_NOT_EXIST; |
| 50 } | 51 } |
| 52 results.push_back(std::make_pair(request_id, result)); |
| 51 } | 53 } |
| 52 | 54 |
| 53 base::ThreadTaskRunnerHandle::Get()->PostTask( | 55 base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
| 54 FROM_HERE, base::Bind(callback, true, count)); | 56 base::Bind(callback, results)); |
| 55 } | |
| 56 | |
| 57 void RequestQueueInMemoryStore::RemoveRequestsByClientId( | |
| 58 const std::vector<ClientId>& client_ids, | |
| 59 const RemoveCallback& callback) { | |
| 60 int count = 0; | |
| 61 | |
| 62 std::set<ClientId> client_id_set(client_ids.begin(), client_ids.end()); | |
| 63 for (auto iter = requests_.begin(); iter != requests_.end(); ) { | |
| 64 if (client_id_set.find(iter->second.client_id()) != client_id_set.end()) { | |
| 65 requests_.erase(iter++); | |
| 66 ++count; | |
| 67 } else { | |
| 68 ++iter; | |
| 69 } | |
| 70 } | |
| 71 | |
| 72 base::ThreadTaskRunnerHandle::Get()->PostTask( | |
| 73 FROM_HERE, base::Bind(callback, true, count)); | |
| 74 } | 57 } |
| 75 | 58 |
| 76 void RequestQueueInMemoryStore::Reset(const ResetCallback& callback) { | 59 void RequestQueueInMemoryStore::Reset(const ResetCallback& callback) { |
| 77 requests_.clear(); | 60 requests_.clear(); |
| 78 base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, | 61 base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
| 79 base::Bind(callback, true)); | 62 base::Bind(callback, true)); |
| 80 } | 63 } |
| 81 | 64 |
| 82 } // namespace offline_pages | 65 } // namespace offline_pages |
| OLD | NEW |