main.misk.logging.StackDriverLogging.kt Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of misk-gcp Show documentation

Open source application container in Kotlin

There is a newer version: 2024.09.17.200749-4708422

package misk.logging

import ch.qos.logback.classic.Logger
import ch.qos.logback.classic.filter.ThresholdFilter
import com.google.cloud.logging.logback.LoggingAppender
import misk.cloud.gcp.logging.StackDriverLoggingConfig
import misk.cloud.gcp.tracing.TracingLoggingEnhancer
import org.slf4j.LoggerFactory

/**
 * Configures Misk to send application logs to StackDriver. If credentials are
 * required to send logging, set the GOOGLE_APPLICATION_CREDENTIALS environment
 * variable with the path to the JSON credentials.
 */
fun enableStackDriverLogging(config: StackDriverLoggingConfig) {
  val rootLogger = LoggerFactory.getLogger(Logger.ROOT_LOGGER_NAME) as Logger
  val context = rootLogger.loggerContext
  context.reset()

  val filter = ThresholdFilter()
  filter.setLevel(config.filter_level.levelStr)
  filter.context = context
  filter.start()

  val appender = LoggingAppender()
  appender.addFilter(filter)
  appender.addEnhancer(TracingLoggingEnhancer::class.qualifiedName)
  appender.setFlushLevel(config.flush_level)
  appender.setLog(config.log)
  appender.context = context
  appender.start()

  rootLogger.addAppender(appender)
}