feat: support single and rotate handler

Author: leeqvip

log/handler/file.go

@@ -20,6 +20,8 @@ type FileHandler struct {
 	filenameFormat string
 	dateFormat     string
 	timedFilename  string
+
+	rotate bool
 }
 
 func NewFileHandler(filename string, level record.Level) *FileHandler {
@@ -60,12 +62,21 @@ func (h *FileHandler) SetLevel(level record.Level) {
 func (h *FileHandler) write(r record.Record) {
 	h.Lock()
 	defer h.Unlock()
-	file, _ := os.OpenFile(h.GetTimedFilename(), os.O_RDWR|os.O_CREATE|os.O_APPEND, 0644)
+	file, _ := os.OpenFile(h.GetFilename(), os.O_RDWR|os.O_CREATE|os.O_APPEND, 0644)
 	defer file.Close()
 	file.Write([]byte(r.Formatted))
 }
 
-// GetTimedFilename Gets the Timed filename.
+// GetFilename Gets the filename.
+func (h *FileHandler) GetFilename() string {
+	if !h.rotate {
+		return h.filename
+	}
+
+	return h.GetTimedFilename()
+}
+
+// GetTimedFilename Gets the timed filename.
 func (h *FileHandler) GetTimedFilename() string {
 	dirname := path.Dir(h.filename)
 	filename := path.Base(h.filename)

log/handler/file_test.go

@@ -15,7 +15,7 @@ func TestNewFileHandler(t *testing.T) {
 	filename := path.Join(os.TempDir(), "thinkgo.log")
 
 	h := NewFileHandler(filename, record.DEBUG)
-	filename = h.GetTimedFilename()
+	filename = h.GetFilename()
 
 	os.Remove(filename)
 
@@ -32,7 +32,7 @@ func TestNewFileHandler(t *testing.T) {
 	if !strings.Contains(content, message) {
 		t.Error("test FileHandler error")
 	}
-
+	t.Log(filename, content)
 }
 
 func getRecord(message string) record.Record {

log/handler/rotate.go

@@ -0,0 +1,13 @@
+package handler
+
+import (
+	"github.com/forgoer/thinkgo/log/record"
+)
+
+func NewRotateHandler(filename string, level record.Level) *FileHandler {
+	// h.timedFilename = h.GetTimedFilename()
+	fileHandler := NewFileHandler(filename, level)
+	fileHandler.rotate = true
+
+	return fileHandler
+}

log/handler/rotate_test.go

@@ -0,0 +1,34 @@
+package handler
+
+import (
+	"github.com/forgoer/thinkgo/log/record"
+	"io/ioutil"
+	"os"
+	"path"
+	"strings"
+	"testing"
+)
+
+func TestRotateHandler(t *testing.T) {
+	filename := path.Join(os.TempDir(), "thinkgo.log")
+
+	h := NewRotateHandler(filename, record.DEBUG)
+	filename = h.GetFilename()
+
+	os.Remove(filename)
+
+	message := "Log write to file"
+	r := getRecord(message)
+	h.Handle(r)
+
+	b, err := ioutil.ReadFile(filename)
+	if err != nil {
+		t.Error(err)
+	}
+	content := string(b)
+
+	if !strings.Contains(content, message) {
+		t.Error("test FileHandler error")
+	}
+	t.Log(filename, content)
+}