package logger
|
|
import (
|
"fmt"
|
"log"
|
"os"
|
"os/exec"
|
"strings"
|
"time"
|
)
|
|
const (
|
PanicLevel int = iota
|
FatalLevel
|
ErrorLevel
|
WarnLevel
|
InfoLevel
|
DebugLevel
|
)
|
|
const (
|
color_red = uint8(iota + 91)
|
color_green // 绿
|
color_yellow // 黄
|
color_blue // 蓝
|
color_magenta // 洋红
|
)
|
|
const (
|
fatalPrefix = "[FATAL] "
|
errorPrefix = "[ERROR] "
|
warnPrefix = "[WARN] "
|
infoPrefix = "[INFO] "
|
debugPrefix = "[DEBUG] "
|
)
|
|
const (
|
ByDay int = iota
|
ByWeek
|
ByMonth
|
BySize
|
)
|
|
type LogFile struct {
|
level int // 日志等级
|
saveMode int // 保存模式
|
saveDays int // 日志保存天数
|
logTime int64 //
|
fileName string // 日志文件名
|
filesize int64 // 文件大小, 需要设置 saveMode 为 BySize 生效
|
fileFd *os.File
|
}
|
|
var logFile LogFile
|
|
func init() {
|
//logFile.logTime = time.Now().Unix()
|
logFile.saveMode = ByDay // 默认按天保存
|
logFile.saveDays = 15 // 默认保存三天的
|
logFile.level = ErrorLevel
|
logFile.filesize = 1024 * 1024 * 10 // 默认10M， 需要设置 saveMode 为 BySize
|
}
|
|
func Config(logFolder string, level int) {
|
logFile.fileName = logFolder
|
logFile.level = level
|
|
log.SetOutput(logFile)
|
//log.SetFlags(log.Lmicroseconds | log.Lshortfile)
|
log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)
|
}
|
func GetLogFile() (*LogFile) {
|
return &logFile
|
}
|
|
func SetLevel(level int) {
|
logFile.level = level
|
}
|
|
func SetSaveMode(saveMode int) {
|
logFile.saveMode = saveMode
|
}
|
|
func SetSaveDays(saveDays int) {
|
logFile.saveDays = saveDays
|
}
|
|
func SetSaveSize(saveSize int64) {
|
logFile.filesize = saveSize
|
}
|
|
func Debug(args ...interface{}) {
|
if logFile.level >= DebugLevel {
|
log.SetPrefix(blue(debugPrefix))
|
_ = log.Output(2, fmt.Sprintln(args...))
|
}
|
}
|
|
func Info(args ...interface{}) {
|
if logFile.level >= InfoLevel {
|
log.SetPrefix(green(infoPrefix))
|
_ = log.Output(2, fmt.Sprintln(args...))
|
}
|
}
|
|
func Warn(args ...interface{}) {
|
if logFile.level >= WarnLevel {
|
log.SetPrefix(magenta(warnPrefix))
|
_ = log.Output(2, fmt.Sprintln(args...))
|
}
|
}
|
|
func Error(args ...interface{}) {
|
if logFile.level >= ErrorLevel {
|
log.SetPrefix(red(errorPrefix))
|
_ = log.Output(2, fmt.Sprintln(args...))
|
}
|
}
|
|
func Fatal(args ...interface{}) {
|
if logFile.level >= FatalLevel {
|
log.SetPrefix(red(fatalPrefix))
|
_ = log.Output(2, fmt.Sprintln(args...))
|
}
|
}
|
|
func GetRedPrefix(s string) string {
|
return fmt.Sprintf("\x1b[%dm%s\x1b[0m", color_red, s)
|
}
|
|
func red(s string) string {
|
return fmt.Sprintf("\x1b[%dm%s\x1b[0m", color_red, s)
|
}
|
|
func green(s string) string {
|
return fmt.Sprintf("\x1b[%dm%s\x1b[0m", color_green, s)
|
}
|
|
func yellow(s string) string {
|
return fmt.Sprintf("\x1b[%dm%s\x1b[0m", color_yellow, s)
|
}
|
|
func blue(s string) string {
|
return fmt.Sprintf("\x1b[%dm%s\x1b[0m", color_blue, s)
|
}
|
|
func magenta(s string) string {
|
return fmt.Sprintf("\x1b[%dm%s\x1b[0m", color_magenta, s)
|
}
|
|
func (me LogFile) Write(buf []byte) (n int, err error) {
|
if me.fileName == "" {
|
fmt.Printf("consol: %s", buf)
|
return len(buf), nil
|
}
|
|
switch logFile.saveMode {
|
case BySize:
|
fileInfo, err := os.Stat(logFile.fileName)
|
if err != nil {
|
logFile.createLogFile()
|
logFile.logTime = time.Now().Unix()
|
}else {
|
filesize:= fileInfo.Size();
|
if logFile.fileFd == nil ||
|
filesize > logFile.filesize {
|
logFile.createLogFile()
|
logFile.logTime = time.Now().Unix()
|
}
|
}
|
default: // 默认按天 ByDay
|
if logFile.logTime+3600*24 < time.Now().Unix() {
|
logFile.createLogFile()
|
logFile.logTime = time.Now().Unix()
|
}
|
}
|
|
if logFile.fileFd == nil {
|
fmt.Printf("log fileFd is nil !\n")
|
return len(buf), nil
|
}
|
|
return logFile.fileFd.Write(buf)
|
}
|
|
func (me *LogFile) createLogFile() {
|
logdir := "./logger/"
|
if index := strings.LastIndex(me.fileName, "/"); index != -1 {
|
logdir = me.fileName[0:index] + "/"
|
os.MkdirAll(me.fileName[0:index], os.ModePerm)
|
}
|
|
now := time.Now()
|
filename := fmt.Sprintf("%s_%04d%02d%02d%02d",
|
me.fileName, now.Year(), now.Month(), now.Day(), now.Hour())
|
if err := os.Rename(me.fileName, filename); err == nil {
|
go func() {
|
filenameTemp := filename + "Temp"
|
rmCpCmd := exec.Command("cp", filename, filenameTemp)
|
rmCpCmd.Run()
|
|
tarCmd := exec.Command("tar", "-zcf", filename + ".tar.gz", filenameTemp, "--remove-files")
|
tarCmd.Run()
|
|
rmSrcCmd := exec.Command("rm", filename)
|
rmSrcCmd.Run()
|
|
rmCmd := exec.Command("/bin/sh", "-c",
|
"find "+logdir+` -type f -mtime +` +string(logFile.saveDays)+ ` -exec rm {} \;`)
|
rmCmd.Run()
|
}()
|
}
|
|
for index := 0; index < 10; index++ {
|
if fd, err := os.OpenFile(me.fileName, os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0666); nil == err {
|
me.fileFd.Sync()
|
me.fileFd.Close()
|
me.fileFd = fd
|
break
|
}else {
|
fmt.Println("Open logfile error! err: ", err.Error())
|
}
|
me.fileFd = nil
|
}
|
}
|
