Function GetUserUsageBySource

core/http/auth/usage.go:266–293

GetUserUsageBySource returns per-source aggregated usage for one user. Legacy is excluded by design (visible to admins only via the admin variant).

(db *gorm.DB, userID, period string)

// GetUserUsageBySource returns per-source aggregated usage for one user. Legacy
// is excluded by design (visible to admins only via the admin variant).
func GetUserUsageBySource(db *gorm.DB, userID, period string) ([]UsageBucket, SourceTotals, error) {
	sqlite := isSQLiteDB(db)
	since, dateFmt := periodToWindow(period, sqlite)
	bucketExpr := fmt.Sprintf("%s as bucket", dateFmt)

	query := db.Model(&UsageRecord{}).
		Select(bucketExpr+", source, COALESCE(api_key_id, '') as api_key_id, api_key_name, "+
			"SUM(prompt_tokens) as prompt_tokens, "+
			"SUM(completion_tokens) as completion_tokens, "+
			"SUM(total_tokens) as total_tokens, "+
			"COUNT(*) as request_count").
		Where("user_id = ?", userID).
		Where("source <> ?", UsageSourceLegacy).
		Group("bucket, source, api_key_id, api_key_name").
		Order("bucket ASC")

	if !since.IsZero() {
		query = query.Where("created_at >= ?", since)
	}

	var buckets []UsageBucket
	if err := query.Find(&buckets).Error; err != nil {
		return nil, SourceTotals{}, err
	}

	totals := computeSourceTotals(db, userID, "", since, false)
	return buckets, totals, nil
}

// computeSourceTotals rolls up by_source / by_key / grand_total.
// userID/apiKeyID are optional filters. includeLegacy controls whether the