-
Notifications
You must be signed in to change notification settings - Fork 58
/
Copy pathProfileAttribute.cs
80 lines (64 loc) · 1.92 KB
/
ProfileAttribute.cs
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
using PostSharp.Aspects;
using PostSharp.Aspects.Advices;
using PostSharp.Serialization;
using System.Reflection;
namespace PostSharp.Samples.Profiling
{
[PSerializable]
public sealed class ProfileAttribute : MethodLevelAspect
{
MetricMetadata _metadata;
public override void RuntimeInitialize(MethodBase method)
{
this._metadata = ProfilingServices.Collector.RegisterMethod(method);
}
[OnMethodEntryAdvice, SelfPointcut]
public void OnEntry([State(StateScope.MethodInvocation)] out MethodCallData callData)
{
callData = default;
if (ProfilingServices.IsEnabled)
{
callData.Start(this._metadata);
}
}
[OnMethodSuccessAdvice(Master = nameof(OnEntry))]
public static void OnSuccess([State(StateScope.MethodInvocation)] ref MethodCallData callData)
{
if (!callData.IsNull)
{
callData.Stop();
Publish(callData);
}
}
[OnMethodExceptionAdvice(Master = nameof(OnEntry))]
public static void OnException([State(StateScope.MethodInvocation)] ref MethodCallData callData)
{
if (!callData.IsNull)
{
callData.AddException();
callData.Stop();
Publish(callData);
}
}
[OnMethodYieldAdvice(Master = nameof(OnEntry))]
public static void OnYield([State(StateScope.MethodInvocation)] ref MethodCallData callData)
{
if (!callData.IsNull)
{
callData.Pause();
}
}
[OnMethodResumeAdvice(Master = nameof(OnEntry))]
public static void OnResume([State(StateScope.MethodInvocation)] ref MethodCallData callData)
{
if (!callData.IsNull)
{
callData.Resume();
}
}
private static void Publish(in MethodCallData data)
{
ProfilingServices.Collector.GetThreadLocalCollector().AddSample(data.Metadata, data.MetricData);
}
}
}