model-gateway/dao/stat_dao.go

package dao

import (
	"context"
	"fmt"
	"time"

	"model-gateway/consts/public"
	"model-gateway/model/entity"

	"gitea.redpowerfuture.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, public.DbNameModelGateway).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.LogsModelStat, 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
}