package services import ( "time" "fmt" "github.com/engigu/baihu-panel/internal/constant" "github.com/engigu/baihu-panel/internal/database" "github.com/engigu/baihu-panel/internal/eventbus" "github.com/engigu/baihu-panel/internal/logger" "github.com/engigu/baihu-panel/internal/models" "github.com/engigu/baihu-panel/internal/utils" ) type LogRetentionConfig struct { Days int `json:"days"` MaxCount int `json:"max_count"` } type AppLogService struct { settingsService *SettingsService } func NewAppLogService() *AppLogService { return &AppLogService{ settingsService: NewSettingsService(), } } func (s *AppLogService) Add(log *models.AppLog) error { if log.ID == "" { log.ID = utils.GenerateID() } return database.DB.Create(log).Error } func (s *AppLogService) List(category string, status string, level string, page, pageSize int, keyword string) ([]models.AppLog, int64, error) { var logs []models.AppLog var total int64 db := database.DB // 如果是推送日志,尝试关联查询渠道名称 if category == constant.LogCategoryPushLog { db = db.Table(models.AppLog{}.TableName() + " AS al"). Select("al.*, nw.name as channel_name"). Joins(fmt.Sprintf("LEFT JOIN %s AS nw ON al.ref_id = nw.id", models.NotifyWay{}.TableName())) } else { db = db.Model(&models.AppLog{}) } if category != "" { if category == constant.LogCategoryPushLog { db = db.Where("al.category = ?", category) } else { db = db.Where("category = ?", category) } } if status != "" { field := "status" if category == constant.LogCategoryPushLog { field = "al.status" } db = db.Where(field+" = ?", status) } if level != "" { field := "level" if category == constant.LogCategoryPushLog { field = "al.level" } db = db.Where(field+" = ?", level) } if keyword != "" { if category == constant.LogCategoryPushLog { db = db.Where("(al.title LIKE ? OR al.content LIKE ?)", "%"+keyword+"%", "%"+keyword+"%") } else { db = db.Where("(title LIKE ? OR content LIKE ?)", "%"+keyword+"%", "%"+keyword+"%") } } db.Count(&total) offset := (page - 1) * pageSize order := "created_at desc" if category == constant.LogCategoryPushLog { order = "al.created_at desc" } err := db.Order(order).Offset(offset).Limit(pageSize).Scan(&logs).Error return logs, total, err } func (s *AppLogService) MarkAsRead(id string) error { now := models.LocalTime(time.Now()) return database.DB.Model(&models.AppLog{}).Where("id = ?", id).Updates(map[string]interface{}{ "status": constant.LogStatusRead, "read_at": &now, }).Error } func (s *AppLogService) MarkAllAsRead(category string) error { now := models.LocalTime(time.Now()) return database.DB.Model(&models.AppLog{}).Where("category = ? AND status = ?", category, constant.LogStatusUnread).Updates(map[string]interface{}{ "status": constant.LogStatusRead, "read_at": &now, }).Error } func (s *AppLogService) Clear(category string) error { query := database.DB.Model(&models.AppLog{}) if category != "" { query = query.Where("category = ?", category) } return query.Delete(&models.AppLog{}).Error } func (s *AppLogService) GetRetentionConfigs() map[string]LogRetentionConfig { configs := map[string]LogRetentionConfig{ constant.LogCategorySystemNotice: { Days: utils.ToInt(s.settingsService.Get(constant.SectionSystem, constant.KeySystemNoticeDays), 30), MaxCount: utils.ToInt(s.settingsService.Get(constant.SectionSystem, constant.KeySystemNoticeMaxCount), 500), }, constant.LogCategoryPushLog: { Days: utils.ToInt(s.settingsService.Get(constant.SectionSystem, constant.KeyPushLogDays), 15), MaxCount: utils.ToInt(s.settingsService.Get(constant.SectionSystem, constant.KeyPushLogMaxCount), 5000), }, constant.LogCategoryLoginLog: { Days: utils.ToInt(s.settingsService.Get(constant.SectionSystem, constant.KeyLoginLogDays), 30), MaxCount: utils.ToInt(s.settingsService.Get(constant.SectionSystem, constant.KeyLoginLogMaxCount), 1000), }, constant.LogCategoryDefault: { Days: 30, MaxCount: 10000, }, } return configs } func (s *AppLogService) CleanUp() { configs := s.GetRetentionConfigs() categories := []string{constant.LogCategorySystemNotice, constant.LogCategoryPushLog, constant.LogCategoryLoginLog} for _, cat := range categories { cfg, ok := configs[cat] if !ok { cfg = configs[constant.LogCategoryDefault] } if cfg.Days > 0 { deadline := time.Now().AddDate(0, 0, -cfg.Days) database.DB.Where("category = ? AND created_at < ?", cat, deadline).Delete(&models.AppLog{}) } if cfg.MaxCount > 0 { var total int64 database.DB.Model(&models.AppLog{}).Where("category = ?", cat).Count(&total) if total > int64(cfg.MaxCount) { deleteCount := total - int64(cfg.MaxCount) var ids []string database.DB.Model(&models.AppLog{}).Where("category = ?", cat).Order("created_at asc").Limit(int(deleteCount)).Pluck("id", &ids) if len(ids) > 0 { database.DB.Where("id IN ?", ids).Delete(&models.AppLog{}) } } } } logger.Debugf("[AppLog] 完成应用日志清理策略") } func (s *AppLogService) SubscribeEvents(bus *eventbus.EventBus) { // 1. [订阅] 系统通知 -> 存储到数据库表现为红点消息 bus.Subscribe(constant.EventSystemNotice, func(e eventbus.Event) { payload, ok := e.Payload.(map[string]interface{}) if !ok { return } title, _ := payload["title"].(string) content, _ := payload["content"].(string) level, _ := payload["level"].(string) if level == "" { level = constant.LogLevelInfo } s.Add(&models.AppLog{ Category: constant.LogCategorySystemNotice, Title: title, Content: models.BigText(content), Level: level, Status: constant.LogStatusUnread, }) }) // 2. [订阅] 推送结果 -> 存储到数据库供推送日志查看 bus.Subscribe(constant.EventNotifySent, func(e eventbus.Event) { payload, ok := e.Payload.(map[string]interface{}) if !ok { return } title, _ := payload["title"].(string) content, _ := payload["content"].(string) success, _ := payload["success"].(bool) errorMsg, _ := payload["error_msg"].(string) channelID, _ := payload["channel_id"].(string) status := constant.LogStatusSuccess level := constant.LogLevelInfo if !success { status = constant.LogStatusFailed level = constant.LogLevelError } s.Add(&models.AppLog{ Category: constant.LogCategoryPushLog, Title: title, Content: models.BigText(content), Level: level, Status: status, RefID: channelID, ErrorMsg: models.BigText(errorMsg), }) }) // 3. 将某些业务事件转化为系统内部通知 (自动出现在小铃铛) /* bus.Subscribe(constant.EventTaskFailed, func(e eventbus.Event) { payload, ok := e.Payload.(map[string]interface{}) if !ok { return } taskName, _ := payload["task_name"].(string) errMsg, _ := payload["error"].(string) bus.Publish(eventbus.Event{ Type: constant.EventSystemNotice, Payload: map[string]interface{}{ "title": fmt.Sprintf("任务 [%s] 执行失败", taskName), "content": fmt.Sprintf("错误详情: %s", errMsg), "level": constant.LogLevelError, }, }) }) bus.Subscribe(constant.EventTaskTimeout, func(e eventbus.Event) { payload, ok := e.Payload.(map[string]interface{}) if !ok { return } taskName, _ := payload["task_name"].(string) bus.Publish(eventbus.Event{ Type: constant.EventSystemNotice, Payload: map[string]interface{}{ "title": fmt.Sprintf("任务 [%s] 执行超时", taskName), "content": "任务已经超过预设的运行时间并被系统强制中止。", "level": constant.LogLevelWarning, }, }) }) */ bus.Subscribe(constant.EventPasswordChanged, func(e eventbus.Event) { payload, ok := e.Payload.(map[string]interface{}) if !ok { return } username, _ := payload["username"].(string) bus.Publish(eventbus.Event{ Type: constant.EventSystemNotice, Payload: map[string]interface{}{ "title": "账户安全提醒", "content": fmt.Sprintf("用户 %s 的账号密码已被修改。", username), "level": constant.LogLevelInfo, }, }) }) }