All Downloads are FREE. Search and download functionalities are using the official Maven repository.

akka.event.LoggingReceive.scala Maven / Gradle / Ivy

There is a newer version: 2.2.6.3
Show newest version
/*
 * Copyright (C) 2009-2019 Lightbend Inc. 
 */

package akka.event

import language.existentials
import akka.actor.Actor.Receive
import akka.actor.ActorContext
import akka.actor.ActorCell
import akka.actor.DiagnosticActorLogging
import akka.event.Logging.{ LogEvent, LogLevel }
import akka.actor.AbstractActor
import scala.runtime.BoxedUnit

object LoggingReceive {

  /**
   * Wrap a Receive partial function in a logging enclosure, which sends a
   * debug message to the event bus each time before a message is matched.
   * This includes messages which are not handled.
   *
   * 

   * def receive = LoggingReceive {
   *   case x => ...
   * }
   * 
* * This method does NOT modify the given Receive unless * `akka.actor.debug.receive` is set in configuration. */ def apply(r: Receive)(implicit context: ActorContext): Receive = withLabel(null)(r) /** * Wrap a Receive partial function in a logging enclosure, which sends a * message with given log level to the event bus each time before a message is matched. * This includes messages which are not handled. */ def apply(logLevel: LogLevel)(r: Receive)(implicit context: ActorContext): Receive = withLabel(null, logLevel)(r) /** * Java API: compatible with lambda expressions */ def create(r: AbstractActor.Receive, context: AbstractActor.ActorContext): AbstractActor.Receive = new AbstractActor.Receive( apply(r.onMessage.asInstanceOf[PartialFunction[Any, Unit]])(context) .asInstanceOf[PartialFunction[Any, BoxedUnit]]) /** * Create a decorated logger which will append `" in state " + label` to each message it logs. */ def withLabel(label: String, logLevel: LogLevel)(r: Receive)(implicit context: ActorContext): Receive = r match { case _: LoggingReceive => r case _ => if (context.system.settings.AddLoggingReceive) new LoggingReceive(None, r, Option(label), logLevel) else r } /** * Create a decorated logger which will append `" in state " + label` to each message it logs. */ def withLabel(label: String)(r: Receive)(implicit context: ActorContext): Receive = withLabel(label, Logging.DebugLevel)(r) } /** * This decorator adds invocation logging to a Receive function. * @param source the log source, if not defined the actor of the context will be used */ class LoggingReceive(source: Option[AnyRef], r: Receive, label: Option[String], logLevel: LogLevel)( implicit context: ActorContext) extends Receive { def this(source: Option[AnyRef], r: Receive, label: Option[String])(implicit context: ActorContext) = this(source, r, label, Logging.DebugLevel) def this(source: Option[AnyRef], r: Receive)(implicit context: ActorContext) = this(source, r, None, Logging.DebugLevel) def isDefinedAt(o: Any): Boolean = { val handled = r.isDefinedAt(o) if (context.system.eventStream.logLevel >= logLevel) { val src = source.getOrElse(context.asInstanceOf[ActorCell].actor) val (str, clazz) = LogSource.fromAnyRef(src) val message = "received " + (if (handled) "handled" else "unhandled") + " message " + o + " from " + context .sender() + (label match { case Some(l) => " in state " + l case _ => "" }) val event = src match { case a: DiagnosticActorLogging => LogEvent(logLevel, str, clazz, message, a.log.mdc) case _ => LogEvent(logLevel, str, clazz, message) } context.system.eventStream.publish(event) } handled } def apply(o: Any): Unit = r(o) }




© 2015 - 2024 Weber Informatics LLC | Privacy Policy