UNPKG

durable-execution-storage-drizzle

Version:

Drizzle ORM storage implementation for durable-execution

553 lines 19.3 kB
import { type TablesRelationalConfig } from 'drizzle-orm'; import { type PgDatabase, type PgQueryResultHKT } from 'drizzle-orm/pg-core'; import type { DurableChildTaskExecution, DurableChildTaskExecutionErrorStorageObject, DurableExecutionErrorStorageObject, DurableStorage, DurableTaskExecutionStatusStorageObject, DurableTaskRetryOptions } from 'durable-execution'; /** * Create a pg table for durable task executions. * * @param tableName - The name of the table. * @returns The pg table. */ export declare function createDurableTaskExecutionsPgTable(tableName?: string): import("drizzle-orm/pg-core").PgTableWithColumns<{ name: string; schema: undefined; columns: { id: import("drizzle-orm/pg-core").PgColumn<{ name: "id"; tableName: string; dataType: "number"; columnType: "PgBigInt53"; data: number; driverParam: string | number; notNull: true; hasDefault: true; isPrimaryKey: true; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: "always"; generated: undefined; }, {}, {}>; rootTaskId: import("drizzle-orm/pg-core").PgColumn<{ name: "root_task_id"; tableName: string; dataType: "string"; columnType: "PgText"; data: string; driverParam: string; notNull: false; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: [string, ...string[]]; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; rootExecutionId: import("drizzle-orm/pg-core").PgColumn<{ name: "root_execution_id"; tableName: string; dataType: "string"; columnType: "PgText"; data: string; driverParam: string; notNull: false; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: [string, ...string[]]; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; parentTaskId: import("drizzle-orm/pg-core").PgColumn<{ name: "parent_task_id"; tableName: string; dataType: "string"; columnType: "PgText"; data: string; driverParam: string; notNull: false; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: [string, ...string[]]; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; parentExecutionId: import("drizzle-orm/pg-core").PgColumn<{ name: "parent_execution_id"; tableName: string; dataType: "string"; columnType: "PgText"; data: string; driverParam: string; notNull: false; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: [string, ...string[]]; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; isFinalizeTask: import("drizzle-orm/pg-core").PgColumn<{ name: "is_finalize_task"; tableName: string; dataType: "boolean"; columnType: "PgBoolean"; data: boolean; driverParam: boolean; notNull: false; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; taskId: import("drizzle-orm/pg-core").PgColumn<{ name: "task_id"; tableName: string; dataType: "string"; columnType: "PgText"; data: string; driverParam: string; notNull: true; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: [string, ...string[]]; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; executionId: import("drizzle-orm/pg-core").PgColumn<{ name: "execution_id"; tableName: string; dataType: "string"; columnType: "PgText"; data: string; driverParam: string; notNull: true; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: [string, ...string[]]; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; retryOptions: import("drizzle-orm/pg-core").PgColumn<{ name: "retry_options"; tableName: string; dataType: "json"; columnType: "PgJson"; data: DurableTaskRetryOptions; driverParam: unknown; notNull: true; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, { $type: DurableTaskRetryOptions; }>; timeoutMs: import("drizzle-orm/pg-core").PgColumn<{ name: "timeout_ms"; tableName: string; dataType: "number"; columnType: "PgInteger"; data: number; driverParam: string | number; notNull: true; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; sleepMsBeforeRun: import("drizzle-orm/pg-core").PgColumn<{ name: "sleep_ms_before_run"; tableName: string; dataType: "number"; columnType: "PgInteger"; data: number; driverParam: string | number; notNull: true; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; runInput: import("drizzle-orm/pg-core").PgColumn<{ name: "run_input"; tableName: string; dataType: "string"; columnType: "PgText"; data: string; driverParam: string; notNull: true; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: [string, ...string[]]; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; runOutput: import("drizzle-orm/pg-core").PgColumn<{ name: "run_output"; tableName: string; dataType: "string"; columnType: "PgText"; data: string; driverParam: string; notNull: false; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: [string, ...string[]]; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; output: import("drizzle-orm/pg-core").PgColumn<{ name: "output"; tableName: string; dataType: "string"; columnType: "PgText"; data: string; driverParam: string; notNull: false; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: [string, ...string[]]; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; childrenTasksCompletedCount: import("drizzle-orm/pg-core").PgColumn<{ name: "children_tasks_completed_count"; tableName: string; dataType: "number"; columnType: "PgInteger"; data: number; driverParam: string | number; notNull: true; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; childrenTasks: import("drizzle-orm/pg-core").PgColumn<{ name: "children_tasks"; tableName: string; dataType: "json"; columnType: "PgJson"; data: DurableChildTaskExecution[]; driverParam: unknown; notNull: false; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, { $type: DurableChildTaskExecution[]; }>; childrenTasksErrors: import("drizzle-orm/pg-core").PgColumn<{ name: "children_tasks_errors"; tableName: string; dataType: "json"; columnType: "PgJson"; data: DurableChildTaskExecutionErrorStorageObject[]; driverParam: unknown; notNull: false; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, { $type: DurableChildTaskExecutionErrorStorageObject[]; }>; finalizeTask: import("drizzle-orm/pg-core").PgColumn<{ name: "finalize_task"; tableName: string; dataType: "json"; columnType: "PgJson"; data: DurableChildTaskExecution; driverParam: unknown; notNull: false; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, { $type: DurableChildTaskExecution; }>; finalizeTaskError: import("drizzle-orm/pg-core").PgColumn<{ name: "finalize_task_error"; tableName: string; dataType: "json"; columnType: "PgJson"; data: DurableExecutionErrorStorageObject; driverParam: unknown; notNull: false; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, { $type: DurableExecutionErrorStorageObject; }>; error: import("drizzle-orm/pg-core").PgColumn<{ name: "error"; tableName: string; dataType: "json"; columnType: "PgJson"; data: DurableExecutionErrorStorageObject; driverParam: unknown; notNull: false; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, { $type: DurableExecutionErrorStorageObject; }>; status: import("drizzle-orm/pg-core").PgColumn<{ name: "status"; tableName: string; dataType: "string"; columnType: "PgText"; data: DurableTaskExecutionStatusStorageObject; driverParam: string; notNull: true; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: [string, ...string[]]; baseColumn: never; identity: undefined; generated: undefined; }, {}, { $type: DurableTaskExecutionStatusStorageObject; }>; isClosed: import("drizzle-orm/pg-core").PgColumn<{ name: "is_closed"; tableName: string; dataType: "boolean"; columnType: "PgBoolean"; data: boolean; driverParam: boolean; notNull: true; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; needsPromiseCancellation: import("drizzle-orm/pg-core").PgColumn<{ name: "needs_promise_cancellation"; tableName: string; dataType: "boolean"; columnType: "PgBoolean"; data: boolean; driverParam: boolean; notNull: true; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; retryAttempts: import("drizzle-orm/pg-core").PgColumn<{ name: "retry_attempts"; tableName: string; dataType: "number"; columnType: "PgInteger"; data: number; driverParam: string | number; notNull: true; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; startAt: import("drizzle-orm/pg-core").PgColumn<{ name: "start_at"; tableName: string; dataType: "date"; columnType: "PgTimestamp"; data: Date; driverParam: string; notNull: true; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; startedAt: import("drizzle-orm/pg-core").PgColumn<{ name: "started_at"; tableName: string; dataType: "date"; columnType: "PgTimestamp"; data: Date; driverParam: string; notNull: false; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; finishedAt: import("drizzle-orm/pg-core").PgColumn<{ name: "finished_at"; tableName: string; dataType: "date"; columnType: "PgTimestamp"; data: Date; driverParam: string; notNull: false; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; expiresAt: import("drizzle-orm/pg-core").PgColumn<{ name: "expires_at"; tableName: string; dataType: "date"; columnType: "PgTimestamp"; data: Date; driverParam: string; notNull: false; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; createdAt: import("drizzle-orm/pg-core").PgColumn<{ name: "created_at"; tableName: string; dataType: "date"; columnType: "PgTimestamp"; data: Date; driverParam: string; notNull: true; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; updatedAt: import("drizzle-orm/pg-core").PgColumn<{ name: "updated_at"; tableName: string; dataType: "date"; columnType: "PgTimestamp"; data: Date; driverParam: string; notNull: true; hasDefault: false; isPrimaryKey: false; isAutoincrement: false; hasRuntimeDefault: false; enumValues: undefined; baseColumn: never; identity: undefined; generated: undefined; }, {}, {}>; }; dialect: "pg"; }>; /** * The type of the pg table for durable task executions. */ export type DurableTaskExecutionsPgTable = ReturnType<typeof createDurableTaskExecutionsPgTable>; /** * Create a pg durable storage. * * @param db - The pg database. * @param table - The pg task executions table. * @returns The pg durable storage. */ export declare function createPgDurableStorage<TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig>(db: PgDatabase<TQueryResult, TFullSchema, TSchema>, table: DurableTaskExecutionsPgTable): DurableStorage; //# sourceMappingURL=pg.d.ts.map