mirror of https://github.com/dtm-labs/dtm.git
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