1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
3 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
4 // Use of this source code is governed by a BSD-style license that can be
5 // found in the LICENSE file.
6
7 #include "base/message_pump_default.h"
8
9 #include "base/logging.h"
10 #include "base/message_loop.h"
11 #include "base/scoped_nsautorelease_pool.h"
12 #include "GeckoProfiler.h"
13
14 #include "mozilla/BackgroundHangMonitor.h"
15
16 namespace base {
17
MessagePumpDefault()18 MessagePumpDefault::MessagePumpDefault()
19 : keep_running_(true), event_(false, false) {}
20
Run(Delegate * delegate)21 void MessagePumpDefault::Run(Delegate* delegate) {
22 AUTO_PROFILER_LABEL("MessagePumpDefault::Run", OTHER);
23
24 DCHECK(keep_running_) << "Quit must have been called outside of Run!";
25
26 const MessageLoop* const loop = MessageLoop::current();
27 mozilla::BackgroundHangMonitor hangMonitor(loop->thread_name().c_str(),
28 loop->transient_hang_timeout(),
29 loop->permanent_hang_timeout());
30
31 for (;;) {
32 ScopedNSAutoreleasePool autorelease_pool;
33
34 hangMonitor.NotifyActivity();
35 bool did_work = delegate->DoWork();
36 if (!keep_running_) break;
37
38 hangMonitor.NotifyActivity();
39 did_work |= delegate->DoDelayedWork(&delayed_work_time_);
40 if (!keep_running_) break;
41
42 if (did_work) continue;
43
44 hangMonitor.NotifyActivity();
45 did_work = delegate->DoIdleWork();
46 if (!keep_running_) break;
47
48 if (did_work) continue;
49
50 if (delayed_work_time_.is_null()) {
51 hangMonitor.NotifyWait();
52 AUTO_PROFILER_LABEL("MessagePumpDefault::Run:Wait", IDLE);
53 {
54 AUTO_PROFILER_THREAD_SLEEP;
55 event_.Wait();
56 }
57 } else {
58 TimeDelta delay = delayed_work_time_ - TimeTicks::Now();
59 if (delay > TimeDelta()) {
60 hangMonitor.NotifyWait();
61 AUTO_PROFILER_LABEL("MessagePumpDefault::Run:Wait", IDLE);
62 {
63 AUTO_PROFILER_THREAD_SLEEP;
64 event_.TimedWait(delay);
65 }
66 } else {
67 // It looks like delayed_work_time_ indicates a time in the past, so we
68 // need to call DoDelayedWork now.
69 delayed_work_time_ = TimeTicks();
70 }
71 }
72 // Since event_ is auto-reset, we don't need to do anything special here
73 // other than service each delegate method.
74 }
75
76 keep_running_ = true;
77 }
78
Quit()79 void MessagePumpDefault::Quit() { keep_running_ = false; }
80
ScheduleWork()81 void MessagePumpDefault::ScheduleWork() {
82 // Since this can be called on any thread, we need to ensure that our Run
83 // loop wakes up.
84 event_.Signal();
85 }
86
ScheduleDelayedWork(const TimeTicks & delayed_work_time)87 void MessagePumpDefault::ScheduleDelayedWork(
88 const TimeTicks& delayed_work_time) {
89 // We know that we can't be blocked on Wait right now since this method can
90 // only be called on the same thread as Run, so we only need to update our
91 // record of how long to sleep when we do sleep.
92 delayed_work_time_ = delayed_work_time;
93 }
94
95 } // namespace base
96