eu.stratosphere.api.scala.codegen.Logger.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of stratosphere-scala Show documentation
There is a newer version: 0.5.2-hadoop2
/***********************************************************************************************************************
 * Copyright (C) 2010-2013 by the Stratosphere project (http://stratosphere.eu)
 *
 * 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 eu.stratosphere.api.scala.codegen
import scala.reflect.macros.Context

trait Loggers[C <: Context] { this: MacroContextHolder[C] =>
  import c.universe._

  abstract sealed class LogLevel extends Ordered[LogLevel] {
    protected[Loggers] val toInt: Int
    override def compare(that: LogLevel) = this.toInt.compare(that.toInt)
  }

  object LogLevel {
    def unapply(name: String): Option[LogLevel] = name match {
      case "error" | "Error"     => Some(Error)
      case "warn" | "Warn"       => Some(Warn)
      case "debug" | "Debug"     => Some(Debug)
      case "inspect" | "Inspect" => Some(Inspect)
      case _                     => None
    }
    case object Error extends LogLevel { override val toInt = 1 }
    case object Warn extends LogLevel { override val toInt = 2 }
    case object Debug extends LogLevel { override val toInt = 3 }
    case object Inspect extends LogLevel { override val toInt = 4 }
  }

  object logger { var level: LogLevel = LogLevel.Warn }
  private val counter = new Counter

  trait Logger {

    abstract sealed class Severity {
      protected val toInt: Int
      protected def reportInner(msg: String, pos: Position)

      protected def formatMsg(msg: String) = msg

      def isEnabled = this.toInt <= logger.level.toInt

      def report(msg: String) = {
        if (isEnabled) {
          reportInner(formatMsg(msg), c.enclosingPosition)
        }
      }
    }

    case object Error extends Severity {
      override val toInt = LogLevel.Error.toInt
      override def reportInner(msg: String, pos: Position) = c.error(pos, msg)
    }

    case object Warn extends Severity {
      override val toInt = LogLevel.Warn.toInt
      override def reportInner(msg: String, pos: Position) = c.warning(pos, msg)
    }

    case object Debug extends Severity {
      override val toInt = LogLevel.Debug.toInt
      override def reportInner(msg: String, pos: Position) = c.info(pos, msg, true)
    }

    def getMsgAndStackLine(e: Throwable) = {
      val lines = e.getStackTrace.map(_.toString)
      val relevant = lines filter { _.contains("eu.stratosphere") }
      val stackLine = relevant.headOption getOrElse e.getStackTrace.toString
      e.getMessage() + " @ " + stackLine
    }

    def posString(pos: Position): String = pos match {
      case NoPosition => "?:?"
      case _          => pos.line + ":" + pos.column
    }

    def safely(default: => Tree, inspect: Boolean)(onError: Throwable => String)(block: => Tree): Tree = {
      try {
        block
      } catch {
        case e:Throwable => {
          Error.report(onError(e));
          val ret = default
          ret
        }
      }
    }

    def verbosely[T](obs: T => String)(block: => T): T = {
      val ret = block
      Debug.report(obs(ret))
      ret
    }

    def maybeVerbosely[T](guard: T => Boolean)(obs: T => String)(block: => T): T = {
      val ret = block
      if (guard(ret)) Debug.report(obs(ret))
      ret
    }
  }
}