@tidblink/server 0.3.0-rc.0 → 0.4.0-rc.0

package/dist/backend-pool.js

@@ -0,0 +1,419 @@
+import { BackendTransport, backendConnectionUrl, } from "./backend-connection.js";
+import { BackendConnectionClosedError, BackendRequestError, TiDBLinkError, } from "./errors.js";
+import { sleep } from "./http.js";
+const DEFAULT_IDLE_CLOSE_MS = 5_000;
+/** Attempts of a first attach before its caller sees the failure. */
+const FIRST_ATTACH_ATTEMPTS = 3;
+/** Attaches in flight across the pool; the rest wait their turn. */
+const ATTACH_CONCURRENCY = 32;
+const MAX_BACKOFF_MS = 30_000;
+/** Spread of re-attaches after a lost connection, so that peers do not arrive at once. */
+const RECOVERY_JITTER_MS = 1_000;
+/** How long `close()` waits for detach answers before closing the sockets. */
+const CLOSE_GRACE_MS = 2_000;
+const backoff = (failures) => {
+    const base = Math.min(MAX_BACKOFF_MS, 250 * 2 ** Math.min(failures, 10));
+    return Math.round(base / 2 + (Math.random() * base) / 2);
+};
+/** Failures a later attempt, possibly on another connection, can fix. */
+const isTransient = (error) => (error instanceof BackendConnectionClosedError && error.retryable) ||
+    (error instanceof BackendRequestError && error.code === -32603);
+const isAttachResult = (value) => typeof value === "object" &&
+    value !== null &&
+    Number.isSafeInteger(value.generation) &&
+    value.generation >= 0 &&
+    typeof value.cursor === "string" &&
+    value.cursor !== "";
+/**
+ * The Backend Connections of one OrganizationClient. Connections open on
+ * demand and hold up to the `maxSessions` the core announces; further
+ * attachments open another connection, which the load balancer may place on
+ * another replica. A lost connection's attachments re-attach elsewhere with
+ * their latest parameters; `connection.drain` moves them before the core
+ * closes the connection. Connections that hold nothing close after a short
+ * idle period.
+ */
+export class BackendConnectionPool {
+    #url;
+    #apiKey;
+    #WebSocket;
+    #idleCloseMs;
+    #openTimeoutMs;
+    #slots = new Set();
+    /** Attachments the application keeps, attached or being re-attached. */
+    #wanted = new Map();
+    /** The connection currently routing each attachment. */
+    #where = new Map();
+    #opening;
+    #openFailures = 0;
+    #attaching = 0;
+    #attachQueue = [];
+    #closed = false;
+    constructor(options) {
+        this.#url = backendConnectionUrl(options.baseUrl);
+        this.#apiKey = options.apiKey;
+        this.#WebSocket = options.WebSocket;
+        this.#idleCloseMs = options.idleCloseMs ?? DEFAULT_IDLE_CLOSE_MS;
+        this.#openTimeoutMs = options.openTimeoutMs;
+    }
+    /** Open Backend Connections, for diagnostics and tests. */
+    get connections() {
+        return this.#slots.size;
+    }
+    /**
+     * Attach a Session and keep it attached until `detach()` or until the pool
+     * gives it up through `ended()`. Rejects when the first attach fails; one
+     * attachment per Session is allowed at a time.
+     */
+    async attach(attachment) {
+        if (this.#closed)
+            throw new TiDBLinkError("This OrganizationClient is closed");
+        if (this.#wanted.has(attachment.sessionId))
+            throw new TiDBLinkError(`Session ${attachment.sessionId} is already connected through this OrganizationClient; reuse that connection or close it first`);
+        this.#wanted.set(attachment.sessionId, attachment);
+        try {
+            return await this.#attach(attachment, FIRST_ATTACH_ATTEMPTS);
+        }
+        catch (error) {
+            if (this.#wanted.get(attachment.sessionId) === attachment)
+                this.#wanted.delete(attachment.sessionId);
+            throw error;
+        }
+    }
+    /** Release an attachment. Best effort: a lost connection releases it anyway. */
+    async detach(attachment) {
+        if (this.#wanted.get(attachment.sessionId) !== attachment)
+            return;
+        this.#wanted.delete(attachment.sessionId);
+        const slot = this.#where.get(attachment);
+        this.#where.delete(attachment);
+        if (!slot || slot.held.get(attachment.sessionId) !== attachment)
+            return;
+        this.#release(slot, attachment.sessionId);
+        try {
+            await slot.transport.request("session.detach", { sessionId: attachment.sessionId });
+        }
+        catch {
+            // The attachment lapses with its connection.
+        }
+        finally {
+            this.#maybeIdle(slot);
+        }
+    }
+    /**
+     * Send a request that any connection of the Organization may carry, such as
+     * `tool.result`, retrying on another connection after transient failures
+     * until `retryUntil` (epoch milliseconds). Only idempotent requests belong here.
+     */
+    async request(method, params, options = {}) {
+        const until = options.retryUntil ?? 0;
+        for (let failures = 0;; failures++) {
+            let slot;
+            try {
+                slot = await this.#anySlot();
+            }
+            catch (error) {
+                if (!isTransient(error) || Date.now() >= until)
+                    throw error;
+                await sleep(Math.min(backoff(failures), Math.max(0, until - Date.now())));
+                continue;
+            }
+            this.#cancelIdle(slot);
+            try {
+                return await slot.transport.request(method, params);
+            }
+            catch (error) {
+                if (!isTransient(error) || Date.now() >= until)
+                    throw error;
+                if (error instanceof BackendRequestError)
+                    await sleep(Math.min(backoff(failures), Math.max(0, until - Date.now())));
+            }
+            finally {
+                this.#maybeIdle(slot);
+            }
+        }
+    }
+    /**
+     * End every attachment with reason `closed`, release them and close the
+     * connections. Later attaches are refused.
+     */
+    async close() {
+        if (this.#closed)
+            return;
+        this.#closed = true;
+        const attachments = [...this.#wanted.values()];
+        this.#wanted.clear();
+        const detaches = [];
+        for (const attachment of attachments) {
+            const slot = this.#where.get(attachment);
+            this.#where.delete(attachment);
+            if (slot && slot.held.get(attachment.sessionId) === attachment) {
+                this.#release(slot, attachment.sessionId);
+                detaches.push(slot.transport
+                    .request("session.detach", { sessionId: attachment.sessionId })
+                    .catch(() => undefined));
+            }
+            safely(() => attachment.ended({ reason: "closed" }));
+        }
+        const grace = new AbortController();
+        await Promise.race([
+            Promise.allSettled(detaches),
+            sleep(CLOSE_GRACE_MS, grace.signal).catch(() => undefined),
+        ]);
+        grace.abort();
+        for (const slot of this.#slots) {
+            clearTimeout(slot.idleTimer);
+            slot.transport.close(1000, "client closed");
+        }
+        this.#slots.clear();
+    }
+    async #attach(attachment, attempts) {
+        for (let failures = 0;;) {
+            if (this.#wanted.get(attachment.sessionId) !== attachment)
+                throw new TiDBLinkError("The Session connection was closed while attaching");
+            let slot;
+            try {
+                slot = await this.#reserveSlot();
+                const result = await this.#limited(() => slot.transport.request("session.attach", attachment.attachParams()));
+                if (!isAttachResult(result))
+                    throw new TiDBLinkError("session.attach returned an invalid result");
+                if (this.#wanted.get(attachment.sessionId) !== attachment) {
+                    // Closed while the attach was in flight: release what it took.
+                    slot.transport
+                        .request("session.detach", { sessionId: attachment.sessionId })
+                        .catch(() => undefined);
+                    throw new TiDBLinkError("The Session connection was closed while attaching");
+                }
+                slot.held.set(attachment.sessionId, attachment);
+                this.#where.set(attachment, slot);
+                attachment.attached(result, slot.transport);
+                return result;
+            }
+            catch (error) {
+                failures += 1;
+                if (error instanceof BackendRequestError &&
+                    slot &&
+                    (error.reason === "connection_full" || error.reason === "connection_draining")) {
+                    // Try another connection at once; a core that refuses every one still ends the loop.
+                    if (error.reason === "connection_full")
+                        slot.full = true;
+                    else
+                        slot.draining = true;
+                    if (failures >= attempts)
+                        throw error;
+                    if (failures > 2)
+                        await sleep(backoff(failures));
+                    continue;
+                }
+                if (!isTransient(error) || failures >= attempts)
+                    throw error;
+                await sleep(backoff(failures));
+            }
+            finally {
+                if (slot) {
+                    slot.reserved -= 1;
+                    this.#maybeIdle(slot);
+                }
+            }
+        }
+    }
+    /** Re-attach after a lost or draining connection, until it works or is given up. */
+    async #recover(attachment, jitterMs) {
+        try {
+            if (jitterMs)
+                await sleep(Math.round(Math.random() * jitterMs));
+            await this.#attach(attachment, Number.POSITIVE_INFINITY);
+        }
+        catch (error) {
+            if (this.#wanted.get(attachment.sessionId) !== attachment)
+                return;
+            this.#wanted.delete(attachment.sessionId);
+            safely(() => attachment.ended({ reason: "error", error }));
+        }
+    }
+    async #limited(task) {
+        if (this.#attaching >= ATTACH_CONCURRENCY)
+            await new Promise((resolve) => this.#attachQueue.push(resolve));
+        this.#attaching += 1;
+        try {
+            return await task();
+        }
+        finally {
+            this.#attaching -= 1;
+            this.#attachQueue.shift()?.();
+        }
+    }
+    /** A connection with room for one more attachment, reserved for the caller. */
+    async #reserveSlot() {
+        for (;;) {
+            for (const slot of this.#slots)
+                if (slot.transport.open &&
+                    !slot.draining &&
+                    !slot.full &&
+                    slot.held.size + slot.reserved < slot.transport.hello.maxSessions) {
+                    // Reserve before yielding, so that concurrent attaches count each other.
+                    slot.reserved += 1;
+                    this.#cancelIdle(slot);
+                    return slot;
+                }
+            await this.#openShared();
+        }
+    }
+    async #anySlot() {
+        let fallback;
+        for (const slot of this.#slots) {
+            if (!slot.transport.open)
+                continue;
+            if (!slot.draining)
+                return slot;
+            fallback ??= slot;
+        }
+        return fallback ?? (await this.#openShared());
+    }
+    /** One connection opens at a time; concurrent callers share it. */
+    #openShared() {
+        if (this.#closed)
+            return Promise.reject(new TiDBLinkError("This OrganizationClient is closed"));
+        this.#opening ??= this.#open().finally(() => {
+            this.#opening = undefined;
+        });
+        return this.#opening;
+    }
+    async #open() {
+        if (this.#openFailures)
+            await sleep(backoff(this.#openFailures));
+        let slot;
+        let transport;
+        try {
+            transport = await BackendTransport.open({
+                url: this.#url,
+                apiKey: this.#apiKey,
+                WebSocket: this.#WebSocket,
+                ...(this.#openTimeoutMs === undefined ? {} : { openTimeoutMs: this.#openTimeoutMs }),
+                onNotification: (method, params) => {
+                    if (slot)
+                        this.#notify(slot, method, params);
+                },
+            });
+        }
+        catch (error) {
+            this.#openFailures += 1;
+            throw error;
+        }
+        this.#openFailures = 0;
+        slot = {
+            transport,
+            held: new Map(),
+            reserved: 0,
+            full: false,
+            draining: false,
+            idleTimer: undefined,
+        };
+        if (this.#closed) {
+            transport.close(1000, "client closed");
+            throw new TiDBLinkError("This OrganizationClient is closed");
+        }
+        this.#slots.add(slot);
+        const opened = slot;
+        void transport.closed.then((error) => this.#lost(opened, error));
+        this.#maybeIdle(slot);
+        return slot;
+    }
+    #notify(slot, method, params) {
+        if (method === "connection.drain") {
+            const closeAfterMs = Number(params.closeAfterMs);
+            this.#drain(slot, Number.isFinite(closeAfterMs) ? closeAfterMs : 0);
+            return;
+        }
+        const sessionId = params.sessionId;
+        if (typeof sessionId !== "string")
+            return;
+        const attachment = slot.held.get(sessionId);
+        if (!attachment)
+            return;
+        if (method === "session.detached") {
+            this.#release(slot, sessionId);
+            this.#where.delete(attachment);
+            if (this.#wanted.get(sessionId) === attachment)
+                this.#wanted.delete(sessionId);
+            safely(() => attachment.ended({
+                reason: params.reason === "session_deleted" ? "session_deleted" : "superseded",
+            }));
+            this.#maybeIdle(slot);
+            return;
+        }
+        safely(() => attachment.notify(method, params, slot.transport));
+    }
+    /** Move every attachment to another connection, then close this one. */
+    #drain(slot, closeAfterMs) {
+        if (slot.draining && slot.held.size === 0)
+            return;
+        slot.draining = true;
+        const deadline = Date.now() + closeAfterMs;
+        const moving = [...slot.held.values()];
+        // Stop routing now: each re-attach resumes from what its Session processed.
+        slot.held.clear();
+        const moves = moving.map((attachment) => {
+            this.#where.delete(attachment);
+            return this.#recover(attachment, 0);
+        });
+        void (async () => {
+            await Promise.allSettled(moves);
+            // Let claims and results in flight finish, then close before the core does.
+            while (slot.transport.open && slot.transport.pendingRequests > 0 && Date.now() < deadline)
+                await sleep(50);
+            slot.transport.close(1000, "drained");
+        })();
+    }
+    #lost(slot, error) {
+        this.#slots.delete(slot);
+        clearTimeout(slot.idleTimer);
+        const held = [...slot.held.values()];
+        slot.held.clear();
+        for (const attachment of held) {
+            this.#where.delete(attachment);
+            if (this.#wanted.get(attachment.sessionId) !== attachment)
+                continue;
+            if (error.retryable && !this.#closed) {
+                void this.#recover(attachment, RECOVERY_JITTER_MS);
+            }
+            else {
+                this.#wanted.delete(attachment.sessionId);
+                safely(() => attachment.ended({ reason: "error", error }));
+            }
+        }
+    }
+    #release(slot, sessionId) {
+        slot.held.delete(sessionId);
+        slot.full = false;
+    }
+    #cancelIdle(slot) {
+        clearTimeout(slot.idleTimer);
+        slot.idleTimer = undefined;
+    }
+    #maybeIdle(slot) {
+        if (slot.idleTimer || !this.#slots.has(slot) || slot.held.size || slot.reserved)
+            return;
+        slot.idleTimer = setTimeout(() => {
+            slot.idleTimer = undefined;
+            if (!this.#slots.has(slot) || slot.held.size || slot.reserved)
+                return;
+            // Claims and results may still be in flight; look again later.
+            if (slot.transport.pendingRequests)
+                return this.#maybeIdle(slot);
+            this.#slots.delete(slot);
+            slot.transport.close(1000, "idle");
+        }, this.#idleCloseMs);
+    }
+}
+function safely(callback) {
+    try {
+        callback();
+    }
+    catch (error) {
+        queueMicrotask(() => {
+            throw error;
+        });
+    }
+}
+//# sourceMappingURL=backend-pool.js.map

package/dist/backend-pool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"backend-pool.js","sourceRoot":"","sources":["../src/backend-pool.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,gBAAgB,EAChB,oBAAoB,GAErB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,4BAA4B,EAC5B,mBAAmB,EACnB,aAAa,GACd,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,KAAK,EAAE,MAAM,WAAW,CAAC;AAkElC,MAAM,qBAAqB,GAAG,KAAK,CAAC;AACpC,qEAAqE;AACrE,MAAM,qBAAqB,GAAG,CAAC,CAAC;AAChC,oEAAoE;AACpE,MAAM,kBAAkB,GAAG,EAAE,CAAC;AAC9B,MAAM,cAAc,GAAG,MAAM,CAAC;AAC9B,0FAA0F;AAC1F,MAAM,kBAAkB,GAAG,KAAK,CAAC;AACjC,8EAA8E;AAC9E,MAAM,cAAc,GAAG,KAAK,CAAC;AAE7B,MAAM,OAAO,GAAG,CAAC,QAAgB,EAAU,EAAE;IAC3C,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,GAAG,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC;IACzE,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AAC3D,CAAC,CAAC;AAEF,yEAAyE;AACzE,MAAM,WAAW,GAAG,CAAC,KAAc,EAAW,EAAE,CAC9C,CAAC,KAAK,YAAY,4BAA4B,IAAI,KAAK,CAAC,SAAS,CAAC;IAClE,CAAC,KAAK,YAAY,mBAAmB,IAAI,KAAK,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC;AAElE,MAAM,cAAc,GAAG,CAAC,KAAc,EAAyB,EAAE,CAC/D,OAAO,KAAK,KAAK,QAAQ;IACzB,KAAK,KAAK,IAAI;IACd,MAAM,CAAC,aAAa,CAAE,KAAsB,CAAC,UAAU,CAAC;IACvD,KAAsB,CAAC,UAAU,IAAI,CAAC;IACvC,OAAQ,KAAsB,CAAC,MAAM,KAAK,QAAQ;IACjD,KAAsB,CAAC,MAAM,KAAK,EAAE,CAAC;AAExC;;;;;;;;GAQG;AACH,MAAM,OAAO,qBAAqB;IACvB,IAAI,CAAS;IACb,OAAO,CAAS;IAChB,UAAU,CAAmC;IAC7C,YAAY,CAAS;IACrB,cAAc,CAAqB;IACnC,MAAM,GAAG,IAAI,GAAG,EAAQ,CAAC;IAClC,wEAAwE;IAC/D,OAAO,GAAG,IAAI,GAAG,EAA0B,CAAC;IACrD,wDAAwD;IAC/C,MAAM,GAAG,IAAI,GAAG,EAAwB,CAAC;IAClD,QAAQ,CAA4B;IACpC,aAAa,GAAG,CAAC,CAAC;IAClB,UAAU,GAAG,CAAC,CAAC;IACN,YAAY,GAAmB,EAAE,CAAC;IAC3C,OAAO,GAAG,KAAK,CAAC;IAEhB,YAAY,OAAqC;QAC/C,IAAI,CAAC,IAAI,GAAG,oBAAoB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAClD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC;QAC9B,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC;QACpC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,WAAW,IAAI,qBAAqB,CAAC;QACjE,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,aAAa,CAAC;IAC9C,CAAC;IAED,2DAA2D;IAC3D,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;IAC1B,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,MAAM,CAAC,UAA0B;QACrC,IAAI,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,aAAa,CAAC,mCAAmC,CAAC,CAAC;QAC/E,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC;YACxC,MAAM,IAAI,aAAa,CACrB,WAAW,UAAU,CAAC,SAAS,gGAAgG,CAChI,CAAC;QACJ,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;QACnD,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,qBAAqB,CAAC,CAAC;QAC/D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,KAAK,UAAU;gBACvD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YAC5C,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,gFAAgF;IAChF,KAAK,CAAC,MAAM,CAAC,UAA0B;QACrC,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,KAAK,UAAU;YAAE,OAAO;QAClE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACzC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAC/B,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,KAAK,UAAU;YAAE,OAAO;QACxE,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC;QAC1C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,gBAAgB,EAAE,EAAE,SAAS,EAAE,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC;QACtF,CAAC;QAAC,MAAM,CAAC;YACP,6CAA6C;QAC/C,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,OAAO,CACX,MAAc,EACd,MAA+B,EAC/B,UAAmC,EAAE;QAErC,MAAM,KAAK,GAAG,OAAO,CAAC,UAAU,IAAI,CAAC,CAAC;QACtC,KAAK,IAAI,QAAQ,GAAG,CAAC,GAAI,QAAQ,EAAE,EAAE,CAAC;YACpC,IAAI,IAAU,CAAC;YACf,IAAI,CAAC;gBACH,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC/B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,KAAK;oBAAE,MAAM,KAAK,CAAC;gBAC5D,MAAM,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC1E,SAAS;YACX,CAAC;YACD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACvB,IAAI,CAAC;gBACH,OAAO,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAAI,MAAM,EAAE,MAAM,CAAC,CAAC;YACzD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,KAAK;oBAAE,MAAM,KAAK,CAAC;gBAC5D,IAAI,KAAK,YAAY,mBAAmB;oBACtC,MAAM,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;YAC9E,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,MAAM,WAAW,GAAG,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,MAAM,QAAQ,GAAuB,EAAE,CAAC;QACxC,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACzC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAC/B,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,KAAK,UAAU,EAAE,CAAC;gBAC/D,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC;gBAC1C,QAAQ,CAAC,IAAI,CACX,IAAI,CAAC,SAAS;qBACX,OAAO,CAAC,gBAAgB,EAAE,EAAE,SAAS,EAAE,UAAU,CAAC,SAAS,EAAE,CAAC;qBAC9D,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAC1B,CAAC;YACJ,CAAC;YACD,MAAM,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QACvD,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;QACpC,MAAM,OAAO,CAAC,IAAI,CAAC;YACjB,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;YAC5B,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;SAC3D,CAAC,CAAC;QACH,KAAK,CAAC,KAAK,EAAE,CAAC;QACd,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAC/B,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC7B,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;QAC9C,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,UAA0B,EAAE,QAAgB;QACxD,KAAK,IAAI,QAAQ,GAAG,CAAC,IAAM,CAAC;YAC1B,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,KAAK,UAAU;gBACvD,MAAM,IAAI,aAAa,CAAC,mDAAmD,CAAC,CAAC;YAC/E,IAAI,IAAsB,CAAC;YAC3B,IAAI,CAAC;gBACH,IAAI,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;gBACjC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,CACtC,IAAK,CAAC,SAAS,CAAC,OAAO,CAAC,gBAAgB,EAAE,UAAU,CAAC,YAAY,EAAE,CAAC,CACrE,CAAC;gBACF,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC;oBACzB,MAAM,IAAI,aAAa,CAAC,2CAA2C,CAAC,CAAC;gBACvE,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,KAAK,UAAU,EAAE,CAAC;oBAC1D,+DAA+D;oBAC/D,IAAI,CAAC,SAAS;yBACX,OAAO,CAAC,gBAAgB,EAAE,EAAE,SAAS,EAAE,UAAU,CAAC,SAAS,EAAE,CAAC;yBAC9D,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;oBAC1B,MAAM,IAAI,aAAa,CAAC,mDAAmD,CAAC,CAAC;gBAC/E,CAAC;gBACD,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;gBAChD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;gBAClC,UAAU,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC5C,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,QAAQ,IAAI,CAAC,CAAC;gBACd,IACE,KAAK,YAAY,mBAAmB;oBACpC,IAAI;oBACJ,CAAC,KAAK,CAAC,MAAM,KAAK,iBAAiB,IAAI,KAAK,CAAC,MAAM,KAAK,qBAAqB,CAAC,EAC9E,CAAC;oBACD,qFAAqF;oBACrF,IAAI,KAAK,CAAC,MAAM,KAAK,iBAAiB;wBAAE,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;;wBACpD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;oBAC1B,IAAI,QAAQ,IAAI,QAAQ;wBAAE,MAAM,KAAK,CAAC;oBACtC,IAAI,QAAQ,GAAG,CAAC;wBAAE,MAAM,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;oBACjD,SAAS;gBACX,CAAC;gBACD,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,QAAQ,IAAI,QAAQ;oBAAE,MAAM,KAAK,CAAC;gBAC7D,MAAM,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;YACjC,CAAC;oBAAS,CAAC;gBACT,IAAI,IAAI,EAAE,CAAC;oBACT,IAAI,CAAC,QAAQ,IAAI,CAAC,CAAC;oBACnB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;gBACxB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,oFAAoF;IACpF,KAAK,CAAC,QAAQ,CAAC,UAA0B,EAAE,QAAgB;QACzD,IAAI,CAAC;YACH,IAAI,QAAQ;gBAAE,MAAM,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,QAAQ,CAAC,CAAC,CAAC;YAChE,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,CAAC,iBAAiB,CAAC,CAAC;QAC3D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,KAAK,UAAU;gBAAE,OAAO;YAClE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YAC1C,MAAM,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ,CAAI,IAAsB;QACtC,IAAI,IAAI,CAAC,UAAU,IAAI,kBAAkB;YACvC,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;QACxE,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC;QACrB,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,EAAE,CAAC;QACtB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC;YACrB,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,EAAE,EAAE,CAAC;QAChC,CAAC;IACH,CAAC;IAED,+EAA+E;IAC/E,KAAK,CAAC,YAAY;QAChB,SAAS,CAAC;YACR,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,MAAM;gBAC5B,IACE,IAAI,CAAC,SAAS,CAAC,IAAI;oBACnB,CAAC,IAAI,CAAC,QAAQ;oBACd,CAAC,IAAI,CAAC,IAAI;oBACV,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,WAAW,EACjE,CAAC;oBACD,yEAAyE;oBACzE,IAAI,CAAC,QAAQ,IAAI,CAAC,CAAC;oBACnB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;oBACvB,OAAO,IAAI,CAAC;gBACd,CAAC;YACH,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,IAAI,QAA0B,CAAC;QAC/B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAC/B,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI;gBAAE,SAAS;YACnC,IAAI,CAAC,IAAI,CAAC,QAAQ;gBAAE,OAAO,IAAI,CAAC;YAChC,QAAQ,KAAK,IAAI,CAAC;QACpB,CAAC;QACD,OAAO,QAAQ,IAAI,CAAC,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IAChD,CAAC;IAED,mEAAmE;IACnE,WAAW;QACT,IAAI,IAAI,CAAC,OAAO;YACd,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,aAAa,CAAC,mCAAmC,CAAC,CAAC,CAAC;QAChF,IAAI,CAAC,QAAQ,KAAK,IAAI,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;YAC1C,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAC;QAC5B,CAAC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,aAAa;YAAE,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;QACjE,IAAI,IAAsB,CAAC;QAC3B,IAAI,SAA2B,CAAC;QAChC,IAAI,CAAC;YACH,SAAS,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC;gBACtC,GAAG,EAAE,IAAI,CAAC,IAAI;gBACd,MAAM,EAAE,IAAI,CAAC,OAAO;gBACpB,SAAS,EAAE,IAAI,CAAC,UAAU;gBAC1B,GAAG,CAAC,IAAI,CAAC,cAAc,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC;gBACpF,cAAc,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE;oBACjC,IAAI,IAAI;wBAAE,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;gBAC/C,CAAC;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,aAAa,IAAI,CAAC,CAAC;YACxB,MAAM,KAAK,CAAC;QACd,CAAC;QACD,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC;QACvB,IAAI,GAAG;YACL,SAAS;YACT,IAAI,EAAE,IAAI,GAAG,EAAE;YACf,QAAQ,EAAE,CAAC;YACX,IAAI,EAAE,KAAK;YACX,QAAQ,EAAE,KAAK;YACf,SAAS,EAAE,SAAS;SACrB,CAAC;QACF,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;YACvC,MAAM,IAAI,aAAa,CAAC,mCAAmC,CAAC,CAAC;QAC/D,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACtB,MAAM,MAAM,GAAG,IAAI,CAAC;QACpB,KAAK,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;QACjE,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,CAAC,IAAU,EAAE,MAAc,EAAE,MAA+B;QACjE,IAAI,MAAM,KAAK,kBAAkB,EAAE,CAAC;YAClC,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;YACjD,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACpE,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QACnC,IAAI,OAAO,SAAS,KAAK,QAAQ;YAAE,OAAO;QAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC5C,IAAI,CAAC,UAAU;YAAE,OAAO;QACxB,IAAI,MAAM,KAAK,kBAAkB,EAAE,CAAC;YAClC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;YAC/B,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAC/B,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,UAAU;gBAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAC/E,MAAM,CAAC,GAAG,EAAE,CACV,UAAU,CAAC,KAAK,CAAC;gBACf,MAAM,EAAE,MAAM,CAAC,MAAM,KAAK,iBAAiB,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,YAAY;aAC/E,CAAC,CACH,CAAC;YACF,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YACtB,OAAO;QACT,CAAC;QACD,MAAM,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;IAClE,CAAC;IAED,wEAAwE;IACxE,MAAM,CAAC,IAAU,EAAE,YAAoB;QACrC,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC;YAAE,OAAO;QAClD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,YAAY,CAAC;QAC3C,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACvC,4EAA4E;QAC5E,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QAClB,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;YACtC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAC/B,OAAO,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,KAAK,IAAI,EAAE;YACf,MAAM,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAChC,4EAA4E;YAC5E,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,eAAe,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ;gBACvF,MAAM,KAAK,CAAC,EAAE,CAAC,CAAC;YAClB,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QACxC,CAAC,CAAC,EAAE,CAAC;IACP,CAAC;IAED,KAAK,CAAC,IAAU,EAAE,KAAmC;QACnD,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACzB,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC7B,MAAM,IAAI,GAAG,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACrC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QAClB,KAAK,MAAM,UAAU,IAAI,IAAI,EAAE,CAAC;YAC9B,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAC/B,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,KAAK,UAAU;gBAAE,SAAS;YACpE,IAAI,KAAK,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACrC,KAAK,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC;YACrD,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;gBAC1C,MAAM,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,IAAU,EAAE,SAAiB;QACpC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC5B,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC;IACpB,CAAC;IAED,WAAW,CAAC,IAAU;QACpB,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,UAAU,CAAC,IAAU;QACnB,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ;YAAE,OAAO;QACxF,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;YAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;YAC3B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ;gBAAE,OAAO;YACtE,+DAA+D;YAC/D,IAAI,IAAI,CAAC,SAAS,CAAC,eAAe;gBAAE,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YACjE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACzB,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACrC,CAAC,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;IACxB,CAAC;CACF;AAED,SAAS,MAAM,CAAC,QAAoB;IAClC,IAAI,CAAC;QACH,QAAQ,EAAE,CAAC;IACb,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,cAAc,CAAC,GAAG,EAAE;YAClB,MAAM,KAAK,CAAC;QACd,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC"}

package/dist/client-tool-executor.d.ts

@@ -0,0 +1,31 @@
+import type { BackendConnectionPool, BackendLink } from "./backend-pool.js";
+import { TiDBLinkError } from "./errors.js";
+import type { ClientToolHandlers } from "./session-connection.js";
+/**
+ * Raised through `onError` when the core did not accept a Client Tool
+ * result. `reason` is the core's: `call_settled` (the call already had an
+ * outcome, for example its deadline passed), `not_claimed` or
+ * `call_not_found`.
+ */
+export declare class ClientToolResultRefusedError extends TiDBLinkError {
+    readonly callId: string;
+    readonly reason: string;
+    constructor(callId: string, reason: string);
+}
+/**
+ * Executes the Client Tool calls an executing Session connection receives:
+ * claim first, run only after `claimed: true`, then submit the result over any
+ * Backend Connection until the call's deadline. Each callId runs at most once.
+ */
+export declare class ClientToolExecutor {
+    #private;
+    constructor(input: {
+        pool: BackendConnectionPool;
+        sessionId: string;
+        handlers: ClientToolHandlers;
+        report: (error: unknown) => void;
+    });
+    /** Handle one `tool.call` from the connection that delivered it. */
+    receive(params: Record<string, unknown>, link: BackendLink): void;
+}
+//# sourceMappingURL=client-tool-executor.d.ts.map

package/dist/client-tool-executor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"client-tool-executor.d.ts","sourceRoot":"","sources":["../src/client-tool-executor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAC5E,OAAO,EAGL,aAAa,EACd,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EAEV,kBAAkB,EACnB,MAAM,yBAAyB,CAAC;AAkCjC;;;;;GAKG;AACH,qBAAa,4BAA6B,SAAQ,aAAa;IAE3D,QAAQ,CAAC,MAAM,EAAE,MAAM;IACvB,QAAQ,CAAC,MAAM,EAAE,MAAM;gBADd,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,MAAM;CAI1B;AA4CD;;;;GAIG;AACH,qBAAa,kBAAkB;;gBAOjB,KAAK,EAAE;QACjB,IAAI,EAAE,qBAAqB,CAAC;QAC5B,SAAS,EAAE,MAAM,CAAC;QAClB,QAAQ,EAAE,kBAAkB,CAAC;QAC7B,MAAM,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;KAClC;IAOD,oEAAoE;IACpE,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,GAAG,IAAI;CA6HlE"}

package/dist/client-tool-executor.js

@@ -0,0 +1,190 @@
+import { BackendConnectionClosedError, BackendRequestError, TiDBLinkError, } from "./errors.js";
+const OUTPUT_MAX = 65_536;
+const ERROR_MAX = 4_096;
+/** How long a settled callId is remembered to ignore late duplicates. */
+const SETTLED_MEMORY_MS = 60_000;
+/**
+ * Raised through `onError` when the core did not accept a Client Tool
+ * result. `reason` is the core's: `call_settled` (the call already had an
+ * outcome, for example its deadline passed), `not_claimed` or
+ * `call_not_found`.
+ */
+export class ClientToolResultRefusedError extends TiDBLinkError {
+    callId;
+    reason;
+    constructor(callId, reason) {
+        super(`The core did not accept the result of Client Tool call ${callId}: ${reason}`);
+        this.callId = callId;
+        this.reason = reason;
+    }
+}
+const isString = (value) => typeof value === "string" && value !== "";
+const isObject = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
+const parseCall = (params) => isString(params.callId) &&
+    isString(params.sessionId) &&
+    isString(params.userId) &&
+    isString(params.turnId) &&
+    isString(params.name) &&
+    isObject(params.arguments) &&
+    isString(params.claimBy) &&
+    isString(params.deadlineAt) &&
+    Number.isFinite(Date.parse(params.deadlineAt))
+    ? params
+    : undefined;
+const truncate = (text, max) => text.length <= max ? text : `${text.slice(0, max - 1)}…`;
+/** The text the model receives for a handler's return value. */
+const outputOf = (value) => {
+    let output;
+    try {
+        output = typeof value === "string" ? value : value === undefined ? "" : JSON.stringify(value);
+    }
+    catch (cause) {
+        return { error: { message: `The tool returned a value that is not JSON: ${String(cause)}` } };
+    }
+    if (output === undefined)
+        return { error: { message: "The tool returned a value that is not JSON" } };
+    if (output.length > OUTPUT_MAX)
+        return {
+            error: { message: `The tool output exceeds ${OUTPUT_MAX} characters` },
+        };
+    return { output };
+};
+const errorOf = (error) => {
+    const message = error instanceof Error ? error.message : String(error);
+    return { error: { message: truncate(message || "The tool failed", ERROR_MAX) } };
+};
+/**
+ * Executes the Client Tool calls an executing Session connection receives:
+ * claim first, run only after `claimed: true`, then submit the result over any
+ * Backend Connection until the call's deadline. Each callId runs at most once.
+ */
+export class ClientToolExecutor {
+    #pool;
+    #sessionId;
+    #handlers;
+    #report;
+    #calls = new Map();
+    constructor(input) {
+        this.#pool = input.pool;
+        this.#sessionId = input.sessionId;
+        this.#handlers = input.handlers;
+        this.#report = input.report;
+    }
+    /** Handle one `tool.call` from the connection that delivered it. */
+    receive(params, link) {
+        const call = parseCall(params);
+        if (!call || call.sessionId !== this.#sessionId) {
+            this.#report(new TiDBLinkError("Received an invalid Client Tool call"));
+            return;
+        }
+        // The core may send a call again after a reconnect; it runs once.
+        if (this.#calls.has(call.callId))
+            return;
+        // Never sent: the core sends it again to the next attachment while unclaimed.
+        if (!link.open)
+            return;
+        this.#calls.set(call.callId, "claiming");
+        void this.#claimAndRun(call, link);
+    }
+    async #claimAndRun(call, link) {
+        let answer;
+        try {
+            answer = await link.request("tool.claim", { callId: call.callId });
+        }
+        catch (error) {
+            if (error instanceof BackendConnectionClosedError ||
+                (error instanceof BackendRequestError && error.code === -32603)) {
+                // The claim may have been granted; the call must not run, so close it as failed.
+                await this.#submit(call, {
+                    error: {
+                        message: "The Application Server lost the claim's answer and did not execute the call",
+                    },
+                });
+                return;
+            }
+            this.#calls.set(call.callId, "settled");
+            this.#forget(call);
+            this.#report(error);
+            return;
+        }
+        if (!answer || answer.claimed !== true) {
+            if (answer?.reason === "superseded") {
+                // Another generation holds the Session; this process may receive it again there.
+                this.#calls.delete(call.callId);
+            }
+            else {
+                this.#calls.set(call.callId, "settled");
+                this.#forget(call);
+            }
+            return;
+        }
+        this.#calls.set(call.callId, "running");
+        await this.#submit(call, await this.#execute(call));
+    }
+    async #execute(call) {
+        const handler = Object.hasOwn(this.#handlers, call.name)
+            ? this.#handlers[call.name]
+            : undefined;
+        if (!handler)
+            return {
+                error: {
+                    message: `Client Tool ${call.name} has no handler in this Application Server`,
+                },
+            };
+        const deadlineAt = new Date(call.deadlineAt);
+        const controller = new AbortController();
+        let timer;
+        const expired = new Promise((resolve) => {
+            timer = setTimeout(() => {
+                controller.abort(new TiDBLinkError("The Client Tool call reached its deadline"));
+                resolve({ error: { message: "The tool did not finish before its deadline" } });
+            }, Math.max(0, deadlineAt.getTime() - Date.now()));
+        });
+        const context = {
+            callId: call.callId,
+            sessionId: call.sessionId,
+            userId: call.userId,
+            turnId: call.turnId,
+            name: call.name,
+            deadlineAt,
+            signal: controller.signal,
+        };
+        const run = (async () => {
+            try {
+                return outputOf(await handler.run(call.arguments, context));
+            }
+            catch (error) {
+                return errorOf(error);
+            }
+        })();
+        try {
+            return await Promise.race([run, expired]);
+        }
+        finally {
+            clearTimeout(timer);
+        }
+    }
+    async #submit(call, result) {
+        try {
+            const answer = await this.#pool.request("tool.result", { callId: call.callId, ...result }, { retryUntil: Date.parse(call.deadlineAt) });
+            if (answer?.accepted !== true) {
+                if (answer?.reason === "not_claimed") {
+                    // The lost claim never took effect: the call may arrive again and run then.
+                    this.#calls.delete(call.callId);
+                    return;
+                }
+                this.#report(new ClientToolResultRefusedError(call.callId, answer?.reason ?? "unknown"));
+            }
+        }
+        catch (error) {
+            this.#report(error);
+        }
+        this.#calls.set(call.callId, "settled");
+        this.#forget(call);
+    }
+    #forget(call) {
+        const timer = setTimeout(() => this.#calls.delete(call.callId), Math.max(0, Date.parse(call.deadlineAt) - Date.now()) + SETTLED_MEMORY_MS);
+        timer.unref?.();
+    }
+}
+//# sourceMappingURL=client-tool-executor.js.map