From ce98b833ae9e23fa844f694ccc84cd778c3a5e38 Mon Sep 17 00:00:00 2001 From: yanweidong Date: Fri, 21 Feb 2025 15:11:34 +0800 Subject: [PATCH] fix print module --- print/print.go | 20 ++++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) diff --git a/print/print.go b/print/print.go index 8311065..7aef2c9 100644 --- a/print/print.go +++ b/print/print.go @@ -3,32 +3,36 @@ package print import ( "fmt" "log" + "os" ) +var logger *log.Logger + func init() { - log.SetFlags(log.Ldate | log.Ltime) + logger = log.New(os.Stdout, "", 0) // 创建一个新的 Logger 实例,不带任何标志 + logger.SetFlags(log.Ldate | log.Ltime) } // record INFO message. Color White func Info(format string, a ...interface{}) { - message := fmt.Sprintf("\033[37m[Info] "+format+"\033[0m\n", a...) - log.Print(message) + message := fmt.Sprintf("\033[37m[Info] "+format+"\033[0m\n", a...) + logger.Print(message) } // record Warn message. Color Orange func Warn(format string, a ...interface{}) { - message := fmt.Sprintf("\033[33m[Warn] "+format+"\033[0m\n", a...) - log.Print(message) + message := fmt.Sprintf("\033[33m[Warn] "+format+"\033[0m\n", a...) + logger.Print(message) } // record Success message. Color Green func Success(format string, a ...interface{}) { message := fmt.Sprintf("\033[32m[Success] "+format+"\033[0m\n", a...) - log.Print(message) + logger.Print(message) } // record ERROR message. Color Red func Error(format string, a ...interface{}) { - message := fmt.Sprintf("\033[31m[Error] "+format+"\033[0m\n", a...) - log.Print(message) + message := fmt.Sprintf("\033[31m[Error] "+format+"\033[0m\n", a...) + logger.Print(message) }