feat(node): add genericPool integration using opentelemetry instrumentation

Signed-off-by: Kaung Zin Hein <[email protected]>

Author: Zen-cronic

packages/node/package.json

@@ -88,6 +88,7 @@
     "@opentelemetry/resources": "^1.25.1",
     "@opentelemetry/sdk-trace-base": "^1.25.1",
     "@opentelemetry/semantic-conventions": "^1.25.1",
+    "@opentelemetry/instrumentation-generic-pool": "^0.38.0",
     "@prisma/instrumentation": "5.17.0",
     "@sentry/core": "8.25.0",
     "@sentry/opentelemetry": "8.25.0",

packages/node/src/integrations/tracing/genericPool.ts

@@ -0,0 +1,33 @@
+import { GenericPoolInstrumentation } from '@opentelemetry/instrumentation-generic-pool';
+import { defineIntegration, SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, spanToJSON } from '@sentry/core';
+import type { IntegrationFn } from '@sentry/types';
+import { generateInstrumentOnce } from '../../otel/instrument';
+
+const INTEGRATION_NAME = 'GenericPool';
+
+export const instrumentGenericPool = generateInstrumentOnce(INTEGRATION_NAME, () => new GenericPoolInstrumentation({}));
+
+const _genericPoolIntegration = (() => {
+  return {
+    name: INTEGRATION_NAME,
+    setupOnce() {
+      instrumentGenericPool();
+    },
+
+    setup(client) {
+      client.on('spanStart', span => {
+        const spanJSON = spanToJSON(span);
+        if (spanJSON.description === 'generic-pool.acquire') {
+          span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, 'auto.db.otel.generic-pool');
+        }
+      });
+    },
+  };
+}) satisfies IntegrationFn;
+
+/**
+ * GenericPool integration
+ *
+ * Capture tracing data for GenericPool.
+ */
+export const genericPoolIntegration = defineIntegration(_genericPoolIntegration);

packages/node/src/integrations/tracing/index.ts

@@ -14,6 +14,7 @@ import { instrumentMysql2, mysql2Integration } from './mysql2';
 import { instrumentNest, nestIntegration } from './nest/nest';
 import { instrumentPostgres, postgresIntegration } from './postgres';
 import { instrumentRedis, redisIntegration } from './redis';
+import { instrumentGenericPool, genericPoolIntegration} from './genericPool'
 
 /**
  * With OTEL, all performance integrations will be added, as OTEL only initializes them when the patched package is actually required.
@@ -37,6 +38,7 @@ export function getAutoPerformanceIntegrations(): Integration[] {
     hapiIntegration(),
     koaIntegration(),
     connectIntegration(),
+    genericPoolIntegration(),
   ];
 }
 
@@ -61,5 +63,6 @@ export function getOpenTelemetryInstrumentationToPreload(): (((options?: any) =>
     instrumentHapi,
     instrumentGraphql,
     instrumentRedis,
+    instrumentGenericPool,
   ];
 }