[Sync] Signal UIModelWorker to abort on sync shutdown.

components/sync/driver/glue/ui_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 "components/sync/driver/glue/ui_model_worker.h"
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/callback.h"
-#include "base/synchronization/waitable_event.h"
 #include "components/sync/base/scoped_event_signal.h"
 
 namespace syncer {
 
 namespace {
 
-void CallDoWorkAndSignalEvent(const WorkCallback& work,
-                              syncer::ScopedEventSignal scoped_event_signal,
-                              SyncerError* error_info) {
+// |worker| owns the event backing |scoped_event_signal|, so it's important that
+// it comes after in the parameter list so it gets destroyed after.
+void DoWorkAndSignalEvent(const WorkCallback& work,
+                          syncer::ScopedEventSignal scoped_event_signal,
+                          scoped_refptr<ModelSafeWorker> worker,
+                          SyncerError* error_info) {
+  DCHECK(worker->IsOnModelThread());
+  if (worker->IsStopped()) {
+    // Don't set the error here; the pointer may no longer be valid.
+    return;
+  }
+
   *error_info = work.Run();
   // The event in |scoped_event_signal| is signaled at the end of this scope.
 }
 
 }  // namespace
 
 UIModelWorker::UIModelWorker(
     scoped_refptr<base::SingleThreadTaskRunner> ui_thread)
-    : ui_thread_(std::move(ui_thread)) {}
+    : ui_thread_(std::move(ui_thread)),
+      work_done_or_abandoned_(base::WaitableEvent::ResetPolicy::AUTOMATIC,
+                              base::WaitableEvent::InitialState::NOT_SIGNALED) {
+}
 
 SyncerError UIModelWorker::DoWorkAndWaitUntilDoneImpl(
     const WorkCallback& work) {
   SyncerError error_info;
   if (ui_thread_->BelongsToCurrentThread()) {
     DLOG(WARNING) << "DoWorkAndWaitUntilDone called from "
                   << "ui_loop_. Probably a nested invocation?";
     return work.Run();
   }
 
-  // Signaled when the task is deleted, i.e. after it runs or when it is
-  // abandoned.
-  base::WaitableEvent work_done_or_abandoned(
-      base::WaitableEvent::ResetPolicy::AUTOMATIC,
-      base::WaitableEvent::InitialState::NOT_SIGNALED);
-
-  if (!ui_thread_->PostTask(FROM_HERE,
-                            base::Bind(&CallDoWorkAndSignalEvent, work,
-                                       base::Passed(syncer::ScopedEventSignal(
-                                           &work_done_or_abandoned)),
-                                       &error_info))) {
+  if (!ui_thread_->PostTask(
+          FROM_HERE, base::Bind(&DoWorkAndSignalEvent, work,
+                                base::Passed(syncer::ScopedEventSignal(
+                                    &work_done_or_abandoned_)),
+                                make_scoped_refptr(this), &error_info))) {
     DLOG(WARNING) << "Could not post work to UI loop.";
     error_info = CANNOT_DO_WORK;
     return error_info;
   }
-  work_done_or_abandoned.Wait();
+  work_done_or_abandoned_.Wait();
 
   return error_info;
 }
 
+void UIModelWorker::RequestStop() {
+  DCHECK(ui_thread_->BelongsToCurrentThread());
+  ModelSafeWorker::RequestStop();
+  work_done_or_abandoned_.Signal();
+}
+
 ModelSafeGroup UIModelWorker::GetModelSafeGroup() {
   return GROUP_UI;
 }
 
 bool UIModelWorker::IsOnModelThread() {
   return ui_thread_->BelongsToCurrentThread();
 }
 
 UIModelWorker::~UIModelWorker() {}
 
 }  // namespace syncer