Compare commits

...

7 Commits

Author SHA1 Message Date
ehlxr a15a40a425 upgrade dependencies 2021-01-04 17:14:30 +08:00
ehlxr 08e6fdbd53 update at 2021-01-04 15:51:39 by ehlxr 2021-01-04 15:51:39 +08:00
ehlxr 7eee167d91 update at 2021-01-01 23:12:00 by ehlxr 2021-01-01 23:12:00 +08:00
ehlxr c03451395f modify text encoder format bug 2020-12-31 17:26:35 +08:00
ehlxr 48e8f68955 format field content style 2020-12-31 17:16:11 +08:00
ehlxr 7b9983504f format field content style 2020-12-31 17:15:59 +08:00
ehlxr 6ac40ca1a2 format field content style 2020-12-31 17:15:40 +08:00
6 changed files with 82 additions and 26 deletions

View File

@ -68,12 +68,13 @@ func (enc textEncoder) EncodeEntry(ent zapcore.Entry, fields []zapcore.Field) (*
if enc.TimeKey != "" && enc.EncodeTime != nil {
enc.EncodeTime(ent.Time, arr)
}
if enc.LevelKey != "" && enc.EncodeLevel != nil {
enc.EncodeLevel(ent.Level, arr)
}
if ent.LoggerName != "" && enc.NameKey != "" {
nameEncoder := enc.EncodeName
if nameEncoder == nil {
// Fall back to FullNameEncoder for backward compatibility.
nameEncoder = zapcore.FullNameEncoder
@ -81,17 +82,23 @@ func (enc textEncoder) EncodeEntry(ent zapcore.Entry, fields []zapcore.Field) (*
nameEncoder(ent.LoggerName, arr)
}
if ent.Caller.Defined && enc.CallerKey != "" && enc.EncodeCaller != nil {
enc.EncodeCaller(ent.Caller, arr)
}
for i := range arr.elems {
// if i > 0 {
// line.AppendByte('\t')
// }
_, _ = fmt.Fprint(line, arr.elems[i])
}
putSliceEncoder(arr)
// Add any structured context.
enc.writeContext(line, fields)
// Add the message itself.
if enc.MessageKey != "" {
// c.addTabIfNecessary(line)
@ -99,9 +106,6 @@ func (enc textEncoder) EncodeEntry(ent zapcore.Entry, fields []zapcore.Field) (*
line.AppendString(ent.Message)
}
// Add any structured context.
enc.writeContext(line, fields)
// If there's no stacktrace key, honor that; this allows users to force
// single-line output.
if ent.Stack != "" && enc.StacktraceKey != "" {
@ -114,6 +118,7 @@ func (enc textEncoder) EncodeEntry(ent zapcore.Entry, fields []zapcore.Field) (*
} else {
line.AppendString(zapcore.DefaultLineEnding)
}
return line, nil
}
@ -128,9 +133,10 @@ func (enc textEncoder) writeContext(line *buffer.Buffer, extra []zapcore.Field)
}
// c.addTabIfNecessary(line)
line.AppendByte('{')
// line.AppendByte('{')
line.AppendByte(' ')
_, _ = line.Write(context.buf.Bytes())
line.AppendByte('}')
// line.AppendByte('}')
}
func (enc textEncoder) addTabIfNecessary(line *buffer.Buffer) {
@ -216,11 +222,13 @@ func (enc *textEncoder) AddDuration(key string, val time.Duration) {
func (enc *textEncoder) AddFloat64(key string, val float64) {
enc.addKey(key)
enc.appendFloat(val, 64)
enc.buf.AppendByte('}')
}
func (enc *textEncoder) AddInt64(key string, val int64) {
enc.addKey(key)
enc.addElementSeparator()
enc.buf.AppendInt(val)
enc.buf.AppendByte('}')
}
func (enc *textEncoder) AddReflected(key string, obj interface{}) error {
enc.resetReflectBuf()
@ -241,9 +249,10 @@ func (enc *textEncoder) OpenNamespace(key string) {
func (enc *textEncoder) AddString(key, val string) {
enc.addKey(key)
enc.addElementSeparator()
enc.buf.AppendByte('"')
// enc.buf.AppendByte('"')
enc.safeAddString(val)
enc.buf.AppendByte('"')
// enc.buf.AppendByte('"')
enc.buf.AppendByte('}')
}
func (enc *textEncoder) AddTime(key string, val time.Time) {
enc.addKey(key)
@ -261,6 +270,7 @@ func (enc *textEncoder) AddUint64(key string, val uint64) {
enc.addKey(key)
enc.addElementSeparator()
enc.buf.AppendUint(val)
enc.buf.AppendByte('}')
}
//noinspection GoRedundantConversion
@ -293,15 +303,39 @@ func (enc *textEncoder) clone() *textEncoder {
func (enc *textEncoder) addKey(key string) {
enc.addElementSeparator()
enc.buf.AppendByte('"')
// enc.buf.AppendByte('"')
if enc.replaceSeparator('}') {
enc.buf.AppendByte(',')
enc.buf.AppendByte(' ')
} else {
enc.buf.AppendByte('{')
}
enc.safeAddString(key)
enc.buf.AppendByte('"')
enc.buf.AppendByte(':')
// enc.buf.AppendByte('"')
// enc.buf.AppendByte(':')
enc.buf.AppendByte('=')
if enc.spaced {
enc.buf.AppendByte(' ')
}
}
func (enc *textEncoder) replaceSeparator(v byte) bool {
last := enc.buf.Len() - 1
if last < 0 {
return false
}
if enc.buf.Bytes()[last] == v {
t := enc.buf.Bytes()[:last]
enc.buf.Reset()
_, _ = enc.buf.Write(t)
return true
}
return false
}
func (enc *textEncoder) addElementSeparator() {
last := enc.buf.Len() - 1
if last < 0 {
@ -311,7 +345,7 @@ func (enc *textEncoder) addElementSeparator() {
case '{', '[', ':', ',', ' ':
return
default:
enc.buf.AppendByte(',')
// enc.buf.AppendByte(',')
if enc.spaced {
enc.buf.AppendByte(' ')
}

8
go.mod
View File

@ -4,6 +4,10 @@ go 1.13
require (
github.com/ehlxr/lumberjack v0.0.2-0.20200107093220-2a579f1b2e4d
github.com/robfig/cron/v3 v3.0.0
go.uber.org/zap v1.12.0
github.com/robfig/cron/v3 v3.0.1
go.uber.org/multierr v1.6.0 // indirect
go.uber.org/zap v1.16.0
golang.org/x/lint v0.0.0-20201208152925-83fdc39ff7b5 // indirect
golang.org/x/tools v0.0.0-20210104081019-d8d6ddbec6ee // indirect
honnef.co/go/tools v0.1.0 // indirect
)

11
log.go
View File

@ -35,6 +35,7 @@ type logConfig struct {
CrashLogFilename string
ErrorLogFilename string
EnableLineNumber bool
AddCallerSkip int
// enable the truncation of the level text to 4 characters.
EnableLevelTruncation bool
@ -62,6 +63,10 @@ func (config *logConfig) Init() {
logger = config.newLogger().Sugar()
}
func (config *logConfig) New() *zap.SugaredLogger {
return config.newLogger().Sugar()
}
func NewLogConfig() *logConfig {
return &logConfig{
Level: DebugLevel,
@ -113,7 +118,7 @@ func (config *logConfig) newLogger() *zap.Logger {
var options []zap.Option
if config.EnableLineNumber {
options = append(options, zap.AddCaller(), zap.AddCallerSkip(1))
options = append(options, zap.AddCaller(), zap.AddCallerSkip(config.AddCallerSkip))
}
if config.EnableErrorStacktrace {
@ -265,3 +270,7 @@ func writeCrashLog(file string) {
func Fields(args ...interface{}) {
logger = logger.With(args...)
}
func With(l *zap.SugaredLogger, args ...interface{}) *zap.SugaredLogger {
return l.With(args...)
}

View File

@ -1,9 +1,10 @@
package log
import (
"go.uber.org/zap"
"testing"
"time"
"go.uber.org/zap"
)
func TestLog(t *testing.T) {
@ -15,21 +16,33 @@ func TestLog(t *testing.T) {
func TestLogWithConfig(t *testing.T) {
config := NewLogConfig()
_ = config.Level.Set("d")
_ = config.Level.Set("debug")
config.Name = "main"
//config.Fields = []zap.Field{zap.String("traceid", "12123123123")}
// config.Fields = []zap.Field{zap.String("traceid", "12123123123")}
config.Init()
With("foo", "baz")
Fields("traceid", float64(21221212122))
Debugf("this is %s message", "debug")
config.Init()
With(zap.String("traceid", "12123123123"))
Fields(zap.String("traceid", "12123123123"))
Infof("this is %s message", "info")
//Errorf("this is %s message", "error")
// Errorf("this is %s message", "error")
// Panicf("this is %s message", "panic")
}
func TestLogWithNew(t *testing.T) {
config := NewLogConfig()
_ = config.Level.Set("debug")
config.Name = "main"
logger := config.New()
log := With(logger, "traceid", float64(21221212122), "request", "[POST]/hello/v2")
log.Debugf("this is %s message", "debug")
log.Infof("this is %s message", "info")
}
func TestLogRote(t *testing.T) {
lc := NewLogConfig()
lc.MaxSize = 1

View File

@ -1,9 +1,5 @@
package log
func With(args ...interface{}) {
logger = logger.With(args...)
}
func Debug(args ...interface{}) {
logger.Debug(args...)
}

View File

@ -1 +1 @@
v0.0.6
v0.0.11