blob: 9b6b6866f1956228a33ad028dea295292e2de24b (
plain) (
tree)
|
|
/*
* =========================================================================================
* 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 akka.dispatch.sysmsg.EarliestFirstSystemMessageList
import kamon.trace.{ TraceContextAware, TraceRecorder }
import org.aspectj.lang.ProceedingJoinPoint
import org.aspectj.lang.annotation._
@Aspect
class ActorSystemMessageInstrumentation {
@Pointcut("execution(* akka.actor.ActorCell.invokeAll$1(..)) && args(messages, *)")
def systemMessageProcessing(messages: EarliestFirstSystemMessageList): Unit = {}
@Around("systemMessageProcessing(messages)")
def aroundSystemMessageInvoke(pjp: ProceedingJoinPoint, messages: EarliestFirstSystemMessageList): Any = {
if (messages.nonEmpty) {
val ctx = messages.head.asInstanceOf[TraceContextAware].traceContext
TraceRecorder.withTraceContext(ctx)(pjp.proceed())
} else pjp.proceed()
}
}
@Aspect
class TraceContextIntoSystemMessageMixin {
@DeclareMixin("akka.dispatch.sysmsg.SystemMessage+")
def mixinTraceContextAwareToSystemMessage: TraceContextAware = TraceContextAware.default
@Pointcut("execution(akka.dispatch.sysmsg.SystemMessage+.new(..)) && this(ctx)")
def envelopeCreation(ctx: TraceContextAware): Unit = {}
@After("envelopeCreation(ctx)")
def afterEnvelopeCreation(ctx: TraceContextAware): Unit = {
// Necessary to force the initialization of ContextAware at the moment of creation.
ctx.traceContext
}
}
@Aspect
class TraceContextIntoRepointableActorRefMixin {
@DeclareMixin("akka.actor.RepointableActorRef")
def mixinTraceContextAwareToRepointableActorRef: TraceContextAware = TraceContextAware.default
@Pointcut("execution(akka.actor.RepointableActorRef.new(..)) && this(ctx)")
def envelopeCreation(ctx: TraceContextAware): Unit = {}
@After("envelopeCreation(ctx)")
def afterEnvelopeCreation(ctx: TraceContextAware): Unit = {
// Necessary to force the initialization of ContextAware at the moment of creation.
ctx.traceContext
}
@Pointcut("execution(* akka.actor.RepointableActorRef.point(..)) && this(repointableActorRef)")
def repointableActorRefCreation(repointableActorRef: TraceContextAware): Unit = {}
@Around("repointableActorRefCreation(repointableActorRef)")
def afterRepointableActorRefCreation(pjp: ProceedingJoinPoint, repointableActorRef: TraceContextAware): Any = {
TraceRecorder.withTraceContext(repointableActorRef.traceContext) {
pjp.proceed()
}
}
}
|