package util import ( "fmt" "github.com/fatih/color" "time" ) const ( // LevelError 错误 LevelError = iota // LevelWarning 警告 LevelWarning // LevelInformational 提示 LevelInformational // LevelDebug 除错 LevelDebug ) var logger *Logger // Logger 日志 type Logger struct { level int } // 日志颜色 var colors = map[string]func(a ...interface{}) string{ "Warning": color.New(color.FgYellow).Add(color.Bold).SprintFunc(), "Panic": color.New(color.BgRed).Add(color.Bold).SprintFunc(), "Error": color.New(color.FgRed).Add(color.Bold).SprintFunc(), "Info": color.New(color.FgCyan).Add(color.Bold).SprintFunc(), "Debug": color.New(color.FgWhite).Add(color.Bold).SprintFunc(), } // 不同级别前缀与时间的间隔,保持宽度一致 var spaces = map[string]string{ "Warning": "", "Panic": " ", "Error": " ", "Info": " ", "Debug": " ", } // Println 打印 func (ll *Logger) Println(prefix string, msg string) { // TODO Release时去掉 color.NoColor = false c := color.New() _, _ = c.Printf( "%s%s %s %s\n", colors[prefix]("["+prefix+"]"), spaces[prefix], time.Now().Format("2006-01-02 15:04:05"), msg, ) } // Panic 极端错误 func (ll *Logger) Panic(format string, v ...interface{}) { if LevelError > ll.level { return } msg := fmt.Sprintf(format, v...) ll.Println("Panic", msg) panic(msg) } // Error 错误 func (ll *Logger) Error(format string, v ...interface{}) { if LevelError > ll.level { return } msg := fmt.Sprintf(format, v...) ll.Println("Error", msg) } // Warning 警告 func (ll *Logger) Warning(format string, v ...interface{}) { if LevelWarning > ll.level { return } msg := fmt.Sprintf(format, v...) ll.Println("Warning", msg) } // Info 信息 func (ll *Logger) Info(format string, v ...interface{}) { if LevelInformational > ll.level { return } msg := fmt.Sprintf(format, v...) ll.Println("Info", msg) } // Debug 校验 func (ll *Logger) Debug(format string, v ...interface{}) { if LevelDebug > ll.level { return } msg := fmt.Sprintf(format, v...) ll.Println("Debug", msg) } // Print GORM 的 Logger实现 //func (ll *Logger) Print(v ...interface{}) { // if LevelDebug > ll.level { // return // } // msg := fmt.Sprintf("[SQL] %s", v...) // ll.Println(msg) //} // BuildLogger 构建logger func BuildLogger(level string) { intLevel := LevelError switch level { case "error": intLevel = LevelError case "warning": intLevel = LevelWarning case "info": intLevel = LevelInformational case "debug": intLevel = LevelDebug } l := Logger{ level: intLevel, } logger = &l } // Log 返回日志对象 func Log() *Logger { if logger == nil { l := Logger{ level: LevelDebug, } logger = &l } return logger }