76 lines
1.7 KiB
Go
Executable File
76 lines
1.7 KiB
Go
Executable File
package console
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
|
|
"encoding/json"
|
|
|
|
"github.com/fatih/color"
|
|
"github.com/snail007/mini-logger"
|
|
)
|
|
|
|
const (
|
|
T_JSON = iota
|
|
T_TEXT
|
|
)
|
|
|
|
type ConsoleWriterConfig struct {
|
|
Format string
|
|
Type int
|
|
}
|
|
type ConsoleWriter struct {
|
|
c ConsoleWriterConfig
|
|
}
|
|
|
|
func New(config ConsoleWriterConfig) *ConsoleWriter {
|
|
return &ConsoleWriter{
|
|
c: config,
|
|
}
|
|
}
|
|
func NewDefault() *ConsoleWriter {
|
|
return &ConsoleWriter{
|
|
c: ConsoleWriterConfig{
|
|
Format: "[{level}] [{date} {time}.{mili}] {fields} {text}",
|
|
Type: T_TEXT,
|
|
},
|
|
}
|
|
}
|
|
func (w *ConsoleWriter) Init() (err error) {
|
|
return
|
|
}
|
|
func (w *ConsoleWriter) Write(e logger.Entry) {
|
|
fg := color.New(color.FgHiWhite).SprintFunc()
|
|
switch e.Level {
|
|
case logger.InfoLevel:
|
|
fg = color.New(color.FgHiGreen).SprintFunc()
|
|
case logger.WarnLevel:
|
|
fg = color.New(color.FgHiYellow).SprintFunc()
|
|
case logger.ErrorLevel:
|
|
fg = color.New(color.FgHiRed).SprintFunc()
|
|
case logger.FatalLevel:
|
|
fg = color.New(color.FgHiRed, color.FgHiMagenta).SprintFunc()
|
|
}
|
|
date := time.Unix(e.Timestamp, 0).Format("2006/01/02")
|
|
time := time.Unix(e.Timestamp, 0).Format("15:04:05")
|
|
var c string
|
|
if w.c.Type == T_TEXT {
|
|
c = strings.Replace(w.c.Format, "{level}", fmt.Sprintf("%-5s", e.LevelString), -1)
|
|
c = strings.Replace(c, "{date}", date, -1)
|
|
c = strings.Replace(c, "{time}", time, -1)
|
|
c = strings.Replace(c, "{fields}", e.FieldsString, -1)
|
|
c = strings.Replace(c, "{mili}", fmt.Sprintf("%03d", e.Milliseconds), -1)
|
|
c = strings.Replace(c, "{text}", e.Content, -1)
|
|
} else if w.c.Type == T_JSON {
|
|
e.LevelString = strings.TrimRight(e.LevelString, " ")
|
|
e.FieldsString = ""
|
|
v, _ := json.Marshal(e)
|
|
c = string(v)
|
|
} else {
|
|
return
|
|
}
|
|
fmt.Fprintln(os.Stdout, fg(c))
|
|
}
|