init

2026-03-13 11:00:01 +08:00
commit fa4c458eda
51 changed files with 8843 additions and 0 deletions
--- a/lib/redis/index.ts
+++ b/lib/redis/index.ts
@@ -0,0 +1,107 @@
+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_LOBSTERS_KEY = "active:lobsters";
+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 setLobsterOnline(
+  lobsterId: string,
+  ip: string
+): Promise<void> {
+  await redis.set(`lobster:online:${lobsterId}`, ip, "EX", 300);
+}
+
+export async function isLobsterOnline(lobsterId: string): Promise<boolean> {
+  const result = await redis.exists(`lobster:online:${lobsterId}`);
+  return result === 1;
+}
+
+export async function updateActiveLobsters(lobsterId: string): Promise<void> {
+  const now = Date.now();
+  await redis.zadd(ACTIVE_LOBSTERS_KEY, now, lobsterId);
+}
+
+export async function getActiveLobsterIds(
+  limit: number = 100
+): Promise<string[]> {
+  return redis.zrevrange(ACTIVE_LOBSTERS_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));
+}