identifier/payload type for DbJobSpec

graphile · Jan 17, 2024 · 68c5d9f · 68c5d9f
1 parent b050c84
commit 68c5d9f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/__tests__/runner.helpers.getTaskName.test.ts b/__tests__/runner.helpers.getTaskName.test.ts
@@ -1,6 +1,6 @@
 import { Pool, PoolClient } from "pg";
 
-import { DbJobSpec, Job, Runner, RunnerOptions } from "../src/interfaces";
+import { DbJobSpec, Runner, RunnerOptions } from "../src/interfaces";
 import { run } from "../src/runner";
 import {
   ESCAPED_GRAPHILE_WORKER_SCHEMA,

diff --git a/src/interfaces.ts b/src/interfaces.ts
@@ -526,9 +526,13 @@ export interface TaskSpec {
 }
 
 /** Equivalent of graphile_worker.job_spec DB type */
-export interface DbJobSpec {
-  identifier: string;
-  payload?: Record<string, any>;
+export interface DbJobSpec<
+  TIdentifier extends keyof GraphileWorker.Tasks | (string & {}) = string,
+> {
+  identifier: TIdentifier;
+  payload: TIdentifier extends keyof GraphileWorker.Tasks
+    ? GraphileWorker.Tasks[TIdentifier]
+    : unknown;
   queue_name?: string | null;
   run_at?: string | null;
   max_attempts?: number | null;