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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim:set ts=2 sw=2 sts=2 et cindent: */
/* 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/. */
#ifndef TRACED_TASK_COMMON_H
#define TRACED_TASK_COMMON_H
#include "GeckoTaskTracer.h"
#include "nsCOMPtr.h"
#include "nsThreadUtils.h"
namespace mozilla {
namespace tasktracer {
class TracedTaskCommon {
public:
TracedTaskCommon();
TracedTaskCommon(const TracedTaskCommon& aSrc)
: mSourceEventType(aSrc.mSourceEventType),
mSourceEventId(aSrc.mSourceEventId),
mParentTaskId(aSrc.mParentTaskId),
mTaskId(aSrc.mTaskId),
mIsTraceInfoInit(aSrc.mIsTraceInfoInit) {}
virtual ~TracedTaskCommon();
void DispatchTask(int aDelayTimeMs = 0);
void SetTLSTraceInfo() {
if (mIsTraceInfoInit) {
DoSetTLSTraceInfo();
}
}
void GetTLSTraceInfo() {
if (IsStartLogging()) {
DoGetTLSTraceInfo();
}
}
void ClearTLSTraceInfo();
private:
void DoSetTLSTraceInfo();
void DoGetTLSTraceInfo();
protected:
void Init();
// TraceInfo of TLS will be set by the following parameters, including source
// event type, source event ID, parent task ID, and task ID of this traced
// task/runnable.
SourceEventType mSourceEventType;
uint64_t mSourceEventId;
uint64_t mParentTaskId;
uint64_t mTaskId;
bool mIsTraceInfoInit;
};
class TracedRunnable : public TracedTaskCommon, public nsIRunnable {
public:
NS_DECL_THREADSAFE_ISUPPORTS
NS_DECL_NSIRUNNABLE
explicit TracedRunnable(already_AddRefed<nsIRunnable>&& aOriginalObj);
private:
virtual ~TracedRunnable();
nsCOMPtr<nsIRunnable> mOriginalObj;
};
/**
* This class is used to create a logical task, without a real
* runnable.
*/
class VirtualTask : public TracedTaskCommon {
public:
VirtualTask() : TracedTaskCommon() {}
VirtualTask(const VirtualTask& aSrc) : TracedTaskCommon(aSrc) {}
/**
* Initialize the task to create an unique ID, and store other
* information.
*
* This method may be called for one or more times.
*/
void Init(uintptr_t* aVPtr = nullptr) {
TracedTaskCommon::Init();
if (aVPtr) {
extern void LogVirtualTablePtr(uint64_t aTaskId, uint64_t aSourceEventId,
uintptr_t * aVptr);
LogVirtualTablePtr(mTaskId, mSourceEventId, aVPtr);
}
DispatchTask();
}
/**
* Define the life-span of a VirtualTask.
*
* VirtualTask is not a real task, goes without a runnable, it's
* instances are never dispatched and ran by event loops. This
* class used to define running time as the life-span of it's
* instance.
*/
class AutoRunTask : public AutoSaveCurTraceInfo {
VirtualTask* mTask;
void StartScope(VirtualTask* aTask);
void StopScope();
public:
explicit AutoRunTask(VirtualTask* aTask)
: AutoSaveCurTraceInfo(), mTask(aTask) {
if (HasSavedTraceInfo()) {
StartScope(aTask);
}
}
~AutoRunTask() {
if (HasSavedTraceInfo()) {
StopScope();
}
}
};
};
} // namespace tasktracer
} // namespace mozilla
#endif
|
