rake-db 2.10.72 → 2.11.0

package/dist/index.mjs

@@ -1394,7 +1394,7 @@ var __spreadValues$3 = (a, b) => {
   return a;
 };
 var __spreadProps$1 = (a, b) => __defProps$1(a, __getOwnPropDescs$1(b));
-const createMigrationInterface = (tx, up, config) => {
+const createMigrationInterface = (tx, up, config, asts) => {
   const adapter = new TransactionAdapter(tx, tx.client, tx.types);
   const { query, arrays } = adapter;
   const log = logParamToLogObject(config.logger || console, config.log);
@@ -1413,7 +1413,7 @@ const createMigrationInterface = (tx, up, config) => {
   for (const key of Object.getOwnPropertyNames(proto)) {
     db[key] = proto[key];
   }
-  db.migratedAsts = [];
+  db.migratedAsts = asts;
   return Object.assign(db, {
     adapter,
     log,
@@ -2198,8 +2198,7 @@ const removeMigratedVersion = async (db, version, config) => {
 const getMigratedVersionsMap = async (db, config) => {
   try {
     const result = await db.arrays(
-      `SELECT *
-       FROM ${quoteWithSchema({ name: config.migrationsTable })}`
+      `SELECT * FROM ${quoteWithSchema({ name: config.migrationsTable })}`
     );
     return Object.fromEntries(result.rows.map((row) => [row[0], true]));
   } catch (err) {
@@ -2227,128 +2226,156 @@ var __spreadValues$2 = (a, b) => {
     }
   return a;
 };
+const RAKE_DB_LOCK_KEY = "8582141715823621641";
+function makeMigrateFn(defaultCount, up, fn) {
+  return async (options, config, args = []) => {
+    const files = await getMigrations(config, up);
+    const count = getCount(args);
+    const conf = prepareConfig(config, args, count);
+    const asts = [];
+    const appCodeUpdaterCache = {};
+    const { appCodeUpdater } = conf;
+    const arrOptions = toArray(options);
+    let localAsts = asts;
+    for (const opts of arrOptions) {
+      const adapter = new Adapter(opts);
+      try {
+        await adapter.transaction(begin, async (trx) => {
+          await trx.query(
+            `SELECT pg_advisory_xact_lock('${RAKE_DB_LOCK_KEY}')`
+          );
+          await fn(
+            trx,
+            conf,
+            files,
+            count != null ? count : defaultCount,
+            localAsts
+          );
+        });
+      } finally {
+        await adapter.close();
+      }
+      localAsts = [];
+    }
+    await runCodeUpdaterAfterAll(
+      arrOptions[0],
+      config,
+      appCodeUpdater,
+      asts,
+      appCodeUpdaterCache
+    );
+  };
+}
+const migrate = makeMigrateFn(
+  Infinity,
+  true,
+  (trx, configs, files, count, asts) => migrateOrRollback(trx, configs, files, count, asts, true)
+);
+const rollback = makeMigrateFn(
+  1,
+  false,
+  (trx, config, files, count, asts) => migrateOrRollback(trx, config, files, count, asts, false)
+);
+const redo = makeMigrateFn(
+  1,
+  false,
+  async (trx, config, files, count, asts) => {
+    await migrateOrRollback(trx, config, files, count, asts, false);
+    files.reverse();
+    await migrateOrRollback(trx, config, files, count, asts, true);
+    files.reverse();
+  }
+);
 const getDb = (adapter) => createDb$1({ adapter });
-const migrateOrRollback = async (options, config, args, up) => {
-  var _a, _b, _c, _d, _e, _f;
-  config = __spreadValues$2({}, config);
-  const files = await getMigrations(config, up);
-  let count = up ? Infinity : 1;
-  let argI = 0;
+const getCount = (args) => {
   const num = args[0] === "all" ? Infinity : parseInt(args[0]);
-  if (!isNaN(num)) {
-    argI++;
-    count = num;
-  }
-  const arg = args[argI];
+  return isNaN(num) ? void 0 : num;
+};
+function prepareConfig(config, args, count) {
+  config = __spreadValues$2({}, config);
+  const i = count === void 0 ? 0 : 1;
+  const arg = args[i];
   if (arg === "--code") {
-    config.useCodeUpdater = args[argI + 1] !== "false";
+    config.useCodeUpdater = args[i + 1] !== "false";
   }
   if (!config.useCodeUpdater)
     delete config.appCodeUpdater;
-  const appCodeUpdaterCache = {};
-  for (const opts of toArray(options)) {
-    const adapter = new Adapter(opts);
-    let db;
-    if (up) {
-      await ((_a = config.beforeMigrate) == null ? void 0 : _a.call(config, db != null ? db : db = getDb(adapter)));
-    } else {
-      await ((_b = config.beforeRollback) == null ? void 0 : _b.call(config, db != null ? db : db = getDb(adapter)));
-    }
-    const migratedVersions = await getMigratedVersionsMap(adapter, config);
-    try {
-      for (const file of files) {
-        if (up && migratedVersions[file.version] || !up && !migratedVersions[file.version]) {
-          continue;
-        }
-        if (count-- <= 0)
-          break;
-        await processMigration(
-          adapter,
-          up,
-          file,
-          config,
-          opts,
-          appCodeUpdaterCache
-        );
-        (_c = config.logger) == null ? void 0 : _c.log(
-          `${up ? "Migrated" : "Rolled back"} ${pathToLog(file.path)}`
-        );
-      }
-      if (up) {
-        await ((_d = config.afterMigrate) == null ? void 0 : _d.call(config, db != null ? db : db = getDb(adapter)));
-      } else {
-        await ((_e = config.afterRollback) == null ? void 0 : _e.call(config, db != null ? db : db = getDb(adapter)));
-      }
-    } finally {
-      await ((_f = config.appCodeUpdater) == null ? void 0 : _f.afterAll({
-        options: opts,
-        basePath: config.basePath,
-        cache: appCodeUpdaterCache,
-        logger: config.logger,
-        // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
-        baseTable: config.baseTable,
-        import: config.import
-      }));
-      await adapter.close();
+  return config;
+}
+const migrateOrRollback = async (trx, config, files, count, asts, up) => {
+  var _a, _b, _c;
+  let db;
+  await ((_a = config[up ? "beforeMigrate" : "beforeRollback"]) == null ? void 0 : _a.call(config, db != null ? db : db = getDb(trx)));
+  const migratedVersions = await getMigratedVersionsMap(trx, config);
+  for (const file of files) {
+    if (up && migratedVersions[file.version] || !up && !migratedVersions[file.version]) {
+      continue;
     }
-    delete config.appCodeUpdater;
+    if (count-- <= 0)
+      break;
+    await runMigration(trx, up, file, config, asts);
+    (_b = config.logger) == null ? void 0 : _b.log(
+      `${up ? "Migrated" : "Rolled back"} ${pathToLog(file.path)}`
+    );
   }
+  await ((_c = config[up ? "afterMigrate" : "afterRollback"]) == null ? void 0 : _c.call(config, db != null ? db : db = getDb(trx)));
 };
-const changeCache = {};
-const begin = {
-  text: "BEGIN",
-  values: emptyArray
-};
-const processMigration = async (db, up, file, config, options, appCodeUpdaterCache) => {
-  var _a;
-  const asts = await db.transaction(begin, async (tx) => {
-    clearChanges();
-    let changes = changeCache[file.path];
-    if (!changes) {
-      const module = await file.load();
-      const exported = (module == null ? void 0 : module.default) && toArray(module.default);
-      if (config.forceDefaultExports && !exported) {
-        throw new RakeDbError(
-          `Missing a default export in ${file.path} migration`
-        );
-      }
-      changes = exported || getCurrentChanges();
-      changeCache[file.path] = changes;
-    }
-    const db2 = createMigrationInterface(tx, up, config);
-    if (changes.length) {
-      const from = up ? 0 : changes.length - 1;
-      const to = up ? changes.length : -1;
-      const step = up ? 1 : -1;
-      for (let i = from; i !== to; i += step) {
-        await changes[i](db2, up);
-      }
-    }
-    await (up ? saveMigratedVersion : removeMigratedVersion)(
-      db2.adapter,
-      file.version,
-      config
-    );
-    return db2.migratedAsts;
-  });
+async function runCodeUpdaterAfterAll(options, config, appCodeUpdater, asts, cache) {
   for (const ast of asts) {
-    await ((_a = config.appCodeUpdater) == null ? void 0 : _a.process({
+    await (appCodeUpdater == null ? void 0 : appCodeUpdater.process({
       ast,
       options,
       basePath: config.basePath,
-      cache: appCodeUpdaterCache,
+      cache,
       logger: config.logger,
       // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
       baseTable: config.baseTable,
       import: config.import
     }));
   }
+  await (appCodeUpdater == null ? void 0 : appCodeUpdater.afterAll({
+    options,
+    basePath: config.basePath,
+    cache,
+    logger: config.logger,
+    // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+    baseTable: config.baseTable,
+    import: config.import
+  }));
+}
+const changeCache = {};
+const begin = {
+  text: "BEGIN",
+  values: emptyArray
 };
-const migrate = (options, config, args = []) => migrateOrRollback(options, config, args, true);
-const rollback = (options, config, args = []) => migrateOrRollback(options, config, args, false);
-const redo = async (options, config, args = []) => {
-  await migrateOrRollback(options, config, args, false);
-  await migrateOrRollback(options, config, args, true);
+const runMigration = async (trx, up, file, config, asts) => {
+  clearChanges();
+  let changes = changeCache[file.path];
+  if (!changes) {
+    const module = await file.load();
+    const exported = (module == null ? void 0 : module.default) && toArray(module.default);
+    if (config.forceDefaultExports && !exported) {
+      throw new RakeDbError(
+        `Missing a default export in ${file.path} migration`
+      );
+    }
+    changes = exported || getCurrentChanges();
+    changeCache[file.path] = changes;
+  }
+  const db = createMigrationInterface(trx, up, config, asts);
+  if (changes.length) {
+    const from = up ? 0 : changes.length - 1;
+    const to = up ? changes.length : -1;
+    const step = up ? 1 : -1;
+    for (let i = from; i !== to; i += step) {
+      await changes[i](db, up);
+    }
+  }
+  await (up ? saveMigratedVersion : removeMigratedVersion)(
+    db.adapter,
+    file.version,
+    config
+  );
 };
 
 const execute = async (options, sql) => {
@@ -3857,5 +3884,5 @@ Migrate and rollback common arguments:
   --code false            do not run code updater
 `;
 
-export { Migration, changeCache, createDb, createMigrationInterface, dropDb, generate, getMigratedVersionsMap, makeFileTimeStamp, migrate, migrateOrRollback, rakeDb, redo, removeMigratedVersion, resetDb, rollback, saveMigratedVersion, writeMigrationFile };
+export { Migration, RAKE_DB_LOCK_KEY, changeCache, createDb, createMigrationInterface, dropDb, generate, getMigratedVersionsMap, makeFileTimeStamp, migrate, migrateOrRollback, rakeDb, redo, removeMigratedVersion, resetDb, rollback, saveMigratedVersion, writeMigrationFile };
 //# sourceMappingURL=index.mjs.map