path: root/kamon-core/src/main/scala/kamon/instrumentation/akka/RoutedActorCellInstrumentation.scala

/*
 * =========================================================================================
 * 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.actor.{ActorRef, ActorSystem, Props}
import akka.dispatch.MessageDispatcher
import akka.routing.RoutedActorCell
import kamon.metric.RouterMetrics
import kamon.metric.RouterMetrics.RouterMetricsRecorder
import org.aspectj.lang.annotation._

@Aspect
class RoutedActorCellInstrumentation {

  @Pointcut("execution(akka.routing.RoutedActorCell.new(..)) && this(cell) && args(system, ref, routerProps, routerDispatcher, supervisor)")
  def actorCellCreation(cell: RoutedActorCell, system: ActorSystem, ref: ActorRef, routerProps: Props, routerDispatcher: MessageDispatcher, supervisor: ActorRef ): Unit = {}

//  @After("actorCellCreation(cell, system, ref, routerProps, routerDispatcher, supervisor)")
  @After("execution(akka.routing.RoutedActorCell.new(..)) && this(cell) && args(*, ref, *, *, *, *)")
//  def afterCreation(cell: RoutedActorCell, system: ActorSystem, ref: ActorRef, routerProps: Props, routerDispatcher: MessageDispatcher, supervisor: ActorRef): Unit = {
  def a(cell: RoutedActorCell, ref: ActorRef)  = {

  print("adf;kjaskadjlfaj"+ ref)
//    cell.router.routees
//    val metricsExtension = Kamon(Metrics)(system)
//    val metricIdentity = RouterMetrics(ref.path.elements.mkString("/"))
//    val cellWithMetrics = cell.asInstanceOf[RoutedActorCellMetrics]
//
//    cellWithMetrics.metricIdentity = metricIdentity
//    cellWithMetrics.routerMetricsRecorder = metricsExtension.register(metricIdentity, RouterMetrics.Factory)
  }
}

trait RoutedActorCellMetrics {
  var metricIdentity: RouterMetrics = _
  var routerMetricsRecorder: Option[RouterMetricsRecorder] = _
}

@Aspect
class RoutedActorCellMetricsIntoRoutedActorCellMixin {

  @DeclareMixin("akka.routing.RoutedActorCell")
  def mixinRoutedActorCellMetricsToRoutedActorCell: RoutedActorCellMetrics = new RoutedActorCellMetrics {}
}

//  @Pointcut("(execution(* akka.actor.ActorCell.invoke(*)) || execution(* akka.routing.RoutedActorCell.sendMessage(*))) && this(cell) && args(envelope)")
//  def invokingActorBehaviourAtActorCell(cell: ActorCell, envelope: Envelope) = {}
//
//  @Around("invokingActorBehaviourAtActorCell(cell, envelope)")
//  def aroundBehaviourInvoke(pjp: ProceedingJoinPoint, cell: ActorCell, envelope: Envelope): Any = {
//    val timestampBeforeProcessing = System.nanoTime()
//    val contextAndTimestamp = envelope.asInstanceOf[TraceContextAware]
//    val cellWithMetrics = cell.asInstanceOf[ActorCellMetrics]
//
//    try {
//      TraceRecorder.withInlineTraceContextReplacement(contextAndTimestamp.traceContext) {
//        pjp.proceed()
//      }
//    } finally {
//      cellWithMetrics.actorMetricsRecorder.map {
//        am ⇒
//          am.processingTime.record(System.nanoTime() - timestampBeforeProcessing)
//          am.timeInMailbox.record(timestampBeforeProcessing - contextAndTimestamp.captureNanoTime)
//          am.mailboxSize.decrement()
//      }
//    }
//  }
//
//  @Pointcut("execution(* akka.actor.ActorCell.sendMessage(*)) && this(cell)")
//  def sendingMessageToActorCell(cell: ActorCell): Unit = {}
//
//  @After("sendingMessageToActorCell(cell)")
//  def afterSendMessageToActorCell(cell: ActorCell): Unit = {
//    val cellWithMetrics = cell.asInstanceOf[ActorCellMetrics]
//    cellWithMetrics.actorMetricsRecorder.map(am ⇒ am.mailboxSize.increment())
//  }
//
//  @Pointcut("execution(* akka.actor.ActorCell.stop()) && this(cell)")
//  def actorStop(cell: ActorCell): Unit = {}
//
//  @After("actorStop(cell)")
//  def afterStop(cell: ActorCell): Unit = {
//    val cellWithMetrics = cell.asInstanceOf[ActorCellMetrics]
//
//    cellWithMetrics.actorMetricsRecorder.map { p ⇒
//      Kamon(Metrics)(cell.system).unregister(cellWithMetrics.metricIdentity)
//    }
//  }
//
//  @Pointcut("execution(* akka.actor.ActorCell.handleInvokeFailure(..)) && this(cell)")
//  def actorInvokeFailure(cell: ActorCell): Unit = {}
//
//  @Before("actorInvokeFailure(cell)")
//  def beforeInvokeFailure(cell: ActorCell): Unit = {
//    val cellWithMetrics = cell.asInstanceOf[ActorCellMetrics]
//
//    cellWithMetrics.actorMetricsRecorder.map {
//      am ⇒ am.errors.increment()
//    }
//  }
//}
//
//trait ActorCellMetrics {
//  var metricIdentity: ActorMetrics = _
//  var actorMetricsRecorder: Option[ActorMetricsRecorder] = _
//}
//
//@Aspect
//class ActorCellMetricsIntoActorCellMixin {
//
//  @DeclareMixin("akka.actor.ActorCell")
//  def mixinActorCellMetricsToActorCell: ActorCellMetrics = new ActorCellMetrics {}
//}
//
//@Aspect
//class TraceContextIntoEnvelopeMixin {
//
//  @DeclareMixin("akka.dispatch.Envelope")
//  def mixinTraceContextAwareToEnvelope: TraceContextAware = TraceContextAware.default
//
//  @Pointcut("execution(akka.dispatch.Envelope.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
//  }
//}