Function getFreebuffLiveStats

freebuff/web/src/server/live-stats.ts:56–116

(
  now?: Date,
  options: { cache?: boolean } = {},
)

}

export async function getFreebuffLiveStats(
  now?: Date,
  options: { cache?: boolean } = {},
): Promise<FreebuffLiveStats> {
  const useCache = options.cache ?? now === undefined
  const requestTime = now ?? new Date()

  if (useCache && cachedLiveStats && cachedLiveStats.expiresAt > Date.now()) {
    return cachedLiveStats.stats
  }

  const [countryRows, modelRows] = await Promise.all([
    db
      .select({
        countryCode: schema.freeSession.country_code,
        count: count(),
      })
      .from(schema.freeSession)
      .where(liveSessionWhere(requestTime))
      .groupBy(schema.freeSession.country_code),
    db
      .select({
        modelId: schema.freeSession.model,
        count: count(),
      })
      .from(schema.freeSession)
      .where(liveSessionWhere(requestTime))
      .groupBy(schema.freeSession.model),
  ])

  const countries = sortCounts(
    countryRows.map((row) => ({
      countryCode: row.countryCode ?? 'UNKNOWN',
      count: Number(row.count),
    })),
  )

  const models = sortCounts(
    modelRows.map((row) => ({
      modelId: row.modelId,
      displayName: modelDisplayName(row.modelId),
      count: Number(row.count),
    })),
  )

  const stats = {
    totalLiveUsers: models.reduce((sum, row) => sum + row.count, 0),
    countries,
    models,
    generatedAt: requestTime.toISOString(),
  }

  if (useCache) {
    cachedLiveStats = {
      expiresAt: Date.now() + LIVE_STATS_CACHE_MS,
      stats,
    }
  }