k3s/pkg/util/logger/logger.go

package logger

import (
	"fmt"

	"github.com/go-logr/logr"
	"github.com/sirupsen/logrus"
)

// implicit interface check
var _ logr.LogSink = &logrusSink{}

// mapLevel maps logr log verbosities to logrus log levels
// logr does not have "log levels", but Info prints at verbosity 0
// while logrus's LevelInfo is unit32(4). This means:
// * panic/fatal/warn are unused,
// * 0 is info
// * 1 is debug
// * >=2 are trace
func mapLevel(level int) logrus.Level {
	if level >= 2 {
		return logrus.TraceLevel
	}
	return logrus.Level(level + 4)
}

// mapKV maps a list of keys and values to logrus Fields
func mapKV(kvs []any) logrus.Fields {
	fields := logrus.Fields{}
	for i := 0; i < len(kvs); i += 2 {
		k := kvs[i].(string)
		if len(kvs) > i+1 {
			fields[k] = kvs[i+1]
		} else {
			fields[k] = ""
		}
	}
	return fields
}

// LogrusSink wraps logrus the Logger/Entry types for use as a logr LogSink.
type logrusSink struct {
	e  *logrus.Entry
	ri logr.RuntimeInfo
}

func NewLogrusSink(l *logrus.Logger) *logrusSink {
	if l == nil {
		l = logrus.StandardLogger()
	}
	return &logrusSink{e: logrus.NewEntry(l)}
}

func (ls *logrusSink) AsLogr() logr.Logger {
	return logr.New(ls)
}

func (ls *logrusSink) Init(ri logr.RuntimeInfo) {
	ls.ri = ri
}

func (ls *logrusSink) Enabled(level int) bool {
	return ls.e.Logger.IsLevelEnabled(mapLevel(level))
}

func (ls *logrusSink) Info(level int, msg string, kvs ...any) {
	ls.e.WithFields(mapKV(kvs)).Log(mapLevel(level), msg)
}

func (ls *logrusSink) Error(err error, msg string, kvs ...any) {
	ls.e.WithError(err).WithFields(mapKV(kvs)).Error(msg)
}

func (ls *logrusSink) WithValues(kvs ...any) logr.LogSink {
	return &logrusSink{
		e:  ls.e.WithFields(mapKV(kvs)),
		ri: ls.ri,
	}
}

func (ls *logrusSink) WithName(name string) logr.LogSink {
	if base, ok := ls.e.Data["logger"]; ok {
		name = fmt.Sprintf("%s/%s", base, name)
	}
	return ls.WithValues("logger", name)
}