yedf2
4 years ago
6 changed files with 7 additions and 183 deletions
@ -1,135 +0,0 @@ |
|||||
package logger |
|
||||
|
|
||||
import ( |
|
||||
"encoding/json" |
|
||||
"fmt" |
|
||||
"log" |
|
||||
"net/url" |
|
||||
"os" |
|
||||
"runtime/debug" |
|
||||
"strings" |
|
||||
|
|
||||
"github.com/natefinch/lumberjack" |
|
||||
"go.uber.org/zap" |
|
||||
"go.uber.org/zap/zapcore" |
|
||||
) |
|
||||
|
|
||||
//var logger *zap.SugaredLogger = nil
|
|
||||
|
|
||||
var logger Logger |
|
||||
|
|
||||
const ( |
|
||||
// StdErr is the default configuration for log output.
|
|
||||
StdErr = "stderr" |
|
||||
// StdOut configuration for log output
|
|
||||
StdOut = "stdout" |
|
||||
) |
|
||||
|
|
||||
func init() { |
|
||||
InitLog(os.Getenv("LOG_LEVEL")) |
|
||||
} |
|
||||
|
|
||||
// Logger logger interface
|
|
||||
type Logger interface { |
|
||||
Debugf(format string, args ...interface{}) |
|
||||
Infof(format string, args ...interface{}) |
|
||||
Warnf(format string, args ...interface{}) |
|
||||
Errorf(format string, args ...interface{}) |
|
||||
} |
|
||||
|
|
||||
// WithLogger replaces default logger
|
|
||||
func WithLogger(log Logger) { |
|
||||
logger = log |
|
||||
} |
|
||||
|
|
||||
// InitLog is an initialization for a logger
|
|
||||
// level can be: debug info warn error
|
|
||||
func InitLog(level string) { |
|
||||
InitLog2(level, StdOut, 0, "") |
|
||||
} |
|
||||
|
|
||||
// InitLog2 specify advanced log config
|
|
||||
func InitLog2(level string, outputs string, logRotationEnable int64, logRotateConfigJSON string) { |
|
||||
outputPaths := strings.Split(outputs, ",") |
|
||||
for i, v := range outputPaths { |
|
||||
if logRotationEnable != 0 && v != StdErr && v != StdOut { |
|
||||
outputPaths[i] = fmt.Sprintf("lumberjack://%s", v) |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
if logRotationEnable != 0 { |
|
||||
setupLogRotation(logRotateConfigJSON) |
|
||||
} |
|
||||
|
|
||||
config := loadConfig(level) |
|
||||
config.OutputPaths = outputPaths |
|
||||
p, err := config.Build(zap.AddCallerSkip(1)) |
|
||||
FatalIfError(err) |
|
||||
logger = p.Sugar() |
|
||||
} |
|
||||
|
|
||||
type lumberjackSink struct { |
|
||||
lumberjack.Logger |
|
||||
} |
|
||||
|
|
||||
func (*lumberjackSink) Sync() error { |
|
||||
return nil |
|
||||
} |
|
||||
|
|
||||
// setupLogRotation initializes log rotation for a single file path target.
|
|
||||
func setupLogRotation(logRotateConfigJSON string) { |
|
||||
err := zap.RegisterSink("lumberjack", func(u *url.URL) (zap.Sink, error) { |
|
||||
var conf lumberjackSink |
|
||||
err := json.Unmarshal([]byte(logRotateConfigJSON), &conf) |
|
||||
FatalfIf(err != nil, "bad config LogRotateConfigJSON: %v", err) |
|
||||
conf.Filename = u.Host + u.Path |
|
||||
return &conf, nil |
|
||||
}) |
|
||||
FatalIfError(err) |
|
||||
} |
|
||||
|
|
||||
func loadConfig(logLevel string) zap.Config { |
|
||||
config := zap.NewProductionConfig() |
|
||||
err := config.Level.UnmarshalText([]byte(logLevel)) |
|
||||
FatalIfError(err) |
|
||||
config.EncoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder |
|
||||
if os.Getenv("DTM_DEBUG") != "" { |
|
||||
config.Encoding = "console" |
|
||||
config.EncoderConfig.EncodeLevel = zapcore.CapitalColorLevelEncoder |
|
||||
} |
|
||||
return config |
|
||||
} |
|
||||
|
|
||||
// Debugf log to level debug
|
|
||||
func Debugf(fmt string, args ...interface{}) { |
|
||||
logger.Debugf(fmt, args...) |
|
||||
} |
|
||||
|
|
||||
// Infof log to level info
|
|
||||
func Infof(fmt string, args ...interface{}) { |
|
||||
logger.Infof(fmt, args...) |
|
||||
} |
|
||||
|
|
||||
// Warnf log to level warn
|
|
||||
func Warnf(fmt string, args ...interface{}) { |
|
||||
logger.Warnf(fmt, args...) |
|
||||
} |
|
||||
|
|
||||
// Errorf log to level error
|
|
||||
func Errorf(fmt string, args ...interface{}) { |
|
||||
logger.Errorf(fmt, args...) |
|
||||
} |
|
||||
|
|
||||
// FatalfIf log to level error
|
|
||||
func FatalfIf(cond bool, fmt string, args ...interface{}) { |
|
||||
if !cond { |
|
||||
return |
|
||||
} |
|
||||
debug.PrintStack() |
|
||||
log.Fatalf(fmt, args...) |
|
||||
} |
|
||||
|
|
||||
// FatalIfError if err is not nil, then log to level fatal and call os.Exit
|
|
||||
func FatalIfError(err error) { |
|
||||
FatalfIf(err != nil, "fatal error: %v", err) |
|
||||
} |
|
||||
@ -1,46 +0,0 @@ |
|||||
package logger |
|
||||
|
|
||||
import ( |
|
||||
"os" |
|
||||
"testing" |
|
||||
|
|
||||
"github.com/stretchr/testify/assert" |
|
||||
"go.uber.org/zap" |
|
||||
) |
|
||||
|
|
||||
func TestInitLog(t *testing.T) { |
|
||||
os.Setenv("DTM_DEBUG", "1") |
|
||||
InitLog("debug") |
|
||||
Debugf("a debug msg") |
|
||||
Infof("a info msg") |
|
||||
Warnf("a warn msg") |
|
||||
Errorf("a error msg") |
|
||||
FatalfIf(false, "nothing") |
|
||||
FatalIfError(nil) |
|
||||
|
|
||||
InitLog2("debug", "test.log,stderr", 0, "") |
|
||||
Debugf("a debug msg to console and file") |
|
||||
|
|
||||
InitLog2("debug", "test2.log,/tmp/dtm-test1.log,/tmp/dtm-test.log,stdout,stderr", 1, |
|
||||
"{\"maxsize\": 1, \"maxage\": 1, \"maxbackups\": 1, \"compress\": false}") |
|
||||
Debugf("a debug msg to /tmp/dtm-test.log and test2.log and stdout and stderr") |
|
||||
|
|
||||
// _ = os.Remove("test.log")
|
|
||||
} |
|
||||
|
|
||||
func TestWithLogger(t *testing.T) { |
|
||||
logger := zap.NewExample().Sugar() |
|
||||
WithLogger(logger) |
|
||||
Debugf("a debug msg") |
|
||||
Infof("a info msg") |
|
||||
Warnf("a warn msg") |
|
||||
Errorf("a error msg") |
|
||||
FatalfIf(false, "nothing") |
|
||||
FatalIfError(nil) |
|
||||
} |
|
||||
|
|
||||
func TestCover(t *testing.T) { |
|
||||
sin := lumberjackSink{} |
|
||||
err := sin.Sync() |
|
||||
assert.Nil(t, err) |
|
||||
} |
|
||||
Loading…
Reference in new issue