| Index: components/sync/engine/ui_model_worker.cc
|
| diff --git a/components/sync/engine/ui_model_worker.cc b/components/sync/engine/ui_model_worker.cc
|
| index 08cc1a2502bbd09f396b756bc47bd070eca0782b..77fe9c7799a65db7437df9cc45edf62a0611467c 100644
|
| --- a/components/sync/engine/ui_model_worker.cc
|
| +++ b/components/sync/engine/ui_model_worker.cc
|
| @@ -6,11 +6,90 @@
|
|
|
| #include <utility>
|
|
|
| +#include "base/bind.h"
|
| +#include "base/bind_helpers.h"
|
| +#include "base/callback.h"
|
| +#include "components/sync/base/scoped_event_signal.h"
|
| +
|
| namespace syncer {
|
| +
|
| +namespace {
|
| +
|
| +class ScopedEventSignalWithWorker {
|
| + public:
|
| + ScopedEventSignalWithWorker(scoped_refptr<UIModelWorker> ui_model_worker,
|
| + base::WaitableEvent* event)
|
| + : ui_model_worker_(std::move(ui_model_worker)),
|
| + scoped_event_signal_(event) {}
|
| +
|
| + ScopedEventSignalWithWorker(ScopedEventSignalWithWorker&&) = default;
|
| + ScopedEventSignalWithWorker& operator=(ScopedEventSignalWithWorker&&) =
|
| + default;
|
| +
|
| + bool IsStopped() const { return ui_model_worker_->IsStopped(); }
|
| +
|
| + private:
|
| + // This reference prevents the event in |scoped_event_signal_| from being
|
| + // signaled after being deleted.
|
| + scoped_refptr<UIModelWorker> ui_model_worker_;
|
| +
|
| + ScopedEventSignal scoped_event_signal_;
|
| +
|
| + DISALLOW_COPY_AND_ASSIGN(ScopedEventSignalWithWorker);
|
| +};
|
| +
|
| +void CallDoWorkAndSignalEvent(
|
| + const WorkCallback& work,
|
| + ScopedEventSignalWithWorker scoped_event_signal_with_worker,
|
| + SyncerError* error_info) {
|
| + if (!scoped_event_signal_with_worker.IsStopped())
|
| + *error_info = work.Run();
|
| + // The event in |scoped_event_signal_with_worker| 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::MANUAL,
|
| + base::WaitableEvent::InitialState::NOT_SIGNALED),
|
| + stop_requested_(base::WaitableEvent::ResetPolicy::MANUAL,
|
| + base::WaitableEvent::InitialState::NOT_SIGNALED) {
|
| + sequence_checker_.DetachFromSequence();
|
| +}
|
| +
|
| +SyncerError UIModelWorker::DoWorkAndWaitUntilDoneImpl(
|
| + const WorkCallback& work) {
|
| + DCHECK(sequence_checker_.CalledOnValidSequence());
|
| + DCHECK(!ui_thread_->BelongsToCurrentThread());
|
| +
|
| + SyncerError error_info;
|
| + work_done_or_abandoned_.Reset();
|
| +
|
| + if (!ui_thread_->PostTask(
|
| + FROM_HERE,
|
| + base::Bind(&CallDoWorkAndSignalEvent, work,
|
| + base::Passed(syncer::ScopedEventSignalWithWorker(
|
| + this, &work_done_or_abandoned_)),
|
| + &error_info))) {
|
| + DLOG(WARNING) << "Could not post work to UI loop.";
|
| + error_info = CANNOT_DO_WORK;
|
| + return error_info;
|
| + }
|
| +
|
| + base::WaitableEvent* events[] = {&work_done_or_abandoned_, &stop_requested_};
|
| + base::WaitableEvent::WaitMany(events, arraysize(events));
|
| +
|
| + return error_info;
|
| +}
|
| +
|
| +void UIModelWorker::RequestStop() {
|
| + DCHECK(ui_thread_->BelongsToCurrentThread());
|
| + ModelSafeWorker::RequestStop();
|
| + stop_requested_.Signal();
|
| +}
|
|
|
| ModelSafeGroup UIModelWorker::GetModelSafeGroup() {
|
| return GROUP_UI;
|
| @@ -22,8 +101,4 @@
|
|
|
| UIModelWorker::~UIModelWorker() {}
|
|
|
| -void UIModelWorker::ScheduleWork(base::OnceClosure work) {
|
| - ui_thread_->PostTask(FROM_HERE, std::move(work));
|
| -}
|
| -
|
| } // namespace syncer
|
|
|