feat: Initial commit

internal/logger/logger.go

@@ -0,0 +1,83 @@
+package logger
+
+import (
+	"os"
+	"path/filepath"
+	"time"
+
+	"github.com/sirupsen/logrus"
+)
+
+var Log *logrus.Logger
+
+func init() {
+	Log = logrus.New()
+
+	// 设置日志格式
+	Log.SetFormatter(&logrus.TextFormatter{
+		FullTimestamp:   true,
+		TimestampFormat: "2006-01-02 15:04:05",
+		ForceColors:     true,
+	})
+
+	// 设置日志级别
+	Log.SetLevel(logrus.InfoLevel)
+
+	// 输出到标准输出
+	Log.SetOutput(os.Stdout)
+}
+
+// SetupFileOutput 设置文件输出
+func SetupFileOutput(logDir string) error {
+	if err := os.MkdirAll(logDir, 0755); err != nil {
+		return err
+	}
+
+	logFile := filepath.Join(logDir, time.Now().Format("2006-01-02")+".log")
+	file, err := os.OpenFile(logFile, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
+	if err != nil {
+		return err
+	}
+
+	Log.SetOutput(file)
+	return nil
+}
+
+// SetLevel 设置日志级别
+func SetLevel(level string) {
+	switch level {
+	case "debug":
+		Log.SetLevel(logrus.DebugLevel)
+	case "info":
+		Log.SetLevel(logrus.InfoLevel)
+	case "warn":
+		Log.SetLevel(logrus.WarnLevel)
+	case "error":
+		Log.SetLevel(logrus.ErrorLevel)
+	default:
+		Log.SetLevel(logrus.InfoLevel)
+	}
+}
+
+// 便捷方法
+func Debug(args ...interface{}) { Log.Debug(args...) }
+func Info(args ...interface{})  { Log.Info(args...) }
+func Warn(args ...interface{})  { Log.Warn(args...) }
+func Error(args ...interface{}) { Log.Error(args...) }
+func Fatal(args ...interface{}) { Log.Fatal(args...) }
+
+func Debugf(format string, args ...interface{}) { Log.Debugf(format, args...) }
+func Infof(format string, args ...interface{})  { Log.Infof(format, args...) }
+func Warnf(format string, args ...interface{})  { Log.Warnf(format, args...) }
+func Errorf(format string, args ...interface{}) { Log.Errorf(format, args...) }
+func Fatalf(format string, args ...interface{}) { Log.Fatalf(format, args...) }
+
+// WithField 带字段的日志
+func WithField(key string, value interface{}) *logrus.Entry {
+	return Log.WithField(key, value)
+}
+
+// WithFields 带多个字段的日志
+func WithFields(fields logrus.Fields) *logrus.Entry {
+	return Log.WithFields(fields)
+}