package dao

import (
	"context"
	"time"

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

	"gitea.redpowerfuture.com/red-future/common/db/gfdb"
	"github.com/gogf/gf/v2/os/gtime"
	"github.com/gogf/gf/v2/util/gconv"
)

var ModelGatewayLogsStat = &modelGatewayLogsStatDao{}

type modelGatewayLogsStatDao struct{}

// IncRequestCount 原子累加：按天+租户+创建人+模型 +1
func (d *modelGatewayLogsStatDao) IncRequestCount(ctx context.Context, day time.Time, tenantId uint64, creator, modelName string) error {
	_, err := gfdb.DB(ctx, public.DbNameModelGateway).Model(ctx, public.TableNameStat).
		Data(&entity.ModelGatewayLogsStat{
			Day:          gtime.New(day),
			TenantId:     tenantId,
			Creator:      creator,
			ModelName:    modelName,
			RequestCount: 1,
		}).
		OnDuplicate("request_count", "request_count+1").
		Insert()
	return err
}

// List 分页查询统计
func (d *modelGatewayLogsStatDao) List(ctx context.Context, pageNum, pageSize int, req *entity.ModelGatewayLogsStat) (list []*entity.ModelGatewayLogsStat, total int64, err error) {
	model := gfdb.DB(ctx, public.DbNameModelGateway).Model(ctx, public.TableNameStat).
		OmitEmpty().
		Where(entity.ModelGatewayLogsStatCols.Creator, req.Creator).
		WhereLike(entity.ModelGatewayLogsStatCols.ModelName, "%"+req.ModelName+"%").
		OrderDesc(entity.ModelGatewayLogsStatCols.Day).
		OrderDesc(entity.ModelGatewayLogsStatCols.RequestCount)
	if pageNum > 0 && pageSize > 0 {
		model = model.Page(pageNum, pageSize)
	}
	r, totalInt, err := model.AllAndCount(false)
	if err != nil {
		return nil, 0, err
	}
	total = gconv.Int64(totalInt)
	err = r.Structs(&list)
	return
}