WangLiZhao
4e6b98b7d3
- 新增统计控制器、服务层与数据访问层，提供按天统计接口 - 在 worker 处理任务时原子累加请求计数（仅实际调用模型时计数） - 更新数据库表结构，添加 asynch_model_stat 表及索引 - 更新文档说明统计功能的使用方式与统计口径
62 lines
1.7 KiB
Go
62 lines
1.7 KiB
Go
package dao
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"model-asynch/consts/public"
|
|
"model-asynch/model/entity"
|
|
|
|
"gitea.com/red-future/common/db/gfdb"
|
|
"github.com/gogf/gf/v2/os/gtime"
|
|
)
|
|
|
|
type statDao struct{}
|
|
|
|
var Stat = &statDao{}
|
|
|
|
// IncRequestCount 原子累加（支持分布式/多协程）：按天+租户+创建人+模型 +1
|
|
func (d *statDao) IncRequestCount(ctx context.Context, day time.Time, tenantId int64, creator, modelName string) error {
|
|
sql := fmt.Sprintf(`
|
|
INSERT INTO %s(day, tenant_id, creator, model_name, request_count, created_at, updated_at)
|
|
VALUES(?, ?, ?, ?, 1, NOW(), NOW())
|
|
ON CONFLICT (day, tenant_id, creator, model_name)
|
|
DO UPDATE SET request_count = %s.request_count + 1, updated_at = NOW()`,
|
|
public.TableNameStat, public.TableNameStat,
|
|
)
|
|
_, err := gfdb.DB(ctx).Exec(ctx, sql, gtime.New(day).Format("Y-m-d"), tenantId, creator, modelName)
|
|
return err
|
|
}
|
|
|
|
func (d *statDao) List(ctx context.Context, pageNum, pageSize int, startDay, endDay string, tenantId *int64, creator, modelName string) (list []*entity.AsynchModelStat, total int64, err error) {
|
|
m := gfdb.DB(ctx).Model(ctx, public.TableNameStat).Where("1=1")
|
|
if startDay != "" {
|
|
m = m.Where("day >= ?", startDay)
|
|
}
|
|
if endDay != "" {
|
|
m = m.Where("day <= ?", endDay)
|
|
}
|
|
if tenantId != nil {
|
|
m = m.Where("tenant_id = ?", *tenantId)
|
|
}
|
|
if creator != "" {
|
|
m = m.WhereLike("creator", "%"+creator+"%")
|
|
}
|
|
if modelName != "" {
|
|
m = m.WhereLike("model_name", "%"+modelName+"%")
|
|
}
|
|
m = m.OrderDesc("day").OrderDesc("request_count")
|
|
if pageNum > 0 && pageSize > 0 {
|
|
m = m.Page(pageNum, pageSize)
|
|
}
|
|
r, totalInt, err := m.AllAndCount(false)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
total = int64(totalInt)
|
|
err = r.Structs(&list)
|
|
return
|
|
}
|
|
|