mirror of https://github.com/synctv-org/synctv
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
2.4 KiB
Go
96 lines
2.4 KiB
Go
package bootstrap
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/natefinch/lumberjack"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/synctv-org/synctv/cmd/flags"
|
|
"github.com/synctv-org/synctv/internal/conf"
|
|
"github.com/synctv-org/synctv/utils"
|
|
"github.com/zijiren233/go-colorable"
|
|
)
|
|
|
|
func setLog(l *logrus.Logger) {
|
|
if flags.Dev {
|
|
l.SetLevel(logrus.DebugLevel)
|
|
l.SetReportCaller(true)
|
|
} else {
|
|
l.SetLevel(logrus.InfoLevel)
|
|
l.SetReportCaller(false)
|
|
}
|
|
}
|
|
|
|
func InitLog(ctx context.Context) (err error) {
|
|
setLog(logrus.StandardLogger())
|
|
forceColor := utils.ForceColor()
|
|
if conf.Conf.Log.Enable {
|
|
conf.Conf.Log.FilePath, err = utils.OptFilePath(conf.Conf.Log.FilePath)
|
|
if err != nil {
|
|
logrus.Fatalf("log: log file path error: %v", err)
|
|
}
|
|
var l = &lumberjack.Logger{
|
|
Filename: conf.Conf.Log.FilePath,
|
|
MaxSize: conf.Conf.Log.MaxSize,
|
|
MaxBackups: conf.Conf.Log.MaxBackups,
|
|
MaxAge: conf.Conf.Log.MaxAge,
|
|
Compress: conf.Conf.Log.Compress,
|
|
}
|
|
if err := l.Rotate(); err != nil {
|
|
logrus.Fatalf("log: rotate log file error: %v", err)
|
|
}
|
|
var w io.Writer
|
|
if forceColor {
|
|
w = colorable.NewNonColorableWriter(l)
|
|
} else {
|
|
w = l
|
|
}
|
|
if flags.Dev || flags.LogStd {
|
|
logrus.SetOutput(io.MultiWriter(os.Stdout, w))
|
|
logrus.Infof("log: enable log to stdout and file: %s", conf.Conf.Log.FilePath)
|
|
} else {
|
|
logrus.SetOutput(w)
|
|
logrus.Infof("log: disable log to stdout, only log to file: %s", conf.Conf.Log.FilePath)
|
|
}
|
|
}
|
|
switch conf.Conf.Log.LogFormat {
|
|
case "json":
|
|
logrus.SetFormatter(&logrus.JSONFormatter{
|
|
TimestampFormat: time.DateTime,
|
|
})
|
|
default:
|
|
if conf.Conf.Log.LogFormat != "text" {
|
|
logrus.Warnf("unknown log format: %s, use default: text", conf.Conf.Log.LogFormat)
|
|
}
|
|
logrus.SetFormatter(&logrus.TextFormatter{
|
|
ForceColors: forceColor,
|
|
DisableColors: !forceColor,
|
|
ForceQuote: flags.Dev,
|
|
DisableQuote: !flags.Dev,
|
|
DisableSorting: true,
|
|
FullTimestamp: true,
|
|
TimestampFormat: time.DateTime,
|
|
QuoteEmptyFields: true,
|
|
})
|
|
}
|
|
log.SetOutput(logrus.StandardLogger().Writer())
|
|
return nil
|
|
}
|
|
|
|
func InitStdLog(ctx context.Context) error {
|
|
logrus.StandardLogger().SetOutput(os.Stdout)
|
|
log.SetOutput(os.Stdout)
|
|
setLog(logrus.StandardLogger())
|
|
return nil
|
|
}
|
|
|
|
func InitDiscardLog(ctx context.Context) error {
|
|
logrus.StandardLogger().SetOutput(io.Discard)
|
|
log.SetOutput(io.Discard)
|
|
return nil
|
|
}
|