model-gateway/service/stat/stat_service.go

package stat

import (
	"context"

	"model-gateway/dao"
	"model-gateway/model/dto"
)

type statService struct{}

var Stat = &statService{}

func (s *statService) List(ctx context.Context, req *dto.ListModelStatReq) (res *dto.ListModelStatRes, err error) {
	pageNum, pageSize := 1, 10
	if req != nil {
		if req.PageNum > 0 {
			pageNum = req.PageNum
		}
		if req.PageSize > 0 {
			pageSize = req.PageSize
		}
	}
	startDay, endDay := "", ""
	var tenantID *int64
	creator, modelName := "", ""
	if req != nil {
		startDay = req.StartDay
		endDay = req.EndDay
		tenantID = req.TenantID
		creator = req.Creator
		modelName = req.ModelName
	}
	list, total, err := dao.Stat.List(ctx, pageNum, pageSize, startDay, endDay, tenantID, creator, modelName)
	if err != nil {
		return nil, err
	}
	return &dto.ListModelStatRes{List: list, Total: total}, nil
}