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

logger/telegram_hook.go

@@ -0,0 +1,158 @@
+package logger
+
+import (
+	"fmt"
+	"runtime"
+	"strings"
+
+	"github.com/sirupsen/logrus"
+)
+
+// TelegramHook 实现logrus.Hook接口，将日志推送到Telegram
+type TelegramHook struct {
+	sender  *TelegramSender
+	levels  []logrus.Level
+	enabled bool
+}
+
+// NewTelegramHook 创建Telegram Hook
+func NewTelegramHook(config *TelegramConfig) (*TelegramHook, error) {
+	if !config.Enabled {
+		return &TelegramHook{enabled: false}, nil
+	}
+
+	if config.BotToken == "" || config.ChatID == 0 {
+		return nil, fmt.Errorf("telegram配置不完整: bot_token和chat_id不能为空")
+	}
+
+	// 创建发送器（使用默认参数）
+	sender, err := NewTelegramSender(config.BotToken, config.ChatID)
+	if err != nil {
+		return nil, fmt.Errorf("创建telegram发送器失败: %w", err)
+	}
+
+	hook := &TelegramHook{
+		sender:  sender,
+		levels:  config.GetLogrusLevels(),
+		enabled: true,
+	}
+
+	return hook, nil
+}
+
+// Levels 返回需要触发的日志级别
+func (h *TelegramHook) Levels() []logrus.Level {
+	if !h.enabled {
+		return []logrus.Level{}
+	}
+	return h.levels
+}
+
+// Fire 当日志触发时调用
+func (h *TelegramHook) Fire(entry *logrus.Entry) error {
+	if !h.enabled {
+		return nil
+	}
+
+	// 格式化消息
+	message := h.formatMessage(entry)
+
+	// 异步发送（非阻塞）
+	h.sender.SendAsync(message)
+
+	return nil
+}
+
+// formatMessage 格式化日志消息为Telegram格式
+func (h *TelegramHook) formatMessage(entry *logrus.Entry) string {
+	// 级别emoji
+	levelEmoji := h.getLevelEmoji(entry.Level)
+
+	// 基本信息
+	var builder strings.Builder
+	builder.WriteString(fmt.Sprintf("%s *%s*: 系统日志警报\n", levelEmoji, strings.ToUpper(entry.Level.String())))
+	builder.WriteString(fmt.Sprintf("📝 消息: `%s`\n", escapeMarkdown(entry.Message)))
+
+	// 字段信息
+	if len(entry.Data) > 0 {
+		builder.WriteString("📊 字段:\n")
+		for key, value := range entry.Data {
+			builder.WriteString(fmt.Sprintf("  • %s: `%v`\n", key, value))
+		}
+	}
+
+	// 调用位置
+	if entry.HasCaller() {
+		file := entry.Caller.File
+		// 只保留相对路径
+		if idx := strings.Index(file, "nofx/"); idx >= 0 {
+			file = file[idx:]
+		}
+		builder.WriteString(fmt.Sprintf("📍 位置: `%s:%d`\n", file, entry.Caller.Line))
+	} else {
+		// 如果entry没有caller，手动获取
+		if _, file, line, ok := runtime.Caller(8); ok {
+			if idx := strings.Index(file, "nofx/"); idx >= 0 {
+				file = file[idx:]
+			}
+			builder.WriteString(fmt.Sprintf("📍 位置: `%s:%d`\n", file, line))
+		}
+	}
+
+	// 时间戳
+	builder.WriteString(fmt.Sprintf("🕐 时间: `%s`", entry.Time.Format("2006-01-02 15:04:05")))
+
+	return builder.String()
+}
+
+// getLevelEmoji 获取日志级别对应的emoji
+func (h *TelegramHook) getLevelEmoji(level logrus.Level) string {
+	switch level {
+	case logrus.PanicLevel:
+		return "🔴"
+	case logrus.FatalLevel:
+		return "🔴"
+	case logrus.ErrorLevel:
+		return "🟠"
+	case logrus.WarnLevel:
+		return "🟡"
+	case logrus.InfoLevel:
+		return "🟢"
+	case logrus.DebugLevel:
+		return "🔵"
+	default:
+		return "⚪"
+	}
+}
+
+// escapeMarkdown 转义Markdown特殊字符
+func escapeMarkdown(text string) string {
+	replacer := strings.NewReplacer(
+		"_", "\\_",
+		"*", "\\*",
+		"[", "\\[",
+		"]", "\\]",
+		"(", "\\(",
+		")", "\\)",
+		"~", "\\~",
+		"`", "\\`",
+		">", "\\>",
+		"#", "\\#",
+		"+", "\\+",
+		"-", "\\-",
+		"=", "\\=",
+		"|", "\\|",
+		"{", "\\{",
+		"}", "\\}",
+		".", "\\.",
+		"!", "\\!",
+	)
+	return replacer.Replace(text)
+}
+
+// Stop 停止Hook（优雅关闭）
+func (h *TelegramHook) Stop() {
+	if h.enabled && h.sender != nil {
+		h.sender.Stop()
+	}
+}