Add function to CancelableTaskTracker and convert BootTimeLoader

chrome/common/cancelable_task_tracker.cc

 // Copyright (c) 2012 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 "chrome/common/cancelable_task_tracker.h"
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
+#include "base/location.h"
 #include "base/memory/ref_counted.h"
 #include "base/message_loop_proxy.h"
 #include "base/synchronization/cancellation_flag.h"
 #include "base/task_runner.h"
 
 using base::Bind;
 using base::CancellationFlag;
 using base::Closure;
 using base::hash_map;
 using base::TaskRunner;
 
 namespace {
 
 void RunIfNotCanceled(const CancellationFlag* flag, const Closure& task) {
   if (!flag->IsSet())
     task.Run();
 }
 
 void RunIfNotCanceledThenUntrack(const CancellationFlag* flag,
                                  const Closure& task,
                                  const Closure& untrack) {
   RunIfNotCanceled(flag, task);
   untrack.Run();
 }
 
+bool IsCanceled(const CancellationFlag* flag,
+                base::ScopedClosureRunner* untrack_runner) {
+  return flag->IsSet();
+}
+
+void RunOrPostToTaskRunner(TaskRunner* task_runner, const Closure& closure) {
+  if (task_runner->RunsTasksOnCurrentThread())
+    closure.Run();
+  else
+    task_runner->PostTask(FROM_HERE, closure);
+}
+
 }  // namespace
 
 // static
 const CancelableTaskTracker::TaskId CancelableTaskTracker::kBadTaskId = 0;
 
 CancelableTaskTracker::CancelableTaskTracker()
     : ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)),
       next_id_(1) {}
 
 CancelableTaskTracker::~CancelableTaskTracker() {
@@ skipping 31 matching lines @@
       Bind(&RunIfNotCanceledThenUntrack,
            base::Owned(flag), reply, untrack_closure));
 
   if (!success)
     return kBadTaskId;
 
   Track(id, flag);
   return id;
 }
 
+CancelableTaskTracker::TaskId CancelableTaskTracker::NewTrackedTaskId(
+    IsCanceledCallback* is_canceled_cb) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(base::MessageLoopProxy::current());
+
+  TaskId id = next_id_;
+  next_id_++;  // int64 is big enough that we ignore the potential overflow.
+
+  // Both owned by |is_canceled_cb|.
+  CancellationFlag* flag = new CancellationFlag();
+  base::ScopedClosureRunner* untrack_runner = new base::ScopedClosureRunner(
+      Bind(&RunOrPostToTaskRunner,
+           base::MessageLoopProxy::current(),
+           Bind(&CancelableTaskTracker::Untrack,
+                weak_factory_.GetWeakPtr(), id)));
+
+  *is_canceled_cb = Bind(&IsCanceled,
+                         base::Owned(flag),
+                         base::Owned(untrack_runner));
+  Track(id, flag);
+  return id;
+}
+
 void CancelableTaskTracker::TryCancel(TaskId id) {
   DCHECK(thread_checker_.CalledOnValidThread());
 
   hash_map<TaskId, CancellationFlag*>::const_iterator it = task_flags_.find(id);
   if (it == task_flags_.end()) {
     // 3 possibilities:
     // 1. Task (and reply) has finished running.
     // 2. Task was canceled before and already untracked.
     // 3. The TaskId is bad or never used.
     // Since we only try best to cancel task and reply, it's OK to ignore these.
@@ skipping 17 matching lines @@
   bool success = task_flags_.insert(std::make_pair(id, flag)).second;
   DCHECK(success);
 }
 
 void CancelableTaskTracker::Untrack(TaskId id) {
   DCHECK(thread_checker_.CalledOnValidThread());
 
   size_t num = task_flags_.erase(id);
   DCHECK_EQ(1u, num);
 }