@iterativeflow/mysql 2.0.0-alpha.9 → 2.0.1

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Ahmed Rowaihi
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/dist/index.d.mts

@@ -18,8 +18,18 @@ interface Sql {
   exec(text: string, params?: readonly unknown[]): Promise<WriteResult>;
   tx<T>(fn: (t: Sql) => Promise<T>): Promise<T>;
 }
+/** Options for {@link mysqlPool}. */
+interface MysqlPoolOpts {
+  /**
+   * Set READ COMMITTED per transaction (default `true`). The engine needs READ COMMITTED so a
+   * first-writer-wins checkpoint's re-read sees a concurrent winner's just-committed row. On
+   * PlanetScale/Vitess a per-connection `SET` taints the connection into a reserved pool slot, so
+   * set this `false` there and configure the server's default isolation to READ COMMITTED instead.
+   */
+  setIsolation?: boolean;
+}
 /** Adapt a `mysql2/promise` {@link Pool} to {@link Sql}. `tx` checks out one connection for the unit. */
-declare const mysqlPool: (pool: Pool) => Sql;
+declare const mysqlPool: (pool: Pool, opts?: MysqlPoolOpts) => Sql;
 //#endregion
 //#region src/backend.d.ts
 interface MysqlBackendOpts {

package/dist/index.mjs

@@ -77,11 +77,26 @@ const createMysqlQueue = (sql, t, id) => {
 			await sql.exec(`UPDATE ${t.job} SET lease_token = NULL, lease_expires = NULL, run_at = 0
          WHERE run_id = ? AND lease_token = ? AND lease_expires > ? AND version <> ?`, params);
 		},
-		async depth(now) {
-			return queueDepthOf((await sql.query(`SELECT run_at, lease_expires FROM ${t.job}`)).map((r) => ({
-				runAt: Number(r.run_at),
-				leaseExpires: r.lease_expires == null ? void 0 : Number(r.lease_expires)
-			})), now.getTime());
+		async depth(now, names) {
+			if (names && names.length === 0) return queueDepthOf([], now.getTime());
+			const nowMs = now.getTime();
+			const namePredicate = names ? ` WHERE r.name IN (${names.map(() => "?").join(",")})` : "";
+			const r = (await sql.query(`SELECT SUM(claimable) AS claimable, SUM(leased) AS leased,
+                MIN(CASE WHEN claimable = 1 THEN run_at END) AS oldest
+           FROM (SELECT j.run_at AS run_at,
+                        (j.run_at <= ? AND (j.lease_expires IS NULL OR j.lease_expires <= ?)) AS claimable,
+                        (j.lease_expires > ?) AS leased
+                   FROM ${t.job} j LEFT JOIN ${t.run} r ON r.id = j.run_id${namePredicate}) x`, [
+				nowMs,
+				nowMs,
+				nowMs,
+				...names ?? []
+			]))[0];
+			return {
+				claimable: Number(r?.claimable ?? 0),
+				leased: Number(r?.leased ?? 0),
+				oldestClaimableAgeMs: r?.oldest == null ? null : nowMs - Number(r.oldest)
+			};
 		}
 	};
 };
@@ -536,6 +551,13 @@ const createMysqlStore = (sql, t, id) => {
 		async dueCrons(now, limit) {
 			return (await sql.query(`SELECT * FROM ${t.cron} WHERE next_run_at <= ? ORDER BY next_run_at LIMIT ?`, [now.getTime(), limit])).map(mapCron);
 		},
+		async dueCronCount(now, names) {
+			if (names && names.length === 0) return 0;
+			const namePredicate = names ? ` AND flow_name IN ${inList(names.length)}` : "";
+			const params = names ? [now.getTime(), ...names] : [now.getTime()];
+			const rows = await sql.query(`SELECT count(*) AS n FROM ${t.cron} WHERE next_run_at <= ?${namePredicate}`, params);
+			return Number(rows[0]?.n ?? 0);
+		},
 		async advanceCron(name, expectedNextRunAt, nextRunAt, lastRunAt) {
 			return (await sql.exec(`UPDATE ${t.cron} SET next_run_at = ?, last_run_at = ?
          WHERE name = ? AND next_run_at = ?`, [
@@ -567,6 +589,14 @@ const createMysqlTimer = (sql, t) => {
 				return ids;
 			});
 		},
+		async dueCount(now, names) {
+			if (names && names.length === 0) return 0;
+			const namePredicate = names ? ` AND r.name IN (${names.map(() => "?").join(",")})` : "";
+			const params = names ? [now.getTime(), ...names] : [now.getTime()];
+			const rows = await sql.query(`SELECT count(*) AS n FROM ${t.timer} tm LEFT JOIN ${t.run} r ON r.id = tm.run_id
+         WHERE tm.fire_at <= ?${namePredicate}`, params);
+			return Number(rows[0]?.n ?? 0);
+		},
 		async cancel(runId) {
 			await sql.exec(`DELETE FROM ${t.timer} WHERE run_id = ?`, [runId]);
 		},
@@ -611,17 +641,14 @@ const onConn = (c) => ({
 	tx: (fn) => fn(onConn(c))
 });
 /** Adapt a `mysql2/promise` {@link Pool} to {@link Sql}. `tx` checks out one connection for the unit. */
-const mysqlPool = (pool) => {
-	const configured = /* @__PURE__ */ new WeakSet();
+const mysqlPool = (pool, opts = {}) => {
+	const setIsolation = opts.setIsolation ?? true;
 	return {
 		...bind(pool),
 		async tx(fn) {
 			const c = await pool.getConnection();
 			try {
-				if (!configured.has(c)) {
-					await c.query("SET SESSION TRANSACTION ISOLATION LEVEL READ COMMITTED");
-					configured.add(c);
-				}
+				if (setIsolation) await c.query("SET TRANSACTION ISOLATION LEVEL READ COMMITTED");
 				await c.beginTransaction();
 				const out = await fn(onConn(c));
 				await c.commit();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@iterativeflow/mysql",
-  "version": "2.0.0-alpha.9",
+  "version": "2.0.1",
   "license": "MIT",
   "repository": {
     "type": "git",
@@ -24,13 +24,13 @@
   "publishConfig": {
     "access": "public"
   },
-  "scripts": {
-    "build": "tsdown"
-  },
   "dependencies": {
-    "@iterativeflow/core": "2.0.0-alpha.9"
+    "@iterativeflow/core": "2.0.1"
   },
   "peerDependencies": {
     "mysql2": ">=3.0.0"
+  },
+  "scripts": {
+    "build": "tsdown"
   }
-}
+}