Global PrismaService 구현

prisma/prisma#18628 (comment)

apps/api/src/app/build/build.service.ts

@@ -0,0 +1,7 @@
+import { Injectable } from "@nestjs/common";
+import { PrismaService } from "../prisma/prisma.service";
+
+@Injectable()
+export class BuildService {
+    private constructor(prisma: PrismaService) {}
+}

apps/api/src/app/prisma/extended-client.ts

@@ -0,0 +1,23 @@
+
+import { PrismaClient } from "@prisma/client";
+
+import { existsExtension } from "./extensions/exists.extension";
+
+function extendClient(base: PrismaClient) {
+  // Add as many as you'd like - no ugly types required!
+  return base.$extends(existsExtension);
+}
+
+class UntypedExtendedClient extends PrismaClient {
+  constructor(options?: ConstructorParameters<typeof PrismaClient>[0]) {
+    super(options);
+
+    return extendClient(this) as this;
+  }
+}
+
+const ExtendedPrismaClient = UntypedExtendedClient as unknown as new (
+  options?: ConstructorParameters<typeof PrismaClient>[0]
+) => ReturnType<typeof extendClient>;
+
+export { ExtendedPrismaClient };

apps/api/src/app/prisma/extensions/exists.extension.ts

@@ -0,0 +1,22 @@
+import { Prisma } from "@prisma/client";
+
+export const existsExtension = Prisma.defineExtension({
+  name: "exists",
+  model: {
+    $allModels: {
+      async exists<TModel, TArgs extends Prisma.Args<TModel, "findUniqueOrThrow">>(
+        this: TModel,
+        args?: Prisma.Exact<TArgs, Prisma.Args<TModel, "findUniqueOrThrow">>
+      ): Promise<boolean> {
+        const context = Prisma.getExtensionContext(this);
+
+        try {
+          await (context as any).findUniqueOrThrow(args);
+          return true;
+        } catch {
+          return false;
+        }
+      },
+    },
+  },
+});

apps/api/src/app/prisma/prisma.module.ts

@@ -0,0 +1,10 @@
+import { Global, Module } from "@nestjs/common";
+
+import { PrismaService } from "./prisma.service";
+
+@Global()
+@Module({
+  providers: [PrismaService],
+  exports: [PrismaService],
+})
+export class PrismaModule {}

apps/api/src/app/prisma/prisma.service.ts

@@ -0,0 +1,22 @@
+import type { OnModuleInit, OnModuleDestroy } from "@nestjs/common";
+import { Injectable } from "@nestjs/common";
+
+import { ExtendedPrismaClient } from "./extended-client";
+
+@Injectable()
+export class PrismaService extends ExtendedPrismaClient implements OnModuleInit, OnModuleDestroy {
+  constructor() {
+    super({
+      log: ['query', 'info'],
+      errorFormat: 'colorless',
+    });
+  }
+
+  async onModuleInit() {
+    await this.$connect();
+  }
+
+  async onModuleDestroy() {
+    await this.$disconnect();
+  }
+}