| Index: components/offline_pages/background/request_queue_in_memory_store.cc
|
| diff --git a/components/offline_pages/background/request_queue_in_memory_store.cc b/components/offline_pages/background/request_queue_in_memory_store.cc
|
| index 32219a91616eef459ea08ab4fb7116258f781355..07802400d6f86c49c65662d441d54eda5839ef7c 100644
|
| --- a/components/offline_pages/background/request_queue_in_memory_store.cc
|
| +++ b/components/offline_pages/background/request_queue_in_memory_store.cc
|
| @@ -73,6 +73,46 @@ void RequestQueueInMemoryStore::RemoveRequestsByClientId(
|
| FROM_HERE, base::Bind(callback, true, count));
|
| }
|
|
|
| +void RequestQueueInMemoryStore::PauseRequests(
|
| + const std::vector<int64_t>& request_ids,
|
| + const UpdateCallback& callback) {
|
| + UpdateStatus status = SetStateForRequestIds(
|
| + request_ids, SavePageRequest::RequestState::PAUSED);
|
| +
|
| + base::ThreadTaskRunnerHandle::Get()->PostTask(
|
| + FROM_HERE, base::Bind(callback, status));
|
| +}
|
| +
|
| +void RequestQueueInMemoryStore::ResumeRequests(
|
| + const std::vector<int64_t>& request_ids,
|
| + const UpdateCallback& callback) {
|
| +
|
| + UpdateStatus status = SetStateForRequestIds(
|
| + request_ids, SavePageRequest::RequestState::AVAILABLE);
|
| +
|
| + base::ThreadTaskRunnerHandle::Get()->PostTask(
|
| + FROM_HERE, base::Bind(callback, status));
|
| +}
|
| +
|
| +RequestQueueStore::UpdateStatus
|
| +RequestQueueInMemoryStore::SetStateForRequestIds(
|
| + const std::vector<int64_t>& request_ids,
|
| + SavePageRequest::RequestState newState) {
|
| + int count = 0;
|
| +
|
| + std::set<int64_t> request_id_set(request_ids.begin(), request_ids.end());
|
| + for (auto iter = requests_.begin(); iter != requests_.end(); ++iter) {
|
| + if (request_id_set.find(iter->second.request_id()) !=
|
| + request_id_set.end()) {
|
| + // Set the state if the client ID matches.
|
| + iter->second.set_request_state(newState);
|
| + ++count;
|
| + }
|
| + }
|
| +
|
| + return count > 0 ? UpdateStatus::UPDATED : UpdateStatus::FAILED;
|
| +}
|
| +
|
| void RequestQueueInMemoryStore::Reset(const ResetCallback& callback) {
|
| requests_.clear();
|
| base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE,
|
|
|