fm.common.ColorPatternLayoutEncoder.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2014 Frugal Mechanic (http://frugalmechanic.com)
 *
 * 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 fm.common

import ch.qos.logback.classic.Level.{ERROR,WARN}
import ch.qos.logback.classic.spi.ILoggingEvent
import java.nio.charset.StandardCharsets.UTF_8

/**
 * A simple Logback extension to the PatternLayoutEncoder that adds simple coloring for WARN and ERROR levels
 */
final class ColorPatternLayoutEncoder extends ch.qos.logback.classic.encoder.PatternLayoutEncoder {
  private[this] val END_COLOR:     Array[Byte] = "\u001b[m".getBytes(UTF_8)
  private[this] val ERROR_COLOR:   Array[Byte] = "\u001b[0;31m".getBytes(UTF_8)
  private[this] val WARN_COLOR:    Array[Byte] = "\u001b[0;33m".getBytes(UTF_8)
  private[this] val DEFAULT_COLOR: Array[Byte] = "".getBytes(UTF_8)

  override def encode(e: ILoggingEvent): Array[Byte] = {
    val startColor: Array[Byte] = e.getLevel match {
      case ERROR => ERROR_COLOR
      case WARN  => WARN_COLOR
      case _     => DEFAULT_COLOR
    }

    if (startColor.length === 0) super.encode(e)
    else startColor ++ super.encode(e) ++ END_COLOR
  }
}