feat(tracing): Add Prisma ORM integration.

Author: onurtemizkan

packages/tracing/src/hubextensions.ts

@@ -261,6 +261,12 @@ function _autoloadDatabaseIntegrations(): void {
       };
       return new integration.Postgres();
     },
+    prisma() {
+      const integration = dynamicRequire(module, './integrations/node/prisma') as {
+        Prisma: IntegrationClass<Integration>;
+      };
+      return new integration.Prisma();
+    },
   };
 
   const mappedPackages = Object.keys(packageToIntegrationMapping)

packages/tracing/src/integrations/index.ts

@@ -2,6 +2,7 @@ export { Express } from './node/express';
 export { Postgres } from './node/postgres';
 export { Mysql } from './node/mysql';
 export { Mongo } from './node/mongo';
+export { Prisma } from './node/prisma';
 
 // TODO(v7): Remove this export
 // Please see `src/index.ts` for more details.

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

@@ -0,0 +1,59 @@
+import { Hub } from '@sentry/hub';
+import { EventProcessor, Integration } from '@sentry/types';
+import { fill, isThenable, loadModule, logger } from '@sentry/utils';
+
+import { IS_DEBUG_BUILD } from '../../flags';
+
+interface PrismaClient {
+  prototype: {
+    query: () => void | Promise<unknown>;
+  };
+}
+
+/** Tracing integration for @prisma/client package */
+export class Prisma implements Integration {
+  /**
+   * @inheritDoc
+   */
+  public static id: string = 'Prisma';
+
+  /**
+   * @inheritDoc
+   */
+  public name: string = Prisma.id;
+
+  /**
+   * @inheritDoc
+   */
+  public setupOnce(_: (callback: EventProcessor) => void, getCurrentHub: () => Hub): void {
+    const pkg = loadModule<{ PrismaClient: PrismaClient }>('@prisma/client');
+
+    if (!pkg) {
+      IS_DEBUG_BUILD && logger.error('Prisma integration was unable to require `@prisma/client` package.');
+      return;
+    }
+
+    fill(pkg.PrismaClient.prototype, '_executeRequest', function (orig: () => void | Promise<unknown>) {
+      return function (this: unknown, config: unknown) {
+        const scope = getCurrentHub().getScope();
+        const parentSpan = scope?.getSpan();
+        const span = parentSpan?.startChild({
+          description: typeof config === 'string' ? config : (config as { clientMethod: string }).clientMethod,
+          op: 'db',
+        });
+
+        const rv = orig.call(this, config);
+
+        if (isThenable(rv)) {
+          return rv.then((res: unknown) => {
+            span?.finish();
+            return res;
+          });
+        }
+
+        span?.finish();
+        return rv;
+      };
+    });
+  }
+}