Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(540)

Side by Side Diff: components/offline_pages/background/request_queue_in_memory_store.cc

Issue 2228813003: Changes to fit better with the needs of the download manager. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: CR fixes for Dimich Created 4 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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
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 std::vector<RequestQueue::UpdateRequestResult> results;
44 RequestsMap::iterator iter; 42 RequestsMap::iterator iter;
45 for (auto request_id : request_ids) { 43 for (auto request_id : request_ids) {
46 iter = requests_.find(request_id); 44 iter = requests_.find(request_id);
47 if (iter != requests_.end()) { 45 if (iter != requests_.end()) {
48 requests_.erase(iter); 46 requests_.erase(iter);
49 ++count; 47 results.push_back(RequestQueue::UpdateRequestResult::SUCCESS);
48 } else {
49 results.push_back(
50 RequestQueue::UpdateRequestResult::REQUEST_DOES_NOT_EXIST);
50 } 51 }
51 } 52 }
52 53
53 base::ThreadTaskRunnerHandle::Get()->PostTask( 54 base::ThreadTaskRunnerHandle::Get()->PostTask(
54 FROM_HERE, base::Bind(callback, true, count)); 55 FROM_HERE, base::Bind(callback, request_ids, 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 } 56 }
75 57
76 void RequestQueueInMemoryStore::Reset(const ResetCallback& callback) { 58 void RequestQueueInMemoryStore::Reset(const ResetCallback& callback) {
77 requests_.clear(); 59 requests_.clear();
78 base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, 60 base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE,
79 base::Bind(callback, true)); 61 base::Bind(callback, true));
80 } 62 }
81 63
82 } // namespace offline_pages 64 } // namespace offline_pages
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698