provider-rke2/pkg/log/log.go

package log

import (
	"os"

	"github.com/kairos-io/provider-rke2/pkg/version"
	"github.com/sirupsen/logrus"
	"gopkg.in/natefinch/lumberjack.v2"
)

func InitLogger(path string) {
	f, err := os.OpenFile(path, os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0644)
	if err != nil {
		panic(err)
	}

	logfile := &lumberjack.Logger{
		Filename:   f.Name(),
		MaxSize:    10,
		MaxBackups: 5,
		Compress:   true,
	}

	logrus.SetOutput(logfile)
	logrus.SetFormatter(RKE2Logger{
		Version:   version.Version,
		Formatter: logrus.StandardLogger().Formatter,
	})
}

type RKE2Logger struct {
	Version   string
	Formatter logrus.Formatter
}

func (l RKE2Logger) Format(entry *logrus.Entry) ([]byte, error) {
	entry.Data["version"] = l.Version
	return l.Formatter.Format(entry)
}