feat: OTEL

Author: izadoesdev

apps/basket/package.json

@@ -18,8 +18,13 @@
     "@databuddy/shared": "workspace:*",
     "@databuddy/validation": "workspace:*",
     "@elysiajs/cors": "^1.3.3",
+    "@elysiajs/opentelemetry": "^1.4.6",
     "@elysiajs/server-timing": "^1.3.0",
     "@maxmind/geoip2-node": "^6.1.0",
+    "@opentelemetry/exporter-trace-otlp-proto": "^0.208.0",
+    "@opentelemetry/resources": "^2.2.0",
+    "@opentelemetry/sdk-trace-node": "^2.2.0",
+    "@opentelemetry/semantic-conventions": "^1.38.0",
     "@types/ua-parser-js": "^0.7.39",
     "async-mutex": "^0.5.0",
     "autumn-js": "catalog:",

apps/basket/src/hooks/auth.ts

@@ -7,6 +7,7 @@
 
 import { and, db, eq, member, websites } from "@databuddy/db";
 import { cacheable } from "@databuddy/redis";
+import { record, setAttributes } from "@elysiajs/opentelemetry";
 import { logger } from "../lib/logger";
 
 type Website = typeof websites.$inferSelect;
@@ -354,17 +355,39 @@ const getWebsiteByIdWithOwnerCached = cacheable(
 	}
 );
 
-export async function getWebsiteByIdV2(
+export function getWebsiteByIdV2(
 	id: string
 ): Promise<WebsiteWithOwner | null> {
-	console.time("getWebsiteByIdV2");
-	try {
-		const result = await getWebsiteByIdWithOwnerCached(id);
-		console.timeEnd("getWebsiteByIdV2");
-		return result;
-	} catch (error) {
-		logger.error({ error, websiteId: id }, "Failed to get website by ID V2");
-		console.timeEnd("getWebsiteByIdV2");
-		return null;
-	}
+	return record("getWebsiteByIdV2", async () => {
+		console.time("getWebsiteByIdV2");
+		setAttributes({
+			"website.id": id,
+		});
+
+		try {
+			const result = await getWebsiteByIdWithOwnerCached(id);
+			console.timeEnd("getWebsiteByIdV2");
+
+			if (result) {
+				setAttributes({
+					"website.found": true,
+					"website.status": result.status,
+					"website.has_owner": Boolean(result.ownerId),
+				});
+			} else {
+				setAttributes({
+					"website.found": false,
+				});
+			}
+
+			return result;
+		} catch (error) {
+			logger.error({ error, websiteId: id }, "Failed to get website by ID V2");
+			setAttributes({
+				"website.lookup_failed": true,
+			});
+			console.timeEnd("getWebsiteByIdV2");
+			return null;
+		}
+	});
 }

apps/basket/src/index.ts

@@ -1,12 +1,13 @@
-// import './polyfills/compression'
+import "./polyfills/compression";
 
+import { opentelemetry } from "@elysiajs/opentelemetry";
+import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-proto";
+import { BatchSpanProcessor } from "@opentelemetry/sdk-trace-node";
 import { Elysia } from "elysia";
 import { logger } from "./lib/logger";
-// import stripeRouter from './routes/stripe';
 import { getProducerStats } from "./lib/producer";
 import basketRouter from "./routes/basket";
 import emailRouter from "./routes/email";
-import "./polyfills/compression";
 
 function getKafkaHealth() {
 	const stats = getProducerStats();
@@ -34,16 +35,31 @@ function getKafkaHealth() {
 		lastErrorTime: stats.lastErrorTime,
 	};
 }
-// import { checkBotId } from "botid/server";
 
 const app = new Elysia()
-	.onError(({ error, code }) => {
+	.use(
+		opentelemetry({
+			serviceName: "basket",
+			spanProcessors: [
+				new BatchSpanProcessor(
+					new OTLPTraceExporter({
+						url: "https://api.axiom.co/v1/traces",
+						headers: {
+							Authorization: `Bearer ${process.env.AXIOM_TOKEN}`,
+							"X-Axiom-Dataset": process.env.AXIOM_DATASET ?? "basket",
+						},
+					})
+				),
+			],
+		})
+	)
+	.onError(function handleError({ error, code }) {
 		if (code === "NOT_FOUND") {
 			return new Response(null, { status: 404 });
 		}
 		logger.error({ error }, "Error in basket service");
 	})
-	.onBeforeHandle(({ request, set }) => {
+	.onBeforeHandle(function handleCors({ request, set }) {
 		const origin = request.headers.get("origin");
 		if (origin) {
 			set.headers ??= {};
@@ -58,17 +74,18 @@ const app = new Elysia()
 	.options("*", () => new Response(null, { status: 204 }))
 	.use(basketRouter)
 	.use(emailRouter)
-	.get("/health", () => ({
-		status: "ok",
-		version: "1.0.0",
-		producer_stats: getProducerStats(),
-		kafka: getKafkaHealth(),
-	}));
+	.get("/health", function healthCheck() {
+		return {
+			status: "ok",
+			version: "1.0.0",
+			producer_stats: getProducerStats(),
+			kafka: getKafkaHealth(),
+		};
+	});
 
 const port = process.env.PORT || 4000;
 
 console.log(`Starting basket service on port ${port}`);
-console.log(`Basket service running on http://localhost:${port}`);
 
 export default {
 	fetch: app.fetch,