Alexey Khit
77 lines
1.3 KiB
Go
77 lines
1.3 KiB
Go
package app
|
|
|
|
import (
|
|
"github.com/rs/zerolog"
|
|
"gopkg.in/yaml.v3"
|
|
"io"
|
|
"os"
|
|
"runtime"
|
|
)
|
|
|
|
func Init() {
|
|
data, _ = os.ReadFile("go2rtc.yaml")
|
|
|
|
var cfg struct {
|
|
Mod map[string]string `yaml:"log"`
|
|
}
|
|
|
|
LoadConfig(&cfg)
|
|
|
|
var writer io.Writer = os.Stdout
|
|
|
|
// styles
|
|
format := cfg.Mod["format"]
|
|
if format != "json" {
|
|
writer = zerolog.ConsoleWriter{
|
|
Out: writer, TimeFormat: "15:04:05.000",
|
|
NoColor: writer != os.Stdout || format == "text",
|
|
}
|
|
}
|
|
|
|
zerolog.TimeFieldFormat = zerolog.TimeFormatUnixMs
|
|
|
|
lvl, err := zerolog.ParseLevel(cfg.Mod["level"])
|
|
if err != nil || lvl == zerolog.NoLevel {
|
|
lvl = zerolog.InfoLevel
|
|
}
|
|
|
|
log = zerolog.New(writer).With().Timestamp().Logger().Level(lvl)
|
|
|
|
modules = cfg.Mod
|
|
|
|
log.Info().Msgf("go2rtc %s/%s", runtime.GOOS, runtime.GOARCH)
|
|
}
|
|
|
|
func LoadConfig(v interface{}) {
|
|
if data != nil {
|
|
_ = yaml.Unmarshal(data, v)
|
|
}
|
|
}
|
|
|
|
func GetLogger(module string) zerolog.Logger {
|
|
if s, ok := modules[module]; ok {
|
|
lvl, err := zerolog.ParseLevel(s)
|
|
if err != nil {
|
|
log.Warn().Err(err).Msg("[log]")
|
|
return log
|
|
}
|
|
|
|
return log.Level(lvl)
|
|
}
|
|
|
|
log.Warn().Msgf("[log] wrong module: %s", module)
|
|
|
|
return log
|
|
}
|
|
|
|
// internal
|
|
|
|
// data - config content
|
|
var data []byte
|
|
|
|
// log - main logger
|
|
var log zerolog.Logger
|
|
|
|
// modules log levels
|
|
var modules map[string]string
|