fix: replace PostgresJsDatabase with DB type in worker constructors for consistency

Author: fahimfaisaal

controlplane/src/core/workers/AIGraphReadme.ts

@@ -1,11 +1,10 @@
-import { PostgresJsDatabase } from 'drizzle-orm/postgres-js';
 import { ConnectionOptions, Job } from 'bullmq';
 import pino from 'pino';
-import * as schema from '../../db/schema.js';
 import { OpenAIGraphql } from '../openai-graphql/index.js';
 import { SubgraphRepository } from '../repositories/SubgraphRepository.js';
 import { FederatedGraphRepository } from '../repositories/FederatedGraphRepository.js';
 import { BaseQueue, BaseWorker } from './base/index.js';
+import { DB } from 'src/db/index.js';
 
 const QueueName = 'ai.graph-readme-generator';
 const WorkerName = 'AIGraphReadmeWorker';
@@ -61,7 +60,7 @@ export class AIGraphReadmeWorker extends BaseWorker<CreateReadmeInputEvent> {
   constructor(
     private input: {
       redisConnection: ConnectionOptions;
-      db: PostgresJsDatabase<typeof schema>;
+      db: DB;
       logger: pino.Logger;
       openAiApiKey: string;
     },

controlplane/src/core/workers/CacheWarmer.ts

@@ -1,12 +1,11 @@
 import { ConnectionOptions, Job, JobsOptions } from 'bullmq';
-import { PostgresJsDatabase } from 'drizzle-orm/postgres-js';
 import pino from 'pino';
-import * as schema from '../../db/schema.js';
 import { BlobStorage } from '../blobstorage/index.js';
 import { ClickHouseClient } from '../clickhouse/index.js';
 import { S3RouterConfigMetadata } from '../composition/composer.js';
 import { CacheWarmerRepository } from '../repositories/CacheWarmerRepository.js';
 import { BaseQueue, BaseWorker } from './base/index.js';
+import { DB } from 'src/db/index.js';
 
 const QueueName = 'cache.warmer';
 const WorkerName = 'CacheWarmerWorker';
@@ -50,7 +49,7 @@ export class CacheWarmerWorker extends BaseWorker<CacheWarmerInput> {
   constructor(
     private input: {
       redisConnection: ConnectionOptions;
-      db: PostgresJsDatabase<typeof schema>;
+      db: DB;
       logger: pino.Logger;
       chClient: ClickHouseClient | undefined;
       blobStorage: BlobStorage;

controlplane/src/core/workers/DeactivateOrganization.ts

@@ -1,11 +1,10 @@
 import { ConnectionOptions, Job, JobsOptions } from 'bullmq';
-import { PostgresJsDatabase } from 'drizzle-orm/postgres-js';
 import pino from 'pino';
-import * as schema from '../../db/schema.js';
 import { OrganizationRepository } from '../repositories/OrganizationRepository.js';
 import Keycloak from '../services/Keycloak.js';
 import { DeleteOrganizationQueue } from './DeleteOrganization.js';
 import { BaseQueue, BaseWorker } from './base/index.js';
+import { DB } from 'src/db/index.js';
 
 const QueueName = 'organization.deactivate';
 const WorkerName = 'DeactivateOrganizationWorker';
@@ -42,7 +41,7 @@ export class DeactivateOrganizationWorker extends BaseWorker<DeactivateOrganizat
   constructor(
     private input: {
       redisConnection: ConnectionOptions;
-      db: PostgresJsDatabase<typeof schema>;
+      db: DB;
       logger: pino.Logger;
       keycloakClient: Keycloak;
       keycloakRealm: string;

controlplane/src/core/workers/DeleteOrganization.ts

@@ -1,14 +1,13 @@
 import { ConnectionOptions, Job, JobsOptions } from 'bullmq';
-import { PostgresJsDatabase } from 'drizzle-orm/postgres-js';
 import pino from 'pino';
-import * as schema from '../../db/schema.js';
 import { OrganizationRepository } from '../repositories/OrganizationRepository.js';
 import Keycloak from '../services/Keycloak.js';
 import { OidcRepository } from '../repositories/OidcRepository.js';
 import OidcProvider from '../services/OidcProvider.js';
 import { BlobStorage } from '../blobstorage/index.js';
 import { DeleteOrganizationAuditLogsQueue } from './DeleteOrganizationAuditLogs.js';
 import { BaseQueue, BaseWorker } from './base/index.js';
+import { DB } from 'src/db/index.js';
 
 const QueueName = 'organization.delete';
 const WorkerName = 'DeleteOrganizationWorker';
@@ -42,7 +41,7 @@ export class DeleteOrganizationWorker extends BaseWorker<DeleteOrganizationInput
   constructor(
     private input: {
       redisConnection: ConnectionOptions;
-      db: PostgresJsDatabase<typeof schema>;
+      db: DB;
       logger: pino.Logger;
       keycloakClient: Keycloak;
       keycloakRealm: string;

controlplane/src/core/workers/DeleteOrganizationAuditLogs.ts

@@ -1,9 +1,8 @@
 import { ConnectionOptions, Job, JobsOptions } from 'bullmq';
-import { PostgresJsDatabase } from 'drizzle-orm/postgres-js';
 import pino from 'pino';
-import * as schema from '../../db/schema.js';
 import { AuditLogRepository } from '../repositories/AuditLogRepository.js';
 import { BaseQueue, BaseWorker } from './base/index.js';
+import { DB } from 'src/db/index.js';
 
 const QueueName = 'organization.delete_audit_logs';
 const WorkerName = 'DeleteOrganizationAuditLogsWorker';
@@ -37,7 +36,7 @@ export class DeleteOrganizationAuditLogsWorker extends BaseWorker<DeleteOrganiza
   constructor(
     private input: {
       redisConnection: ConnectionOptions;
-      db: PostgresJsDatabase<typeof schema>;
+      db: DB;
       logger: pino.Logger;
     },
   ) {

controlplane/src/core/workers/DeleteUser.ts

@@ -1,15 +1,14 @@
 import { PlatformEventName } from '@wundergraph/cosmo-connect/dist/notifications/events_pb';
 import { ConnectionOptions, Job, JobsOptions } from 'bullmq';
-import { PostgresJsDatabase } from 'drizzle-orm/postgres-js';
 import pino from 'pino';
-import * as schema from '../../db/schema.js';
 import { BlobStorage } from '../blobstorage/index.js';
 import { OrganizationRepository } from '../repositories/OrganizationRepository.js';
 import { UserRepository } from '../repositories/UserRepository.js';
 import Keycloak from '../services/Keycloak.js';
 import { PlatformWebhookService } from '../webhooks/PlatformWebhookService.js';
 import { BaseQueue, BaseWorker } from './base/index.js';
 import { DeleteOrganizationAuditLogsQueue } from './DeleteOrganizationAuditLogs.js';
+import { DB } from 'src/db/index.js';
 
 const QueueName = 'user.delete';
 const WorkerName = 'DeleteUserWorker';
@@ -43,7 +42,7 @@ export class DeleteUserQueue extends BaseQueue<DeleteUserInput> {
 export class DeleteUserWorker extends BaseWorker<DeleteUserInput> {
   constructor(
     private input: {
-      db: PostgresJsDatabase<typeof schema>;
+      db: DB;
       redisConnection: ConnectionOptions;
       logger: pino.Logger;
       keycloakClient: Keycloak;

controlplane/src/core/workers/ReactivateOrganization.ts

@@ -1,10 +1,9 @@
 import { ConnectionOptions, Job, JobsOptions } from 'bullmq';
-import { PostgresJsDatabase } from 'drizzle-orm/postgres-js';
 import pino from 'pino';
-import * as schema from '../../db/schema.js';
 import { OrganizationRepository } from '../repositories/OrganizationRepository.js';
 import { DeleteOrganizationQueue } from './DeleteOrganization.js';
 import { BaseQueue, BaseWorker } from './base/index.js';
+import { DB } from 'src/db/index.js';
 
 const QueueName = 'organization.reactivate';
 const WorkerName = 'ReactivateOrganizationWorker';
@@ -39,7 +38,7 @@ export class ReactivateOrganizationWorker extends BaseWorker<ReactivateOrganizat
   constructor(
     private input: {
       redisConnection: ConnectionOptions;
-      db: PostgresJsDatabase<typeof schema>;
+      db: DB;
       logger: pino.Logger;
       deleteOrganizationQueue: DeleteOrganizationQueue;
     },