[Offline Prefetch] Backoff support for PrefetchBackgroundTask

components/offline_pages/core/prefetch/prefetch_dispatcher_impl.cc

 // Copyright 2017 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/offline_pages/core/prefetch/prefetch_dispatcher_impl.h"
 
+#include "base/bind.h"
+#include "base/task_runner.h"
+#include "base/threading/thread_task_runner_handle.h"
 #include "url/gurl.h"
 
 namespace offline_pages {
 
+namespace {
+void DeleteBackgroundTaskHelper(
+    std::unique_ptr<PrefetchDispatcher::ScopedBackgroundTask> task) {
+  task.reset();
+}
+}  // namespace
+
 PrefetchDispatcherImpl::PrefetchDispatcherImpl() {}
 
 PrefetchDispatcherImpl::~PrefetchDispatcherImpl() = default;
 
 void PrefetchDispatcherImpl::AddCandidatePrefetchURLs(
     const std::vector<PrefetchURL>& url_suggestions) {
   NOTIMPLEMENTED();
 }
 void PrefetchDispatcherImpl::RemoveAllUnprocessedPrefetchURLs(
     const std::string& name_space) {
   NOTIMPLEMENTED();
 }
 
 void PrefetchDispatcherImpl::RemovePrefetchURLsByClientId(
     const ClientId& client_id) {
   NOTIMPLEMENTED();
 }
 
 void PrefetchDispatcherImpl::BeginBackgroundTask(
     std::unique_ptr<ScopedBackgroundTask> task) {
-  NOTIMPLEMENTED();
+  task_ = std::move(task);
 }
 
-void PrefetchDispatcherImpl::StopBackgroundTask(ScopedBackgroundTask* task) {
-  NOTIMPLEMENTED();
+void PrefetchDispatcherImpl::StopBackgroundTask() {
+  DisposeTask();
+}
+
+void PrefetchDispatcherImpl::RequestFinishBackgroundTaskForTest() {
+  DisposeTask();
+}
+
+void PrefetchDispatcherImpl::DisposeTask() {
+  if (!task_)

[dewittj, 2017/06/01 00:22:27]

Should this be DCHECK(task_) ?

[jianli, 2017/06/02 00:48:37]

Done.

+    return;
+  // Delay the deletion till the caller finishes.
+  base::ThreadTaskRunnerHandle::Get()->PostTask(
+      FROM_HERE,
+      base::Bind(&DeleteBackgroundTaskHelper, base::Passed(std::move(task_))));
 }
 
 }  // namespace offline_pages