feat: Add OpenTelemetry instrumentation with Drizzle ORM support

artifacts/api-server/.env.example

@@ -0,0 +1,7 @@
+# OpenTelemetry Configuration for Kubiks
+OTEL_EXPORTER_OTLP_ENDPOINT=https://ingest.kubiks.app
+OTEL_EXPORTER_OTLP_HEADERS=x-kubiks-key=your_kubiks_api_key_here
+OTEL_SERVICE_NAME=api-server
+
+# Other environment variables
+PORT=3000

artifacts/api-server/build.mjs

@@ -15,7 +15,10 @@ async function buildAll() {
   await rm(distDir, { recursive: true, force: true });
 
   await esbuild({
-    entryPoints: [path.resolve(artifactDir, "src/index.ts")],
+    entryPoints: [
+      path.resolve(artifactDir, "src/instrumentation.ts"),
+      path.resolve(artifactDir, "src/index.ts"),
+    ],
     platform: "node",
     bundle: true,
     format: "esm",

artifacts/api-server/package.json

@@ -6,11 +6,16 @@
   "scripts": {
     "dev": "export NODE_ENV=development && pnpm run build && pnpm run start",
     "build": "node ./build.mjs",
-    "start": "node --enable-source-maps ./dist/index.mjs",
+    "start": "node --enable-source-maps --require ./dist/instrumentation.mjs ./dist/index.mjs",
     "typecheck": "tsc -p tsconfig.json --noEmit"
   },
   "dependencies": {
     "@clerk/express": "^2.1.10",
+    "@kubiks/otel-drizzle": "^2.1.0",
+    "@opentelemetry/api": "^1.9.1",
+    "@opentelemetry/auto-instrumentations-node": "^0.76.0",
+    "@opentelemetry/exporter-trace-otlp-http": "^0.218.0",
+    "@opentelemetry/sdk-node": "^0.218.0",
     "@workspace/api-zod": "workspace:*",
     "@workspace/db": "workspace:*",
     "@workspace/integrations-openai-ai-server": "workspace:*",

artifacts/api-server/src/index.ts

@@ -1,3 +1,4 @@
+import "./instrumentation";
 import app from "./app";
 import { logger } from "./lib/logger";
 

artifacts/api-server/src/instrumentation.ts

@@ -0,0 +1,34 @@
+import { NodeSDK } from '@opentelemetry/sdk-node';
+import { getNodeAutoInstrumentations } from '@opentelemetry/auto-instrumentations-node';
+import { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-http';
+import { DrizzleInstrumentation } from '@kubiks/otel-drizzle';
+
+const traceExporter = new OTLPTraceExporter({
+  url: process.env.OTEL_EXPORTER_OTLP_ENDPOINT || 'https://ingest.kubiks.app',
+  headers: {
+    'x-kubiks-key': process.env.OTEL_EXPORTER_OTLP_HEADERS?.split('=')[1] || '',
+  },
+});
+
+const sdk = new NodeSDK({
+  traceExporter,
+  instrumentations: [
+    getNodeAutoInstrumentations({
+      '@opentelemetry/instrumentation-fs': {
+        enabled: false,
+      },
+    }),
+    new DrizzleInstrumentation(),
+  ],
+  serviceName: process.env.OTEL_SERVICE_NAME || 'api-server',
+});
+
+sdk.start();
+
+process.on('SIGTERM', () => {
+  sdk
+    .shutdown()
+    .then(() => console.log('SDK shut down successfully'))
+    .catch((err) => console.error('Error shutting down SDK', err))
+    .finally(() => process.exit(0));
+});