Port HistoryService::ScheduleDBTask to CancelableTaskTracker

chrome/browser/sync/glue/history_model_worker.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/browser/sync/glue/history_model_worker.h"
 
 #include "base/memory/ref_counted.h"
 #include "base/message_loop/message_loop.h"
 #include "base/synchronization/waitable_event.h"
 #include "content/public/browser/browser_thread.h"
@@ skipping 48 matching lines @@
 
   base::Closure register_callback_;
 };
 
 namespace {
 
 // Post the work task on |history_service|'s DB thread from the UI
 // thread.
 void PostWorkerTask(const base::WeakPtr<HistoryService>& history_service,
                     const syncer::WorkCallback& work,
-                    CancelableRequestConsumerT<int, 0>* cancelable_consumer,
+                    base::CancelableTaskTracker* cancelable_tracker,
                     WaitableEvent* done,
                     syncer::SyncerError* error) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   if (history_service.get()) {
     scoped_refptr<WorkerTask> task(new WorkerTask(work, done, error));
-    history_service->ScheduleDBTask(task.get(), cancelable_consumer);
+    history_service->ScheduleDBTask(task.get(), cancelable_tracker);
   } else {
     *error = syncer::CANNOT_DO_WORK;
     done->Signal();
   }
 }
 
 }  // namespace
 
 HistoryModelWorker::HistoryModelWorker(
     const base::WeakPtr<HistoryService>& history_service,
     syncer::WorkerLoopDestructionObserver* observer)
   : syncer::ModelSafeWorker(observer),
     history_service_(history_service) {
   CHECK(history_service.get());
+  CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  cancelable_tracker_.reset(new base::CancelableTaskTracker);
 }
 
 void HistoryModelWorker::RegisterForLoopDestruction() {
   CHECK(history_service_.get());
   history_service_->ScheduleDBTask(
       new AddDBThreadObserverTask(
           base::Bind(&HistoryModelWorker::RegisterOnDBThread, this)),
-      &cancelable_consumer_);
+      cancelable_tracker_.get());
 }
 
 void HistoryModelWorker::RegisterOnDBThread() {
   base::MessageLoop::current()->AddDestructionObserver(this);
   SetWorkingLoopToCurrent();
 }
 
 syncer::SyncerError HistoryModelWorker::DoWorkAndWaitUntilDoneImpl(
     const syncer::WorkCallback& work) {
   syncer::SyncerError error = syncer::UNSET;
-  if (BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
-                              base::Bind(&PostWorkerTask, history_service_,
-                                         work, &cancelable_consumer_,
+  if (BrowserThread::PostTask(BrowserThread::UI,
+                              FROM_HERE,
+                              base::Bind(&PostWorkerTask,
+                                         history_service_,
+                                         work,
+                                         cancelable_tracker_.get(),
                                          work_done_or_stopped(),
                                          &error))) {
     work_done_or_stopped()->Wait();
   } else {
     error = syncer::CANNOT_DO_WORK;
   }
   return error;
 }
 
 syncer::ModelSafeGroup HistoryModelWorker::GetModelSafeGroup() {
   return syncer::GROUP_HISTORY;
 }
 
-HistoryModelWorker::~HistoryModelWorker() {}
+HistoryModelWorker::~HistoryModelWorker() {
+  // The base::CancelableTaskTracker class is not thread-safe and must only be
+  // used from a single thread but the current object may not be destroyed from
+  // the UI thread, so delete it from the UI thread.
+  BrowserThread::DeleteOnUIThread::Destruct(cancelable_tracker_.release());
+}
 
 }  // namespace browser_sync