支持按天滚动写日志

Author: c-wind

log/log.go

@@ -2,10 +2,9 @@ package log
 
 import (
 	"fmt"
-	"os"
 )
 
-var mlog = &Logger{out: os.Stdout, level: LogDebug, color: true}
+var mlog = NewLogger()
 
 func SetLevel(level LogLevel) {
 	mlog.SetLevel(level)
@@ -24,31 +23,31 @@ func Infof(format string, v ...interface{}) {
 }
 
 func Debug(v ...interface{}) {
-	mlog.write(LogDebug,fmt.Sprint( v...))
+	mlog.write(LogDebug, fmt.Sprint(v...))
 }
 
 func Debugf(format string, v ...interface{}) {
 	mlog.write(LogDebug, format, v...)
 }
 
 func Warning(v ...interface{}) {
-	mlog.write(LogWarning, fmt.Sprint( v...))
+	mlog.write(LogWarning, fmt.Sprint(v...))
 }
 
 func Warningf(format string, v ...interface{}) {
 	mlog.write(LogWarning, format, v...)
 }
 
 func Error(v ...interface{}) {
-	mlog.write(LogError, fmt.Sprint( v...))
+	mlog.write(LogError, fmt.Sprint(v...))
 }
 
 func Errorf(format string, v ...interface{}) {
 	mlog.write(LogError, format, v...)
 }
 
 func Fatal(v ...interface{}) {
-	mlog.write(LogFatal, fmt.Sprint( v...))
+	mlog.write(LogFatal, fmt.Sprint(v...))
 }
 
 func Fatalf(format string, v ...interface{}) {

log/log_test.go

@@ -13,7 +13,15 @@ func TestLog(t *testing.T) {
 	l := NewLogger()
 	l.Info("logger 2222222222")
 	l.Errorf("logger color %v xxxxxx", time.Now().UnixNano())
-    l.SetColor(false)
+	l.SetColor(false)
 	l.Errorf("logger no color %v yyyyyy", time.Now().UnixNano())
 	Infof("%v default has color test log", time.Now())
+
+	l.SetOutputFile("./vvv.log").SetRolling(true)
+	l.Infof("vvvvvvvvv")
+
+	for i := 0; i < 999999999; i++ {
+		time.Sleep(time.Second)
+		l.Info(time.Now())
+	}
 }

log/log_writer.go

@@ -1,6 +1,7 @@
 package log
 
 import (
+	"bufio"
 	"bytes"
 	"fmt"
 	"os"
@@ -9,19 +10,30 @@ import (
 	"time"
 )
 
+type pos struct {
+	file     string
+	function string
+}
+
 type Logger struct {
-	out   *os.File
-	level LogLevel
-	color bool
-	mu    sync.Mutex
+	rolling  bool
+	fileName string
+	fileTime time.Time
+	file     *os.File
+	out      *bufio.Writer
+	level    LogLevel
+	color    bool
+	posCache map[uintptr]pos
+	mu       sync.Mutex
 }
 
 //NewLogger 创建日志对象.
 func NewLogger() *Logger {
 	return &Logger{
-		out:   os.Stdout,
-		level: LogDebug,
-		color: true,
+		out:      bufio.NewWriter(os.Stdout),
+		level:    LogDebug,
+		color:    true,
+		posCache: make(map[uintptr]pos),
 	}
 }
 
@@ -31,38 +43,46 @@ func (l *Logger) SetColor(on bool) *Logger {
 	return l
 }
 
-//SetLevel 设置日志级别.
-func (l *Logger) SetLevel(level LogLevel) {
-	l.level = level
+//SetRolling 每天生成一个文件.
+func (l *Logger) SetRolling(on bool) *Logger {
+	l.rolling = on
+	return l
 }
 
-//SetLevelByString 设置字符串格式的日志级别.
-func (l *Logger) SetLevelByString(level string) {
-	l.level = StringToLogLevel(level)
-}
+//SetOutputFile 初始化时设置输出文件.
+func (l *Logger) SetOutputFile(path string) *Logger {
+	f, err := os.OpenFile(path, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
+	if err != nil {
+		panic(fmt.Sprintf("open %s error:%v", path, err.Error()))
+	}
 
-type pos struct {
-	file string
-	line int
-	name string
+	now, _ := time.ParseInLocation("20060102", time.Now().Format("20060102"), time.Local)
+	l.fileTime = now.Add(time.Hour * 24)
+	l.file = f
+	l.fileName = path
+	l.out = bufio.NewWriter(f)
+
+	return l
 }
 
-type posCache struct {
-	ps map[uintptr]pos
-	sync.RWMutex
+//SetLevel 设置日志级别.
+func (l *Logger) SetLevel(level LogLevel) *Logger {
+	l.level = level
+	return l
 }
 
-var mpc = posCache{ps: make(map[uintptr]pos)}
+//SetLevelByString 设置字符串格式的日志级别.
+func (l *Logger) SetLevelByString(level string) *Logger {
+	l.level = StringToLogLevel(level)
+	return l
+}
 
-func (l *Logger) caller() (string, int, string) {
+func (l *Logger) caller() (string, string) {
 	pc, file, line, _ := runtime.Caller(3)
 
-	mpc.RLock()
-	p, ok := mpc.ps[pc]
-	mpc.RUnlock()
-
+	p, ok := l.posCache[pc]
 	if ok {
-		return p.file, p.line, p.name
+		return p.file, p.function
 	}
 
 	name := runtime.FuncForPC(pc).Name()
@@ -73,40 +93,69 @@ func (l *Logger) caller() (string, int, string) {
 		file = file[i+1:]
 	}
 
-	mpc.Lock()
-	mpc.ps[pc] = pos{file: file, line: line, name: name}
-	mpc.Unlock()
+	p = pos{file: fmt.Sprintf("%s:%d", file, line), function: name}
+	l.posCache[pc] = p
+
+	return p.file, p.function
+}
+
+func (l *Logger) rotate(now time.Time) {
+	if !l.rolling || l.file == nil || now.Before(l.fileTime) {
+		return
+	}
+
+	l.out.Flush()
+	l.file.Close()
+
+	oldFile := l.fileName + time.Now().Format("20060102")
+
+	os.Rename(l.fileName, oldFile)
 
-	return file, line, name
+	l.SetOutputFile(l.fileName)
 }
 
 func (l *Logger) write(t LogLevel, format string, argv ...interface{}) {
 	if t > l.level {
 		return
 	}
 
-	date := time.Now().Format("2006/01/02 15:04:05")
+	l.mu.Lock()
+	defer l.mu.Unlock()
 
-	file, line, name := l.caller()
+	now := time.Now()
+	l.rotate(now)
+	date := now.Format("2006/01/02 15:04:05")
+
+	file, function := l.caller()
+
+	//时间，源码文件和行号
+	l.out.WriteString(date)
+	l.out.WriteString(" ")
+	l.out.WriteString(file)
+	l.out.WriteString(" ")
 
-	//时间，源码文件，源码列
-	fmt.Fprintf(l.out, "%s %s:%d ", date, file, line)
 	if l.color {
 		//颜色开始
-		fmt.Fprint(l.out, t.Color())
+		l.out.WriteString(t.Color())
 	}
 
+	//日志级别
+	l.out.WriteString(t.String())
+
 	//函数名
-	fmt.Fprintf(l.out, "%s %s ", t.String(), name)
+	l.out.WriteString(function)
 
+	//日志正文
 	fmt.Fprintf(l.out, format, argv...)
 
 	if l.color {
 		//颜色结束
-		fmt.Fprint(l.out, "\033[0m")
+		l.out.WriteString("\033[0m")
 	}
 
 	l.out.WriteString("\n")
+
+	l.out.Flush()
 }
 
 func (l *Logger) Info(v ...interface{}) {