zio.logging.FileLoggerConfig.scala Maven / Gradle / Ivy
The newest version!
/*
* Copyright 2019-2024 John A. De Goes and the ZIO Contributors
*
* 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 zio.logging
import zio._
import zio.prelude._
import java.net.URI
import java.nio.charset.{ Charset, StandardCharsets }
import java.nio.file.{ Path, Paths }
import scala.util.{ Failure, Success, Try }
final case class FileLoggerConfig(
destination: Path,
format: LogFormat,
filter: LogFilter.LogLevelByNameConfig,
charset: Charset = StandardCharsets.UTF_8,
autoFlushBatchSize: Int = 1,
bufferedIOSize: Option[Int] = None,
rollingPolicy: Option[FileLoggerConfig.FileRollingPolicy] = None
) {
def toFilter[M]: LogFilter[M] = filter.toFilter
}
object FileLoggerConfig {
sealed trait FileRollingPolicy
object FileRollingPolicy {
final case object TimeBasedRollingPolicy extends FileRollingPolicy
val config: Config[FileRollingPolicy] = {
val tpe = Config.string("type")
val timeBased = Config.succeed(TimeBasedRollingPolicy)
tpe.switch("TimeBasedRollingPolicy" -> timeBased)
}
}
private def charsetValue(value: String): Either[Config.Error.InvalidData, Charset] =
Try(Charset.forName(value)) match {
case Success(v) => Right(v)
case Failure(_) =>
Left(Config.Error.InvalidData(Chunk.empty, s"Expected a Charset, but found ${value}"))
}
private def pathValue(value: String): Either[Config.Error.InvalidData, Path] =
Try(Paths.get(URI.create(value))) match {
case Success(v) => Right(v)
case Failure(_) =>
Left(Config.Error.InvalidData(Chunk.empty, s"Expected a Path, but found ${value}"))
}
val config: Config[FileLoggerConfig] = {
val autoFlushBatchSizeConfig = Config.int.nested("autoFlushBatchSize").withDefault(1)
val bufferedIOSizeConfig = Config.int.nested("bufferedIOSize").optional
val filterConfig = LogFilter.LogLevelByNameConfig.config.nested("filter")
val charsetConfig =
Config.string.mapOrFail(charsetValue).nested("charset").withDefault(StandardCharsets.UTF_8)
val pathConfig = Config.string.mapOrFail(pathValue).nested("path")
val formatConfig = LogFormat.config.nested("format").withDefault(LogFormat.default)
val rollingPolicyConfig = FileRollingPolicy.config.nested("rollingPolicy").optional
(pathConfig ++ formatConfig ++ filterConfig ++ charsetConfig ++ autoFlushBatchSizeConfig ++ bufferedIOSizeConfig ++ rollingPolicyConfig).map {
case (path, format, filter, charset, autoFlushBatchSize, bufferedIOSize, rollingPolicy) =>
FileLoggerConfig(
path,
format,
filter,
charset,
autoFlushBatchSize,
bufferedIOSize,
rollingPolicy
)
}
}
implicit val equal: Equal[FileLoggerConfig] = Equal.make { (l, r) =>
l.destination == r.destination &&
l.charset == r.charset &&
l.autoFlushBatchSize == r.autoFlushBatchSize &&
l.bufferedIOSize == r.bufferedIOSize &&
l.rollingPolicy == r.rollingPolicy &&
l.format == r.format &&
l.filter === r.filter
}
def load(configPath: NonEmptyChunk[String] = loggerConfigPath): ZIO[Any, Config.Error, FileLoggerConfig] =
ZIO.config(FileLoggerConfig.config.nested(configPath.head, configPath.tail: _*))
def make(
configPath: NonEmptyChunk[String] = loggerConfigPath
): ZLayer[Any, Config.Error, FileLoggerConfig] =
ZLayer.fromZIO(load(configPath))
}