feat: explicitely connect to postgres and exit if no connection

tippfehlr · Apr 13, 2024 · 952424c · 952424c
1 parent 9bdb0dd
commit 952424c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -1,4 +1,4 @@
-import { db, migrateToLatest } from './modules/db';
+import { db, initDB } from './modules/db';
 import { connect } from './modules/bot';
 import { writeApi } from './modules/metrics';
 
@@ -12,7 +12,7 @@ async function main() {
   process.on('SIGINT', close);
   process.on('SIGTERM', close);
 
-  await migrateToLatest();
+  await initDB();
   await connect();
 }
 

diff --git a/src/modules/db.ts b/src/modules/db.ts
@@ -10,7 +10,7 @@ import { DB, ActivityRoles, Guilds, StatusRoles, Users } from './db.types';
 import config from './config';
 import { writeIntPoint } from './metrics';
 
-export const pool = new Pool({ connectionString: config.DATABASE_URL, max: 10 });
+const pool = new Pool({ connectionString: config.DATABASE_URL, max: 10 });
 const dialect = new PostgresDialect({ pool });
 export const db = new Kysely<DB>({ dialect });
 const migrator = new Migrator({
@@ -22,7 +22,12 @@ const migrator = new Migrator({
   }),
 });
 
-export async function migrateToLatest() {
+export async function initDB() {
+  await pool.connect().catch(err => {
+    log.error('Couldn’t connect to the database: ' + err.code);
+    process.exit(1);
+  });
+
   const { error, results } = await migrator.migrateToLatest();
 
   results?.forEach(it => {