Merge pull request #2645 from JasirZaeem/add-pooled-db-url

Add databse pooling using Pgbouncer
garageScript · Dec 27, 2022 · 9065dce · 9065dce · vercel · Dec 27, 2022
2 parents 1e5d4c0 + 908194d
commit 9065dce
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/prisma/index.ts b/prisma/index.ts
@@ -5,10 +5,20 @@ declare global {
   var prismag: PrismaClient
 }
 
+let databaseUrl = `postgresql://${process.env.DB_USER}:${process.env.DB_PW}@${process.env.DB_HOST}:${process.env.DB_PORT}/${process.env.DB_NAME}?connection_limit=1`
+
+if (
+  process.env.VERCEL &&
+  process.env.VERCEL_ENV === 'production' &&
+  process.env.USE_POOLED_DB
+) {
+  databaseUrl = `postgresql://${process.env.DB_USER}:${process.env.DB_PW}@${process.env.DB_HOST}:${process.env.DB_POOL_PORT}/${process.env.DB_NAME}?connection_limit=1&pgbouncer=true`
+}
+
 const prismaOptions = {
   datasources: {
     db: {
-      url: `postgresql://${process.env.DB_USER}:${process.env.DB_PW}@${process.env.DB_HOST}:${process.env.DB_PORT}/${process.env.DB_NAME}?connection_limit=1`
+      url: databaseUrl
     }
   }
 }