Function GetMonthlyUsageStatistics

model/usage_statistics.go:68–152

GetMonthlyUsageStatistics 获取月度用量统计数据

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

// GetMonthlyUsageStatistics 获取月度用量统计数据
func GetMonthlyUsageStatistics(startDate, endDate string, tokenId int, modelName string, page, pageSize int) ([]*UsageStatistics, int64, error) {
	var statistics []*UsageStatistics
	var total int64

	// 使用原生SQL查询实现按月分组统计
	db := DB.Model(&UsageStatistics{})

	// 构建查询条件
	conditions := ""
	params := []interface{}{}

	// 添加日期范围条件（按月查询）
	if startDate != "" {
		conditions += " AND date >= ?"
		params = append(params, startDate+"-01")
	}
	if endDate != "" {
		// 获取 endDate 所在月份的最后一天
		if len(endDate) >= 7 {
			year := endDate[0:4]
			month := endDate[5:7]
			conditions += " AND date <= ?"
			params = append(params, year+"-"+month+"-31")
		}
	}
	
	if tokenId > 0 {
		conditions += " AND token_id = ?"
		params = append(params, tokenId)
	}
	if modelName != "" {
		conditions += " AND model_name LIKE ?"
		params = append(params, "%"+modelName+"%")
	}

	// 构建完整的SQL查询
	sql := `
		SELECT 
			MAX(id) as id,
			SUBSTR(date, 1, 7) as date,
			token_id,
			token_name,
			model_name,
			SUM(total_requests) as total_requests,
			SUM(successful_requests) as successful_requests,
			SUM(failed_requests) as failed_requests,
			SUM(total_tokens) as total_tokens,
			SUM(prompt_tokens) as prompt_tokens,
			SUM(completion_tokens) as completion_tokens,
			SUM(total_quota) as total_quota,
			MAX(created_time) as created_time,
			MAX(updated_time) as updated_time
		FROM usage_statistics 
		WHERE 1=1` + conditions + `
		GROUP BY SUBSTR(date, 1, 7), token_id, token_name, model_name
		ORDER BY date DESC, token_id ASC, model_name ASC
	`