36f10954cf
- Add token_usage table with composite unique index for claw_id + date - Add API endpoints: POST /token, GET /token/leaderboard, GET /token/stats - Add TokenLeaderboard component with daily/total period toggle - Add CLI commands: `token` and `stats` for reporting and viewing usage - Add Redis caching for leaderboard with 1-minute TTL - Add shared utilities: authenticateRequest, getTodayDateString - Use UPSERT pattern for atomic token updates - Add i18n translations (en/zh) for new UI elements
135 lines
4.2 KiB
TypeScript
135 lines
4.2 KiB
TypeScript
import Redis from "ioredis";
|
|
|
|
const globalForRedis = globalThis as unknown as {
|
|
redis: Redis | undefined;
|
|
redisSub: Redis | undefined;
|
|
};
|
|
|
|
function createRedisClient(): Redis {
|
|
return new Redis(process.env.REDIS_URL!, {
|
|
maxRetriesPerRequest: 3,
|
|
retryStrategy(times) {
|
|
const delay = Math.min(times * 50, 2000);
|
|
return delay;
|
|
},
|
|
});
|
|
}
|
|
|
|
export const redis = globalForRedis.redis ?? createRedisClient();
|
|
export const redisSub = globalForRedis.redisSub ?? createRedisClient();
|
|
|
|
if (process.env.NODE_ENV !== "production") {
|
|
globalForRedis.redis = redis;
|
|
globalForRedis.redisSub = redisSub;
|
|
}
|
|
|
|
const CHANNEL_REALTIME = "channel:realtime";
|
|
const ACTIVE_CLAWS_KEY = "active:claws";
|
|
const STATS_GLOBAL_KEY = "stats:global";
|
|
const STATS_REGION_KEY = "stats:region";
|
|
const HEATMAP_CACHE_KEY = "cache:heatmap";
|
|
const HOURLY_ACTIVITY_KEY = "stats:hourly";
|
|
|
|
export async function setClawOnline(
|
|
clawId: string,
|
|
ip: string
|
|
): Promise<void> {
|
|
await redis.set(`claw:online:${clawId}`, ip, "EX", 300);
|
|
}
|
|
|
|
export async function isClawOnline(clawId: string): Promise<boolean> {
|
|
const result = await redis.exists(`claw:online:${clawId}`);
|
|
return result === 1;
|
|
}
|
|
|
|
export async function updateActiveClaws(clawId: string): Promise<void> {
|
|
const now = Date.now();
|
|
await redis.zadd(ACTIVE_CLAWS_KEY, now, clawId);
|
|
}
|
|
|
|
export async function getActiveClawIds(
|
|
limit: number = 100
|
|
): Promise<string[]> {
|
|
return redis.zrevrange(ACTIVE_CLAWS_KEY, 0, limit - 1);
|
|
}
|
|
|
|
export async function incrementRegionCount(region: string): Promise<void> {
|
|
await redis.hincrby(STATS_REGION_KEY, region, 1);
|
|
}
|
|
|
|
export async function incrementGlobalStat(
|
|
field: string,
|
|
amount: number = 1
|
|
): Promise<void> {
|
|
await redis.hincrby(STATS_GLOBAL_KEY, field, amount);
|
|
}
|
|
|
|
export async function getGlobalStats(): Promise<Record<string, string>> {
|
|
return redis.hgetall(STATS_GLOBAL_KEY);
|
|
}
|
|
|
|
export async function getRegionStats(): Promise<Record<string, string>> {
|
|
return redis.hgetall(STATS_REGION_KEY);
|
|
}
|
|
|
|
export async function incrementHourlyActivity(): Promise<void> {
|
|
const now = new Date();
|
|
const hourKey = `${now.getUTCFullYear()}-${String(now.getUTCMonth() + 1).padStart(2, "0")}-${String(now.getUTCDate()).padStart(2, "0")}T${String(now.getUTCHours()).padStart(2, "0")}`;
|
|
|
|
await redis.hincrby(HOURLY_ACTIVITY_KEY, hourKey, 1);
|
|
await redis.expire(HOURLY_ACTIVITY_KEY, 48 * 60 * 60);
|
|
}
|
|
|
|
export async function getHourlyActivity(): Promise<Record<string, number>> {
|
|
const allData = await redis.hgetall(HOURLY_ACTIVITY_KEY);
|
|
const now = new Date();
|
|
const result: Record<string, number> = {};
|
|
|
|
for (let i = 23; i >= 0; i--) {
|
|
const date = new Date(now.getTime() - i * 60 * 60 * 1000);
|
|
const hourKey = `${date.getUTCFullYear()}-${String(date.getUTCMonth() + 1).padStart(2, "0")}-${String(date.getUTCDate()).padStart(2, "0")}T${String(date.getUTCHours()).padStart(2, "0")}`;
|
|
result[hourKey] = parseInt(allData[hourKey] || "0", 10);
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
export async function setCacheHeatmap(data: string): Promise<void> {
|
|
await redis.set(HEATMAP_CACHE_KEY, data, "EX", 30);
|
|
}
|
|
|
|
export async function getCacheHeatmap(): Promise<string | null> {
|
|
return redis.get(HEATMAP_CACHE_KEY);
|
|
}
|
|
|
|
export async function publishEvent(event: object): Promise<void> {
|
|
await redis.publish(CHANNEL_REALTIME, JSON.stringify(event));
|
|
}
|
|
|
|
// Token Leaderboard Cache
|
|
const TOKEN_LEADERBOARD_DAILY_KEY = "cache:token_leaderboard:daily";
|
|
const TOKEN_LEADERBOARD_TOTAL_KEY = "cache:token_leaderboard:total";
|
|
|
|
export async function setTokenLeaderboardCache(
|
|
period: "daily" | "total",
|
|
date: string,
|
|
data: string
|
|
): Promise<void> {
|
|
const key = period === "daily" ? TOKEN_LEADERBOARD_DAILY_KEY : TOKEN_LEADERBOARD_TOTAL_KEY;
|
|
await redis.set(`${key}:${date}`, data, "EX", 60); // 1 minute cache
|
|
}
|
|
|
|
export async function getTokenLeaderboardCache(
|
|
period: "daily" | "total",
|
|
date: string
|
|
): Promise<string | null> {
|
|
const key = period === "daily" ? TOKEN_LEADERBOARD_DAILY_KEY : TOKEN_LEADERBOARD_TOTAL_KEY;
|
|
return redis.get(`${key}:${date}`);
|
|
}
|
|
|
|
export async function invalidateTokenLeaderboardCache(): Promise<void> {
|
|
const today = new Date().toISOString().split("T")[0];
|
|
await redis.del(`${TOKEN_LEADERBOARD_DAILY_KEY}:${today}`);
|
|
await redis.del(`${TOKEN_LEADERBOARD_TOTAL_KEY}:${today}`);
|
|
}
|