mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-09-18 01:06:39 +08:00
121 lines
3.7 KiB
Go
121 lines
3.7 KiB
Go
package admin
|
|
|
|
import (
|
|
"chatplus/core"
|
|
"chatplus/core/types"
|
|
"chatplus/handler"
|
|
"chatplus/store/model"
|
|
"chatplus/utils/resp"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/shopspring/decimal"
|
|
"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 int `json:"tokens"`
|
|
Income float64 `json:"income"`
|
|
Chart map[string]map[string]float64 `json:"chart"`
|
|
}
|
|
|
|
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 historyMessages []model.ChatMessage
|
|
res = h.db.Where("created_at > ?", zeroTime).Find(&historyMessages)
|
|
for _, item := range historyMessages {
|
|
stats.Tokens += item.Tokens
|
|
}
|
|
|
|
// 众筹收入
|
|
var rewards []model.Reward
|
|
res = h.db.Where("created_at > ?", zeroTime).Find(&rewards)
|
|
for _, item := range rewards {
|
|
stats.Income += item.Amount
|
|
}
|
|
|
|
// 订单收入
|
|
var orders []model.Order
|
|
res = h.db.Where("status = ?", types.OrderPaidSuccess).Where("created_at > ?", zeroTime).Find(&orders)
|
|
for _, item := range orders {
|
|
stats.Income += item.Amount
|
|
}
|
|
|
|
// 统计7天的订单的图表
|
|
startDate := now.Add(-7 * 24 * time.Hour).Format("2006-01-02")
|
|
var statsChart = make(map[string]map[string]float64)
|
|
//// 初始化
|
|
var userStatistic, historyMessagesStatistic, incomeStatistic = make(map[string]float64), make(map[string]float64), make(map[string]float64)
|
|
for i := 0; i < 7; i++ {
|
|
var initTime = time.Date(now.Year(), now.Month(), now.Day()-i, 0, 0, 0, 0, now.Location()).Format("2006-01-02")
|
|
userStatistic[initTime] = float64(0)
|
|
historyMessagesStatistic[initTime] = float64(0)
|
|
incomeStatistic[initTime] = float64(0)
|
|
}
|
|
|
|
// 统计用户7天增加的曲线
|
|
var users []model.User
|
|
res = h.db.Model(&model.User{}).Where("created_at > ?", startDate).Find(&users)
|
|
if res.Error == nil {
|
|
for _, item := range users {
|
|
userStatistic[item.CreatedAt.Format("2006-01-02")] += 1
|
|
}
|
|
}
|
|
|
|
// 统计7天Token 消耗
|
|
res = h.db.Where("created_at > ?", startDate).Find(&historyMessages)
|
|
for _, item := range historyMessages {
|
|
historyMessagesStatistic[item.CreatedAt.Format("2006-01-02")] += float64(item.Tokens)
|
|
}
|
|
|
|
// 浮点数相加？
|
|
// 统计最近7天的众筹
|
|
res = h.db.Where("created_at > ?", startDate).Find(&rewards)
|
|
for _, item := range rewards {
|
|
incomeStatistic[item.CreatedAt.Format("2006-01-02")], _ = decimal.NewFromFloat(incomeStatistic[item.CreatedAt.Format("2006-01-02")]).Add(decimal.NewFromFloat(item.Amount)).Float64()
|
|
}
|
|
|
|
// 统计最近7天的订单
|
|
res = h.db.Where("status = ?", types.OrderPaidSuccess).Where("created_at > ?", startDate).Find(&orders)
|
|
for _, item := range orders {
|
|
incomeStatistic[item.CreatedAt.Format("2006-01-02")], _ = decimal.NewFromFloat(incomeStatistic[item.CreatedAt.Format("2006-01-02")]).Add(decimal.NewFromFloat(item.Amount)).Float64()
|
|
}
|
|
|
|
statsChart["users"] = userStatistic
|
|
statsChart["historyMessage"] = historyMessagesStatistic
|
|
statsChart["orders"] = incomeStatistic
|
|
|
|
stats.Chart = statsChart
|
|
|
|
resp.SUCCESS(c, stats)
|
|
}
|