Adding custom ZLogger to capture unhandled defects

CustomZLogger.scala

// https://github.com/zio/zio/issues/6278

import zio.{Cause, FiberId, FiberRef, LogLevel, LogSpan, RuntimeConfigAspect, ZIO, ZIOAppDefault, ZLogger, ZTraceElement}

object Example extends ZIOAppDefault {

  val specialLogger: ZLogger[Cause[Any], Unit] =
    (
      trace: ZTraceElement,
      fiberId: FiberId,
      level: LogLevel,
      message: () => Cause[Any],
      context: Map[FiberRef.Runtime[_], AnyRef],
      spans: List[LogSpan],
      location: ZTraceElement
    ) => {
      try {
        System.err.println(message().prettyPrint)
        ()
      } catch {
        case t: Throwable => t.printStackTrace()
      }
    }

  override def hook = RuntimeConfigAspect.addLogger(specialLogger)

  val run =
    for {
      f <- ZIO.succeed(throw new Exception("bang")).fork
      _ <- f.await
    } yield ()


}