package log import ( "bytes" "encoding/json" "fmt" "io" "os" "path" "runtime" "strconv" "sync" "sync/atomic" "time" "github.com/labstack/gommon/color" "github.com/mattn/go-isatty" "github.com/valyala/fasttemplate" ) type ( Logger struct { prefix string level uint32 skip int output io.Writer template *fasttemplate.Template levels []string color *color.Color bufferPool sync.Pool mutex sync.Mutex } Lvl uint8 JSON map[string]interface{} ) const ( DEBUG Lvl = iota + 1 INFO WARN ERROR OFF panicLevel fatalLevel ) var ( global = New("-") defaultHeader = `{"time":"${time_rfc3339_nano}","level":"${level}","prefix":"${prefix}",` + `"file":"${short_file}","line":"${line}"}` ) func init() { global.skip = 3 } func New(prefix string) (l *Logger) { l = &Logger{ level: uint32(INFO), skip: 2, prefix: prefix, template: l.newTemplate(defaultHeader), color: color.New(), bufferPool: sync.Pool{ New: func() interface{} { return bytes.NewBuffer(make([]byte, 256)) }, }, } l.initLevels() l.SetOutput(output()) return } func (l *Logger) initLevels() { l.levels = []string{ "-", l.color.Blue("DEBUG"), l.color.Green("INFO"), l.color.Yellow("WARN"), l.color.Red("ERROR"), "", l.color.Yellow("PANIC", color.U), l.color.Red("FATAL", color.U), } } func (l *Logger) newTemplate(format string) *fasttemplate.Template { return fasttemplate.New(format, "${", "}") } func (l *Logger) DisableColor() { l.color.Disable() l.initLevels() } func (l *Logger) EnableColor() { l.color.Enable() l.initLevels() } func (l *Logger) Prefix() string { return l.prefix } func (l *Logger) SetPrefix(p string) { l.prefix = p } func (l *Logger) Level() Lvl { return Lvl(atomic.LoadUint32(&l.level)) } func (l *Logger) SetLevel(level Lvl) { atomic.StoreUint32(&l.level, uint32(level)) } func (l *Logger) Output() io.Writer { return l.output } func (l *Logger) SetOutput(w io.Writer) { l.output = w if w, ok := w.(*os.File); !ok || !isatty.IsTerminal(w.Fd()) { l.DisableColor() } } func (l *Logger) Color() *color.Color { return l.color } func (l *Logger) SetHeader(h string) { l.template = l.newTemplate(h) } func (l *Logger) Print(i ...interface{}) { l.log(0, "", i...) // fmt.Fprintln(l.output, i...) } func (l *Logger) Printf(format string, args ...interface{}) { l.log(0, format, args...) } func (l *Logger) Printj(j JSON) { l.log(0, "json", j) } func (l *Logger) Debug(i ...interface{}) { l.log(DEBUG, "", i...) } func (l *Logger) Debugf(format string, args ...interface{}) { l.log(DEBUG, format, args...) } func (l *Logger) Debugj(j JSON) { l.log(DEBUG, "json", j) } func (l *Logger) Info(i ...interface{}) { l.log(INFO, "", i...) } func (l *Logger) Infof(format string, args ...interface{}) { l.log(INFO, format, args...) } func (l *Logger) Infoj(j JSON) { l.log(INFO, "json", j) } func (l *Logger) Warn(i ...interface{}) { l.log(WARN, "", i...) } func (l *Logger) Warnf(format string, args ...interface{}) { l.log(WARN, format, args...) } func (l *Logger) Warnj(j JSON) { l.log(WARN, "json", j) } func (l *Logger) Error(i ...interface{}) { l.log(ERROR, "", i...) } func (l *Logger) Errorf(format string, args ...interface{}) { l.log(ERROR, format, args...) } func (l *Logger) Errorj(j JSON) { l.log(ERROR, "json", j) } func (l *Logger) Fatal(i ...interface{}) { l.log(fatalLevel, "", i...) os.Exit(1) } func (l *Logger) Fatalf(format string, args ...interface{}) { l.log(fatalLevel, format, args...) os.Exit(1) } func (l *Logger) Fatalj(j JSON) { l.log(fatalLevel, "json", j) os.Exit(1) } func (l *Logger) Panic(i ...interface{}) { l.log(panicLevel, "", i...) panic(fmt.Sprint(i...)) } func (l *Logger) Panicf(format string, args ...interface{}) { l.log(panicLevel, format, args...) panic(fmt.Sprintf(format, args...)) } func (l *Logger) Panicj(j JSON) { l.log(panicLevel, "json", j) panic(j) } func DisableColor() { global.DisableColor() } func EnableColor() { global.EnableColor() } func Prefix() string { return global.Prefix() } func SetPrefix(p string) { global.SetPrefix(p) } func Level() Lvl { return global.Level() } func SetLevel(level Lvl) { global.SetLevel(level) } func Output() io.Writer { return global.Output() } func SetOutput(w io.Writer) { global.SetOutput(w) } func SetHeader(h string) { global.SetHeader(h) } func Print(i ...interface{}) { global.Print(i...) } func Printf(format string, args ...interface{}) { global.Printf(format, args...) } func Printj(j JSON) { global.Printj(j) } func Debug(i ...interface{}) { global.Debug(i...) } func Debugf(format string, args ...interface{}) { global.Debugf(format, args...) } func Debugj(j JSON) { global.Debugj(j) } func Info(i ...interface{}) { global.Info(i...) } func Infof(format string, args ...interface{}) { global.Infof(format, args...) } func Infoj(j JSON) { global.Infoj(j) } func Warn(i ...interface{}) { global.Warn(i...) } func Warnf(format string, args ...interface{}) { global.Warnf(format, args...) } func Warnj(j JSON) { global.Warnj(j) } func Error(i ...interface{}) { global.Error(i...) } func Errorf(format string, args ...interface{}) { global.Errorf(format, args...) } func Errorj(j JSON) { global.Errorj(j) } func Fatal(i ...interface{}) { global.Fatal(i...) } func Fatalf(format string, args ...interface{}) { global.Fatalf(format, args...) } func Fatalj(j JSON) { global.Fatalj(j) } func Panic(i ...interface{}) { global.Panic(i...) } func Panicf(format string, args ...interface{}) { global.Panicf(format, args...) } func Panicj(j JSON) { global.Panicj(j) } func (l *Logger) log(level Lvl, format string, args ...interface{}) { if level >= l.Level() || level == 0 { buf := l.bufferPool.Get().(*bytes.Buffer) buf.Reset() defer l.bufferPool.Put(buf) _, file, line, _ := runtime.Caller(l.skip) message := "" if format == "" { message = fmt.Sprint(args...) } else if format == "json" { b, err := json.Marshal(args[0]) if err != nil { panic(err) } message = string(b) } else { message = fmt.Sprintf(format, args...) } _, err := l.template.ExecuteFunc(buf, func(w io.Writer, tag string) (int, error) { switch tag { case "time_rfc3339": return w.Write([]byte(time.Now().Format(time.RFC3339))) case "time_rfc3339_nano": return w.Write([]byte(time.Now().Format(time.RFC3339Nano))) case "level": return w.Write([]byte(l.levels[level])) case "prefix": return w.Write([]byte(l.prefix)) case "long_file": return w.Write([]byte(file)) case "short_file": return w.Write([]byte(path.Base(file))) case "line": return w.Write([]byte(strconv.Itoa(line))) } return 0, nil }) if err == nil { s := buf.String() i := buf.Len() - 1 if i >= 0 && s[i] == '}' { // JSON header buf.Truncate(i) buf.WriteByte(',') if format == "json" { buf.WriteString(message[1:]) } else { buf.WriteString(`"message":`) buf.WriteString(strconv.Quote(message)) buf.WriteString(`}`) } } else { // Text header if len(s) > 0 { buf.WriteByte(' ') } buf.WriteString(message) } buf.WriteByte('\n') l.mutex.Lock() defer l.mutex.Unlock() l.output.Write(buf.Bytes()) } } }