1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "DelayedRunnable.h"
#include "mozilla/ProfilerRunnable.h"
namespace mozilla {
DelayedRunnable::DelayedRunnable(already_AddRefed<nsISerialEventTarget> aTarget,
already_AddRefed<nsIRunnable> aRunnable,
uint32_t aDelay)
: mozilla::Runnable("DelayedRunnable"),
mTarget(aTarget),
mDelayedFrom(TimeStamp::NowLoRes()),
mDelay(aDelay),
mWrappedRunnable(aRunnable) {}
nsresult DelayedRunnable::Init() {
MutexAutoLock lock(mMutex);
if (!mWrappedRunnable) {
MOZ_ASSERT_UNREACHABLE();
return NS_ERROR_INVALID_ARG;
}
nsresult rv = mTarget->RegisterShutdownTask(this);
if (NS_FAILED(rv)) {
MOZ_DIAGNOSTIC_ASSERT(
rv == NS_ERROR_UNEXPECTED,
"DelayedRunnable target must support RegisterShutdownTask");
NS_WARNING("DelayedRunnable init after target is shutdown");
return rv;
}
rv = NS_NewTimerWithCallback(getter_AddRefs(mTimer), this, mDelay,
nsITimer::TYPE_ONE_SHOT, mTarget);
if (NS_WARN_IF(NS_FAILED(rv))) {
mTarget->UnregisterShutdownTask(this);
}
return rv;
}
NS_IMETHODIMP DelayedRunnable::Run() {
MOZ_ASSERT(mTarget->IsOnCurrentThread());
nsCOMPtr<nsIRunnable> runnable;
{
MutexAutoLock lock(mMutex);
MOZ_ASSERT(mTimer, "Init() must have been called");
// Already ran?
if (!mWrappedRunnable) {
return NS_OK;
}
// Are we too early?
if ((mozilla::TimeStamp::NowLoRes() - mDelayedFrom).ToMilliseconds() <
mDelay) {
return NS_OK; // Let the nsITimer run us.
}
mTimer->Cancel();
mTarget->UnregisterShutdownTask(this);
runnable = mWrappedRunnable.forget();
}
AUTO_PROFILE_FOLLOWING_RUNNABLE(runnable);
return runnable->Run();
}
NS_IMETHODIMP DelayedRunnable::Notify(nsITimer* aTimer) {
MOZ_ASSERT(mTarget->IsOnCurrentThread());
nsCOMPtr<nsIRunnable> runnable;
{
MutexAutoLock lock(mMutex);
MOZ_ASSERT(mTimer, "Init() must have been called");
// We may have already run due to races
if (!mWrappedRunnable) {
return NS_OK;
}
mTarget->UnregisterShutdownTask(this);
runnable = mWrappedRunnable.forget();
}
AUTO_PROFILE_FOLLOWING_RUNNABLE(runnable);
return runnable->Run();
}
void DelayedRunnable::TargetShutdown() {
MOZ_ASSERT(mTarget->IsOnCurrentThread());
// Called at shutdown
MutexAutoLock lock(mMutex);
if (!mWrappedRunnable) {
return;
}
mWrappedRunnable = nullptr;
if (mTimer) {
mTimer->Cancel();
}
}
NS_IMPL_ISUPPORTS_INHERITED(DelayedRunnable, Runnable, nsITimerCallback,
nsITargetShutdownTask)
} // namespace mozilla
|
