almond.logger.internal.PrintStreamLogger.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of logger_3 Show documentation

logger

There is a newer version: 0.14.0-RC15

package almond.logger.internal

import java.io.PrintStream
import java.lang.management.ManagementFactory

import almond.logger.Level

import scala.annotation.tailrec

final class PrintStreamLogger(
  val level: Level,
  out: PrintStream,
  colored: Boolean,
  addPid: Boolean
) extends ActualLogger {

  def this(level: Level, out: PrintStream) =
    this(level, out, colored = true, addPid = false)

  def log(level: Level, message: String, exception: Throwable = null): Unit = {
    val b = new StringBuilder

    if (addPid) {
      if (colored)
        b ++= Console.CYAN
      b ++= PrintStreamLogger.pid
      if (colored)
        b ++= Console.RESET
      b += ' '
    }

    b ++= (if (colored) level.coloredName else level.name)
    b += ' '
    b ++= message

    @tailrec
    def addException(ex: Throwable): Unit =
      if (ex != null) {
        b += '\n' // FIXME Not portable
        b ++= ex.toString
        for (elem <- ex.getStackTrace) {
          b ++= "\n  " // FIXME Not portable
          b ++= elem.toString
        }
        addException(ex.getCause)
      }

    addException(exception)

    out.println(b.result())
  }
}

object PrintStreamLogger {
  lazy val pid: String = ManagementFactory.getRuntimeMXBean.getName.takeWhile(_ != '@')
}