refactor: better postgres connection param typing (#13800)

immich-app · Oct 29, 2024 · 68a4cc2 · 68a4cc2
1 parent e74ddca
commit 68a4cc2
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 3 deletions.
diff --git a/server/src/interfaces/config.interface.ts b/server/src/interfaces/config.interface.ts
@@ -3,7 +3,7 @@ import { QueueOptions } from 'bullmq';
 import { RedisOptions } from 'ioredis';
 import { OpenTelemetryModuleOptions } from 'nestjs-otel/lib/interfaces';
 import { ImmichEnvironment, ImmichTelemetry, ImmichWorker, LogLevel } from 'src/enum';
-import { VectorExtension } from 'src/interfaces/database.interface';
+import { DatabaseConnectionParams, VectorExtension } from 'src/interfaces/database.interface';
 import { PostgresConnectionOptions } from 'typeorm/driver/postgres/PostgresConnectionOptions.js';
 
 export const IConfigRepository = 'IConfigRepository';
@@ -37,7 +37,7 @@ export interface EnvData {
   };
 
   database: {
-    config: PostgresConnectionOptions;
+    config: PostgresConnectionOptions & DatabaseConnectionParams;
     skipMigrations: boolean;
     vectorExtension: VectorExtension;
   };

diff --git a/server/src/interfaces/database.interface.ts b/server/src/interfaces/database.interface.ts
@@ -7,6 +7,22 @@ export enum DatabaseExtension {
 
 export type VectorExtension = DatabaseExtension.VECTOR | DatabaseExtension.VECTORS;
 
+export type DatabaseConnectionURL = {
+  connectionType: 'url';
+  url: string;
+};
+
+export type DatabaseConnectionParts = {
+  connectionType: 'parts';
+  host: string;
+  port: number;
+  username: string;
+  password: string;
+  database: string;
+};
+
+export type DatabaseConnectionParams = DatabaseConnectionURL | DatabaseConnectionParts;
+
 export enum VectorIndex {
   CLIP = 'clip_index',
   FACE = 'face_index',

diff --git a/server/src/repositories/config.repository.ts b/server/src/repositories/config.repository.ts
@@ -132,8 +132,9 @@ const getEnv = (): EnvData => {
         connectTimeoutMS: 10_000, // 10 seconds
         parseInt8: true,
         ...(databaseUrl
-          ? { url: databaseUrl }
+          ? { connectionType: 'url', url: databaseUrl }
           : {
+              connectionType: 'parts',
               host: process.env.DB_HOSTNAME || 'database',
               port: Number(process.env.DB_PORT) || 5432,
               username: process.env.DB_USERNAME || 'postgres',

diff --git a/server/src/services/database.service.spec.ts b/server/src/services/database.service.spec.ts
@@ -61,6 +61,7 @@ describe(DatabaseService.name, () => {
           mockEnvData({
             database: {
               config: {
+                connectionType: 'parts',
                 type: 'postgres',
                 host: 'database',
                 port: 5432,
@@ -290,6 +291,7 @@ describe(DatabaseService.name, () => {
         mockEnvData({
           database: {
             config: {
+              connectionType: 'parts',
               type: 'postgres',
               host: 'database',
               port: 5432,
@@ -313,6 +315,7 @@ describe(DatabaseService.name, () => {
         mockEnvData({
           database: {
             config: {
+              connectionType: 'parts',
               type: 'postgres',
               host: 'database',
               port: 5432,

diff --git a/server/test/repositories/config.repository.mock.ts b/server/test/repositories/config.repository.mock.ts
@@ -17,6 +17,8 @@ const envData: EnvData = {
 
   database: {
     config: {
+      connectionType: 'parts',
+      database: 'immich',
       type: 'postgres',
       host: 'database',
       port: 5432,