migraguard 0.10.2 → 0.10.4

package/dist/cli.js

@@ -520,6 +520,7 @@ var init_dsl = __esm({
 // src/db-mysql.ts
 var ADVISORY_LOCK_KEY = "migraguard-apply";
 var CONNECTION_TIMEOUT_MS = 1e4;
+var SESSION_MAX_EXECUTION_TIME_MS = 3e4;
 var DDL_LOCK_TIMEOUT_SEC = 5;
 var CREATE_TABLE_SQL = `
 CREATE TABLE IF NOT EXISTS schema_migrations (
@@ -563,6 +564,10 @@ var MigraguardDbMysql = class {
       password: this.config.connection.password,
       connectTimeout: CONNECTION_TIMEOUT_MS
     });
+    try {
+      await this.exec(`SET SESSION max_execution_time = ${SESSION_MAX_EXECUTION_TIME_MS}`);
+    } catch {
+    }
   }
   async close() {
     await this.connection?.end();
@@ -581,6 +586,10 @@ var MigraguardDbMysql = class {
       }
     } finally {
       await this.exec(`SET SESSION lock_wait_timeout = DEFAULT`);
+      try {
+        await this.exec(`SET SESSION max_execution_time = ${SESSION_MAX_EXECUTION_TIME_MS}`);
+      } catch {
+      }
     }
   }
   async tableHasAllColumns() {
@@ -801,7 +810,24 @@ function createDb(config) {
       return new MigraguardDb(config);
   }
 }
+async function safeGetAllRecords(db) {
+  try {
+    return await db.getAllRecords();
+  } catch (err) {
+    if (isTableNotFound(err)) return [];
+    throw err;
+  }
+}
+function isTableNotFound(err) {
+  if (!(err instanceof Error)) return false;
+  const code = err["code"];
+  if (code === "42P01") return true;
+  if (code === "ER_NO_SUCH_TABLE") return true;
+  if (err.message.includes("no such table")) return true;
+  return false;
+}
 var CONNECTION_TIMEOUT_MS2 = 1e4;
+var SESSION_STATEMENT_TIMEOUT_MS = 3e4;
 var DDL_LOCK_TIMEOUT = "5s";
 var DDL_STATEMENT_TIMEOUT = "10s";
 var REQUIRED_COLUMNS = ["migration_class", "phase", "group_name", "tag"];
@@ -819,6 +845,7 @@ var MigraguardDb = class {
   }
   async connect() {
     await this.client.connect();
+    await this.client.query(`SET statement_timeout = ${SESSION_STATEMENT_TIMEOUT_MS}`);
   }
   async close() {
     await this.client.end();
@@ -832,7 +859,7 @@ var MigraguardDb = class {
       await this.client.query(ALTER_TABLE_SQL);
     } finally {
       await this.client.query("RESET lock_timeout");
-      await this.client.query("RESET statement_timeout");
+      await this.client.query(`SET statement_timeout = ${SESSION_STATEMENT_TIMEOUT_MS}`);
     }
   }
   async tableHasAllColumns() {
@@ -2490,8 +2517,7 @@ async function commandGroupStatus(config, groupName) {
   const db = createDb(config);
   try {
     await db.connect();
-    await db.ensureTable();
-    const allRecords = await db.getAllRecords();
+    const allRecords = await safeGetAllRecords(db);
     let groups;
     if (groupName) {
       const state = deriveGroupState(allRecords, groupName);
@@ -2607,8 +2633,7 @@ async function commandGate(config, options) {
   const db = createDb(config);
   try {
     await db.connect();
-    await db.ensureTable();
-    const allRecords = await db.getAllRecords();
+    const allRecords = await safeGetAllRecords(db);
     const groupStates = deriveAllGroupStates(allRecords);
     const reasons = [];
     for (const req of contract.required) {
@@ -5102,8 +5127,7 @@ async function commandEditable(config) {
     await db.connect();
     dbConnected = true;
     try {
-      await db.ensureTable();
-      const allRecords = await db.getAllRecords();
+      const allRecords = await safeGetAllRecords(db);
       const recordsByFile = /* @__PURE__ */ new Map();
       for (const r of allRecords) {
         const list = recordsByFile.get(r.fileName) ?? [];
@@ -5155,9 +5179,8 @@ async function commandStatus(config) {
   let groups = [];
   try {
     await db.connect();
-    await db.ensureTable();
     const files = await scanMigrations(config);
-    const allRecords = await db.getAllRecords();
+    const allRecords = await safeGetAllRecords(db);
     const recordsByFile = /* @__PURE__ */ new Map();
     for (const r of allRecords) {
       const list = recordsByFile.get(r.fileName) ?? [];
@@ -5369,9 +5392,11 @@ async function createPgShadow(conn, dbName) {
     port: conn.port,
     database: "postgres",
     user: conn.user,
-    password: conn.password
+    password: conn.password,
+    connectionTimeoutMillis: 1e4
   });
   await client.connect();
+  await client.query("SET statement_timeout = 30000");
   try {
     await client.query(`CREATE DATABASE "${dbName}"`);
   } finally {
@@ -5384,9 +5409,11 @@ async function dropPgShadow(conn, dbName) {
     port: conn.port,
     database: "postgres",
     user: conn.user,
-    password: conn.password
+    password: conn.password,
+    connectionTimeoutMillis: 1e4
   });
   await client.connect();
+  await client.query("SET statement_timeout = 30000");
   try {
     await client.query(`DROP DATABASE IF EXISTS "${dbName}"`);
   } finally {
@@ -5541,8 +5568,7 @@ async function getAppliedFiles(config) {
   const db = createDb(config);
   try {
     await db.connect();
-    await db.ensureTable();
-    const records = await db.getAllRecords();
+    const records = await safeGetAllRecords(db);
     return new Set(
       records.filter((r) => r.status === "applied" || r.status === "skipped").map((r) => r.fileName)
     );
@@ -6608,6 +6634,7 @@ program.command("explain").description("Explain command output in human-readable
     reportFormat: opts.reportFormat
   });
 }));
-program.parse();
+await program.parseAsync();
+process.exit(0);
 //# sourceMappingURL=cli.js.map
 //# sourceMappingURL=cli.js.map