Function GetTokenStatistics

model/statistics_charts.go:163–265

GetTokenStatistics 获取按令牌统计的数据

(startTimestamp, endTimestamp int, username, tokenName, modelName string, channel int, group, defaultTime string)

// GetTokenStatistics 获取按令牌统计的数据
func GetTokenStatistics(startTimestamp, endTimestamp int, username, tokenName, modelName string, channel int, group, defaultTime string) ([]TokenStatistics, error) {
	var statistics []TokenStatistics

	// 构建查询条件
	tx := LOG_DB.Table("logs").Select(`
		created_at,
		token_name,
		COUNT(*) as count,
		SUM(quota) as quota
	`).Where("type = ?", 2) // LogTypeConsume = 2

	if username != "" {
		tx = tx.Where("username = ?", username)
	}
	if tokenName != "" {
		tx = tx.Where("token_name = ?", tokenName)
	}
	if modelName != "" {
		tx = tx.Where("model_name LIKE ?", "%"+modelName+"%")
	}
	if channel != 0 {
		tx = tx.Where("channel_id = ?", channel)
	}
	if group != "" {
		tx = tx.Where("`group` = ?", group)
	}
	if startTimestamp != 0 {
		tx = tx.Where("created_at >= ?", startTimestamp)
	}
	if endTimestamp != 0 {
		tx = tx.Where("created_at <= ?", endTimestamp)
	}

	// 按时间粒度分组 - 使用数据库无关的函数
	var groupBy string
	switch defaultTime {
	case "hour":
		if common.UsingMySQL {
			groupBy = "token_name, DATE_FORMAT(FROM_UNIXTIME(created_at), '%Y-%m-%d %H:00:00')"
		} else if common.UsingPostgreSQL {
			groupBy = "token_name, TO_CHAR(TO_TIMESTAMP(created_at), 'YYYY-MM-DD HH24:00:00')"
		} else {
			// SQLite
			groupBy = "token_name, strftime('%Y-%m-%d %H:00:00', datetime(created_at, 'unixepoch'))"
		}
	case "day":
		if common.UsingMySQL {
			groupBy = "token_name, DATE_FORMAT(FROM_UNIXTIME(created_at), '%Y-%m-%d')"
		} else if common.UsingPostgreSQL {
			groupBy = "token_name, TO_CHAR(TO_TIMESTAMP(created_at), 'YYYY-MM-DD')"
		} else {
			// SQLite
			groupBy = "token_name, strftime('%Y-%m-%d', datetime(created_at, 'unixepoch'))"
		}
	case "week":
		if common.UsingMySQL {
			groupBy = "token_name, DATE_FORMAT(FROM_UNIXTIME(created_at), '%Y-%U')"
		} else if common.UsingPostgreSQL {