migraguard 0.10.2 → 0.10.3

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() {
@@ -802,6 +811,7 @@ function createDb(config) {
   }
 }
 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 +829,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 +843,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() {
@@ -5369,9 +5380,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 +5397,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 {
@@ -6608,6 +6623,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