feat(api): add analytics

Author: ToastedDev

api/package.json

@@ -6,14 +6,15 @@
     "dev": "BUN_CONFIG_NO_CLEAR_TERMINAL_ON_RELOAD=1 bun with-env bun --watch src/index.ts",
     "start": "bun with-env bun src/index.ts",
     "typecheck": "tsc --noEmit",
-    "db:push": "bun with-env drizzle-kit push",
+    "db:push": "bun with-env drizzle-kit push && bun with-env bun src/utils/db/postpush.ts",
     "db:studio": "bun with-env drizzle-kit studio",
     "with-env": "dotenv -e ../.env --"
   },
   "dependencies": {
     "@hono/trpc-server": "^0.3.2",
     "@hono/zod-openapi": "^0.15.1",
     "@trpc/server": "^10.45.2",
+    "cron": "^3.1.7",
     "drizzle-orm": "^0.32.1",
     "hono": "^4.5.2",
     "postgres": "^3.4.4",

api/src/analytics.ts

@@ -0,0 +1,24 @@
+import { CronJob } from "cron";
+import { db } from "./utils/db";
+import { analytics } from "./utils/db/schema";
+
+export async function createAnalyticsCronJob() {
+  new CronJob(
+    "* * * * *",
+    async () => {
+      const currentDate = new Date();
+      const channels = await db.query.channels.findMany();
+      await db.insert(analytics).values(
+        channels.map((channel) => ({
+          timestamp: currentDate,
+          guildId: channel.guildId,
+          channelId: channel.id,
+          count: channel.count,
+        }))
+      );
+    },
+    null,
+    true,
+    "Africa/Abidjan"
+  );
+}

api/src/index.ts

@@ -4,9 +4,12 @@ import { Hono } from "hono";
 import { appRouter } from "./router";
 import { createContext } from "./utils/trpc";
 import { restRouter } from "./rest";
+import { createAnalyticsCronJob } from "./analytics";
 
 const app = new Hono();
 
+createAnalyticsCronJob();
+
 app.get("/", (c) => {
   return c.json({
     message: "Hello World",

api/src/utils/db/postpush.ts

@@ -0,0 +1,17 @@
+import { sql } from "drizzle-orm";
+import { db } from ".";
+
+async function main() {
+  try {
+    await db.execute(
+      sql`SELECT create_hypertable('analytics', by_range('timestamp'))`
+    );
+  } catch (err) {
+    console.log(
+      "An error occured while creating hypertable. Hypertable probably already exists."
+    );
+  }
+  process.exit(0);
+}
+
+main();

api/src/utils/db/schema.ts

@@ -1,5 +1,5 @@
 import { relations } from "drizzle-orm";
-import { integer, pgTable, text } from "drizzle-orm/pg-core";
+import { integer, pgTable, text, timestamp } from "drizzle-orm/pg-core";
 
 export const guilds = pgTable("guilds", {
   id: text("id").primaryKey(),
@@ -25,3 +25,14 @@ export const channelRelations = relations(channels, ({ one }) => ({
     references: [guilds.id],
   }),
 }));
+
+export const analytics = pgTable("analytics", {
+  timestamp: timestamp("timestamp", { withTimezone: true }).primaryKey(),
+  guildId: text("guild_id")
+    .notNull()
+    .references(() => guilds.id),
+  channelId: text("channel_id")
+    .notNull()
+    .references(() => channels.id),
+  count: integer("count").notNull().default(0),
+});

bun.lockb

@@ -1,6 +1,6 @@
 services:
   db:
-    image: postgres:16
+    image: timescale/timescaledb-ha:pg16
     restart: always
     environment:
       POSTGRES_PASSWORD: ${DATABASE_PASSWORD}