mirror of
https://github.com/kairos-io/kairos-sdk.git
synced 2025-09-21 02:47:55 +00:00
123
logger/log.go
Normal file
123
logger/log.go
Normal file
@@ -0,0 +1,123 @@
|
||||
package logger
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
|
||||
"github.com/rs/zerolog"
|
||||
)
|
||||
|
||||
// KairosLog implements the bridge between zerolog and the logger.Interface that yip needs.
|
||||
// We also use that interface across all kairos libs, so its easier to bridge it here
|
||||
type KairosLog struct {
|
||||
zerolog.Logger
|
||||
}
|
||||
|
||||
func (k KairosLog) Infof(tpl string, args ...interface{}) {
|
||||
k.Logger.Info().Msg(fmt.Sprintf(tpl, args...))
|
||||
}
|
||||
func (k KairosLog) Info(args ...interface{}) {
|
||||
k.Logger.Info().Msg(fmt.Sprint(args...))
|
||||
}
|
||||
func (k KairosLog) Warnf(tpl string, args ...interface{}) {
|
||||
k.Logger.Warn().Msg(fmt.Sprintf(tpl, args...))
|
||||
}
|
||||
func (k KairosLog) Warn(args ...interface{}) {
|
||||
k.Logger.Warn().Msg(fmt.Sprint(args...))
|
||||
}
|
||||
func (k KairosLog) Debugf(tpl string, args ...interface{}) {
|
||||
k.Logger.Debug().Msg(fmt.Sprintf(tpl, args...))
|
||||
}
|
||||
func (k KairosLog) Debug(args ...interface{}) {
|
||||
k.Logger.Debug().Msg(fmt.Sprint(args...))
|
||||
}
|
||||
func (k KairosLog) Errorf(tpl string, args ...interface{}) {
|
||||
k.Logger.Error().Msg(fmt.Sprintf(tpl, args...))
|
||||
}
|
||||
func (k KairosLog) Error(args ...interface{}) {
|
||||
k.Logger.Error().Msg(fmt.Sprint(args...))
|
||||
}
|
||||
func (k KairosLog) Fatalf(tpl string, args ...interface{}) {
|
||||
k.Logger.Fatal().Msg(fmt.Sprintf(tpl, args...))
|
||||
}
|
||||
func (k KairosLog) Fatal(args ...interface{}) {
|
||||
k.Logger.Fatal().Msg(fmt.Sprint(args...))
|
||||
}
|
||||
func (k KairosLog) Panicf(tpl string, args ...interface{}) {
|
||||
k.Logger.Panic().Msg(fmt.Sprintf(tpl, args...))
|
||||
}
|
||||
func (k KairosLog) Panic(args ...interface{}) {
|
||||
k.Logger.Panic().Msg(fmt.Sprint(args...))
|
||||
}
|
||||
func (k KairosLog) Tracef(tpl string, args ...interface{}) {
|
||||
k.Logger.Trace().Msg(fmt.Sprintf(tpl, args...))
|
||||
}
|
||||
func (k KairosLog) Trace(args ...interface{}) {
|
||||
k.Logger.Trace().Msg(fmt.Sprint(args...))
|
||||
}
|
||||
func (k KairosLog) SetLevel(level Level) {
|
||||
k.Logger.Level(zerolog.Level(level))
|
||||
}
|
||||
func (k KairosLog) IsDebugLevel() bool {
|
||||
return k.Logger.GetLevel() == zerolog.DebugLevel
|
||||
}
|
||||
|
||||
// NewKairosLog provides a normal console log
|
||||
func NewKairosLog(opts ...LogOption) (*KairosLog, error) {
|
||||
k := &KairosLog{zerolog.New(zerolog.NewConsoleWriter())}
|
||||
for _, o := range opts {
|
||||
o(k)
|
||||
}
|
||||
return k, nil
|
||||
}
|
||||
|
||||
// NewKairosLogToFile provides a log that logs to a given file
|
||||
// file is closed automatically by GC on application exit
|
||||
func NewKairosLogToFile(logfile string, opts ...LogOption) (*KairosLog, error) {
|
||||
f, err := os.Open(logfile)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
k := &KairosLog{zerolog.New(f)}
|
||||
for _, o := range opts {
|
||||
o(k)
|
||||
}
|
||||
return k, nil
|
||||
}
|
||||
|
||||
// NewKairosMultiLog provides a logger that logs to both console and a given file
|
||||
func NewKairosMultiLog(logfile string, opts ...LogOption) (*KairosLog, error) {
|
||||
f, err := os.Open(logfile)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
multiWriter := zerolog.MultiLevelWriter(zerolog.New(zerolog.NewConsoleWriter()), zerolog.New(f))
|
||||
k := &KairosLog{zerolog.New(multiWriter).With().Logger()}
|
||||
return k, nil
|
||||
}
|
||||
|
||||
// NewKairosNullLog provides a logger that discards all output
|
||||
func NewKairosNullLog(opts ...LogOption) (*KairosLog, error) {
|
||||
k := &KairosLog{zerolog.New(nil).Level(zerolog.Level(Disabled))}
|
||||
return k, nil
|
||||
}
|
||||
|
||||
// NewKairosBufferLog will return a logger that stores all logs in a buffer, used mainly for testing
|
||||
func NewKairosBufferLog(b *bytes.Buffer) (*KairosLog, error) {
|
||||
k := &KairosLog{zerolog.New(zerolog.NewConsoleWriter(func(w *zerolog.ConsoleWriter) {
|
||||
w.Out = b
|
||||
}))}
|
||||
return k, nil
|
||||
}
|
||||
|
||||
// NewKairosMultiCustomLogTargets provides a multi log with custom writers so we can override the writers easily
|
||||
// The burden of generating the writers is on the caller
|
||||
// This should be called with writers that implement the zerolog.LevelWriter interface so zerolog can properly write to
|
||||
// them based on the level and such. Otherwise it will use the Writer method directly and miss a lot fo stuff
|
||||
func NewKairosMultiCustomLogTargets(writers ...io.Writer) (*KairosLog, error) {
|
||||
multiWriter := zerolog.MultiLevelWriter(writers...)
|
||||
k := &KairosLog{zerolog.New(multiWriter)}
|
||||
return k, nil
|
||||
}
|
Reference in New Issue
Block a user