Function GetUserUsageStatistics

model/usage_statistics.go:404–440

GetUserUsageStatistics 获取特定用户的用量统计数据

(userId int, startDate, endDate string, tokenId int, modelName string, page, pageSize int)

// GetUserUsageStatistics 获取特定用户的用量统计数据
func GetUserUsageStatistics(userId int, startDate, endDate string, tokenId int, modelName string, page, pageSize int) ([]*UsageStatistics, int64, error) {
	var statistics []*UsageStatistics
	var total int64

	// 基本查询，需要通过token表连接过滤用户
	query := DB.Table("usage_statistics").
		Joins("JOIN tokens ON usage_statistics.token_id = tokens.id").
		Where("tokens.user_id = ?", userId)

	// 添加查询条件
	if startDate != "" {
		query = query.Where("usage_statistics.date >= ?", startDate)
	}
	if endDate != "" {
		query = query.Where("usage_statistics.date <= ?", endDate)
	}
	if tokenId > 0 {
		query = query.Where("usage_statistics.token_id = ?", tokenId)
	}
	if modelName != "" {
		query = query.Where("usage_statistics.model_name LIKE ?", "%"+modelName+"%")
	}

	// 获取总数
	err := query.Count(&total).Error
	if err != nil {
		return nil, 0, err
	}

	// 分页查询
	offset := (page - 1) * pageSize
	err = query.Select("usage_statistics.*").
		Order("usage_statistics.date DESC, usage_statistics.token_id ASC, usage_statistics.model_name ASC").
		Offset(offset).Limit(pageSize).Find(&statistics).Error

	return statistics, total, err
}

// GetUserMonthlyUsageStatistics 获取特定用户的月度用量统计数据
func GetUserMonthlyUsageStatistics(userId int, startDate, endDate string, tokenId int, modelName string, page, pageSize int) ([]*UsageStatistics, int64, error) {