mirror of
https://github.com/kairos-io/immucore.git
synced 2025-09-01 22:57:33 +00:00
@@ -1,47 +0,0 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/rs/zerolog"
|
||||
"golang.org/x/sys/unix"
|
||||
"io"
|
||||
"os"
|
||||
)
|
||||
|
||||
var Log zerolog.Logger
|
||||
|
||||
func SetLogger() {
|
||||
var loggers []io.Writer
|
||||
devKmsg, err := os.OpenFile("/dev/kmsg", unix.O_WRONLY, 0o600)
|
||||
if err == nil {
|
||||
err := os.WriteFile("/proc/sys/kernel/printk_ratelimit", []byte("100"), os.ModeType)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
err = os.WriteFile("/proc/sys/kernel/printk_ratelimit_burst", []byte("1000"), os.ModeType)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
loggers = append(loggers, zerolog.ConsoleWriter{Out: devKmsg})
|
||||
}
|
||||
logFile, err := os.Create("/run/immucore.log")
|
||||
if err == nil {
|
||||
loggers = append(loggers, zerolog.ConsoleWriter{Out: logFile})
|
||||
}
|
||||
|
||||
// No loggers? Then stdout ¯\_(ツ)_/¯
|
||||
if len(loggers) == 0 {
|
||||
loggers = append(loggers, zerolog.ConsoleWriter{Out: os.Stdout})
|
||||
}
|
||||
multi := zerolog.MultiLevelWriter(loggers...)
|
||||
Log = zerolog.New(multi).With().Logger()
|
||||
Log.WithLevel(zerolog.InfoLevel)
|
||||
|
||||
// Set debug logger
|
||||
debug := len(ReadCMDLineArg("rd.immucore.debug")) > 0
|
||||
debugFromEnv := os.Getenv("IMMUCORE_DEBUG") != ""
|
||||
if debug || debugFromEnv {
|
||||
Log = zerolog.New(multi).With().Caller().Logger()
|
||||
Log.WithLevel(zerolog.DebugLevel)
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user