init

app/api/v1/heatmap/route.ts

@@ -0,0 +1,61 @@
+import { NextResponse } from "next/server";
+import { gte, and, isNotNull, sql } from "drizzle-orm";
+import { db } from "@/lib/db";
+import { lobsters } from "@/lib/db/schema";
+import { getCacheHeatmap, setCacheHeatmap } from "@/lib/redis";
+
+export async function GET() {
+  try {
+    const cached = await getCacheHeatmap();
+    if (cached) {
+      return NextResponse.json(JSON.parse(cached));
+    }
+
+    const fiveMinutesAgo = new Date(Date.now() - 5 * 60 * 1000);
+
+    const activeLobsters = await db
+      .select({
+        city: lobsters.city,
+        country: lobsters.country,
+        latitude: lobsters.latitude,
+        longitude: lobsters.longitude,
+        count: sql<number>`count(*)`,
+      })
+      .from(lobsters)
+      .where(
+        and(
+          gte(lobsters.lastHeartbeat, fiveMinutesAgo),
+          isNotNull(lobsters.latitude),
+          isNotNull(lobsters.longitude)
+        )
+      )
+      .groupBy(
+        lobsters.city,
+        lobsters.country,
+        lobsters.latitude,
+        lobsters.longitude
+      );
+
+    const points = activeLobsters.map((row) => ({
+      lat: Number(row.latitude),
+      lng: Number(row.longitude),
+      weight: row.count,
+      lobsterCount: row.count,
+      city: row.city,
+      country: row.country,
+    }));
+
+    const lastUpdated = new Date().toISOString();
+    const responseData = { points, lastUpdated };
+
+    await setCacheHeatmap(JSON.stringify(responseData));
+
+    return NextResponse.json(responseData);
+  } catch (error) {
+    console.error("Heatmap error:", error);
+    return NextResponse.json(
+      { error: "Internal server error" },
+      { status: 500 }
+    );
+  }
+}