Drop postgres (#2912)

Author: CarsonF

package.json

@@ -85,7 +85,6 @@
     "nestjs-console": "^9.0.0",
     "p-retry": "^5.1.2",
     "pako": "^2.1.0",
-    "pg": "^8.11.3",
     "pkg-up": "^4.0.0",
     "plur": "^5.1.0",
     "prismjs-terminal": "^1.2.2",
@@ -118,7 +117,6 @@
     "@types/luxon": "^3.3.1",
     "@types/node": "^18.17.7",
     "@types/pako": "^2.0.0",
-    "@types/pg": "^8.10.2",
     "@types/prismjs": "^1.26.0",
     "@types/react": "^18.2.20",
     "@types/stack-trace": "^0.0.30",

src/core/config/config.service.ts

@@ -9,7 +9,6 @@ import LRUCache from 'lru-cache';
 import { Duration, DurationLike } from 'luxon';
 import { nanoid } from 'nanoid';
 import { Config as Neo4JDriverConfig } from 'neo4j-driver';
-import { PoolConfig } from 'pg';
 import { keys as keysOf } from 'ts-transformer-keys';
 import { Class, Merge, ReadonlyDeep } from 'type-fest';
 import { csv, ID, ServerException } from '../../common';
@@ -177,30 +176,8 @@ export const makeConfig = (env: EnvironmentService) =>
       };
     })();
 
-    postgres = (() => {
-      // Put the PG* env vars in the global env, so the library can use them.
-      // There's dozens of them, so we'll just pass them through implicitly.
-      for (const [key, value] of env) {
-        if (key.startsWith('PG')) {
-          process.env[key] = value;
-        }
-      }
-      /* eslint-disable @typescript-eslint/naming-convention */
-      const config: PoolConfig = {
-        application_name: env.string('PGAPPNAME').optional('cord_api'),
-        connectionString:
-          env.string('PGURI').optional() ??
-          env
-            .string('PGURL')
-            .optional('postgres://postgres:postgres@localhost/cord'),
-      };
-      /* eslint-enable @typescript-eslint/naming-convention */
-      return config;
-    })();
-
-    // Control which database is prioritized, while we migrate to postgres
-    database = env.string('DATABASE').optional('neo4j');
-    usePostgres = this.database.toLowerCase() === 'postgres';
+    // Control which database is prioritized, while we migrate.
+    databaseEngine = env.string('DATABASE').optional('neo4j').toLowerCase();
 
     dbIndexesCreate = env.boolean('DB_CREATE_INDEXES').optional(true);
     dbAutoMigrate = env

src/core/core.module.ts

@@ -14,7 +14,6 @@ import { EventsModule } from './events';
 import { ExceptionFilter } from './exception/exception.filter';
 import { ExceptionNormalizer } from './exception/exception.normalizer';
 import { GraphqlModule } from './graphql';
-import { PostgresModule } from './postgres/postgres.module';
 import { ResourceModule } from './resources/resource.module';
 import { ScalarProviders } from './scalars.resolver';
 import { TimeoutInterceptor } from './timeout.interceptor';
@@ -34,7 +33,6 @@ import { WaitResolver } from './wait.resolver';
     GraphqlModule,
     EventsModule,
     TracingModule,
-    PostgresModule,
     ResourceModule,
   ],
   providers: [
@@ -59,7 +57,6 @@ import { WaitResolver } from './wait.resolver';
     EventsModule,
     ResourceModule,
     TracingModule,
-    PostgresModule,
   ],
 })
 export class CoreModule {}

src/core/database/index.ts

@@ -8,3 +8,4 @@ export * from './indexer';
 export * from './migration';
 export * from './db-type';
 export * from './rollback-manager';
+export * from './split-db.provider';

src/core/database/split-db.provider.ts

@@ -0,0 +1,17 @@
+import type { Provider, Type } from '@nestjs/common';
+import { ModuleRef } from '@nestjs/core';
+import type { PublicOf } from '~/common';
+import { ConfigService } from '../config/config.service';
+
+export const splitDb = <T>(
+  neo4jRepository: Type<T>,
+  edgeDbRepository: Type<PublicOf<T>>,
+): Provider => ({
+  provide: neo4jRepository,
+  inject: [ModuleRef, ConfigService],
+  useFactory: async (moduleRef: ModuleRef, config: ConfigService) => {
+    const cls =
+      config.databaseEngine === 'edgedb' ? edgeDbRepository : neo4jRepository;
+    return await moduleRef.create<T>(cls);
+  },
+});

src/core/index.ts

@@ -5,7 +5,6 @@ export * from './core.module';
 export * from './cache';
 export * from './graphql';
 export * from './database';
-export * from './postgres';
 export * from './events';
 export * from './resources';
 export * from './data-loader';