Add quickAddJob

Author: madeindjs

.vscode/settings.json

@@ -0,0 +1,6 @@
+{
+  "cSpell.words": [
+    "Graphile",
+    "postgres"
+  ]
+}

docker-compose.yml

@@ -0,0 +1,26 @@
+version: '3.3'
+services:
+  backend:
+    image: node:16-alpine
+    working_dir: /usr/src/app
+    command: sh -c "npm install && npm run start:dev"
+    volumes:
+      - ./.:/usr/src/app
+    ports:
+      - 3000:3000
+    depends_on:
+      - postgres
+
+  postgres:
+    image: postgres:13
+    environment:
+      POSTGRES_USER: example
+      POSTGRES_PASSWORD: password
+      POSTGRES_DB: example
+    # command: postgres -c max_locks_per_transaction=1000
+    ports:
+      - 5432:5432
+    expose:
+      - 5432
+    volumes:
+      - ./docker/postgres:/docker-entrypoint-initdb.d

libs/graphile-worker/src/graphile-worker.module.ts

@@ -0,0 +1,30 @@
+import { DynamicModule, Module, Provider } from '@nestjs/common';
+import { GraphileWorkerService } from './graphile-worker.service';
+
+export const GRAPHILE_WORKER_TOKEN = Symbol.for('NestJsGraphileWorker');
+
+@Module({
+  providers: [GraphileWorkerService],
+  exports: [GraphileWorkerService],
+})
+export class GraphileWorkerModule {
+  /**
+   *
+   * @param connectionString `${env.TYPEORM_CONNECTION}://${env.TYPEORM_USERNAME}:${env.TYPEORM_PASSWORD}@${env.TYPEORM_HOST}/${env.TYPEORM_DATABASE}`
+  }
+   * @returns
+   */
+  static forRoot(connectionString: string): DynamicModule {
+    const graphileWorkerService: Provider = {
+      provide: GraphileWorkerService,
+      useValue: new GraphileWorkerService(connectionString),
+    };
+
+    return {
+      global: true,
+      module: GraphileWorkerModule,
+      providers: [graphileWorkerService],
+      exports: [graphileWorkerService],
+    };
+  }
+}

libs/graphile-worker/src/graphile-worker.service.spec.ts

@@ -0,0 +1,18 @@
+import { Test, TestingModule } from '@nestjs/testing';
+import { GraphileWorkerService } from './graphile-worker.service';
+
+describe('GraphileWorkerService', () => {
+  let service: GraphileWorkerService;
+
+  beforeEach(async () => {
+    const module: TestingModule = await Test.createTestingModule({
+      providers: [GraphileWorkerService],
+    }).compile();
+
+    service = module.get<GraphileWorkerService>(GraphileWorkerService);
+  });
+
+  it('should be defined', () => {
+    expect(service).toBeDefined();
+  });
+});

libs/graphile-worker/src/graphile-worker.service.ts

@@ -0,0 +1,49 @@
+import { Injectable, Logger } from '@nestjs/common';
+import {
+  Job,
+  quickAddJob,
+  runMigrations,
+  TaskSpec,
+  WorkerUtilsOptions,
+} from 'graphile-worker';
+
+@Injectable()
+export class GraphileWorkerService {
+  private readonly logger = new Logger(GraphileWorkerService.name);
+  private isMigrationDone: boolean;
+
+  constructor(private readonly connectionString: string) {}
+
+  async quickAddJob(
+    options: WorkerUtilsOptions,
+    identifier: string,
+    payload?: unknown,
+    spec?: TaskSpec,
+  ): Promise<Job> {
+    await this.runMigrations();
+
+    const job = await quickAddJob(
+      {
+        connectionString: this.connectionString,
+        ...options,
+      },
+      identifier,
+      payload,
+      spec,
+    );
+
+    this.logger.debug(`quickAddJob add job #${job.id}`);
+
+    return job;
+  }
+
+  private async runMigrations() {
+    if (this.isMigrationDone) {
+      return;
+    }
+
+    await runMigrations({ connectionString: this.connectionString });
+    this.logger.debug('Run migrations');
+    this.isMigrationDone = true;
+  }
+}

libs/graphile-worker/src/index.ts

@@ -0,0 +1,2 @@
+export * from './graphile-worker.module';
+export * from './graphile-worker.service';

libs/graphile-worker/tsconfig.lib.json

@@ -0,0 +1,9 @@
+{
+  "extends": "../../tsconfig.json",
+  "compilerOptions": {
+    "declaration": true,
+    "outDir": "../../dist/libs/graphile-worker"
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "dist", "test", "**/*spec.ts"]
+}

nest-cli.json

@@ -1,4 +1,18 @@
 {
   "collection": "@nestjs/schematics",
-  "sourceRoot": "src"
-}
+  "sourceRoot": "src",
+  "projects": {
+    "graphile-worker": {
+      "type": "library",
+      "root": "libs/graphile-worker",
+      "entryFile": "index",
+      "sourceRoot": "libs/graphile-worker/src",
+      "compilerOptions": {
+        "tsConfigPath": "libs/graphile-worker/tsconfig.lib.json"
+      }
+    }
+  },
+  "compilerOptions": {
+    "webpack": true
+  }
+}