log: add logrus log lib and add telegram notification push as an option

logger/logger.go

@@ -0,0 +1,210 @@
+package logger
+
+import (
+	"nofx/config"
+	"os"
+
+	"github.com/sirupsen/logrus"
+)
+
+var (
+	// Log 全局logger实例
+	Log *logrus.Logger
+
+	// telegramHook 保存hook引用，用于优雅关闭
+	telegramHook *TelegramHook
+)
+
+// ============================================================================
+// 初始化函数
+// ============================================================================
+
+// Init 初始化全局logger
+// 如果config为nil，使用默认配置（console输出，info级别）
+func Init(cfg *Config) error {
+	Log = logrus.New()
+
+	// 如果没有配置，使用默认值
+	if cfg == nil {
+		cfg = &Config{Level: "info"}
+	}
+
+	// 设置默认值
+	cfg.SetDefaults()
+
+	// 设置日志级别
+	level, err := logrus.ParseLevel(cfg.Level)
+	if err != nil {
+		level = logrus.InfoLevel
+	}
+	Log.SetLevel(level)
+
+	// 设置格式化器（固定使用彩色文本格式）
+	Log.SetFormatter(&logrus.TextFormatter{
+		FullTimestamp:   true,
+		TimestampFormat: "2006-01-02 15:04:05",
+		ForceColors:     true,
+	})
+
+	// 设置输出目标（默认stdout）
+	Log.SetOutput(os.Stdout)
+
+	// 启用调用位置信息
+	Log.SetReportCaller(true)
+
+	// 添加Telegram Hook（可选）
+	if cfg.Telegram != nil && cfg.Telegram.Enabled {
+		if err := setupTelegramHook(cfg.Telegram); err != nil {
+			Log.Warnf("初始化Telegram推送失败，将继续使用普通日志: %v", err)
+		}
+	}
+
+	return nil
+}
+
+// setupTelegramHook 设置Telegram Hook
+func setupTelegramHook(telegramCfg *TelegramConfig) error {
+	hook, err := NewTelegramHook(telegramCfg)
+	if err != nil {
+		return err
+	}
+
+	Log.AddHook(hook)
+	telegramHook = hook
+	Log.Info("✅ Telegram日志推送已启用")
+	return nil
+}
+
+// InitWithSimpleConfig 使用简化配置初始化logger
+// 适用于只需要基本功能的场景
+func InitWithSimpleConfig(level string) error {
+	return Init(&Config{Level: level})
+}
+
+// InitWithTelegram 使用Telegram配置初始化logger
+func InitWithTelegram(botToken string, chatID int64) error {
+	return Init(&Config{
+		Level: "info",
+		Telegram: &TelegramConfig{
+			Enabled:  true,
+			BotToken: botToken,
+			ChatID:   chatID,
+		},
+	})
+}
+
+// InitFromLogConfig 从config.LogConfig初始化logger
+func InitFromLogConfig(logConfig *config.LogConfig) error {
+	if logConfig == nil {
+		return InitWithSimpleConfig("info")
+	}
+
+	cfg := &Config{
+		Level: logConfig.Level,
+	}
+
+	if cfg.Level == "" {
+		cfg.Level = "info"
+	}
+
+	// 如果启用了Telegram，添加配置
+	if logConfig.Telegram != nil && logConfig.Telegram.Enabled {
+		if botToken := logConfig.Telegram.BotToken; botToken != "" && logConfig.Telegram.ChatID != 0 {
+			cfg.Telegram = &TelegramConfig{
+				Enabled:  true,
+				BotToken: botToken,
+				ChatID:   logConfig.Telegram.ChatID,
+				MinLevel: logConfig.Telegram.MinLevel,
+			}
+		}
+	}
+
+	return Init(cfg)
+}
+
+// InitFromParams 从参数初始化logger
+// 适用于不依赖config包的场景
+func InitFromParams(level string, telegramEnabled bool, botToken string, chatID int64) error {
+	cfg := &Config{Level: level}
+
+	if telegramEnabled && botToken != "" && chatID != 0 {
+		cfg.Telegram = &TelegramConfig{
+			Enabled:  true,
+			BotToken: botToken,
+			ChatID:   chatID,
+		}
+	}
+
+	return Init(cfg)
+}
+
+// Shutdown 优雅关闭logger（主要用于关闭Telegram发送器）
+func Shutdown() {
+	if telegramHook != nil {
+		telegramHook.Stop()
+		telegramHook = nil
+	}
+}
+
+// ============================================================================
+// 日志记录函数
+// ============================================================================
+
+// WithFields 创建带字段的logger entry
+func WithFields(fields logrus.Fields) *logrus.Entry {
+	return Log.WithFields(fields)
+}
+
+// WithField 创建带单个字段的logger entry
+func WithField(key string, value interface{}) *logrus.Entry {
+	return Log.WithField(key, value)
+}
+
+// add debug, info, warn
+func Debug(args ...interface{}) {
+	Log.Debug(args...)
+}
+
+func Info(args ...interface{}) {
+	Log.Info(args...)
+}
+
+func Warn(args ...interface{}) {
+	Log.Warn(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 Error(args ...interface{}) {
+	Log.Error(args...)
+}
+
+func Errorf(format string, args ...interface{}) {
+	Log.Errorf(format, args...)
+}
+
+func Fatal(args ...interface{}) {
+	Log.Fatal(args...)
+}
+
+func Fatalf(format string, args ...interface{}) {
+	Log.Fatalf(format, args...)
+}
+
+func Panic(args ...interface{}) {
+	Log.Panic(args...)
+}
+
+func Panicf(format string, args ...interface{}) {
+	Log.Panicf(format, args...)
+}