93 lines
1.8 KiB
Go
93 lines
1.8 KiB
Go
package jsonlog
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io"
|
|
"os"
|
|
"runtime/debug"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
type Level int8
|
|
|
|
const (
|
|
LevelInfo Level = iota
|
|
LevelError
|
|
LevelFatal
|
|
LevelOff
|
|
)
|
|
|
|
func (l Level) String() string {
|
|
switch l {
|
|
case LevelInfo:
|
|
return "INFO"
|
|
case LevelError:
|
|
return "ERROR"
|
|
case LevelFatal:
|
|
return "FATAL"
|
|
default:
|
|
return ""
|
|
}
|
|
}
|
|
|
|
type Logger struct {
|
|
out io.Writer
|
|
minLevel Level
|
|
mu sync.Mutex
|
|
}
|
|
|
|
func New(out io.Writer, minLevel Level) *Logger {
|
|
return &Logger{
|
|
out: out,
|
|
minLevel: minLevel,
|
|
}
|
|
}
|
|
|
|
func (l *Logger) PrintInfo(message string, properties map[string]string) {
|
|
l.print(LevelInfo, message, properties)
|
|
}
|
|
|
|
func (l *Logger) PrintError(err error, properties map[string]string) {
|
|
l.print(LevelError, err.Error(), properties)
|
|
}
|
|
|
|
func (l *Logger) PrintFatal(err error, properties map[string]string) {
|
|
l.print(LevelFatal, err.Error(), properties)
|
|
os.Exit(1)
|
|
}
|
|
|
|
func (l *Logger) print(level Level, message string, properties map[string]string) (int, error) {
|
|
if level < l.minLevel {
|
|
return 0, nil
|
|
}
|
|
aux := struct {
|
|
Level string `json:"level"`
|
|
TIme string `json:"time"`
|
|
Message string `json:"message"`
|
|
Properties map[string]string `json:"properties,omitempty"`
|
|
Trace string `json:"trace,omitempty"`
|
|
}{
|
|
Level: level.String(),
|
|
TIme: time.Now().UTC().Format(time.RFC3339),
|
|
Message: message,
|
|
Properties: properties,
|
|
}
|
|
|
|
if level >= LevelError {
|
|
aux.Trace = string(debug.Stack())
|
|
}
|
|
var line []byte
|
|
line, err := json.Marshal(aux)
|
|
if err != nil {
|
|
line = []byte(LevelError.String() + ": unable to marshal log message:" + err.Error())
|
|
}
|
|
l.mu.Lock()
|
|
defer l.mu.Unlock()
|
|
return l.out.Write(append(line, '\n'))
|
|
}
|
|
|
|
func (l *Logger) Write(message []byte) (n int, err error) {
|
|
return l.print(LevelError, string(message), nil)
|
|
}
|