add logger root method (#20)

* add logger root methoe * rotate date * modify rotate date * rotate by date --------- Co-authored-by: chuyali <[email protected]>
GuanceCloud · Sep 26, 2023 · abf20c8 · abf20c8
1 parent db5a1bc
commit abf20c8
Show file tree

Hide file tree

Showing 3 changed files with 67 additions and 8 deletions.
diff --git a/logger/customize_output.go b/logger/customize_output.go
@@ -70,3 +70,26 @@ func newCustomizeRootLogger(level string, options int, ws io.Writer) (*zap.Logge
 	l := zap.New(core, zap.AddCaller())
 	return l, nil
 }
+
+func newOnlyMessageRootLogger(ws io.Writer) (*zap.Logger, error) {
+	// use lumberjack.Logger for rotate
+	w := zapcore.AddSync(ws)
+
+	cfg := zapcore.EncoderConfig{
+		MessageKey: NameKeyMsg,
+
+		EncodeLevel:  zapcore.CapitalLevelEncoder,
+		EncodeTime:   zapcore.ISO8601TimeEncoder,
+		EncodeCaller: zapcore.FullCallerEncoder,
+	}
+
+	//enc := zapcore.NewJSONEncoder(cfg)
+	enc := zapcore.NewConsoleEncoder(cfg)
+	lvl := zap.DebugLevel
+
+	core := zapcore.NewCore(enc, w, lvl)
+	// NOTE: why need add another option while
+	// zapcore.ShortCallerEncoder/FullCallerEncoder been set
+	l := zap.New(core, zap.AddCaller())
+	return l, nil
+}
diff --git a/logger/logger.go b/logger/logger.go
@@ -48,9 +48,11 @@ type Logger struct {
 }
 
 type Option struct {
-	Path  string
-	Level string
-	Flags int
+	Path     string
+	Level    string
+	MaxSize  int
+	Flags    int
+	Compress bool
 }
 
 func Reset() {

diff --git a/logger/root.go b/logger/root.go
@@ -13,6 +13,7 @@ import (
 	"os"
 	"path/filepath"
 	"strings"
+	"time"
 
 	"go.uber.org/zap"
 	"gopkg.in/natefinch/lumberjack.v2"
@@ -64,11 +65,11 @@ func SetGlobalRootLogger(fpath, level string, options int) error {
 }
 
 // InitRoot used to setup global root logger, include
-//	- log level
-//	- log path
-//		- set to disk file(with or without rotate)
-//		- set to some remtoe TCP/UDP server
-//	- a bounch of other OPT_XXXs
+//   - log level
+//   - log path
+//   - set to disk file(with or without rotate)
+//   - set to some remtoe TCP/UDP server
+//   - a bounch of other OPT_XXXs
 func InitRoot(opt *Option) error {
 	if opt == nil {
 		opt = defaultOption
@@ -149,3 +150,36 @@ func newRootLogger(fpath, level string, options int) (*zap.Logger, error) {
 
 	return newNormalRootLogger(fpath, level, options)
 }
+
+// InitCustomizeRoot used to setup global root logger, include
+//   - log path
+//   - log maxsize
+//   - log compress
+
+func InitCustomizeRoot(opt *Option) (*zap.Logger, error) {
+	mtx.Lock()
+	defer mtx.Unlock()
+
+	lumberLog := &lumberjack.Logger{
+		Filename: opt.Path,
+		MaxSize:  opt.MaxSize,
+		Compress: opt.Compress,
+	}
+	go func() {
+		next := time.Date(time.Now().Year(), time.Now().Month(), time.Now().Day()+1, 0, 0, 0, 0, time.Local)
+		after := next.Unix() - time.Now().Unix() - 1
+
+		time.Sleep(time.Duration(after) * time.Second)
+		lumberLog.Rotate()
+
+		ticker := time.NewTicker(24 * time.Hour)
+		defer ticker.Stop()
+		for {
+			select {
+			case <-ticker.C:
+				lumberLog.Rotate()
+			}
+		}
+	}()
+	return newOnlyMessageRootLogger(lumberLog)
+}