@koralabs/kora-labs-common 6.14.0 → 6.14.2

package/cronLock/cqlCronLock.d.ts

@@ -36,6 +36,8 @@ type Mode = 'serial' | 'localSerial' | 'standDown';
 export declare const resolveMode: (offline: string[], topology: CronLockTopology) => Mode;
 /** Only an UNAVAILABLE response proves that no SERIAL proposal was accepted. */
 export declare const isPeerUnavailableError: (error: unknown) => boolean;
+/** A write timeout leaves an LWT's outcome unknown: the proposal may or may not have been applied. */
+export declare const isWriteTimeoutError: (error: unknown) => boolean;
 /**
  * The directly testable LWT operation. Each invocation owns a unique token, so
  * concurrent ticks on one node cannot both treat the node identity as ownership.

package/cronLock/cqlCronLock.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.release = exports.acquire = exports.acquireWithExecutor = exports.isPeerUnavailableError = exports.resolveMode = exports.parseOffline = exports.leaseMs = exports.resolveTopology = void 0;
+exports.release = exports.acquire = exports.acquireWithExecutor = exports.isWriteTimeoutError = exports.isPeerUnavailableError = exports.resolveMode = exports.parseOffline = exports.leaseMs = exports.resolveTopology = void 0;
 /**
  * Cross-datacenter cron mutex for the self-hosted, multi-DC Scylla topology.
  * (Moved from minting.handle.me src/helpers/cronLock/cqlCronLock.ts @ 6ecb9c3 so every box cron
@@ -114,6 +114,9 @@ const createToken = (topology) => `${topology.nodeCode}:${(0, node_crypto_1.rand
 /** Only an UNAVAILABLE response proves that no SERIAL proposal was accepted. */
 const isPeerUnavailableError = (error) => error instanceof cassandra_driver_1.errors.ResponseError && error.code === cassandra_driver_1.types.responseErrorCodes.unavailableException;
 exports.isPeerUnavailableError = isPeerUnavailableError;
+/** A write timeout leaves an LWT's outcome unknown: the proposal may or may not have been applied. */
+const isWriteTimeoutError = (error) => error instanceof cassandra_driver_1.errors.ResponseError && error.code === cassandra_driver_1.types.responseErrorCodes.writeTimeout;
+exports.isWriteTimeoutError = isWriteTimeoutError;
 /**
  * The directly testable LWT operation. Each invocation owns a unique token, so
  * concurrent ticks on one node cannot both treat the node identity as ownership.
@@ -141,10 +144,32 @@ const acquireWithExecutor = async (executor, name, offline, topology, token = cr
         if (mode === 'serial' && (0, exports.isPeerUnavailableError)(error)) {
             return { status: 'peerUnavailable', token, message };
         }
+        if ((0, exports.isWriteTimeoutError)(error)) {
+            // The LWT timed out: it may have been applied WITH OUR TOKEN, in which case nobody (not even
+            // us) could run until the lease expired. A read at the same serial consistency completes any
+            // in-flight Paxos round, so it tells us the real owner.
+            const owner = await readOwner(executor, key, serialConsistency);
+            if (owner === token)
+                return { status: 'acquired', token };
+            if (owner)
+                return { status: 'held' };
+        }
         return { status: 'unavailable', message };
     }
 };
 exports.acquireWithExecutor = acquireWithExecutor;
+/** The lock's owner as a linearizable (serial) read sees it; undefined when absent or unreadable. */
+const readOwner = async (executor, key, serialConsistency) => {
+    var _a;
+    try {
+        const result = await executor.execute(`SELECT owner FROM ${KEYSPACE}.${LOCK_TABLE} WHERE name = ?`, [key], { prepare: true, consistency: serialConsistency });
+        const owner = (_a = result.rows[0]) === null || _a === void 0 ? void 0 : _a.owner;
+        return typeof owner === 'string' ? owner : undefined;
+    }
+    catch (_b) {
+        return undefined;
+    }
+};
 const acquire = async (name, offline, options = {}) => {
     try {
         const topology = (0, exports.resolveTopology)();

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@koralabs/kora-labs-common",
-    "version": "6.14.0",
+    "version": "6.14.2",
     "description": "Kora Labs Common Utilities",
     "main": "index.js",
     "types": "index.d.ts",