geekai/api/handler/admin/dashboard_handler.go

package admin

import (
	"chatplus/core"
	"chatplus/handler"
	"chatplus/store/model"
	"chatplus/utils/resp"
	"github.com/gin-gonic/gin"
	"gorm.io/gorm"
	"time"
)

type DashboardHandler struct {
	handler.BaseHandler
	db *gorm.DB
}

func NewDashboardHandler(app *core.AppServer, db *gorm.DB) *DashboardHandler {
	h := DashboardHandler{db: db}
	h.App = app
	return &h
}

type statsVo struct {
	Users  int64 `json:"users"`
	Chats  int64 `json:"chats"`
	Tokens int64 `json:"tokens"`
}

func (h *DashboardHandler) Stats(c *gin.Context) {
	stats := statsVo{}
	// new users statistic
	var userCount int64
	now := time.Now()
	zeroTime := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location())
	res := h.db.Model(&model.User{}).Where("created_at > ?", zeroTime).Count(&userCount)
	if res.Error == nil {
		stats.Users = userCount
	}

	// new chats statistic
	var chatCount int64
	res = h.db.Model(&model.ChatItem{}).Where("created_at > ?", zeroTime).Count(&chatCount)
	if res.Error == nil {
		stats.Chats = chatCount
	}

	// tokens took stats
	var tokenCount int64
	res = h.db.Model(&model.HistoryMessage{}).Select("sum(tokens) as tokens_total").Where("created_at > ?", zeroTime).Scan(&tokenCount)
	if res.Error == nil {
		stats.Tokens = tokenCount
	}
	resp.SUCCESS(c, stats)
}