+ core: provide metrics for routers

* processing-time
* errors
* time-in-mailbox
closes #62

1 files changed, 143 insertions, 0 deletions

diff --git a/kamon-core/src/main/scala/kamon/instrumentation/akka/RoutedActorCellInstrumentation.scala b/kamon-core/src/main/scala/kamon/instrumentation/akka/RoutedActorCellInstrumentation.scala
new file mode 100644
index 00000000..f75080db
--- /dev/null
+++ b/kamon-core/src/main/scala/kamon/instrumentation/akka/RoutedActorCellInstrumentation.scala
@@ -0,0 +1,143 @@
+/*
+ * =========================================================================================
+ * 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
+//  }
+//}
\ No newline at end of file