47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
package logrus
|
|
|
|
import (
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"github.com/micro/go-micro/v2/logger"
|
|
)
|
|
|
|
type Options struct {
|
|
logger.Options
|
|
Formatter logrus.Formatter
|
|
Hooks logrus.LevelHooks
|
|
// Flag for whether to log caller info (off by default)
|
|
ReportCaller bool
|
|
// Exit Function to call when FatalLevel log
|
|
ExitFunc func(int)
|
|
}
|
|
|
|
type formatterKey struct{}
|
|
|
|
func WithTextTextFormatter(formatter *logrus.TextFormatter) logger.Option {
|
|
return logger.SetOption(formatterKey{}, formatter)
|
|
}
|
|
func WithJSONFormatter(formatter *logrus.JSONFormatter) logger.Option {
|
|
return logger.SetOption(formatterKey{}, formatter)
|
|
}
|
|
|
|
type hooksKey struct{}
|
|
|
|
func WithLevelHooks(hooks logrus.LevelHooks) logger.Option {
|
|
return logger.SetOption(hooksKey{}, hooks)
|
|
}
|
|
|
|
type reportCallerKey struct{}
|
|
|
|
// warning to use this option. because logrus doest not open CallerDepth option
|
|
// this will only print this package
|
|
func ReportCaller() logger.Option {
|
|
return logger.SetOption(reportCallerKey{}, true)
|
|
}
|
|
|
|
type exitKey struct{}
|
|
|
|
func WithExitFunc(exit func(int)) logger.Option {
|
|
return logger.SetOption(exitKey{}, exit)
|
|
}
|