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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
|
/*
* =========================================================================================
* Copyright © 2013-2015 the kamon project <http://kamon.io/>
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
* except in compliance with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software distributed under the
* License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
* either express or implied. See the License for the specific language governing permissions
* and limitations under the License.
* =========================================================================================
*/
package kamon.annotation.instrumentation
import java.util.concurrent.atomic.AtomicReferenceArray
import javax.el.ELProcessor
import kamon.Kamon
import kamon.annotation.el.{ EnhancedELProcessor, ELProcessorFactory }
import kamon.annotation.{ Histogram, _ }
import kamon.metric.instrument.Histogram.DynamicRange
import kamon.metric.instrument.{ Counter, MinMaxCounter }
import kamon.metric._
import kamon.trace.Tracer
import kamon.util.Latency
import org.aspectj.lang.{ JoinPoint, ProceedingJoinPoint }
import org.aspectj.lang.annotation.{ Aspect, DeclareMixin }
import org.aspectj.lang.reflect.MethodSignature
import EnhancedELProcessor.Syntax
class BaseAnnotationInstrumentation {
@inline final def registerTime(jps: JoinPoint.StaticPart, histograms: AtomicReferenceArray[instrument.Histogram], evalString: StringEvaluator, evalTags: TagsEvaluator): instrument.Histogram = {
val method = jps.getSignature.asInstanceOf[MethodSignature].getMethod
val time = method.getAnnotation(classOf[Time])
val name = evalString(time.name())
val tags = evalTags(time.tags())
val currentHistogram = Kamon.metrics.histogram(name, tags)
histograms.set(jps.getId, currentHistogram)
currentHistogram
}
@inline final def registerHistogram(jps: JoinPoint.StaticPart, histograms: AtomicReferenceArray[instrument.Histogram], evalString: StringEvaluator, evalTags: TagsEvaluator): instrument.Histogram = {
val method = jps.getSignature.asInstanceOf[MethodSignature].getMethod
val histogram = method.getAnnotation(classOf[Histogram])
val name = evalString(histogram.name())
val tags = evalTags(histogram.tags())
val dynamicRange = DynamicRange(histogram.lowestDiscernibleValue(), histogram.highestTrackableValue(), histogram.precision())
val currentHistogram = Kamon.metrics.histogram(name, tags, dynamicRange)
histograms.set(jps.getId, currentHistogram)
currentHistogram
}
@inline final def registerMinMaxCounter(jps: JoinPoint.StaticPart, minMaxCounters: AtomicReferenceArray[MinMaxCounter], evalString: StringEvaluator, evalTags: TagsEvaluator): instrument.MinMaxCounter = {
val method = jps.getSignature.asInstanceOf[MethodSignature].getMethod
val minMaxCount = method.getAnnotation(classOf[MinMaxCount])
val name = evalString(minMaxCount.name())
val tags = evalTags(minMaxCount.tags())
val minMaxCounter = Kamon.metrics.minMaxCounter(name, tags)
minMaxCounters.set(jps.getId, minMaxCounter)
minMaxCounter
}
@inline final def registerCounter(jps: JoinPoint.StaticPart, counters: AtomicReferenceArray[Counter], evalString: StringEvaluator, evalTags: TagsEvaluator): instrument.Counter = {
val method = jps.getSignature.asInstanceOf[MethodSignature].getMethod
val count = method.getAnnotation(classOf[Count])
val name = evalString(count.name())
val tags = evalTags(count.tags())
val counter = Kamon.metrics.counter(name, tags)
counters.set(jps.getId, counter)
counter
}
@inline final def registerTrace(jps: JoinPoint.StaticPart, traces: AtomicReferenceArray[TraceContextInfo], evalString: StringEvaluator, evalTags: TagsEvaluator): TraceContextInfo = {
val method = jps.getSignature.asInstanceOf[MethodSignature].getMethod
val trace = method.getAnnotation(classOf[Trace])
val name = evalString(trace.value())
val tags = evalTags(trace.tags())
val traceContextInfo = TraceContextInfo(name, tags)
traces.set(jps.getId, traceContextInfo)
traceContextInfo
}
@inline final def registerSegment(jps: JoinPoint.StaticPart, segments: AtomicReferenceArray[SegmentInfo], evalString: StringEvaluator, evalTags: TagsEvaluator): SegmentInfo = {
val method = jps.getSignature.asInstanceOf[MethodSignature].getMethod
val segment = method.getAnnotation(classOf[Segment])
val name = evalString(segment.name())
val category = evalString(segment.category())
val library = evalString(segment.library())
val tags = evalTags(segment.tags())
val segmentInfo = SegmentInfo(name, category, library, tags)
segments.set(jps.getId, segmentInfo)
segmentInfo
}
@inline final def processTrace(traceInfo: TraceContextInfo, pjp: ProceedingJoinPoint): AnyRef = {
Tracer.withContext(Kamon.tracer.newContext(traceInfo.name)) {
traceInfo.tags.foreach { case (key, value) ⇒ Tracer.currentContext.addTag(key, value) }
val result = pjp.proceed()
Tracer.currentContext.finish()
result
}
}
@inline final def processSegment(segmentInfo: SegmentInfo, pjp: ProceedingJoinPoint): AnyRef = {
Tracer.currentContext.collect { ctx ⇒
val currentSegment = ctx.startSegment(segmentInfo.name, segmentInfo.category, segmentInfo.library)
segmentInfo.tags.foreach { case (key, value) ⇒ currentSegment.addMetadata(key, value) }
val result = pjp.proceed()
currentSegment.finish()
result
} getOrElse pjp.proceed()
}
@inline final def processTime(histogram: instrument.Histogram, pjp: ProceedingJoinPoint): AnyRef = {
Latency.measure(histogram)(pjp.proceed)
}
@inline final def processHistogram(histogram: instrument.Histogram, result: AnyRef, jps: JoinPoint.StaticPart): Unit = {
histogram.record(result.asInstanceOf[Number].longValue())
}
final def processCount(counter: instrument.Counter, pjp: ProceedingJoinPoint): AnyRef = {
try pjp.proceed() finally counter.increment()
}
final def processMinMax(minMaxCounter: instrument.MinMaxCounter, pjp: ProceedingJoinPoint): AnyRef = {
minMaxCounter.increment()
try pjp.proceed() finally minMaxCounter.decrement()
}
private[annotation] def declaringType(signature: org.aspectj.lang.Signature) = signature.getDeclaringType
}
@Aspect
class ClassToAnnotationInstrumentsMixin {
@DeclareMixin("(@kamon.annotation.EnableKamon *)")
def mixinClassToAnnotationInstruments: AnnotationInstruments = new AnnotationInstruments {}
}
trait AnnotationInstruments {
var traces: AtomicReferenceArray[TraceContextInfo] = _
var segments: AtomicReferenceArray[SegmentInfo] = _
var histograms: AtomicReferenceArray[instrument.Histogram] = _
var timeHistograms: AtomicReferenceArray[instrument.Histogram] = _
var counters: AtomicReferenceArray[Counter] = _
var minMaxCounters: AtomicReferenceArray[MinMaxCounter] = _
}
case class SegmentInfo(name: String, category: String, library: String, tags: Map[String, String])
case class TraceContextInfo(name: String, tags: Map[String, String])
abstract class StringEvaluator(val processor: ELProcessor) extends (String ⇒ String)
object StringEvaluator {
def apply(obj: AnyRef) = new StringEvaluator(ELProcessorFactory.withObject(obj)) {
def apply(str: String) = processor.evalToString(str)
}
def apply(clazz: Class[_]) = new StringEvaluator(ELProcessorFactory.withClass(clazz)) {
def apply(str: String) = processor.evalToString(str)
}
}
abstract class TagsEvaluator(val processor: ELProcessor) extends (String ⇒ Map[String, String])
object TagsEvaluator {
def apply(obj: AnyRef) = new TagsEvaluator(ELProcessorFactory.withObject(obj)) {
def apply(str: String) = processor.evalToMap(str)
}
def apply(clazz: Class[_]) = new TagsEvaluator(ELProcessorFactory.withClass(clazz)) {
def apply(str: String) = processor.evalToMap(str)
}
}
|
