Golang 按大小 or 时间切割日志

package log

import (
	"fmt"
	"os"
	"strings"
	"time"

	"github.com/natefinch/lumberjack"
	"go.uber.org/zap"
	"go.uber.org/zap/zapcore"
)

var Sugar *zap.SugaredLogger = nil
var Raw *zap.Logger = nil


// LogInit 初始化日志库
// dev 开发模式，日志不入文件
// logPath 日志文件存放路径
// logName 日志文件名称
// maxSize 单个文件大小，超过后切割，单位 M
// maxBackups 旧的日志文件最多保留个数，0 为保存所有
// maxAge 旧的日志文件最多保留天数，0 为保存所有
// MaxBackups 和 maxAge 只要有一个不满足，就不再保留
func LogInit(dev bool, logPath, logName string, maxSize, maxBackups, maxAge int) (err error) {
  if dev {
    Raw, _ = zap.NewDevelopment()
    Sugar = Raw.Sugar()
    return
  }

  if !strings.HasSuffix(logPath, "/") {
    logPath += "/"
  }

  if !strings.HasSuffix(logName, ".log") {
    logName += ".log"
  }

  _, err = os.Stat(logPath)
  if os.IsNotExist(err) {
    // Create parent directory if it does not exist
    if err = os.MkdirAll(logPath, 0744); err != nil {
      fmt.Println("os.MkdirAll failed, err=", err)
      return
    }
  }
  
  if err != nil {
    fmt.Println("os.Stat failed, err=", err)
    return
  }

  w := zapcore.AddSync(&lumberjack.Logger{
    Filename:   logPath + logName,
    MaxSize:    maxSize,
    MaxBackups: maxBackups,
    MaxAge:     maxAge,
  })

  core := zapcore.NewCore(
    zapcore.NewConsoleEncoder(zap.NewProductionEncoderConfig()),
    w,
    zap.InfoLevel,
  )
  Raw = zap.New(core)
  Sugar = Raw.Sugar()

  return
}

core := zapcore.NewCore(
  zapcore.NewConsoleEncoder(MyEncoderConfig()),
  w,
  zap.InfoLevel,
)

func MyEncoderConfig() zapcore.EncoderConfig {
  return zapcore.EncoderConfig{
    TimeKey:       "ts",
    LevelKey:      "level",
    NameKey:       "logger",
    CallerKey:     "caller",
    FunctionKey:   zapcore.OmitKey,
    MessageKey:    "msg",
    StacktraceKey: "stacktrace",
    LineEnding:    zapcore.DefaultLineEnding,
    EncodeLevel:   zapcore.LowercaseLevelEncoder,
    EncodeTime: func(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
      enc.AppendString(t.Format("2006/01/02T15:04:05"))
    },
    EncodeDuration: zapcore.SecondsDurationEncoder,
    EncodeCaller:   zapcore.ShortCallerEncoder,
  }
}

package log

import (
	"fmt"
	"os"
	"strings"
	"time"

	rotatelogs "github.com/lestrrat/go-file-rotatelogs"
	"go.uber.org/zap"
	"go.uber.org/zap/zapcore"
)

var Sugar *zap.SugaredLogger = nil
var Raw *zap.Logger = nil

// LogInit 初始化日志库
// logPath 日志文件存放路径
// logName 日志文件名称
// rotationTime 按时间分割，分割的时间长度
func LogInit(logPath, logName string, rotationTime time.Duration) (err error) {
  if !strings.HasSuffix(logPath, "/") {
    logPath += "/"
  }

  if !strings.HasSuffix(logName, ".log") {
    logName += ".log"
  }

  _, err = os.Stat(logPath)
  if os.IsNotExist(err) {
    // Create parent directory if it does not exist
    if err = os.MkdirAll(logPath, 0744); err != nil {
      fmt.Println("os.MkdirAll failed, err=", err)
      return
    }
  }

  if err != nil {
    fmt.Println("os.Stat failed, err=", err)
    return
  }

  rotate, err := rotatelogs.New(
    logPath+logName+".%Y%m%d",
    rotatelogs.WithLinkName(logPath+logName),
    rotatelogs.WithMaxAge(rotationTime), // 这里不写默认 24 hour
  )
  if err != nil {
    fmt.Println("rotatelogs.New failed, err=", err)
    return
  }

  writer := zapcore.AddSync(rotate)

  core := zapcore.NewCore(
    zapcore.NewConsoleEncoder(zap.NewProductionEncoderConfig()),
    writer,
    zap.InfoLevel,
  )
  Raw = zap.New(core)
  Sugar = Raw.Sugar()

  return
}