Function computeSourceTotals

core/http/auth/usage.go:298–387

computeSourceTotals rolls up by_source / by_key / grand_total. userID/apiKeyID are optional filters. includeLegacy controls whether the legacy bucket is exposed (admin-only).

(db *gorm.DB, userID, apiKeyID string, since time.Time, includeLegacy bool)

// userID/apiKeyID are optional filters. includeLegacy controls whether the
// legacy bucket is exposed (admin-only).
func computeSourceTotals(db *gorm.DB, userID, apiKeyID string, since time.Time, includeLegacy bool) SourceTotals {
	totals := SourceTotals{BySource: map[string]TotalsEntry{}}

	bySourceQ := db.Model(&UsageRecord{}).
		Select("source, SUM(total_tokens) as tokens, COUNT(*) as requests").
		Group("source")
	bySourceQ = applyFilters(bySourceQ, userID, apiKeyID, since, includeLegacy)

	var bySourceRows []struct {
		Source   string
		Tokens   int64
		Requests int64
	}
	if err := bySourceQ.Scan(&bySourceRows).Error; err != nil {
		xlog.Warn("computeSourceTotals: by-source Scan failed", "error", err)
		return totals
	}
	for _, r := range bySourceRows {
		totals.BySource[r.Source] = TotalsEntry{Tokens: r.Tokens, Requests: r.Requests}
		totals.GrandTotal.Tokens += r.Tokens
		totals.GrandTotal.Requests += r.Requests
	}

	byKeyQ := db.Model(&UsageRecord{}).
		Select("COALESCE(api_key_id, '') as api_key_id, api_key_name, "+
			"user_id, user_name, "+
			"SUM(total_tokens) as tokens, COUNT(*) as requests, MAX(created_at) as last_used").
		Where("api_key_id IS NOT NULL AND api_key_id <> ''").
		Group("api_key_id, api_key_name, user_id, user_name").
		Order("tokens DESC").
		Limit(maxKeyTotals)
	byKeyQ = applyFilters(byKeyQ, userID, apiKeyID, since, includeLegacy)

	// Iterate Rows() manually because MAX(created_at) is returned as a string by
	// the SQLite driver, and Go's database/sql refuses to scan that into
	// *time.Time. Postgres returns a proper timestamp. We accept both shapes
	// via a Rows.Scan into a string column, then parse uniformly.
	rows, err := byKeyQ.Rows()
	if err != nil {
		xlog.Warn("computeSourceTotals: by-key Rows() failed", "error", err)
	} else {
		defer func() { _ = rows.Close() }()
		out := make([]KeyTotal, 0)
		for rows.Next() {
			var (
				apiKeyID, apiKeyName, userIDCol, userName, lastUsedRaw string
				tokens, requests                                       int64
			)
			if scanErr := rows.Scan(&apiKeyID, &apiKeyName, &userIDCol, &userName, &tokens, &requests, &lastUsedRaw); scanErr != nil {
				continue
			}
			out = append(out, KeyTotal{
				APIKeyID:   apiKeyID,
				APIKeyName: apiKeyName,
				UserID:     userIDCol,
				UserName:   userName,
				Tokens:     tokens,
				Requests:   requests,