Move DATABASE_MODE to environment variable

colonial-heritage · Sep 15, 2023 · 7e2e7de · 7e2e7de
1 parent ee2fa79
commit 7e2e7de
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/packages/datahub-data/src/db/connection.ts b/packages/datahub-data/src/db/connection.ts
@@ -1,9 +1,14 @@
+import {Mode} from 'drizzle-orm/mysql-core';
 import * as schema from './schema';
 import {drizzle} from 'drizzle-orm/mysql2';
 import mysql from 'mysql2/promise';
+import {env} from 'node:process';
 
 const poolConnection = mysql.createPool({
-  uri: process.env['DATABASE_URL'],
+  uri: env['DATABASE_URL'],
 });
 
-export default drizzle(poolConnection, {schema, mode: 'planetscale'});
+export default drizzle(poolConnection, {
+  schema,
+  mode: (env['DATABASE_MODE'] || 'default') as Mode,
+});