⚡ update qq/gitee/github oauth2 login

middleware/logger_date.go

@@ -0,0 +1,157 @@
+package middleware
+
+import (
+	"bytes"
+	"fmt"
+	"io"
+	"os"
+	"path/filepath"
+	"schisandra-cloud-album/global"
+	"strings"
+	"time"
+
+	"github.com/sirupsen/logrus"
+)
+
+var (
+	ccRed    = 1
+	ccYellow = 3
+	ccBlue   = 4
+	ccCyan   = 6
+	ccGray   = 7
+)
+
+type LogConsoleFormatter struct{}
+
+func (s *LogConsoleFormatter) Format(entry *logrus.Entry) ([]byte, error) {
+	timestamp := time.Now().Local().Format("2006-01-02 15:04:05")
+	var color int
+	switch entry.Level {
+
+	case logrus.ErrorLevel:
+		color = ccRed
+	case logrus.WarnLevel:
+		color = ccYellow
+	case logrus.InfoLevel:
+		color = ccBlue
+	case logrus.DebugLevel:
+		color = ccCyan
+	default:
+		color = ccGray
+	}
+	// 设置 buffer 缓冲区
+	var b *bytes.Buffer
+	if entry.Buffer == nil {
+		b = &bytes.Buffer{}
+	} else {
+		b = entry.Buffer
+	}
+	fileVal := fmt.Sprintf("%s:%d", filepath.Base(entry.Caller.File), entry.Caller.Line)
+	fmt.Fprintf(b, "[%s] \033[3%dm[%s]\033[0m [%s] %s %s\n", global.CONFIG.Logger.Prefix, color, entry.Level, timestamp, fileVal, entry.Message)
+	return b.Bytes(), nil
+}
+
+type LogFileFormatter struct{}
+
+func (s *LogFileFormatter) Format(entry *logrus.Entry) ([]byte, error) {
+	timestamp := time.Now().Local().Format("2006-01-02 15:04:05")
+	var file string
+	var length int
+	if entry.Caller != nil {
+		file = filepath.Base(entry.Caller.File)
+		length = entry.Caller.Line
+	}
+	msg := fmt.Sprintf("[%s] %s [%s:%d] %s\n", strings.ToUpper(entry.Level.String()), timestamp, file, length, entry.Message)
+	return []byte(msg), nil
+}
+
+type ConsoleHook struct {
+	formatter logrus.Formatter
+}
+
+func (hook *ConsoleHook) Levels() []logrus.Level {
+	return logrus.AllLevels
+}
+func (hook *ConsoleHook) Fire(entry *logrus.Entry) error {
+	originalFormatter := entry.Logger.Formatter
+	entry.Logger.Formatter = hook.formatter
+	defer func() { entry.Logger.Formatter = originalFormatter }()
+	line, err := entry.Logger.Formatter.Format(entry)
+	if err != nil {
+		return err
+	}
+	_, err = os.Stdout.Write(line)
+	return err
+}
+
+type FileHook struct {
+	formatter logrus.Formatter
+	file      *os.File
+}
+
+func (hook *FileHook) Levels() []logrus.Level {
+	return logrus.AllLevels
+}
+
+func (hook *FileHook) Fire(entry *logrus.Entry) error {
+	originalFormatter := entry.Logger.Formatter
+	entry.Logger.Formatter = hook.formatter
+	defer func() { entry.Logger.Formatter = originalFormatter }()
+	line, err := entry.Logger.Formatter.Format(entry)
+	if err != nil {
+		return err
+	}
+	_, err = hook.file.Write(line)
+	return err
+}
+
+type DateLogConfig struct {
+	Date string
+	Path string
+	Name string
+}
+
+func NewDateLog(d *DateLogConfig) *DateLogConfig {
+	return &DateLogConfig{
+		Date: d.Date,
+		Path: d.Path,
+		Name: d.Name,
+	}
+}
+
+func (d *DateLogConfig) Init() *logrus.Logger {
+	// 实例化 logrus
+	log := logrus.New()
+	// 设置是否输出文件名和行号信息
+	log.SetReportCaller(global.CONFIG.Logger.ShowLine)
+	// 将 logrus 的默认输出丢弃，确保日志只通过 hooks 输出
+	log.SetOutput(io.Discard)
+
+	// 控制台输出的 hook
+	consoleHook := &ConsoleHook{
+		formatter: &LogConsoleFormatter{},
+	}
+	// 添加控制台输出的 hook
+	log.AddHook(consoleHook)
+
+	// 文件路径
+	filename := fmt.Sprintf("%s/%s/%s.log", d.Path, d.Date, d.Name)
+	// 创建目录
+	if err := os.MkdirAll(fmt.Sprintf("%s/%s", d.Path, d.Date), os.ModePerm); err != nil {
+		log.Fatal(err)
+	}
+	// 打开文件，如果文件不存在，则创建文件
+	file, err := os.OpenFile(filename, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0600)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	// 文件输出的 hook
+	fileHook := &FileHook{
+		formatter: &LogFileFormatter{},
+		file:      file,
+	}
+	// 添加文件输出的 hook
+	log.AddHook(fileHook)
+	return log
+}