diff options
author | Ivan Topolnjak <ivantopo@gmail.com> | 2014-07-29 00:56:39 -0300 |
---|---|---|
committer | Ivan Topolnjak <ivantopo@gmail.com> | 2014-07-29 00:56:39 -0300 |
commit | 96f31cbac82bf350db61e4b48ea95bb2706cc02b (patch) | |
tree | 482a479a898e77f2484a48daeafae962613bd468 /kamon-core/src/main/scala/kamon/instrumentation/akka/ActorLoggingInstrumentation.scala | |
parent | ec97bccd2346aa8c50997ec1f198a5eb6c79eab5 (diff) | |
parent | e83babb5f88e91661bec2f1013fcb6b03612bea9 (diff) | |
download | Kamon-96f31cbac82bf350db61e4b48ea95bb2706cc02b.tar.gz Kamon-96f31cbac82bf350db61e4b48ea95bb2706cc02b.tar.bz2 Kamon-96f31cbac82bf350db61e4b48ea95bb2706cc02b.zip |
Merge branch 'master' into release-0.3_scala-2.11
Conflicts:
.travis.yml
project/Settings.scala
Diffstat (limited to 'kamon-core/src/main/scala/kamon/instrumentation/akka/ActorLoggingInstrumentation.scala')
-rw-r--r-- | kamon-core/src/main/scala/kamon/instrumentation/akka/ActorLoggingInstrumentation.scala | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/kamon-core/src/main/scala/kamon/instrumentation/akka/ActorLoggingInstrumentation.scala b/kamon-core/src/main/scala/kamon/instrumentation/akka/ActorLoggingInstrumentation.scala new file mode 100644 index 00000000..ee9d442f --- /dev/null +++ b/kamon-core/src/main/scala/kamon/instrumentation/akka/ActorLoggingInstrumentation.scala @@ -0,0 +1,47 @@ +/* + * ========================================================================================= + * Copyright © 2013-2014 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 akka.instrumentation + +import kamon.trace.{ TraceContextAware, TraceRecorder } +import org.aspectj.lang.ProceedingJoinPoint +import org.aspectj.lang.annotation._ + +@Aspect +class ActorLoggingInstrumentation { + + @DeclareMixin("akka.event.Logging.LogEvent+") + def mixinTraceContextAwareToLogEvent: TraceContextAware = TraceContextAware.default + + @Pointcut("execution(akka.event.Logging.LogEvent+.new(..)) && this(event)") + def logEventCreation(event: TraceContextAware): Unit = {} + + @After("logEventCreation(event)") + def captureTraceContext(event: TraceContextAware): Unit = { + // Force initialization of TraceContextAware + event.traceContext + } + + @Pointcut("execution(* akka.event.slf4j.Slf4jLogger.withMdc(..)) && args(logSource, logEvent, logStatement)") + def withMdcInvocation(logSource: String, logEvent: TraceContextAware, logStatement: () ⇒ _): Unit = {} + + @Around("withMdcInvocation(logSource, logEvent, logStatement)") + def aroundWithMdcInvocation(pjp: ProceedingJoinPoint, logSource: String, logEvent: TraceContextAware, logStatement: () ⇒ _): Unit = { + TraceRecorder.withTraceContext(logEvent.traceContext) { + pjp.proceed() + } + } +} |