goodBaby/api/dashboard/dashboard.go

// Package dashboard 汇总首页需要的统计数据
package dashboard

import (
	"time"

	"github.com/gin-gonic/gin"
	"github.com/ssdomei232/goodBaby/api/response"
	"github.com/ssdomei232/goodBaby/api/user"
	"github.com/ssdomei232/goodBaby/handler/db"
	"github.com/ssdomei232/goodBaby/model"
)

// HandleGetOverview 返回首页概览：各类对象数量、最紧急的 Timer、最近的执行日志
func HandleGetOverview(c *gin.Context) {
	userInfo, err := user.GetUserInfoByGinCtx(c)
	if err != nil {
		response.Unauthorized(c, "获取用户信息失败")
		return
	}

	gormDB, err := db.GetGormDB()
	if err != nil {
		response.ServerError(c, "服务器内部错误")
		return
	}

	timers := []model.Timer{}
	if err := gormDB.Where("uid = ?", userInfo.ID).Find(&timers).Error; err != nil {
		response.ServerError(c, "获取统计数据失败")
		return
	}

	var ruleCount, gatewayCount, gatewayRuleCount, accountCount int64
	gormDB.Model(&model.Rule{}).Where("uid = ?", userInfo.ID).Count(&ruleCount)
	gormDB.Model(&model.MessageGateway{}).Where("uid = ?", userInfo.ID).Count(&gatewayCount)
	gormDB.Model(&model.GatewayRule{}).Where("uid = ?", userInfo.ID).Count(&gatewayRuleCount)
	gormDB.Model(&model.Account{}).Where("uid = ?", userInfo.ID).Count(&accountCount)

	recentLogs := []model.ExecutionLog{}
	gormDB.Where("uid = ?", userInfo.ID).Order("id DESC").Limit(10).Find(&recentLogs)

	now := time.Now().Unix()
	var (
		enabledCount   int
		triggeredCount int
		urgent         *model.Timer
	)

	for i := range timers {
		timer := timers[i]
		if !timer.Enabled {
			continue
		}
		enabledCount++
		if timer.Triggered {
			triggeredCount++
			continue
		}
		// 剩余时间最短的启用中 Timer
		if urgent == nil || timer.NextSignTime() < urgent.NextSignTime() {
			urgent = &timers[i]
		}
	}

	overview := gin.H{
		"timer_count":        len(timers),
		"enabled_timers":     enabledCount,
		"triggered_count":    triggeredCount,
		"rule_count":         ruleCount,
		"gateway_count":      gatewayCount,
		"gateway_rule_count": gatewayRuleCount,
		"account_count":      accountCount,
		"server_time":        now,
		"recent_logs":        recentLogs,
	}

	if urgent != nil {
		overview["urgent_timer"] = urgent
		overview["urgent_seconds_left"] = urgent.NextSignTime() - now
	}

	response.OK(c, overview)
}