@prairielearn/migrations 2.1.0 → 3.0.0

package/dist/batched-migrations/batched-migration.js

@@ -1,10 +1,7 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.retryFailedBatchedMigrationJobs = exports.updateBatchedMigrationStatus = exports.selectBatchedMigrationForTimestamp = exports.selectBatchedMigration = exports.selectAllBatchedMigrations = exports.insertBatchedMigration = exports.validateBatchedMigrationImplementation = exports.makeBatchedMigration = exports.BatchedMigrationRowSchema = exports.BatchedMigrationStatusSchema = void 0;
-const postgres_1 = require("@prairielearn/postgres");
-const zod_1 = require("zod");
-const sql = (0, postgres_1.loadSqlEquiv)(__filename);
-exports.BatchedMigrationStatusSchema = zod_1.z.enum([
+import { loadSqlEquiv, queryAsync, queryRow, queryRows, queryOptionalRow, } from '@prairielearn/postgres';
+import { z } from 'zod';
+const sql = loadSqlEquiv(import.meta.filename);
+export const BatchedMigrationStatusSchema = z.enum([
     'pending',
     'paused',
     'running',
@@ -12,28 +9,27 @@ exports.BatchedMigrationStatusSchema = zod_1.z.enum([
     'failed',
     'succeeded',
 ]);
-exports.BatchedMigrationRowSchema = zod_1.z.object({
-    id: zod_1.z.string(),
-    project: zod_1.z.string(),
-    filename: zod_1.z.string(),
-    timestamp: zod_1.z.string(),
-    batch_size: zod_1.z.number(),
-    min_value: zod_1.z.bigint({ coerce: true }),
-    max_value: zod_1.z.bigint({ coerce: true }),
-    status: exports.BatchedMigrationStatusSchema,
-    created_at: zod_1.z.date(),
-    updated_at: zod_1.z.date(),
-    started_at: zod_1.z.date().nullable(),
+export const BatchedMigrationRowSchema = z.object({
+    id: z.string(),
+    project: z.string(),
+    filename: z.string(),
+    timestamp: z.string(),
+    batch_size: z.number(),
+    min_value: z.bigint({ coerce: true }),
+    max_value: z.bigint({ coerce: true }),
+    status: BatchedMigrationStatusSchema,
+    created_at: z.date(),
+    updated_at: z.date(),
+    started_at: z.date().nullable(),
 });
 /**
  * Identity function that helps to write correct batched migrations.
  */
-function makeBatchedMigration(fns) {
+export function makeBatchedMigration(fns) {
     validateBatchedMigrationImplementation(fns);
     return fns;
 }
-exports.makeBatchedMigration = makeBatchedMigration;
-function validateBatchedMigrationImplementation(fns) {
+export function validateBatchedMigrationImplementation(fns) {
     if (typeof fns.getParameters !== 'function') {
         throw new Error('getParameters() must be a function');
     }
@@ -41,33 +37,26 @@ function validateBatchedMigrationImplementation(fns) {
         throw new Error('execute() must be a function');
     }
 }
-exports.validateBatchedMigrationImplementation = validateBatchedMigrationImplementation;
 /**
  * Inserts a new batched migration. If one already exists for the given
  * project/timestamp pair, returns null, otherwise returns the inserted row.
  */
-async function insertBatchedMigration(migration) {
-    return await (0, postgres_1.queryOptionalRow)(sql.insert_batched_migration, migration, exports.BatchedMigrationRowSchema);
+export async function insertBatchedMigration(migration) {
+    return await queryOptionalRow(sql.insert_batched_migration, migration, BatchedMigrationRowSchema);
 }
-exports.insertBatchedMigration = insertBatchedMigration;
-async function selectAllBatchedMigrations(project) {
-    return await (0, postgres_1.queryRows)(sql.select_all_batched_migrations, { project }, exports.BatchedMigrationRowSchema);
+export async function selectAllBatchedMigrations(project) {
+    return await queryRows(sql.select_all_batched_migrations, { project }, BatchedMigrationRowSchema);
 }
-exports.selectAllBatchedMigrations = selectAllBatchedMigrations;
-async function selectBatchedMigration(project, id) {
-    return await (0, postgres_1.queryRow)(sql.select_batched_migration, { project, id }, exports.BatchedMigrationRowSchema);
+export async function selectBatchedMigration(project, id) {
+    return await queryRow(sql.select_batched_migration, { project, id }, BatchedMigrationRowSchema);
 }
-exports.selectBatchedMigration = selectBatchedMigration;
-async function selectBatchedMigrationForTimestamp(project, timestamp) {
-    return await (0, postgres_1.queryRow)(sql.select_batched_migration_for_timestamp, { project, timestamp }, exports.BatchedMigrationRowSchema);
+export async function selectBatchedMigrationForTimestamp(project, timestamp) {
+    return await queryRow(sql.select_batched_migration_for_timestamp, { project, timestamp }, BatchedMigrationRowSchema);
 }
-exports.selectBatchedMigrationForTimestamp = selectBatchedMigrationForTimestamp;
-async function updateBatchedMigrationStatus(id, status) {
-    return await (0, postgres_1.queryRow)(sql.update_batched_migration_status, { id, status }, exports.BatchedMigrationRowSchema);
+export async function updateBatchedMigrationStatus(id, status) {
+    return await queryRow(sql.update_batched_migration_status, { id, status }, BatchedMigrationRowSchema);
 }
-exports.updateBatchedMigrationStatus = updateBatchedMigrationStatus;
-async function retryFailedBatchedMigrationJobs(project, id) {
-    await (0, postgres_1.queryAsync)(sql.retry_failed_jobs, { project, id });
+export async function retryFailedBatchedMigrationJobs(project, id) {
+    await queryAsync(sql.retry_failed_jobs, { project, id });
 }
-exports.retryFailedBatchedMigrationJobs = retryFailedBatchedMigrationJobs;
 //# sourceMappingURL=batched-migration.js.map

package/dist/batched-migrations/batched-migration.js.map

@@ -1 +1 @@
-{"version":3,"file":"batched-migration.js","sourceRoot":"","sources":["../../src/batched-migrations/batched-migration.ts"],"names":[],"mappings":";;;AAAA,qDAMgC;AAChC,6BAAwB;AAExB,MAAM,GAAG,GAAG,IAAA,uBAAY,EAAC,UAAU,CAAC,CAAC;AAExB,QAAA,4BAA4B,GAAG,OAAC,CAAC,IAAI,CAAC;IACjD,SAAS;IACT,QAAQ;IACR,SAAS;IACT,YAAY;IACZ,QAAQ;IACR,WAAW;CACZ,CAAC,CAAC;AAGU,QAAA,yBAAyB,GAAG,OAAC,CAAC,MAAM,CAAC;IAChD,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE;IACd,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE;IACnB,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE;IACpB,SAAS,EAAE,OAAC,CAAC,MAAM,EAAE;IACrB,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE;IACtB,SAAS,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACrC,SAAS,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACrC,MAAM,EAAE,oCAA4B;IACpC,UAAU,EAAE,OAAC,CAAC,IAAI,EAAE;IACpB,UAAU,EAAE,OAAC,CAAC,IAAI,EAAE;IACpB,UAAU,EAAE,OAAC,CAAC,IAAI,EAAE,CAAC,QAAQ,EAAE;CAChC,CAAC,CAAC;AAcH;;GAEG;AACH,SAAgB,oBAAoB,CAA2C,GAAM;IACnF,sCAAsC,CAAC,GAAG,CAAC,CAAC;IAC5C,OAAO,GAAG,CAAC;AACb,CAAC;AAHD,oDAGC;AAED,SAAgB,sCAAsC,CACpD,GAAmC;IAEnC,IAAI,OAAO,GAAG,CAAC,aAAa,KAAK,UAAU,EAAE,CAAC;QAC5C,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;IACxD,CAAC;IACD,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;QACtC,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;IAClD,CAAC;AACH,CAAC;AATD,wFASC;AAOD;;;GAGG;AACI,KAAK,UAAU,sBAAsB,CAC1C,SAA8B;IAE9B,OAAO,MAAM,IAAA,2BAAgB,EAAC,GAAG,CAAC,wBAAwB,EAAE,SAAS,EAAE,iCAAyB,CAAC,CAAC;AACpG,CAAC;AAJD,wDAIC;AAEM,KAAK,UAAU,0BAA0B,CAAC,OAAe;IAC9D,OAAO,MAAM,IAAA,oBAAS,EAAC,GAAG,CAAC,6BAA6B,EAAE,EAAE,OAAO,EAAE,EAAE,iCAAyB,CAAC,CAAC;AACpG,CAAC;AAFD,gEAEC;AAEM,KAAK,UAAU,sBAAsB,CAC1C,OAAe,EACf,EAAU;IAEV,OAAO,MAAM,IAAA,mBAAQ,EAAC,GAAG,CAAC,wBAAwB,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,iCAAyB,CAAC,CAAC;AAClG,CAAC;AALD,wDAKC;AAEM,KAAK,UAAU,kCAAkC,CACtD,OAAe,EACf,SAAiB;IAEjB,OAAO,MAAM,IAAA,mBAAQ,EACnB,GAAG,CAAC,sCAAsC,EAC1C,EAAE,OAAO,EAAE,SAAS,EAAE,EACtB,iCAAyB,CAC1B,CAAC;AACJ,CAAC;AATD,gFASC;AAEM,KAAK,UAAU,4BAA4B,CAChD,EAAU,EACV,MAA8B;IAE9B,OAAO,MAAM,IAAA,mBAAQ,EACnB,GAAG,CAAC,+BAA+B,EACnC,EAAE,EAAE,EAAE,MAAM,EAAE,EACd,iCAAyB,CAC1B,CAAC;AACJ,CAAC;AATD,oEASC;AAEM,KAAK,UAAU,+BAA+B,CAAC,OAAe,EAAE,EAAU;IAC/E,MAAM,IAAA,qBAAU,EAAC,GAAG,CAAC,iBAAiB,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;AAC3D,CAAC;AAFD,0EAEC","sourcesContent":["import {\n  loadSqlEquiv,\n  queryAsync,\n  queryRow,\n  queryRows,\n  queryOptionalRow,\n} from '@prairielearn/postgres';\nimport { z } from 'zod';\n\nconst sql = loadSqlEquiv(__filename);\n\nexport const BatchedMigrationStatusSchema = z.enum([\n  'pending',\n  'paused',\n  'running',\n  'finalizing',\n  'failed',\n  'succeeded',\n]);\nexport type BatchedMigrationStatus = z.infer<typeof BatchedMigrationStatusSchema>;\n\nexport const BatchedMigrationRowSchema = z.object({\n  id: z.string(),\n  project: z.string(),\n  filename: z.string(),\n  timestamp: z.string(),\n  batch_size: z.number(),\n  min_value: z.bigint({ coerce: true }),\n  max_value: z.bigint({ coerce: true }),\n  status: BatchedMigrationStatusSchema,\n  created_at: z.date(),\n  updated_at: z.date(),\n  started_at: z.date().nullable(),\n});\nexport type BatchedMigrationRow = z.infer<typeof BatchedMigrationRowSchema>;\n\nexport interface BatchedMigrationParameters {\n  min?: bigint | string | null;\n  max: bigint | string | null;\n  batchSize?: number;\n}\n\nexport interface BatchedMigrationImplementation {\n  getParameters(): Promise<BatchedMigrationParameters>;\n  execute(start: bigint, end: bigint): Promise<void>;\n}\n\n/**\n * Identity function that helps to write correct batched migrations.\n */\nexport function makeBatchedMigration<T extends BatchedMigrationImplementation>(fns: T): T {\n  validateBatchedMigrationImplementation(fns);\n  return fns;\n}\n\nexport function validateBatchedMigrationImplementation(\n  fns: BatchedMigrationImplementation,\n): asserts fns is BatchedMigrationImplementation {\n  if (typeof fns.getParameters !== 'function') {\n    throw new Error('getParameters() must be a function');\n  }\n  if (typeof fns.execute !== 'function') {\n    throw new Error('execute() must be a function');\n  }\n}\n\ntype NewBatchedMigration = Pick<\n  BatchedMigrationRow,\n  'project' | 'filename' | 'timestamp' | 'batch_size' | 'min_value' | 'max_value' | 'status'\n>;\n\n/**\n * Inserts a new batched migration. If one already exists for the given\n * project/timestamp pair, returns null, otherwise returns the inserted row.\n */\nexport async function insertBatchedMigration(\n  migration: NewBatchedMigration,\n): Promise<BatchedMigrationRow | null> {\n  return await queryOptionalRow(sql.insert_batched_migration, migration, BatchedMigrationRowSchema);\n}\n\nexport async function selectAllBatchedMigrations(project: string) {\n  return await queryRows(sql.select_all_batched_migrations, { project }, BatchedMigrationRowSchema);\n}\n\nexport async function selectBatchedMigration(\n  project: string,\n  id: string,\n): Promise<BatchedMigrationRow> {\n  return await queryRow(sql.select_batched_migration, { project, id }, BatchedMigrationRowSchema);\n}\n\nexport async function selectBatchedMigrationForTimestamp(\n  project: string,\n  timestamp: string,\n): Promise<BatchedMigrationRow> {\n  return await queryRow(\n    sql.select_batched_migration_for_timestamp,\n    { project, timestamp },\n    BatchedMigrationRowSchema,\n  );\n}\n\nexport async function updateBatchedMigrationStatus(\n  id: string,\n  status: BatchedMigrationStatus,\n): Promise<BatchedMigrationRow> {\n  return await queryRow(\n    sql.update_batched_migration_status,\n    { id, status },\n    BatchedMigrationRowSchema,\n  );\n}\n\nexport async function retryFailedBatchedMigrationJobs(project: string, id: string): Promise<void> {\n  await queryAsync(sql.retry_failed_jobs, { project, id });\n}\n"]}
+{"version":3,"file":"batched-migration.js","sourceRoot":"","sources":["../../src/batched-migrations/batched-migration.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,YAAY,EACZ,UAAU,EACV,QAAQ,EACR,SAAS,EACT,gBAAgB,GACjB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,MAAM,GAAG,GAAG,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AAE/C,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAAC,CAAC,IAAI,CAAC;IACjD,SAAS;IACT,QAAQ;IACR,SAAS;IACT,YAAY;IACZ,QAAQ;IACR,WAAW;CACZ,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,CAAC;IAChD,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;IACnB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;IACtB,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACrC,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACrC,MAAM,EAAE,4BAA4B;IACpC,UAAU,EAAE,CAAC,CAAC,IAAI,EAAE;IACpB,UAAU,EAAE,CAAC,CAAC,IAAI,EAAE;IACpB,UAAU,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,QAAQ,EAAE;CAChC,CAAC,CAAC;AAcH;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAA2C,GAAM;IACnF,sCAAsC,CAAC,GAAG,CAAC,CAAC;IAC5C,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,sCAAsC,CACpD,GAAmC;IAEnC,IAAI,OAAO,GAAG,CAAC,aAAa,KAAK,UAAU,EAAE,CAAC;QAC5C,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;IACxD,CAAC;IACD,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;QACtC,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;IAClD,CAAC;AACH,CAAC;AAOD;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,SAA8B;IAE9B,OAAO,MAAM,gBAAgB,CAAC,GAAG,CAAC,wBAAwB,EAAE,SAAS,EAAE,yBAAyB,CAAC,CAAC;AACpG,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAAC,OAAe;IAC9D,OAAO,MAAM,SAAS,CAAC,GAAG,CAAC,6BAA6B,EAAE,EAAE,OAAO,EAAE,EAAE,yBAAyB,CAAC,CAAC;AACpG,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,OAAe,EACf,EAAU;IAEV,OAAO,MAAM,QAAQ,CAAC,GAAG,CAAC,wBAAwB,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,yBAAyB,CAAC,CAAC;AAClG,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kCAAkC,CACtD,OAAe,EACf,SAAiB;IAEjB,OAAO,MAAM,QAAQ,CACnB,GAAG,CAAC,sCAAsC,EAC1C,EAAE,OAAO,EAAE,SAAS,EAAE,EACtB,yBAAyB,CAC1B,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAChD,EAAU,EACV,MAA8B;IAE9B,OAAO,MAAM,QAAQ,CACnB,GAAG,CAAC,+BAA+B,EACnC,EAAE,EAAE,EAAE,MAAM,EAAE,EACd,yBAAyB,CAC1B,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,+BAA+B,CAAC,OAAe,EAAE,EAAU;IAC/E,MAAM,UAAU,CAAC,GAAG,CAAC,iBAAiB,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;AAC3D,CAAC","sourcesContent":["import {\n  loadSqlEquiv,\n  queryAsync,\n  queryRow,\n  queryRows,\n  queryOptionalRow,\n} from '@prairielearn/postgres';\nimport { z } from 'zod';\n\nconst sql = loadSqlEquiv(import.meta.filename);\n\nexport const BatchedMigrationStatusSchema = z.enum([\n  'pending',\n  'paused',\n  'running',\n  'finalizing',\n  'failed',\n  'succeeded',\n]);\nexport type BatchedMigrationStatus = z.infer<typeof BatchedMigrationStatusSchema>;\n\nexport const BatchedMigrationRowSchema = z.object({\n  id: z.string(),\n  project: z.string(),\n  filename: z.string(),\n  timestamp: z.string(),\n  batch_size: z.number(),\n  min_value: z.bigint({ coerce: true }),\n  max_value: z.bigint({ coerce: true }),\n  status: BatchedMigrationStatusSchema,\n  created_at: z.date(),\n  updated_at: z.date(),\n  started_at: z.date().nullable(),\n});\nexport type BatchedMigrationRow = z.infer<typeof BatchedMigrationRowSchema>;\n\nexport interface BatchedMigrationParameters {\n  min?: bigint | string | null;\n  max: bigint | string | null;\n  batchSize?: number;\n}\n\nexport interface BatchedMigrationImplementation {\n  getParameters(): Promise<BatchedMigrationParameters>;\n  execute(start: bigint, end: bigint): Promise<void>;\n}\n\n/**\n * Identity function that helps to write correct batched migrations.\n */\nexport function makeBatchedMigration<T extends BatchedMigrationImplementation>(fns: T): T {\n  validateBatchedMigrationImplementation(fns);\n  return fns;\n}\n\nexport function validateBatchedMigrationImplementation(\n  fns: BatchedMigrationImplementation,\n): asserts fns is BatchedMigrationImplementation {\n  if (typeof fns.getParameters !== 'function') {\n    throw new Error('getParameters() must be a function');\n  }\n  if (typeof fns.execute !== 'function') {\n    throw new Error('execute() must be a function');\n  }\n}\n\ntype NewBatchedMigration = Pick<\n  BatchedMigrationRow,\n  'project' | 'filename' | 'timestamp' | 'batch_size' | 'min_value' | 'max_value' | 'status'\n>;\n\n/**\n * Inserts a new batched migration. If one already exists for the given\n * project/timestamp pair, returns null, otherwise returns the inserted row.\n */\nexport async function insertBatchedMigration(\n  migration: NewBatchedMigration,\n): Promise<BatchedMigrationRow | null> {\n  return await queryOptionalRow(sql.insert_batched_migration, migration, BatchedMigrationRowSchema);\n}\n\nexport async function selectAllBatchedMigrations(project: string) {\n  return await queryRows(sql.select_all_batched_migrations, { project }, BatchedMigrationRowSchema);\n}\n\nexport async function selectBatchedMigration(\n  project: string,\n  id: string,\n): Promise<BatchedMigrationRow> {\n  return await queryRow(sql.select_batched_migration, { project, id }, BatchedMigrationRowSchema);\n}\n\nexport async function selectBatchedMigrationForTimestamp(\n  project: string,\n  timestamp: string,\n): Promise<BatchedMigrationRow> {\n  return await queryRow(\n    sql.select_batched_migration_for_timestamp,\n    { project, timestamp },\n    BatchedMigrationRowSchema,\n  );\n}\n\nexport async function updateBatchedMigrationStatus(\n  id: string,\n  status: BatchedMigrationStatus,\n): Promise<BatchedMigrationRow> {\n  return await queryRow(\n    sql.update_batched_migration_status,\n    { id, status },\n    BatchedMigrationRowSchema,\n  );\n}\n\nexport async function retryFailedBatchedMigrationJobs(project: string, id: string): Promise<void> {\n  await queryAsync(sql.retry_failed_jobs, { project, id });\n}\n"]}

package/dist/batched-migrations/batched-migrations-runner.d.ts

@@ -1,4 +1,4 @@
-/// <reference types="node" />
+/// <reference types="node" resolution-mode="require"/>
 import { EventEmitter } from 'node:events';
 interface BatchedMigrationRunnerOptions {
     project: string;

package/dist/batched-migrations/batched-migrations-runner.js

@@ -1,47 +1,18 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || function (mod) {
-    if (mod && mod.__esModule) return mod;
-    var result = {};
-    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
-    __setModuleDefault(result, mod);
-    return result;
-};
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.finalizeBatchedMigration = exports.enqueueBatchedMigration = exports.stopBatchedMigrations = exports.startBatchedMigrations = exports.initBatchedMigrations = exports.BatchedMigrationsRunner = void 0;
-const node_events_1 = require("node:events");
-const node_path_1 = __importDefault(require("node:path"));
-const promises_1 = require("node:timers/promises");
-const postgres_1 = require("@prairielearn/postgres");
-const named_locks_1 = require("@prairielearn/named-locks");
-const load_migrations_1 = require("../load-migrations");
-const batched_migration_1 = require("./batched-migration");
-const batched_migration_runner_1 = require("./batched-migration-runner");
-const sql = (0, postgres_1.loadSqlEquiv)(__filename);
+import { EventEmitter } from 'node:events';
+import path from 'node:path';
+import { setTimeout as sleep } from 'node:timers/promises';
+import { loadSqlEquiv, queryOptionalRow } from '@prairielearn/postgres';
+import { doWithLock } from '@prairielearn/named-locks';
+import { readAndValidateMigrationsFromDirectories } from '../load-migrations.js';
+import { BatchedMigrationRowSchema, insertBatchedMigration, selectBatchedMigrationForTimestamp, updateBatchedMigrationStatus, validateBatchedMigrationImplementation, } from './batched-migration.js';
+import { BatchedMigrationRunner } from './batched-migration-runner.js';
+const sql = loadSqlEquiv(import.meta.filename);
 const DEFAULT_MIN_VALUE = 1n;
 const DEFAULT_BATCH_SIZE = 1_000;
 const DEFAULT_WORK_DURATION_MS = 60_000;
 const DEFAULT_SLEEP_DURATION_MS = 30_000;
 const EXTENSIONS = ['.js', '.ts', '.mjs', '.mts'];
-class BatchedMigrationsRunner extends node_events_1.EventEmitter {
+export class BatchedMigrationsRunner extends EventEmitter {
     options;
     lockName;
     running = false;
@@ -57,7 +28,7 @@ class BatchedMigrationsRunner extends node_events_1.EventEmitter {
     }
     getMigrationFiles = async () => {
         if (!this.migrationFiles) {
-            this.migrationFiles = await (0, load_migrations_1.readAndValidateMigrationsFromDirectories)(this.options.directories, EXTENSIONS);
+            this.migrationFiles = await readAndValidateMigrationsFromDirectories(this.options.directories, EXTENSIONS);
         }
         return this.migrationFiles;
     };
@@ -75,10 +46,10 @@ class BatchedMigrationsRunner extends node_events_1.EventEmitter {
      */
     async loadMigrationImplementation(migrationFile) {
         // We use dynamic imports to handle both CJS and ESM modules.
-        const migrationModulePath = node_path_1.default.join(migrationFile.directory, migrationFile.filename);
-        const migrationModule = await Promise.resolve(`${migrationModulePath}`).then(s => __importStar(require(s)));
+        const migrationModulePath = path.join(migrationFile.directory, migrationFile.filename);
+        const migrationModule = await import(migrationModulePath);
         const migrationImplementation = migrationModule.default;
-        (0, batched_migration_1.validateBatchedMigrationImplementation)(migrationImplementation);
+        validateBatchedMigrationImplementation(migrationImplementation);
         return migrationImplementation;
     }
     async enqueueBatchedMigration(identifier) {
@@ -94,7 +65,7 @@ class BatchedMigrationsRunner extends node_events_1.EventEmitter {
         const minValue = BigInt(migrationParameters.min ?? DEFAULT_MIN_VALUE);
         const maxValue = BigInt(migrationParameters.max ?? minValue);
         const batchSize = migrationParameters.batchSize ?? DEFAULT_BATCH_SIZE;
-        await (0, batched_migration_1.insertBatchedMigration)({
+        await insertBatchedMigration({
             project: this.options.project,
             filename: migrationFile.filename,
             timestamp: migrationFile.timestamp,
@@ -106,20 +77,20 @@ class BatchedMigrationsRunner extends node_events_1.EventEmitter {
     }
     async finalizeBatchedMigration(identifier, options) {
         const timestamp = identifier.split('_')[0];
-        let migration = await (0, batched_migration_1.selectBatchedMigrationForTimestamp)(this.options.project, timestamp);
+        let migration = await selectBatchedMigrationForTimestamp(this.options.project, timestamp);
         if (migration.status === 'succeeded')
             return;
         // If the migration isn't already in the finalizing state, mark it as such.
         if (migration.status !== 'finalizing') {
-            migration = await (0, batched_migration_1.updateBatchedMigrationStatus)(migration.id, 'finalizing');
+            migration = await updateBatchedMigrationStatus(migration.id, 'finalizing');
         }
-        await (0, named_locks_1.doWithLock)(this.lockNameForTimestamp(timestamp), { autoRenew: true }, async () => {
+        await doWithLock(this.lockNameForTimestamp(timestamp), { autoRenew: true }, async () => {
             const migrationFile = await this.getMigrationForIdentifier(identifier);
             if (!migrationFile) {
                 throw new Error(`No migration found for identifier ${identifier}`);
             }
             const migrationImplementation = await this.loadMigrationImplementation(migrationFile);
-            const runner = new batched_migration_runner_1.BatchedMigrationRunner(migration, migrationImplementation, {
+            const runner = new BatchedMigrationRunner(migration, migrationImplementation, {
                 // Always log progress unless explicitly disabled.
                 logProgress: options?.logProgress ?? true,
             });
@@ -127,7 +98,7 @@ class BatchedMigrationsRunner extends node_events_1.EventEmitter {
             // has attempted every job.
             await runner.run();
         });
-        migration = await (0, batched_migration_1.selectBatchedMigrationForTimestamp)(this.options.project, timestamp);
+        migration = await selectBatchedMigrationForTimestamp(this.options.project, timestamp);
         if (migration.status === 'succeeded')
             return;
         throw new Error(`Expected batched migration with identifier ${identifier} to be marked as 'succeeded', but it is '${migration.status}'.`);
@@ -163,7 +134,7 @@ class BatchedMigrationsRunner extends node_events_1.EventEmitter {
                 // We provide the signal here so that we can more quickly stop things
                 // when we're shutting down.
                 try {
-                    await (0, promises_1.setTimeout)(sleepDurationMs, null, { ref: false, signal: this.abortController.signal });
+                    await sleep(sleepDurationMs, null, { ref: false, signal: this.abortController.signal });
                 }
                 catch (err) {
                     // We don't care about errors here, they should only ever occur when
@@ -175,13 +146,13 @@ class BatchedMigrationsRunner extends node_events_1.EventEmitter {
         }
     }
     async getOrStartMigration() {
-        return (0, named_locks_1.doWithLock)(this.lockName, {
+        return doWithLock(this.lockName, {
             // Don't fail if the lock couldn't be acquired immediately.
             onNotAcquired: () => null,
         }, async () => {
-            let migration = await (0, postgres_1.queryOptionalRow)(sql.select_running_migration, { project: this.options.project }, batched_migration_1.BatchedMigrationRowSchema);
+            let migration = await queryOptionalRow(sql.select_running_migration, { project: this.options.project }, BatchedMigrationRowSchema);
             if (!migration) {
-                migration = await (0, postgres_1.queryOptionalRow)(sql.start_next_pending_migration, { project: this.options.project }, batched_migration_1.BatchedMigrationRowSchema);
+                migration = await queryOptionalRow(sql.start_next_pending_migration, { project: this.options.project }, BatchedMigrationRowSchema);
             }
             return migration;
         });
@@ -199,14 +170,14 @@ class BatchedMigrationsRunner extends node_events_1.EventEmitter {
             return false;
         }
         let didWork = false;
-        await (0, named_locks_1.doWithLock)(this.lockNameForTimestamp(migrationFile.timestamp), {
+        await doWithLock(this.lockNameForTimestamp(migrationFile.timestamp), {
             autoRenew: true,
             // Do nothing if the lock could not immediately be acquired.
             onNotAcquired: () => null,
         }, async () => {
             didWork = true;
             const migrationImplementation = await this.loadMigrationImplementation(migrationFile);
-            const runner = new batched_migration_runner_1.BatchedMigrationRunner(migration, migrationImplementation);
+            const runner = new BatchedMigrationRunner(migration, migrationImplementation);
             try {
                 await runner.run({ signal: this.abortController.signal, durationMs });
             }
@@ -220,35 +191,31 @@ class BatchedMigrationsRunner extends node_events_1.EventEmitter {
         this.abortController.abort();
         // Spin until we're no longer running.
         while (this.running) {
-            await (0, promises_1.setTimeout)(1000);
+            await sleep(1000);
         }
     }
 }
-exports.BatchedMigrationsRunner = BatchedMigrationsRunner;
 let runner = null;
 function assertRunner(runner) {
     if (!runner)
         throw new Error('Batched migrations not initialized');
 }
-function initBatchedMigrations(options) {
+export function initBatchedMigrations(options) {
     if (runner)
         throw new Error('Batched migrations already initialized');
     runner = new BatchedMigrationsRunner(options);
     return runner;
 }
-exports.initBatchedMigrations = initBatchedMigrations;
-function startBatchedMigrations(options = {}) {
+export function startBatchedMigrations(options = {}) {
     assertRunner(runner);
     runner.start(options);
     return runner;
 }
-exports.startBatchedMigrations = startBatchedMigrations;
-async function stopBatchedMigrations() {
+export async function stopBatchedMigrations() {
     assertRunner(runner);
     await runner.stop();
     runner = null;
 }
-exports.stopBatchedMigrations = stopBatchedMigrations;
 /**
  * Given a batched migration identifier like `20230406184103_migration`,
  * enqueues it for execution by creating a row in the `batched_migrations`
@@ -260,11 +227,10 @@ exports.stopBatchedMigrations = stopBatchedMigrations;
  *
  * @param identifier The identifier of the batched migration to enqueue.
  */
-async function enqueueBatchedMigration(identifier) {
+export async function enqueueBatchedMigration(identifier) {
     assertRunner(runner);
     await runner.enqueueBatchedMigration(identifier);
 }
-exports.enqueueBatchedMigration = enqueueBatchedMigration;
 /**
  * Given a batched migration identifier like `20230406184103_migration`,
  * synchronously runs it to completion. An error will be thrown if the final
@@ -273,9 +239,8 @@ exports.enqueueBatchedMigration = enqueueBatchedMigration;
  * @param identifier The identifier of the batched migration to finalize.
  * @param options Options for finalizing the batched migration.
  */
-async function finalizeBatchedMigration(identifier, options) {
+export async function finalizeBatchedMigration(identifier, options) {
     assertRunner(runner);
     await runner.finalizeBatchedMigration(identifier, options);
 }
-exports.finalizeBatchedMigration = finalizeBatchedMigration;
 //# sourceMappingURL=batched-migrations-runner.js.map

package/dist/batched-migrations/batched-migrations-runner.js.map

@@ -1 +1 @@
-{"version":3,"file":"batched-migrations-runner.js","sourceRoot":"","sources":["../../src/batched-migrations/batched-migrations-runner.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,6CAA2C;AAC3C,0DAA6B;AAC7B,mDAA2D;AAC3D,qDAAwE;AACxE,2DAAuD;AAEvD,wDAA6F;AAC7F,2DAS6B;AAC7B,yEAAoE;AAEpE,MAAM,GAAG,GAAG,IAAA,uBAAY,EAAC,UAAU,CAAC,CAAC;AAErC,MAAM,iBAAiB,GAAG,EAAE,CAAC;AAC7B,MAAM,kBAAkB,GAAG,KAAK,CAAC;AACjC,MAAM,wBAAwB,GAAG,MAAM,CAAC;AACxC,MAAM,yBAAyB,GAAG,MAAM,CAAC;AACzC,MAAM,UAAU,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;AAgBlD,MAAa,uBAAwB,SAAQ,0BAAY;IACtC,OAAO,CAAgC;IACvC,QAAQ,CAAS;IAC1B,OAAO,GAAG,KAAK,CAAC;IAChB,cAAc,GAA2B,IAAI,CAAC;IAC9C,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;IAEhD,YAAY,OAAsC;QAChD,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,QAAQ,GAAG,sBAAsB,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;IAC/D,CAAC;IAEO,oBAAoB,CAAC,SAAiB;QAC5C,OAAO,GAAG,IAAI,CAAC,QAAQ,IAAI,SAAS,EAAE,CAAC;IACzC,CAAC;IAEO,iBAAiB,GAAG,KAAK,IAAI,EAAE;QACrC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,IAAI,CAAC,cAAc,GAAG,MAAM,IAAA,0DAAwC,EAClE,IAAI,CAAC,OAAO,CAAC,WAAW,EACxB,UAAU,CACX,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC,CAAC;IAEM,KAAK,CAAC,yBAAyB,CAAC,UAAkB;QACxD,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAE3C,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACtD,MAAM,aAAa,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC;QAC5E,OAAO,aAAa,IAAI,IAAI,CAAC;IAC/B,CAAC;IAED;;;;;OAKG;IACK,KAAK,CAAC,2BAA2B,CAAC,aAA4B;QACpE,6DAA6D;QAC7D,MAAM,mBAAmB,GAAG,mBAAI,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC;QACvF,MAAM,eAAe,GAAG,yBAAa,mBAAmB,uCAAC,CAAC;QAE1D,MAAM,uBAAuB,GAAG,eAAe,CAAC,OAAyC,CAAC;QAC1F,IAAA,0DAAsC,EAAC,uBAAuB,CAAC,CAAC;QAChE,OAAO,uBAAuB,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,uBAAuB,CAAC,UAAkB;QAC9C,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,yBAAyB,CAAC,UAAU,CAAC,CAAC;QACvE,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,qCAAqC,UAAU,EAAE,CAAC,CAAC;QACrE,CAAC;QAED,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,aAAa,CAAC,CAAC;QACtF,MAAM,mBAAmB,GAAG,MAAM,uBAAuB,CAAC,aAAa,EAAE,CAAC;QAE1E,uEAAuE;QACvE,qDAAqD;QACrD,MAAM,MAAM,GACV,mBAAmB,CAAC,GAAG,KAAK,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC;QAE7D,MAAM,QAAQ,GAAG,MAAM,CAAC,mBAAmB,CAAC,GAAG,IAAI,iBAAiB,CAAC,CAAC;QACtE,MAAM,QAAQ,GAAG,MAAM,CAAC,mBAAmB,CAAC,GAAG,IAAI,QAAQ,CAAC,CAAC;QAC7D,MAAM,SAAS,GAAG,mBAAmB,CAAC,SAAS,IAAI,kBAAkB,CAAC;QAEtE,MAAM,IAAA,0CAAsB,EAAC;YAC3B,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;YAC7B,QAAQ,EAAE,aAAa,CAAC,QAAQ;YAChC,SAAS,EAAE,aAAa,CAAC,SAAS;YAClC,UAAU,EAAE,SAAS;YACrB,SAAS,EAAE,QAAQ;YACnB,SAAS,EAAE,QAAQ;YACnB,MAAM;SACP,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,wBAAwB,CAAC,UAAkB,EAAE,OAAyC;QAC1F,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAE3C,IAAI,SAAS,GAAG,MAAM,IAAA,sDAAkC,EAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAE1F,IAAI,SAAS,CAAC,MAAM,KAAK,WAAW;YAAE,OAAO;QAE7C,2EAA2E;QAC3E,IAAI,SAAS,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;YACtC,SAAS,GAAG,MAAM,IAAA,gDAA4B,EAAC,SAAS,CAAC,EAAE,EAAE,YAAY,CAAC,CAAC;QAC7E,CAAC;QAED,MAAM,IAAA,wBAAU,EAAC,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,KAAK,IAAI,EAAE;YACrF,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,yBAAyB,CAAC,UAAU,CAAC,CAAC;YACvE,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,MAAM,IAAI,KAAK,CAAC,qCAAqC,UAAU,EAAE,CAAC,CAAC;YACrE,CAAC;YACD,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,aAAa,CAAC,CAAC;YAEtF,MAAM,MAAM,GAAG,IAAI,iDAAsB,CAAC,SAAS,EAAE,uBAAuB,EAAE;gBAC5E,kDAAkD;gBAClD,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,IAAI;aAC1C,CAAC,CAAC;YAEH,uEAAuE;YACvE,2BAA2B;YAC3B,MAAM,MAAM,CAAC,GAAG,EAAE,CAAC;QACrB,CAAC,CAAC,CAAC;QAEH,SAAS,GAAG,MAAM,IAAA,sDAAkC,EAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAEtF,IAAI,SAAS,CAAC,MAAM,KAAK,WAAW;YAAE,OAAO;QAE7C,MAAM,IAAI,KAAK,CACb,8CAA8C,UAAU,4CAA4C,SAAS,CAAC,MAAM,IAAI,CACzH,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAwC,EAAE;QAC9C,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAChE,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,EAAE,cAAc,EAAE,eAAe,EAAgC;QAC1E,cAAc,KAAK,wBAAwB,CAAC;QAC5C,eAAe,KAAK,yBAAyB,CAAC;QAE9C,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;YACpB,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACxC,uEAAuE;gBACvE,mBAAmB;gBACnB,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;gBACrB,OAAO;YACT,CAAC;YAED,IAAI,OAAO,GAAG,KAAK,CAAC;YACpB,IAAI,CAAC;gBACH,OAAO,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;YACxD,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;YAC1B,CAAC;YAED,0EAA0E;YAC1E,0EAA0E;YAC1E,yBAAyB;YACzB,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,qEAAqE;gBACrE,4BAA4B;gBAC5B,IAAI,CAAC;oBACH,MAAM,IAAA,qBAAK,EAAC,eAAe,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC1F,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,oEAAoE;oBACpE,oEAAoE;oBACpE,gCAAgC;oBAChC,SAAS;gBACX,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mBAAmB;QAC/B,OAAO,IAAA,wBAAU,EACf,IAAI,CAAC,QAAQ,EACb;YACE,2DAA2D;YAC3D,aAAa,EAAE,GAAG,EAAE,CAAC,IAAI;SAC1B,EACD,KAAK,IAAI,EAAE;YACT,IAAI,SAAS,GAAG,MAAM,IAAA,2BAAgB,EACpC,GAAG,CAAC,wBAAwB,EAC5B,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EACjC,6CAAyB,CAC1B,CAAC;YAEF,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,SAAS,GAAG,MAAM,IAAA,2BAAgB,EAChC,GAAG,CAAC,4BAA4B,EAChC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EACjC,6CAAyB,CAC1B,CAAC;YACJ,CAAC;YAED,OAAO,SAAS,CAAC;QACnB,CAAC,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,UAAkB;QACvC,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACnD,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,mCAAmC;YACnC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,uEAAuE;QACvE,+CAA+C;QAC/C,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,yBAAyB,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAChF,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,MAAM,IAAA,wBAAU,EACd,IAAI,CAAC,oBAAoB,CAAC,aAAa,CAAC,SAAS,CAAC,EAClD;YACE,SAAS,EAAE,IAAI;YACf,4DAA4D;YAC5D,aAAa,EAAE,GAAG,EAAE,CAAC,IAAI;SAC1B,EACD,KAAK,IAAI,EAAE;YACT,OAAO,GAAG,IAAI,CAAC;YACf,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,aAAa,CAAC,CAAC;YAEtF,MAAM,MAAM,GAAG,IAAI,iDAAsB,CAAC,SAAS,EAAE,uBAAuB,CAAC,CAAC;YAE9E,IAAI,CAAC;gBACH,MAAM,MAAM,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YACxE,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC,CACF,CAAC;QAEF,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAE7B,sCAAsC;QACtC,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,IAAA,qBAAK,EAAC,IAAI,CAAC,CAAC;QACpB,CAAC;IACH,CAAC;CACF;AA9OD,0DA8OC;AAED,IAAI,MAAM,GAAmC,IAAI,CAAC;AAElD,SAAS,YAAY,CACnB,MAAsC;IAEtC,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;AACrE,CAAC;AAED,SAAgB,qBAAqB,CAAC,OAAsC;IAC1E,IAAI,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IACtE,MAAM,GAAG,IAAI,uBAAuB,CAAC,OAAO,CAAC,CAAC;IAC9C,OAAO,MAAM,CAAC;AAChB,CAAC;AAJD,sDAIC;AAED,SAAgB,sBAAsB,CAAC,UAAwC,EAAE;IAC/E,YAAY,CAAC,MAAM,CAAC,CAAC;IACrB,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACtB,OAAO,MAAM,CAAC;AAChB,CAAC;AAJD,wDAIC;AAEM,KAAK,UAAU,qBAAqB;IACzC,YAAY,CAAC,MAAM,CAAC,CAAC;IACrB,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;IACpB,MAAM,GAAG,IAAI,CAAC;AAChB,CAAC;AAJD,sDAIC;AAED;;;;;;;;;;GAUG;AACI,KAAK,UAAU,uBAAuB,CAAC,UAAkB;IAC9D,YAAY,CAAC,MAAM,CAAC,CAAC;IACrB,MAAM,MAAM,CAAC,uBAAuB,CAAC,UAAU,CAAC,CAAC;AACnD,CAAC;AAHD,0DAGC;AAED;;;;;;;GAOG;AACI,KAAK,UAAU,wBAAwB,CAC5C,UAAkB,EAClB,OAAyC;IAEzC,YAAY,CAAC,MAAM,CAAC,CAAC;IACrB,MAAM,MAAM,CAAC,wBAAwB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;AAC7D,CAAC;AAND,4DAMC","sourcesContent":["import { EventEmitter } from 'node:events';\nimport path from 'node:path';\nimport { setTimeout as sleep } from 'node:timers/promises';\nimport { loadSqlEquiv, queryOptionalRow } from '@prairielearn/postgres';\nimport { doWithLock } from '@prairielearn/named-locks';\n\nimport { MigrationFile, readAndValidateMigrationsFromDirectories } from '../load-migrations';\nimport {\n  BatchedMigrationRowSchema,\n  BatchedMigrationRow,\n  insertBatchedMigration,\n  BatchedMigrationStatus,\n  selectBatchedMigrationForTimestamp,\n  updateBatchedMigrationStatus,\n  BatchedMigrationImplementation,\n  validateBatchedMigrationImplementation,\n} from './batched-migration';\nimport { BatchedMigrationRunner } from './batched-migration-runner';\n\nconst sql = loadSqlEquiv(__filename);\n\nconst DEFAULT_MIN_VALUE = 1n;\nconst DEFAULT_BATCH_SIZE = 1_000;\nconst DEFAULT_WORK_DURATION_MS = 60_000;\nconst DEFAULT_SLEEP_DURATION_MS = 30_000;\nconst EXTENSIONS = ['.js', '.ts', '.mjs', '.mts'];\n\ninterface BatchedMigrationRunnerOptions {\n  project: string;\n  directories: string[];\n}\n\ninterface BatchedMigrationStartOptions {\n  workDurationMs?: number;\n  sleepDurationMs?: number;\n}\n\ninterface BatchedMigrationFinalizeOptions {\n  logProgress?: boolean;\n}\n\nexport class BatchedMigrationsRunner extends EventEmitter {\n  private readonly options: BatchedMigrationRunnerOptions;\n  private readonly lockName: string;\n  private running = false;\n  private migrationFiles: MigrationFile[] | null = null;\n  private abortController = new AbortController();\n\n  constructor(options: BatchedMigrationRunnerOptions) {\n    super();\n    this.options = options;\n    this.lockName = `batched-migrations:${this.options.project}`;\n  }\n\n  private lockNameForTimestamp(timestamp: string) {\n    return `${this.lockName}:${timestamp}`;\n  }\n\n  private getMigrationFiles = async () => {\n    if (!this.migrationFiles) {\n      this.migrationFiles = await readAndValidateMigrationsFromDirectories(\n        this.options.directories,\n        EXTENSIONS,\n      );\n    }\n    return this.migrationFiles;\n  };\n\n  private async getMigrationForIdentifier(identifier: string): Promise<MigrationFile | null> {\n    const timestamp = identifier.split('_')[0];\n\n    const migrationFiles = await this.getMigrationFiles();\n    const migrationFile = migrationFiles.find((m) => m.timestamp === timestamp);\n    return migrationFile ?? null;\n  }\n\n  /**\n   * Loads the implementation for the migration with the given identifier. The identifier\n   * must start with a 14-character timestamp. It may optionally be followed by\n   * an underscore with additional characters, which are ignored. These should\n   * typically be used to provide a human-readable name for the migration.\n   */\n  private async loadMigrationImplementation(migrationFile: MigrationFile) {\n    // We use dynamic imports to handle both CJS and ESM modules.\n    const migrationModulePath = path.join(migrationFile.directory, migrationFile.filename);\n    const migrationModule = await import(migrationModulePath);\n\n    const migrationImplementation = migrationModule.default as BatchedMigrationImplementation;\n    validateBatchedMigrationImplementation(migrationImplementation);\n    return migrationImplementation;\n  }\n\n  async enqueueBatchedMigration(identifier: string) {\n    const migrationFile = await this.getMigrationForIdentifier(identifier);\n    if (!migrationFile) {\n      throw new Error(`No migration found for identifier ${identifier}`);\n    }\n\n    const migrationImplementation = await this.loadMigrationImplementation(migrationFile);\n    const migrationParameters = await migrationImplementation.getParameters();\n\n    // If `max` is null, that implies that there are no rows to process, so\n    // we can immediately mark the migration as finished.\n    const status: BatchedMigrationStatus =\n      migrationParameters.max === null ? 'succeeded' : 'pending';\n\n    const minValue = BigInt(migrationParameters.min ?? DEFAULT_MIN_VALUE);\n    const maxValue = BigInt(migrationParameters.max ?? minValue);\n    const batchSize = migrationParameters.batchSize ?? DEFAULT_BATCH_SIZE;\n\n    await insertBatchedMigration({\n      project: this.options.project,\n      filename: migrationFile.filename,\n      timestamp: migrationFile.timestamp,\n      batch_size: batchSize,\n      min_value: minValue,\n      max_value: maxValue,\n      status,\n    });\n  }\n\n  async finalizeBatchedMigration(identifier: string, options?: BatchedMigrationFinalizeOptions) {\n    const timestamp = identifier.split('_')[0];\n\n    let migration = await selectBatchedMigrationForTimestamp(this.options.project, timestamp);\n\n    if (migration.status === 'succeeded') return;\n\n    // If the migration isn't already in the finalizing state, mark it as such.\n    if (migration.status !== 'finalizing') {\n      migration = await updateBatchedMigrationStatus(migration.id, 'finalizing');\n    }\n\n    await doWithLock(this.lockNameForTimestamp(timestamp), { autoRenew: true }, async () => {\n      const migrationFile = await this.getMigrationForIdentifier(identifier);\n      if (!migrationFile) {\n        throw new Error(`No migration found for identifier ${identifier}`);\n      }\n      const migrationImplementation = await this.loadMigrationImplementation(migrationFile);\n\n      const runner = new BatchedMigrationRunner(migration, migrationImplementation, {\n        // Always log progress unless explicitly disabled.\n        logProgress: options?.logProgress ?? true,\n      });\n\n      // Because we don't give any arguments to `run()`, it will run until it\n      // has attempted every job.\n      await runner.run();\n    });\n\n    migration = await selectBatchedMigrationForTimestamp(this.options.project, timestamp);\n\n    if (migration.status === 'succeeded') return;\n\n    throw new Error(\n      `Expected batched migration with identifier ${identifier} to be marked as 'succeeded', but it is '${migration.status}'.`,\n    );\n  }\n\n  start(options: BatchedMigrationStartOptions = {}) {\n    if (this.running) {\n      throw new Error('BatchedMigrationsRunner is already running');\n    }\n\n    this.loop(options);\n  }\n\n  async loop({ workDurationMs, sleepDurationMs }: BatchedMigrationStartOptions) {\n    workDurationMs ??= DEFAULT_WORK_DURATION_MS;\n    sleepDurationMs ??= DEFAULT_SLEEP_DURATION_MS;\n\n    this.running = true;\n    while (this.running) {\n      if (this.abortController.signal.aborted) {\n        // We assign this here so that `stop()` can tell when this loop is done\n        // processing jobs.\n        this.running = false;\n        return;\n      }\n\n      let didWork = false;\n      try {\n        didWork = await this.maybePerformWork(workDurationMs);\n      } catch (err) {\n        this.emit('error', err);\n      }\n\n      // If we did work, we'll immediately try again since there's probably more\n      // work to be done. If not, we'll sleep for a while - maybe some more work\n      // will become available!\n      if (!didWork) {\n        // We provide the signal here so that we can more quickly stop things\n        // when we're shutting down.\n        try {\n          await sleep(sleepDurationMs, null, { ref: false, signal: this.abortController.signal });\n        } catch (err) {\n          // We don't care about errors here, they should only ever occur when\n          // the AbortController is aborted. Continue to the next iteration of\n          // the loop so we can shut down.\n          continue;\n        }\n      }\n    }\n  }\n\n  private async getOrStartMigration(): Promise<BatchedMigrationRow | null> {\n    return doWithLock(\n      this.lockName,\n      {\n        // Don't fail if the lock couldn't be acquired immediately.\n        onNotAcquired: () => null,\n      },\n      async () => {\n        let migration = await queryOptionalRow(\n          sql.select_running_migration,\n          { project: this.options.project },\n          BatchedMigrationRowSchema,\n        );\n\n        if (!migration) {\n          migration = await queryOptionalRow(\n            sql.start_next_pending_migration,\n            { project: this.options.project },\n            BatchedMigrationRowSchema,\n          );\n        }\n\n        return migration;\n      },\n    );\n  }\n\n  async maybePerformWork(durationMs: number): Promise<boolean> {\n    const migration = await this.getOrStartMigration();\n    if (!migration) {\n      // No work to do. Handle this case.\n      return false;\n    }\n\n    // This server may not yet know about the current running migration. If\n    // that's the case, we'll just skip it for now.\n    const migrationFile = await this.getMigrationForIdentifier(migration.timestamp);\n    if (!migrationFile) {\n      return false;\n    }\n\n    let didWork = false;\n    await doWithLock(\n      this.lockNameForTimestamp(migrationFile.timestamp),\n      {\n        autoRenew: true,\n        // Do nothing if the lock could not immediately be acquired.\n        onNotAcquired: () => null,\n      },\n      async () => {\n        didWork = true;\n        const migrationImplementation = await this.loadMigrationImplementation(migrationFile);\n\n        const runner = new BatchedMigrationRunner(migration, migrationImplementation);\n\n        try {\n          await runner.run({ signal: this.abortController.signal, durationMs });\n        } catch (err) {\n          this.emit('error', err);\n        }\n      },\n    );\n\n    return didWork;\n  }\n\n  async stop() {\n    this.abortController.abort();\n\n    // Spin until we're no longer running.\n    while (this.running) {\n      await sleep(1000);\n    }\n  }\n}\n\nlet runner: BatchedMigrationsRunner | null = null;\n\nfunction assertRunner(\n  runner: BatchedMigrationsRunner | null,\n): asserts runner is BatchedMigrationsRunner {\n  if (!runner) throw new Error('Batched migrations not initialized');\n}\n\nexport function initBatchedMigrations(options: BatchedMigrationRunnerOptions) {\n  if (runner) throw new Error('Batched migrations already initialized');\n  runner = new BatchedMigrationsRunner(options);\n  return runner;\n}\n\nexport function startBatchedMigrations(options: BatchedMigrationStartOptions = {}) {\n  assertRunner(runner);\n  runner.start(options);\n  return runner;\n}\n\nexport async function stopBatchedMigrations() {\n  assertRunner(runner);\n  await runner.stop();\n  runner = null;\n}\n\n/**\n * Given a batched migration identifier like `20230406184103_migration`,\n * enqueues it for execution by creating a row in the `batched_migrations`\n * table.\n *\n * Despite taking a full identifier, only the timestamp is used to uniquely\n * identify the batched migration. The remaining part is just used to make\n * calls more human-readable.\n *\n * @param identifier The identifier of the batched migration to enqueue.\n */\nexport async function enqueueBatchedMigration(identifier: string) {\n  assertRunner(runner);\n  await runner.enqueueBatchedMigration(identifier);\n}\n\n/**\n * Given a batched migration identifier like `20230406184103_migration`,\n * synchronously runs it to completion. An error will be thrown if the final\n * status of the migration is not `succeeded`.\n *\n * @param identifier The identifier of the batched migration to finalize.\n * @param options Options for finalizing the batched migration.\n */\nexport async function finalizeBatchedMigration(\n  identifier: string,\n  options?: BatchedMigrationFinalizeOptions,\n) {\n  assertRunner(runner);\n  await runner.finalizeBatchedMigration(identifier, options);\n}\n"]}
+{"version":3,"file":"batched-migrations-runner.js","sourceRoot":"","sources":["../../src/batched-migrations/batched-migrations-runner.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,UAAU,IAAI,KAAK,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AACxE,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAEvD,OAAO,EAAiB,wCAAwC,EAAE,MAAM,uBAAuB,CAAC;AAChG,OAAO,EACL,yBAAyB,EAEzB,sBAAsB,EAEtB,kCAAkC,EAClC,4BAA4B,EAE5B,sCAAsC,GACvC,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AAEvE,MAAM,GAAG,GAAG,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AAE/C,MAAM,iBAAiB,GAAG,EAAE,CAAC;AAC7B,MAAM,kBAAkB,GAAG,KAAK,CAAC;AACjC,MAAM,wBAAwB,GAAG,MAAM,CAAC;AACxC,MAAM,yBAAyB,GAAG,MAAM,CAAC;AACzC,MAAM,UAAU,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;AAgBlD,MAAM,OAAO,uBAAwB,SAAQ,YAAY;IACtC,OAAO,CAAgC;IACvC,QAAQ,CAAS;IAC1B,OAAO,GAAG,KAAK,CAAC;IAChB,cAAc,GAA2B,IAAI,CAAC;IAC9C,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;IAEhD,YAAY,OAAsC;QAChD,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,QAAQ,GAAG,sBAAsB,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;IAC/D,CAAC;IAEO,oBAAoB,CAAC,SAAiB;QAC5C,OAAO,GAAG,IAAI,CAAC,QAAQ,IAAI,SAAS,EAAE,CAAC;IACzC,CAAC;IAEO,iBAAiB,GAAG,KAAK,IAAI,EAAE;QACrC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,IAAI,CAAC,cAAc,GAAG,MAAM,wCAAwC,CAClE,IAAI,CAAC,OAAO,CAAC,WAAW,EACxB,UAAU,CACX,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC,CAAC;IAEM,KAAK,CAAC,yBAAyB,CAAC,UAAkB;QACxD,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAE3C,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACtD,MAAM,aAAa,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC;QAC5E,OAAO,aAAa,IAAI,IAAI,CAAC;IAC/B,CAAC;IAED;;;;;OAKG;IACK,KAAK,CAAC,2BAA2B,CAAC,aAA4B;QACpE,6DAA6D;QAC7D,MAAM,mBAAmB,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC;QACvF,MAAM,eAAe,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAE1D,MAAM,uBAAuB,GAAG,eAAe,CAAC,OAAyC,CAAC;QAC1F,sCAAsC,CAAC,uBAAuB,CAAC,CAAC;QAChE,OAAO,uBAAuB,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,uBAAuB,CAAC,UAAkB;QAC9C,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,yBAAyB,CAAC,UAAU,CAAC,CAAC;QACvE,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,qCAAqC,UAAU,EAAE,CAAC,CAAC;QACrE,CAAC;QAED,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,aAAa,CAAC,CAAC;QACtF,MAAM,mBAAmB,GAAG,MAAM,uBAAuB,CAAC,aAAa,EAAE,CAAC;QAE1E,uEAAuE;QACvE,qDAAqD;QACrD,MAAM,MAAM,GACV,mBAAmB,CAAC,GAAG,KAAK,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC;QAE7D,MAAM,QAAQ,GAAG,MAAM,CAAC,mBAAmB,CAAC,GAAG,IAAI,iBAAiB,CAAC,CAAC;QACtE,MAAM,QAAQ,GAAG,MAAM,CAAC,mBAAmB,CAAC,GAAG,IAAI,QAAQ,CAAC,CAAC;QAC7D,MAAM,SAAS,GAAG,mBAAmB,CAAC,SAAS,IAAI,kBAAkB,CAAC;QAEtE,MAAM,sBAAsB,CAAC;YAC3B,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;YAC7B,QAAQ,EAAE,aAAa,CAAC,QAAQ;YAChC,SAAS,EAAE,aAAa,CAAC,SAAS;YAClC,UAAU,EAAE,SAAS;YACrB,SAAS,EAAE,QAAQ;YACnB,SAAS,EAAE,QAAQ;YACnB,MAAM;SACP,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,wBAAwB,CAAC,UAAkB,EAAE,OAAyC;QAC1F,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAE3C,IAAI,SAAS,GAAG,MAAM,kCAAkC,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAE1F,IAAI,SAAS,CAAC,MAAM,KAAK,WAAW;YAAE,OAAO;QAE7C,2EAA2E;QAC3E,IAAI,SAAS,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;YACtC,SAAS,GAAG,MAAM,4BAA4B,CAAC,SAAS,CAAC,EAAE,EAAE,YAAY,CAAC,CAAC;QAC7E,CAAC;QAED,MAAM,UAAU,CAAC,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,KAAK,IAAI,EAAE;YACrF,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,yBAAyB,CAAC,UAAU,CAAC,CAAC;YACvE,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,MAAM,IAAI,KAAK,CAAC,qCAAqC,UAAU,EAAE,CAAC,CAAC;YACrE,CAAC;YACD,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,aAAa,CAAC,CAAC;YAEtF,MAAM,MAAM,GAAG,IAAI,sBAAsB,CAAC,SAAS,EAAE,uBAAuB,EAAE;gBAC5E,kDAAkD;gBAClD,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,IAAI;aAC1C,CAAC,CAAC;YAEH,uEAAuE;YACvE,2BAA2B;YAC3B,MAAM,MAAM,CAAC,GAAG,EAAE,CAAC;QACrB,CAAC,CAAC,CAAC;QAEH,SAAS,GAAG,MAAM,kCAAkC,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAEtF,IAAI,SAAS,CAAC,MAAM,KAAK,WAAW;YAAE,OAAO;QAE7C,MAAM,IAAI,KAAK,CACb,8CAA8C,UAAU,4CAA4C,SAAS,CAAC,MAAM,IAAI,CACzH,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAwC,EAAE;QAC9C,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAChE,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,EAAE,cAAc,EAAE,eAAe,EAAgC;QAC1E,cAAc,KAAK,wBAAwB,CAAC;QAC5C,eAAe,KAAK,yBAAyB,CAAC;QAE9C,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;YACpB,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACxC,uEAAuE;gBACvE,mBAAmB;gBACnB,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;gBACrB,OAAO;YACT,CAAC;YAED,IAAI,OAAO,GAAG,KAAK,CAAC;YACpB,IAAI,CAAC;gBACH,OAAO,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;YACxD,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;YAC1B,CAAC;YAED,0EAA0E;YAC1E,0EAA0E;YAC1E,yBAAyB;YACzB,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,qEAAqE;gBACrE,4BAA4B;gBAC5B,IAAI,CAAC;oBACH,MAAM,KAAK,CAAC,eAAe,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC1F,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,oEAAoE;oBACpE,oEAAoE;oBACpE,gCAAgC;oBAChC,SAAS;gBACX,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mBAAmB;QAC/B,OAAO,UAAU,CACf,IAAI,CAAC,QAAQ,EACb;YACE,2DAA2D;YAC3D,aAAa,EAAE,GAAG,EAAE,CAAC,IAAI;SAC1B,EACD,KAAK,IAAI,EAAE;YACT,IAAI,SAAS,GAAG,MAAM,gBAAgB,CACpC,GAAG,CAAC,wBAAwB,EAC5B,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EACjC,yBAAyB,CAC1B,CAAC;YAEF,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,SAAS,GAAG,MAAM,gBAAgB,CAChC,GAAG,CAAC,4BAA4B,EAChC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EACjC,yBAAyB,CAC1B,CAAC;YACJ,CAAC;YAED,OAAO,SAAS,CAAC;QACnB,CAAC,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,UAAkB;QACvC,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACnD,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,mCAAmC;YACnC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,uEAAuE;QACvE,+CAA+C;QAC/C,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,yBAAyB,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAChF,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,MAAM,UAAU,CACd,IAAI,CAAC,oBAAoB,CAAC,aAAa,CAAC,SAAS,CAAC,EAClD;YACE,SAAS,EAAE,IAAI;YACf,4DAA4D;YAC5D,aAAa,EAAE,GAAG,EAAE,CAAC,IAAI;SAC1B,EACD,KAAK,IAAI,EAAE;YACT,OAAO,GAAG,IAAI,CAAC;YACf,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,aAAa,CAAC,CAAC;YAEtF,MAAM,MAAM,GAAG,IAAI,sBAAsB,CAAC,SAAS,EAAE,uBAAuB,CAAC,CAAC;YAE9E,IAAI,CAAC;gBACH,MAAM,MAAM,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YACxE,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC,CACF,CAAC;QAEF,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAE7B,sCAAsC;QACtC,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC;QACpB,CAAC;IACH,CAAC;CACF;AAED,IAAI,MAAM,GAAmC,IAAI,CAAC;AAElD,SAAS,YAAY,CACnB,MAAsC;IAEtC,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;AACrE,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,OAAsC;IAC1E,IAAI,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IACtE,MAAM,GAAG,IAAI,uBAAuB,CAAC,OAAO,CAAC,CAAC;IAC9C,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,UAAwC,EAAE;IAC/E,YAAY,CAAC,MAAM,CAAC,CAAC;IACrB,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACtB,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB;IACzC,YAAY,CAAC,MAAM,CAAC,CAAC;IACrB,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;IACpB,MAAM,GAAG,IAAI,CAAC;AAChB,CAAC;AAED;;;;;;;;;;GAUG;AACH,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAAC,UAAkB;IAC9D,YAAY,CAAC,MAAM,CAAC,CAAC;IACrB,MAAM,MAAM,CAAC,uBAAuB,CAAC,UAAU,CAAC,CAAC;AACnD,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAC5C,UAAkB,EAClB,OAAyC;IAEzC,YAAY,CAAC,MAAM,CAAC,CAAC;IACrB,MAAM,MAAM,CAAC,wBAAwB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;AAC7D,CAAC","sourcesContent":["import { EventEmitter } from 'node:events';\nimport path from 'node:path';\nimport { setTimeout as sleep } from 'node:timers/promises';\nimport { loadSqlEquiv, queryOptionalRow } from '@prairielearn/postgres';\nimport { doWithLock } from '@prairielearn/named-locks';\n\nimport { MigrationFile, readAndValidateMigrationsFromDirectories } from '../load-migrations.js';\nimport {\n  BatchedMigrationRowSchema,\n  BatchedMigrationRow,\n  insertBatchedMigration,\n  BatchedMigrationStatus,\n  selectBatchedMigrationForTimestamp,\n  updateBatchedMigrationStatus,\n  BatchedMigrationImplementation,\n  validateBatchedMigrationImplementation,\n} from './batched-migration.js';\nimport { BatchedMigrationRunner } from './batched-migration-runner.js';\n\nconst sql = loadSqlEquiv(import.meta.filename);\n\nconst DEFAULT_MIN_VALUE = 1n;\nconst DEFAULT_BATCH_SIZE = 1_000;\nconst DEFAULT_WORK_DURATION_MS = 60_000;\nconst DEFAULT_SLEEP_DURATION_MS = 30_000;\nconst EXTENSIONS = ['.js', '.ts', '.mjs', '.mts'];\n\ninterface BatchedMigrationRunnerOptions {\n  project: string;\n  directories: string[];\n}\n\ninterface BatchedMigrationStartOptions {\n  workDurationMs?: number;\n  sleepDurationMs?: number;\n}\n\ninterface BatchedMigrationFinalizeOptions {\n  logProgress?: boolean;\n}\n\nexport class BatchedMigrationsRunner extends EventEmitter {\n  private readonly options: BatchedMigrationRunnerOptions;\n  private readonly lockName: string;\n  private running = false;\n  private migrationFiles: MigrationFile[] | null = null;\n  private abortController = new AbortController();\n\n  constructor(options: BatchedMigrationRunnerOptions) {\n    super();\n    this.options = options;\n    this.lockName = `batched-migrations:${this.options.project}`;\n  }\n\n  private lockNameForTimestamp(timestamp: string) {\n    return `${this.lockName}:${timestamp}`;\n  }\n\n  private getMigrationFiles = async () => {\n    if (!this.migrationFiles) {\n      this.migrationFiles = await readAndValidateMigrationsFromDirectories(\n        this.options.directories,\n        EXTENSIONS,\n      );\n    }\n    return this.migrationFiles;\n  };\n\n  private async getMigrationForIdentifier(identifier: string): Promise<MigrationFile | null> {\n    const timestamp = identifier.split('_')[0];\n\n    const migrationFiles = await this.getMigrationFiles();\n    const migrationFile = migrationFiles.find((m) => m.timestamp === timestamp);\n    return migrationFile ?? null;\n  }\n\n  /**\n   * Loads the implementation for the migration with the given identifier. The identifier\n   * must start with a 14-character timestamp. It may optionally be followed by\n   * an underscore with additional characters, which are ignored. These should\n   * typically be used to provide a human-readable name for the migration.\n   */\n  private async loadMigrationImplementation(migrationFile: MigrationFile) {\n    // We use dynamic imports to handle both CJS and ESM modules.\n    const migrationModulePath = path.join(migrationFile.directory, migrationFile.filename);\n    const migrationModule = await import(migrationModulePath);\n\n    const migrationImplementation = migrationModule.default as BatchedMigrationImplementation;\n    validateBatchedMigrationImplementation(migrationImplementation);\n    return migrationImplementation;\n  }\n\n  async enqueueBatchedMigration(identifier: string) {\n    const migrationFile = await this.getMigrationForIdentifier(identifier);\n    if (!migrationFile) {\n      throw new Error(`No migration found for identifier ${identifier}`);\n    }\n\n    const migrationImplementation = await this.loadMigrationImplementation(migrationFile);\n    const migrationParameters = await migrationImplementation.getParameters();\n\n    // If `max` is null, that implies that there are no rows to process, so\n    // we can immediately mark the migration as finished.\n    const status: BatchedMigrationStatus =\n      migrationParameters.max === null ? 'succeeded' : 'pending';\n\n    const minValue = BigInt(migrationParameters.min ?? DEFAULT_MIN_VALUE);\n    const maxValue = BigInt(migrationParameters.max ?? minValue);\n    const batchSize = migrationParameters.batchSize ?? DEFAULT_BATCH_SIZE;\n\n    await insertBatchedMigration({\n      project: this.options.project,\n      filename: migrationFile.filename,\n      timestamp: migrationFile.timestamp,\n      batch_size: batchSize,\n      min_value: minValue,\n      max_value: maxValue,\n      status,\n    });\n  }\n\n  async finalizeBatchedMigration(identifier: string, options?: BatchedMigrationFinalizeOptions) {\n    const timestamp = identifier.split('_')[0];\n\n    let migration = await selectBatchedMigrationForTimestamp(this.options.project, timestamp);\n\n    if (migration.status === 'succeeded') return;\n\n    // If the migration isn't already in the finalizing state, mark it as such.\n    if (migration.status !== 'finalizing') {\n      migration = await updateBatchedMigrationStatus(migration.id, 'finalizing');\n    }\n\n    await doWithLock(this.lockNameForTimestamp(timestamp), { autoRenew: true }, async () => {\n      const migrationFile = await this.getMigrationForIdentifier(identifier);\n      if (!migrationFile) {\n        throw new Error(`No migration found for identifier ${identifier}`);\n      }\n      const migrationImplementation = await this.loadMigrationImplementation(migrationFile);\n\n      const runner = new BatchedMigrationRunner(migration, migrationImplementation, {\n        // Always log progress unless explicitly disabled.\n        logProgress: options?.logProgress ?? true,\n      });\n\n      // Because we don't give any arguments to `run()`, it will run until it\n      // has attempted every job.\n      await runner.run();\n    });\n\n    migration = await selectBatchedMigrationForTimestamp(this.options.project, timestamp);\n\n    if (migration.status === 'succeeded') return;\n\n    throw new Error(\n      `Expected batched migration with identifier ${identifier} to be marked as 'succeeded', but it is '${migration.status}'.`,\n    );\n  }\n\n  start(options: BatchedMigrationStartOptions = {}) {\n    if (this.running) {\n      throw new Error('BatchedMigrationsRunner is already running');\n    }\n\n    this.loop(options);\n  }\n\n  async loop({ workDurationMs, sleepDurationMs }: BatchedMigrationStartOptions) {\n    workDurationMs ??= DEFAULT_WORK_DURATION_MS;\n    sleepDurationMs ??= DEFAULT_SLEEP_DURATION_MS;\n\n    this.running = true;\n    while (this.running) {\n      if (this.abortController.signal.aborted) {\n        // We assign this here so that `stop()` can tell when this loop is done\n        // processing jobs.\n        this.running = false;\n        return;\n      }\n\n      let didWork = false;\n      try {\n        didWork = await this.maybePerformWork(workDurationMs);\n      } catch (err) {\n        this.emit('error', err);\n      }\n\n      // If we did work, we'll immediately try again since there's probably more\n      // work to be done. If not, we'll sleep for a while - maybe some more work\n      // will become available!\n      if (!didWork) {\n        // We provide the signal here so that we can more quickly stop things\n        // when we're shutting down.\n        try {\n          await sleep(sleepDurationMs, null, { ref: false, signal: this.abortController.signal });\n        } catch (err) {\n          // We don't care about errors here, they should only ever occur when\n          // the AbortController is aborted. Continue to the next iteration of\n          // the loop so we can shut down.\n          continue;\n        }\n      }\n    }\n  }\n\n  private async getOrStartMigration(): Promise<BatchedMigrationRow | null> {\n    return doWithLock(\n      this.lockName,\n      {\n        // Don't fail if the lock couldn't be acquired immediately.\n        onNotAcquired: () => null,\n      },\n      async () => {\n        let migration = await queryOptionalRow(\n          sql.select_running_migration,\n          { project: this.options.project },\n          BatchedMigrationRowSchema,\n        );\n\n        if (!migration) {\n          migration = await queryOptionalRow(\n            sql.start_next_pending_migration,\n            { project: this.options.project },\n            BatchedMigrationRowSchema,\n          );\n        }\n\n        return migration;\n      },\n    );\n  }\n\n  async maybePerformWork(durationMs: number): Promise<boolean> {\n    const migration = await this.getOrStartMigration();\n    if (!migration) {\n      // No work to do. Handle this case.\n      return false;\n    }\n\n    // This server may not yet know about the current running migration. If\n    // that's the case, we'll just skip it for now.\n    const migrationFile = await this.getMigrationForIdentifier(migration.timestamp);\n    if (!migrationFile) {\n      return false;\n    }\n\n    let didWork = false;\n    await doWithLock(\n      this.lockNameForTimestamp(migrationFile.timestamp),\n      {\n        autoRenew: true,\n        // Do nothing if the lock could not immediately be acquired.\n        onNotAcquired: () => null,\n      },\n      async () => {\n        didWork = true;\n        const migrationImplementation = await this.loadMigrationImplementation(migrationFile);\n\n        const runner = new BatchedMigrationRunner(migration, migrationImplementation);\n\n        try {\n          await runner.run({ signal: this.abortController.signal, durationMs });\n        } catch (err) {\n          this.emit('error', err);\n        }\n      },\n    );\n\n    return didWork;\n  }\n\n  async stop() {\n    this.abortController.abort();\n\n    // Spin until we're no longer running.\n    while (this.running) {\n      await sleep(1000);\n    }\n  }\n}\n\nlet runner: BatchedMigrationsRunner | null = null;\n\nfunction assertRunner(\n  runner: BatchedMigrationsRunner | null,\n): asserts runner is BatchedMigrationsRunner {\n  if (!runner) throw new Error('Batched migrations not initialized');\n}\n\nexport function initBatchedMigrations(options: BatchedMigrationRunnerOptions) {\n  if (runner) throw new Error('Batched migrations already initialized');\n  runner = new BatchedMigrationsRunner(options);\n  return runner;\n}\n\nexport function startBatchedMigrations(options: BatchedMigrationStartOptions = {}) {\n  assertRunner(runner);\n  runner.start(options);\n  return runner;\n}\n\nexport async function stopBatchedMigrations() {\n  assertRunner(runner);\n  await runner.stop();\n  runner = null;\n}\n\n/**\n * Given a batched migration identifier like `20230406184103_migration`,\n * enqueues it for execution by creating a row in the `batched_migrations`\n * table.\n *\n * Despite taking a full identifier, only the timestamp is used to uniquely\n * identify the batched migration. The remaining part is just used to make\n * calls more human-readable.\n *\n * @param identifier The identifier of the batched migration to enqueue.\n */\nexport async function enqueueBatchedMigration(identifier: string) {\n  assertRunner(runner);\n  await runner.enqueueBatchedMigration(identifier);\n}\n\n/**\n * Given a batched migration identifier like `20230406184103_migration`,\n * synchronously runs it to completion. An error will be thrown if the final\n * status of the migration is not `succeeded`.\n *\n * @param identifier The identifier of the batched migration to finalize.\n * @param options Options for finalizing the batched migration.\n */\nexport async function finalizeBatchedMigration(\n  identifier: string,\n  options?: BatchedMigrationFinalizeOptions,\n) {\n  assertRunner(runner);\n  await runner.finalizeBatchedMigration(identifier, options);\n}\n"]}

package/dist/batched-migrations/batched-migrations-runner.test.js

@@ -1,41 +1,13 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || function (mod) {
-    if (mod && mod.__esModule) return mod;
-    var result = {};
-    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
-    __setModuleDefault(result, mod);
-    return result;
-};
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-const chai_1 = __importStar(require("chai"));
-const chai_as_promised_1 = __importDefault(require("chai-as-promised"));
-const node_path_1 = __importDefault(require("node:path"));
-const postgres_1 = require("@prairielearn/postgres");
-const namedLocks = __importStar(require("@prairielearn/named-locks"));
-const index_1 = require("../index");
-const batched_migrations_runner_1 = require("./batched-migrations-runner");
-const batched_migration_1 = require("./batched-migration");
-chai_1.default.use(chai_as_promised_1.default);
-const postgresTestUtils = (0, postgres_1.makePostgresTestUtils)({
+import chai, { assert } from 'chai';
+import chaiAsPromised from 'chai-as-promised';
+import path from 'node:path';
+import { makePostgresTestUtils } from '@prairielearn/postgres';
+import * as namedLocks from '@prairielearn/named-locks';
+import { SCHEMA_MIGRATIONS_PATH, init } from '../index.js';
+import { BatchedMigrationsRunner } from './batched-migrations-runner.js';
+import { selectAllBatchedMigrations } from './batched-migration.js';
+chai.use(chaiAsPromised);
+const postgresTestUtils = makePostgresTestUtils({
     database: 'prairielearn_migrations',
 });
 describe('BatchedMigrationsRunner', () => {
@@ -44,7 +16,7 @@ describe('BatchedMigrationsRunner', () => {
         await namedLocks.init(postgresTestUtils.getPoolConfig(), (err) => {
             throw err;
         });
-        await (0, index_1.init)([index_1.SCHEMA_MIGRATIONS_PATH], 'prairielearn_migrations');
+        await init([SCHEMA_MIGRATIONS_PATH], 'prairielearn_migrations');
     });
     afterEach(async () => {
         await postgresTestUtils.resetDatabase();
@@ -54,63 +26,63 @@ describe('BatchedMigrationsRunner', () => {
         await postgresTestUtils.dropDatabase();
     });
     it('enqueues migrations', async () => {
-        const runner = new batched_migrations_runner_1.BatchedMigrationsRunner({
+        const runner = new BatchedMigrationsRunner({
             project: 'test',
-            directories: [node_path_1.default.join(__dirname, 'fixtures')],
+            directories: [path.join(import.meta.dirname, 'fixtures')],
         });
         await runner.enqueueBatchedMigration('20230406184103_successful_migration');
         await runner.enqueueBatchedMigration('20230406184107_failing_migration');
         await runner.enqueueBatchedMigration('20230407230446_no_rows_migration');
-        const migrations = await (0, batched_migration_1.selectAllBatchedMigrations)('test');
-        chai_1.assert.lengthOf(migrations, 3);
-        chai_1.assert.equal(migrations[0].timestamp, '20230406184103');
-        chai_1.assert.equal(migrations[0].filename, '20230406184103_successful_migration.ts');
-        chai_1.assert.equal(migrations[0].status, 'pending');
-        chai_1.assert.equal(migrations[1].timestamp, '20230406184107');
-        chai_1.assert.equal(migrations[1].filename, '20230406184107_failing_migration.js');
-        chai_1.assert.equal(migrations[1].status, 'pending');
-        chai_1.assert.equal(migrations[2].timestamp, '20230407230446');
-        chai_1.assert.equal(migrations[2].filename, '20230407230446_no_rows_migration.ts');
-        chai_1.assert.equal(migrations[2].status, 'succeeded');
+        const migrations = await selectAllBatchedMigrations('test');
+        assert.lengthOf(migrations, 3);
+        assert.equal(migrations[0].timestamp, '20230406184103');
+        assert.equal(migrations[0].filename, '20230406184103_successful_migration.ts');
+        assert.equal(migrations[0].status, 'pending');
+        assert.equal(migrations[1].timestamp, '20230406184107');
+        assert.equal(migrations[1].filename, '20230406184107_failing_migration.ts');
+        assert.equal(migrations[1].status, 'pending');
+        assert.equal(migrations[2].timestamp, '20230407230446');
+        assert.equal(migrations[2].filename, '20230407230446_no_rows_migration.ts');
+        assert.equal(migrations[2].status, 'succeeded');
     });
     it('safely enqueues migrations multiple times', async () => {
-        const runner = new batched_migrations_runner_1.BatchedMigrationsRunner({
+        const runner = new BatchedMigrationsRunner({
             project: 'test',
-            directories: [node_path_1.default.join(__dirname, 'fixtures')],
+            directories: [path.join(import.meta.dirname, 'fixtures')],
         });
         await runner.enqueueBatchedMigration('20230406184103_successful_migration');
         await runner.enqueueBatchedMigration('20230406184103_successful_migration');
         await runner.enqueueBatchedMigration('20230406184103_successful_migration');
-        const migrations = await (0, batched_migration_1.selectAllBatchedMigrations)('test');
-        chai_1.assert.lengthOf(migrations, 1);
+        const migrations = await selectAllBatchedMigrations('test');
+        assert.lengthOf(migrations, 1);
     });
     it('finalizes a successful migration', async () => {
-        const runner = new batched_migrations_runner_1.BatchedMigrationsRunner({
+        const runner = new BatchedMigrationsRunner({
             project: 'test',
-            directories: [node_path_1.default.join(__dirname, 'fixtures')],
+            directories: [path.join(import.meta.dirname, 'fixtures')],
         });
         await runner.enqueueBatchedMigration('20230406184103_successful_migration');
         await runner.finalizeBatchedMigration('20230406184103_successful_migration', {
             logProgress: false,
         });
-        const migrations = await (0, batched_migration_1.selectAllBatchedMigrations)('test');
-        chai_1.assert.lengthOf(migrations, 1);
-        chai_1.assert.equal(migrations[0].timestamp, '20230406184103');
-        chai_1.assert.equal(migrations[0].status, 'succeeded');
+        const migrations = await selectAllBatchedMigrations('test');
+        assert.lengthOf(migrations, 1);
+        assert.equal(migrations[0].timestamp, '20230406184103');
+        assert.equal(migrations[0].status, 'succeeded');
     });
     it('finalizes a failing migration', async () => {
-        const runner = new batched_migrations_runner_1.BatchedMigrationsRunner({
+        const runner = new BatchedMigrationsRunner({
             project: 'test',
-            directories: [node_path_1.default.join(__dirname, 'fixtures')],
+            directories: [path.join(import.meta.dirname, 'fixtures')],
         });
         await runner.enqueueBatchedMigration('20230406184107_failing_migration');
-        await chai_1.assert.isRejected(runner.finalizeBatchedMigration('20230406184107_failing_migration', {
+        await assert.isRejected(runner.finalizeBatchedMigration('20230406184107_failing_migration', {
             logProgress: false,
         }), "but it is 'failed'");
-        const migrations = await (0, batched_migration_1.selectAllBatchedMigrations)('test');
-        chai_1.assert.lengthOf(migrations, 1);
-        chai_1.assert.equal(migrations[0].timestamp, '20230406184107');
-        chai_1.assert.equal(migrations[0].status, 'failed');
+        const migrations = await selectAllBatchedMigrations('test');
+        assert.lengthOf(migrations, 1);
+        assert.equal(migrations[0].timestamp, '20230406184107');
+        assert.equal(migrations[0].status, 'failed');
     });
 });
 //# sourceMappingURL=batched-migrations-runner.test.js.map