@aztec/prover-client 0.73.0 → 0.75.0-commit.c03ba01a2a4122e43e90d5133ba017e54b90e9d2

package/dest/prover-agent/memory-proving-queue.js

@@ -1,5 +1,10 @@
-import { __esDecorate, __runInitializers } from "tslib";
-import { ProvingRequestType, } from '@aztec/circuit-types';
+function _ts_decorate(decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for(var i = decorators.length - 1; i >= 0; i--)if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+}
+import { ProvingRequestType } from '@aztec/circuit-types';
 import { randomBytes } from '@aztec/foundation/crypto';
 import { AbortError, TimeoutError } from '@aztec/foundation/error';
 import { createLogger } from '@aztec/foundation/log';
@@ -9,252 +14,236 @@ import { trackSpan } from '@aztec/telemetry-client';
 import { InlineProofStore } from '../proving_broker/proof_store/index.js';
 import { ProvingQueueMetrics } from './queue_metrics.js';
 const MAX_RETRIES = 3;
-const defaultIdGenerator = () => randomBytes(4).toString('hex');
-const defaultTimeSource = () => Date.now();
+const defaultIdGenerator = ()=>randomBytes(4).toString('hex');
+const defaultTimeSource = ()=>Date.now();
 /**
  * A helper class that sits in between services that need proofs created and agents that can create them.
  * The queue accumulates jobs and provides them to agents prioritized by block number.
- */
-let MemoryProvingQueue = (() => {
-    var _a;
-    let _instanceExtraInitializers = [];
-    let _poll_decorators;
-    return _a = class MemoryProvingQueue {
-            constructor(client, 
-            /** Timeout the job if an agent doesn't report back in this time */
-            jobTimeoutMs = 60 * 1000, 
-            /** How often to check for timed out jobs */
-            pollingIntervalMs = 1000, generateId = defaultIdGenerator, timeSource = defaultTimeSource, proofStore = new InlineProofStore()) {
-                this.jobTimeoutMs = (__runInitializers(this, _instanceExtraInitializers), jobTimeoutMs);
-                this.generateId = generateId;
-                this.timeSource = timeSource;
-                this.proofStore = proofStore;
-                this.log = createLogger('prover-client:prover-pool:queue');
-                this.queue = new PriorityMemoryQueue((a, b) => (a.epochNumber ?? 0) - (b.epochNumber ?? 0));
-                this.jobsInProgress = new Map();
-                this.tracer = client.getTracer('MemoryProvingQueue');
-                this.metrics = new ProvingQueueMetrics(client, 'MemoryProvingQueue');
-                this.runningPromise = new RunningPromise(this.poll.bind(this), this.log, pollingIntervalMs);
-            }
-            start() {
-                if (this.runningPromise.isRunning()) {
-                    this.log.warn('Proving queue is already running');
-                    return;
-                }
-                this.runningPromise.start();
-                this.log.info('Proving queue started');
-            }
-            async stop() {
-                if (!this.runningPromise.isRunning()) {
-                    this.log.warn('Proving queue is already stopped');
-                    return;
-                }
-                await this.runningPromise.stop();
-                this.log.info('Proving queue stopped');
-            }
-            async getProvingJob({ timeoutSec = 1 } = {}) {
-                if (!this.runningPromise.isRunning()) {
-                    throw new Error('Proving queue is not running. Start the queue before getting jobs.');
-                }
-                try {
-                    const job = await this.queue.get(timeoutSec);
-                    if (!job) {
-                        return undefined;
-                    }
-                    if (job.signal?.aborted) {
-                        return undefined;
-                    }
-                    job.heartbeat = this.timeSource();
-                    this.jobsInProgress.set(job.id, job);
-                    return {
-                        id: job.id,
-                        type: job.type,
-                        inputsUri: job.inputsUri,
-                        epochNumber: job.epochNumber,
-                    };
-                }
-                catch (err) {
-                    if (err instanceof TimeoutError) {
-                        return undefined;
-                    }
-                    throw err;
-                }
-            }
-            resolveProvingJob(jobId, result) {
-                if (!this.runningPromise.isRunning()) {
-                    throw new Error('Proving queue is not running.');
-                }
-                const job = this.jobsInProgress.get(jobId);
-                if (!job) {
-                    this.log.warn(`Job id=${jobId} not found. Can't resolve`);
-                    return Promise.resolve();
-                }
-                this.jobsInProgress.delete(jobId);
-                if (!job.signal?.aborted) {
-                    job.resolve(result);
-                }
-                return Promise.resolve();
-            }
-            rejectProvingJob(jobId, reason) {
-                if (!this.runningPromise.isRunning()) {
-                    throw new Error('Proving queue is not running.');
-                }
-                const job = this.jobsInProgress.get(jobId);
-                if (!job) {
-                    this.log.warn(`Job id=${jobId} not found. Can't reject`);
-                    return Promise.resolve();
-                }
-                this.jobsInProgress.delete(jobId);
-                if (job.signal?.aborted) {
-                    return Promise.resolve();
-                }
-                // every job should be retried with the exception of the public VM since its in development and can fail
-                if (job.attempts < MAX_RETRIES && job.type !== ProvingRequestType.PUBLIC_VM) {
-                    job.attempts++;
-                    this.log.warn(`Job id=${job.id} type=${ProvingRequestType[job.type]} failed with error: ${reason}. Retry ${job.attempts}/${MAX_RETRIES}`);
-                    this.queue.put(job);
-                }
-                else {
-                    const logFn = job.type === ProvingRequestType.PUBLIC_VM && !process.env.AVM_PROVING_STRICT ? this.log.warn : this.log.error;
-                    logFn(`Job id=${job.id} type=${ProvingRequestType[job.type]} failed with error: ${reason}`);
-                    job.reject(new Error(reason));
-                }
-                return Promise.resolve();
-            }
-            heartbeat(jobId) {
-                if (!this.runningPromise.isRunning()) {
-                    throw new Error('Proving queue is not running.');
-                }
-                const job = this.jobsInProgress.get(jobId);
-                if (job) {
-                    job.heartbeat = this.timeSource();
-                }
-                return Promise.resolve();
-            }
-            isJobRunning(jobId) {
-                return this.jobsInProgress.has(jobId);
-            }
-            poll() {
-                const now = this.timeSource();
-                this.metrics.recordQueueSize(this.queue.length());
-                for (const job of this.jobsInProgress.values()) {
-                    if (job.signal?.aborted) {
-                        this.jobsInProgress.delete(job.id);
-                        continue;
-                    }
-                    if (job.heartbeat + this.jobTimeoutMs < now) {
-                        this.log.warn(`Job ${job.id} type=${ProvingRequestType[job.type]} has timed out`);
-                        this.jobsInProgress.delete(job.id);
-                        job.heartbeat = 0;
-                        this.queue.put(job);
-                    }
-                }
-            }
-            async enqueue(type, inputs, signal, epochNumber) {
-                if (!this.runningPromise.isRunning()) {
-                    return Promise.reject(new Error('Proving queue is not running.'));
-                }
-                const { promise, resolve, reject } = promiseWithResolvers();
-                const id = this.generateId();
-                const inputsUri = await this.proofStore.saveProofInput(id, type, inputs);
-                const item = {
-                    id,
-                    type,
-                    inputsUri,
-                    signal,
-                    promise,
-                    resolve,
-                    reject,
-                    attempts: 1,
-                    heartbeat: 0,
-                    epochNumber: epochNumber ?? 0,
-                };
-                if (signal) {
-                    signal.addEventListener('abort', () => reject(new AbortError('Operation has been aborted')));
-                }
-                this.log.debug(`Adding id=${item.id} type=${ProvingRequestType[type]} proving job to queue depth=${this.queue.length()}`);
-                if (!this.queue.put(item)) {
-                    throw new Error();
-                }
-                return promise.then(({ result }) => result);
-            }
-            getTubeProof(inputs, signal, epochNumber) {
-                return this.enqueue(ProvingRequestType.TUBE_PROOF, inputs, signal, epochNumber);
-            }
-            /**
-             * Creates a proof for the given input.
-             * @param input - Input to the circuit.
-             */
-            getBaseParityProof(inputs, signal, epochNumber) {
-                return this.enqueue(ProvingRequestType.BASE_PARITY, inputs, signal, epochNumber);
-            }
-            /**
-             * Creates a proof for the given input.
-             * @param input - Input to the circuit.
-             */
-            getRootParityProof(inputs, signal, epochNumber) {
-                return this.enqueue(ProvingRequestType.ROOT_PARITY, inputs, signal, epochNumber);
-            }
-            getPrivateBaseRollupProof(inputs, signal, epochNumber) {
-                return this.enqueue(ProvingRequestType.PRIVATE_BASE_ROLLUP, inputs, signal, epochNumber);
-            }
-            getPublicBaseRollupProof(inputs, signal, epochNumber) {
-                return this.enqueue(ProvingRequestType.PUBLIC_BASE_ROLLUP, inputs, signal, epochNumber);
-            }
-            /**
-             * Creates a proof for the given input.
-             * @param input - Input to the circuit.
-             */
-            getMergeRollupProof(inputs, signal, epochNumber) {
-                return this.enqueue(ProvingRequestType.MERGE_ROLLUP, inputs, signal, epochNumber);
-            }
-            /**
-             * Creates a proof for the given input.
-             * @param input - Input to the circuit.
-             */
-            getBlockRootRollupProof(inputs, signal, epochNumber) {
-                return this.enqueue(ProvingRequestType.BLOCK_ROOT_ROLLUP, inputs, signal, epochNumber);
-            }
-            getSingleTxBlockRootRollupProof(inputs, signal, epochNumber) {
-                return this.enqueue(ProvingRequestType.SINGLE_TX_BLOCK_ROOT_ROLLUP, inputs, signal, epochNumber);
-            }
-            getEmptyBlockRootRollupProof(inputs, signal, epochNumber) {
-                return this.enqueue(ProvingRequestType.EMPTY_BLOCK_ROOT_ROLLUP, inputs, signal, epochNumber);
-            }
-            /**
-             * Creates a proof for the given input.
-             * @param input - Input to the circuit.
-             */
-            getBlockMergeRollupProof(inputs, signal, epochNumber) {
-                return this.enqueue(ProvingRequestType.BLOCK_MERGE_ROLLUP, inputs, signal, epochNumber);
-            }
-            /**
-             * Creates a proof for the given input.
-             * @param input - Input to the circuit.
-             */
-            getRootRollupProof(inputs, signal, epochNumber) {
-                return this.enqueue(ProvingRequestType.ROOT_ROLLUP, inputs, signal, epochNumber);
-            }
-            /**
-             * Creates an AVM proof.
-             */
-            getAvmProof(inputs, signal, epochNumber) {
-                return this.enqueue(ProvingRequestType.PUBLIC_VM, inputs, signal, epochNumber);
-            }
-            /**
-             * Verifies a circuit proof
-             */
-            verifyProof() {
-                return Promise.reject('not implemented');
-            }
-        },
-        (() => {
-            const _metadata = typeof Symbol === "function" && Symbol.metadata ? Object.create(null) : void 0;
-            _poll_decorators = [trackSpan('MemoryProvingQueue.poll')];
-            __esDecorate(_a, null, _poll_decorators, { kind: "method", name: "poll", static: false, private: false, access: { has: obj => "poll" in obj, get: obj => obj.poll }, metadata: _metadata }, null, _instanceExtraInitializers);
-            if (_metadata) Object.defineProperty(_a, Symbol.metadata, { enumerable: true, configurable: true, writable: true, value: _metadata });
-        })(),
-        _a;
-})();
-export { MemoryProvingQueue };
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoibWVtb3J5LXByb3ZpbmctcXVldWUuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi9zcmMvcHJvdmVyLWFnZW50L21lbW9yeS1wcm92aW5nLXF1ZXVlLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7QUFBQSxPQUFPLEVBTUwsa0JBQWtCLEdBR25CLE1BQU0sc0JBQXNCLENBQUM7QUEwQjlCLE9BQU8sRUFBRSxXQUFXLEVBQUUsTUFBTSwwQkFBMEIsQ0FBQztBQUN2RCxPQUFPLEVBQUUsVUFBVSxFQUFFLFlBQVksRUFBRSxNQUFNLHlCQUF5QixDQUFDO0FBQ25FLE9BQU8sRUFBRSxZQUFZLEVBQUUsTUFBTSx1QkFBdUIsQ0FBQztBQUNyRCxPQUFPLEVBQTZCLGNBQWMsRUFBRSxvQkFBb0IsRUFBRSxNQUFNLDJCQUEyQixDQUFDO0FBQzVHLE9BQU8sRUFBRSxtQkFBbUIsRUFBRSxNQUFNLHlCQUF5QixDQUFDO0FBQzlELE9BQU8sRUFBcUMsU0FBUyxFQUFFLE1BQU0seUJBQXlCLENBQUM7QUFFdkYsT0FBTyxFQUFFLGdCQUFnQixFQUFtQixNQUFNLHdDQUF3QyxDQUFDO0FBQzNGLE9BQU8sRUFBRSxtQkFBbUIsRUFBRSxNQUFNLG9CQUFvQixDQUFDO0FBVXpELE1BQU0sV0FBVyxHQUFHLENBQUMsQ0FBQztBQUV0QixNQUFNLGtCQUFrQixHQUFHLEdBQUcsRUFBRSxDQUFDLFdBQVcsQ0FBQyxDQUFDLENBQUMsQ0FBQyxRQUFRLENBQUMsS0FBSyxDQUFDLENBQUM7QUFDaEUsTUFBTSxpQkFBaUIsR0FBRyxHQUFHLEVBQUUsQ0FBQyxJQUFJLENBQUMsR0FBRyxFQUFFLENBQUM7QUFDM0M7OztHQUdHO0lBQ1Usa0JBQWtCOzs7O3NCQUFsQixrQkFBa0I7WUFXN0IsWUFDRSxNQUF1QjtZQUN2QixtRUFBbUU7WUFDM0QsZUFBZSxFQUFFLEdBQUcsSUFBSTtZQUNoQyw0Q0FBNEM7WUFDNUMsaUJBQWlCLEdBQUcsSUFBSSxFQUNoQixhQUFhLGtCQUFrQixFQUMvQixhQUFhLGlCQUFpQixFQUM5QixhQUF5QixJQUFJLGdCQUFnQixFQUFFO2dCQUwvQyxpQkFBWSxJQWRYLG1EQUFrQixFQWNuQixZQUFZLEVBQVk7Z0JBR3hCLGVBQVUsR0FBVixVQUFVLENBQXFCO2dCQUMvQixlQUFVLEdBQVYsVUFBVSxDQUFvQjtnQkFDOUIsZUFBVSxHQUFWLFVBQVUsQ0FBcUM7Z0JBbEJqRCxRQUFHLEdBQUcsWUFBWSxDQUFDLGlDQUFpQyxDQUFDLENBQUM7Z0JBQ3RELFVBQUssR0FBRyxJQUFJLG1CQUFtQixDQUNyQyxDQUFDLENBQUMsRUFBRSxDQUFDLEVBQUUsRUFBRSxDQUFDLENBQUMsQ0FBQyxDQUFDLFdBQVcsSUFBSSxDQUFDLENBQUMsR0FBRyxDQUFDLENBQUMsQ0FBQyxXQUFXLElBQUksQ0FBQyxDQUFDLENBQ3RELENBQUM7Z0JBQ00sbUJBQWMsR0FBRyxJQUFJLEdBQUcsRUFBbUMsQ0FBQztnQkFnQmxFLElBQUksQ0FBQyxNQUFNLEdBQUcsTUFBTSxDQUFDLFNBQVMsQ0FBQyxvQkFBb0IsQ0FBQyxDQUFDO2dCQUNyRCxJQUFJLENBQUMsT0FBTyxHQUFHLElBQUksbUJBQW1CLENBQUMsTUFBTSxFQUFFLG9CQUFvQixDQUFDLENBQUM7Z0JBQ3JFLElBQUksQ0FBQyxjQUFjLEdBQUcsSUFBSSxjQUFjLENBQUMsSUFBSSxDQUFDLElBQUksQ0FBQyxJQUFJLENBQUMsSUFBSSxDQUFDLEVBQUUsSUFBSSxDQUFDLEdBQUcsRUFBRSxpQkFBaUIsQ0FBQyxDQUFDO1lBQzlGLENBQUM7WUFFTSxLQUFLO2dCQUNWLElBQUksSUFBSSxDQUFDLGNBQWMsQ0FBQyxTQUFTLEVBQUUsRUFBRSxDQUFDO29CQUNwQyxJQUFJLENBQUMsR0FBRyxDQUFDLElBQUksQ0FBQyxrQ0FBa0MsQ0FBQyxDQUFDO29CQUNsRCxPQUFPO2dCQUNULENBQUM7Z0JBRUQsSUFBSSxDQUFDLGNBQWMsQ0FBQyxLQUFLLEVBQUUsQ0FBQztnQkFDNUIsSUFBSSxDQUFDLEdBQUcsQ0FBQyxJQUFJLENBQUMsdUJBQXVCLENBQUMsQ0FBQztZQUN6QyxDQUFDO1lBRU0sS0FBSyxDQUFDLElBQUk7Z0JBQ2YsSUFBSSxDQUFDLElBQUksQ0FBQyxjQUFjLENBQUMsU0FBUyxFQUFFLEVBQUUsQ0FBQztvQkFDckMsSUFBSSxDQUFDLEdBQUcsQ0FBQyxJQUFJLENBQUMsa0NBQWtDLENBQUMsQ0FBQztvQkFDbEQsT0FBTztnQkFDVCxDQUFDO2dCQUVELE1BQU0sSUFBSSxDQUFDLGNBQWMsQ0FBQyxJQUFJLEVBQUUsQ0FBQztnQkFDakMsSUFBSSxDQUFDLEdBQUcsQ0FBQyxJQUFJLENBQUMsdUJBQXVCLENBQUMsQ0FBQztZQUN6QyxDQUFDO1lBRU0sS0FBSyxDQUFDLGFBQWEsQ0FBQyxFQUFFLFVBQVUsR0FBRyxDQUFDLEVBQUUsR0FBRyxFQUFFO2dCQUNoRCxJQUFJLENBQUMsSUFBSSxDQUFDLGNBQWMsQ0FBQyxTQUFTLEVBQUUsRUFBRSxDQUFDO29CQUNyQyxNQUFNLElBQUksS0FBSyxDQUFDLG9FQUFvRSxDQUFDLENBQUM7Z0JBQ3hGLENBQUM7Z0JBRUQsSUFBSSxDQUFDO29CQUNILE1BQU0sR0FBRyxHQUFHLE1BQU0sSUFBSSxDQUFDLEtBQUssQ0FBQyxHQUFHLENBQUMsVUFBVSxDQUFDLENBQUM7b0JBQzdDLElBQUksQ0FBQyxHQUFHLEVBQUUsQ0FBQzt3QkFDVCxPQUFPLFNBQVMsQ0FBQztvQkFDbkIsQ0FBQztvQkFFRCxJQUFJLEdBQUcsQ0FBQyxNQUFNLEVBQUUsT0FBTyxFQUFFLENBQUM7d0JBQ3hCLE9BQU8sU0FBUyxDQUFDO29CQUNuQixDQUFDO29CQUVELEdBQUcsQ0FBQyxTQUFTLEdBQUcsSUFBSSxDQUFDLFVBQVUsRUFBRSxDQUFDO29CQUNsQyxJQUFJLENBQUMsY0FBYyxDQUFDLEdBQUcsQ0FBQyxHQUFHLENBQUMsRUFBRSxFQUFFLEdBQUcsQ0FBQyxDQUFDO29CQUNyQyxPQUFPO3dCQUNMLEVBQUUsRUFBRSxHQUFHLENBQUMsRUFBRTt3QkFDVixJQUFJLEVBQUUsR0FBRyxDQUFDLElBQUk7d0JBQ2QsU0FBUyxFQUFFLEdBQUcsQ0FBQyxTQUFTO3dCQUN4QixXQUFXLEVBQUUsR0FBRyxDQUFDLFdBQVc7cUJBQzdCLENBQUM7Z0JBQ0osQ0FBQztnQkFBQyxPQUFPLEdBQUcsRUFBRSxDQUFDO29CQUNiLElBQUksR0FBRyxZQUFZLFlBQVksRUFBRSxDQUFDO3dCQUNoQyxPQUFPLFNBQVMsQ0FBQztvQkFDbkIsQ0FBQztvQkFFRCxNQUFNLEdBQUcsQ0FBQztnQkFDWixDQUFDO1lBQ0gsQ0FBQztZQUVELGlCQUFpQixDQUErQixLQUFhLEVBQUUsTUFBa0M7Z0JBQy9GLElBQUksQ0FBQyxJQUFJLENBQUMsY0FBYyxDQUFDLFNBQVMsRUFBRSxFQUFFLENBQUM7b0JBQ3JDLE1BQU0sSUFBSSxLQUFLLENBQUMsK0JBQStCLENBQUMsQ0FBQztnQkFDbkQsQ0FBQztnQkFFRCxNQUFNLEdBQUcsR0FBRyxJQUFJLENBQUMsY0FBYyxDQUFDLEdBQUcsQ0FBQyxLQUFLLENBQUMsQ0FBQztnQkFDM0MsSUFBSSxDQUFDLEdBQUcsRUFBRSxDQUFDO29CQUNULElBQUksQ0FBQyxHQUFHLENBQUMsSUFBSSxDQUFDLFVBQVUsS0FBSywyQkFBMkIsQ0FBQyxDQUFDO29CQUMxRCxPQUFPLE9BQU8sQ0FBQyxPQUFPLEVBQUUsQ0FBQztnQkFDM0IsQ0FBQztnQkFFRCxJQUFJLENBQUMsY0FBYyxDQUFDLE1BQU0sQ0FBQyxLQUFLLENBQUMsQ0FBQztnQkFDbEMsSUFBSSxDQUFDLEdBQUcsQ0FBQyxNQUFNLEVBQUUsT0FBTyxFQUFFLENBQUM7b0JBQ3pCLEdBQUcsQ0FBQyxPQUFPLENBQUMsTUFBTSxDQUFDLENBQUM7Z0JBQ3RCLENBQUM7Z0JBRUQsT0FBTyxPQUFPLENBQUMsT0FBTyxFQUFFLENBQUM7WUFDM0IsQ0FBQztZQUVELGdCQUFnQixDQUFDLEtBQWEsRUFBRSxNQUFjO2dCQUM1QyxJQUFJLENBQUMsSUFBSSxDQUFDLGNBQWMsQ0FBQyxTQUFTLEVBQUUsRUFBRSxDQUFDO29CQUNyQyxNQUFNLElBQUksS0FBSyxDQUFDLCtCQUErQixDQUFDLENBQUM7Z0JBQ25ELENBQUM7Z0JBRUQsTUFBTSxHQUFHLEdBQUcsSUFBSSxDQUFDLGNBQWMsQ0FBQyxHQUFHLENBQUMsS0FBSyxDQUFDLENBQUM7Z0JBQzNDLElBQUksQ0FBQyxHQUFHLEVBQUUsQ0FBQztvQkFDVCxJQUFJLENBQUMsR0FBRyxDQUFDLElBQUksQ0FBQyxVQUFVLEtBQUssMEJBQTBCLENBQUMsQ0FBQztvQkFDekQsT0FBTyxPQUFPLENBQUMsT0FBTyxFQUFFLENBQUM7Z0JBQzNCLENBQUM7Z0JBRUQsSUFBSSxDQUFDLGNBQWMsQ0FBQyxNQUFNLENBQUMsS0FBSyxDQUFDLENBQUM7Z0JBRWxDLElBQUksR0FBRyxDQUFDLE1BQU0sRUFBRSxPQUFPLEVBQUUsQ0FBQztvQkFDeEIsT0FBTyxPQUFPLENBQUMsT0FBTyxFQUFFLENBQUM7Z0JBQzNCLENBQUM7Z0JBRUQsd0dBQXdHO2dCQUN4RyxJQUFJLEdBQUcsQ0FBQyxRQUFRLEdBQUcsV0FBVyxJQUFJLEdBQUcsQ0FBQyxJQUFJLEtBQUssa0JBQWtCLENBQUMsU0FBUyxFQUFFLENBQUM7b0JBQzVFLEdBQUcsQ0FBQyxRQUFRLEVBQUUsQ0FBQztvQkFDZixJQUFJLENBQUMsR0FBRyxDQUFDLElBQUksQ0FDWCxVQUFVLEdBQUcsQ0FBQyxFQUFFLFNBQVMsa0JBQWtCLENBQUMsR0FBRyxDQUFDLElBQUksQ0FBQyx1QkFBdUIsTUFBTSxXQUNoRixHQUFHLENBQUMsUUFDTixJQUFJLFdBQVcsRUFBRSxDQUNsQixDQUFDO29CQUNGLElBQUksQ0FBQyxLQUFLLENBQUMsR0FBRyxDQUFDLEdBQUcsQ0FBQyxDQUFDO2dCQUN0QixDQUFDO3FCQUFNLENBQUM7b0JBQ04sTUFBTSxLQUFLLEdBQ1QsR0FBRyxDQUFDLElBQUksS0FBSyxrQkFBa0IsQ0FBQyxTQUFTLElBQUksQ0FBQyxPQUFPLENBQUMsR0FBRyxDQUFDLGtCQUFrQixDQUFDLENBQUMsQ0FBQyxJQUFJLENBQUMsR0FBRyxDQUFDLElBQUksQ0FBQyxDQUFDLENBQUMsSUFBSSxDQUFDLEdBQUcsQ0FBQyxLQUFLLENBQUM7b0JBQ2hILEtBQUssQ0FBQyxVQUFVLEdBQUcsQ0FBQyxFQUFFLFNBQVMsa0JBQWtCLENBQUMsR0FBRyxDQUFDLElBQUksQ0FBQyx1QkFBdUIsTUFBTSxFQUFFLENBQUMsQ0FBQztvQkFDNUYsR0FBRyxDQUFDLE1BQU0sQ0FBQyxJQUFJLEtBQUssQ0FBQyxNQUFNLENBQUMsQ0FBQyxDQUFDO2dCQUNoQyxDQUFDO2dCQUNELE9BQU8sT0FBTyxDQUFDLE9BQU8sRUFBRSxDQUFDO1lBQzNCLENBQUM7WUFFTSxTQUFTLENBQUMsS0FBYTtnQkFDNUIsSUFBSSxDQUFDLElBQUksQ0FBQyxjQUFjLENBQUMsU0FBUyxFQUFFLEVBQUUsQ0FBQztvQkFDckMsTUFBTSxJQUFJLEtBQUssQ0FBQywrQkFBK0IsQ0FBQyxDQUFDO2dCQUNuRCxDQUFDO2dCQUVELE1BQU0sR0FBRyxHQUFHLElBQUksQ0FBQyxjQUFjLENBQUMsR0FBRyxDQUFDLEtBQUssQ0FBQyxDQUFDO2dCQUMzQyxJQUFJLEdBQUcsRUFBRSxDQUFDO29CQUNSLEdBQUcsQ0FBQyxTQUFTLEdBQUcsSUFBSSxDQUFDLFVBQVUsRUFBRSxDQUFDO2dCQUNwQyxDQUFDO2dCQUVELE9BQU8sT0FBTyxDQUFDLE9BQU8sRUFBRSxDQUFDO1lBQzNCLENBQUM7WUFFTSxZQUFZLENBQUMsS0FBYTtnQkFDL0IsT0FBTyxJQUFJLENBQUMsY0FBYyxDQUFDLEdBQUcsQ0FBQyxLQUFLLENBQUMsQ0FBQztZQUN4QyxDQUFDO1lBR08sSUFBSTtnQkFDVixNQUFNLEdBQUcsR0FBRyxJQUFJLENBQUMsVUFBVSxFQUFFLENBQUM7Z0JBQzlCLElBQUksQ0FBQyxPQUFPLENBQUMsZUFBZSxDQUFDLElBQUksQ0FBQyxLQUFLLENBQUMsTUFBTSxFQUFFLENBQUMsQ0FBQztnQkFFbEQsS0FBSyxNQUFNLEdBQUcsSUFBSSxJQUFJLENBQUMsY0FBYyxDQUFDLE1BQU0sRUFBRSxFQUFFLENBQUM7b0JBQy9DLElBQUksR0FBRyxDQUFDLE1BQU0sRUFBRSxPQUFPLEVBQUUsQ0FBQzt3QkFDeEIsSUFBSSxDQUFDLGNBQWMsQ0FBQyxNQUFNLENBQUMsR0FBRyxDQUFDLEVBQUUsQ0FBQyxDQUFDO3dCQUNuQyxTQUFTO29CQUNYLENBQUM7b0JBRUQsSUFBSSxHQUFHLENBQUMsU0FBUyxHQUFHLElBQUksQ0FBQyxZQUFZLEdBQUcsR0FBRyxFQUFFLENBQUM7d0JBQzVDLElBQUksQ0FBQyxHQUFHLENBQUMsSUFBSSxDQUFDLE9BQU8sR0FBRyxDQUFDLEVBQUUsU0FBUyxrQkFBa0IsQ0FBQyxHQUFHLENBQUMsSUFBSSxDQUFDLGdCQUFnQixDQUFDLENBQUM7d0JBRWxGLElBQUksQ0FBQyxjQUFjLENBQUMsTUFBTSxDQUFDLEdBQUcsQ0FBQyxFQUFFLENBQUMsQ0FBQzt3QkFDbkMsR0FBRyxDQUFDLFNBQVMsR0FBRyxDQUFDLENBQUM7d0JBQ2xCLElBQUksQ0FBQyxLQUFLLENBQUMsR0FBRyxDQUFDLEdBQUcsQ0FBQyxDQUFDO29CQUN0QixDQUFDO2dCQUNILENBQUM7WUFDSCxDQUFDO1lBRU8sS0FBSyxDQUFDLE9BQU8sQ0FDbkIsSUFBTyxFQUNQLE1BQThCLEVBQzlCLE1BQW9CLEVBQ3BCLFdBQW9CO2dCQUVwQixJQUFJLENBQUMsSUFBSSxDQUFDLGNBQWMsQ0FBQyxTQUFTLEVBQUUsRUFBRSxDQUFDO29CQUNyQyxPQUFPLE9BQU8sQ0FBQyxNQUFNLENBQUMsSUFBSSxLQUFLLENBQUMsK0JBQStCLENBQUMsQ0FBQyxDQUFDO2dCQUNwRSxDQUFDO2dCQUVELE1BQU0sRUFBRSxPQUFPLEVBQUUsT0FBTyxFQUFFLE1BQU0sRUFBRSxHQUFHLG9CQUFvQixFQUE4QixDQUFDO2dCQUN4RixNQUFNLEVBQUUsR0FBRyxJQUFJLENBQUMsVUFBVSxFQUFFLENBQUM7Z0JBQzdCLE1BQU0sU0FBUyxHQUFHLE1BQU0sSUFBSSxDQUFDLFVBQVUsQ0FBQyxjQUFjLENBQUMsRUFBRSxFQUFFLElBQUksRUFBRSxNQUFNLENBQUMsQ0FBQztnQkFDekUsTUFBTSxJQUFJLEdBQStCO29CQUN2QyxFQUFFO29CQUNGLElBQUk7b0JBQ0osU0FBUztvQkFDVCxNQUFNO29CQUNOLE9BQU87b0JBQ1AsT0FBTztvQkFDUCxNQUFNO29CQUNOLFFBQVEsRUFBRSxDQUFDO29CQUNYLFNBQVMsRUFBRSxDQUFDO29CQUNaLFdBQVcsRUFBRSxXQUFXLElBQUksQ0FBQztpQkFDOUIsQ0FBQztnQkFFRixJQUFJLE1BQU0sRUFBRSxDQUFDO29CQUNYLE1BQU0sQ0FBQyxnQkFBZ0IsQ0FBQyxPQUFPLEVBQUUsR0FBRyxFQUFFLENBQUMsTUFBTSxDQUFDLElBQUksVUFBVSxDQUFDLDRCQUE0QixDQUFDLENBQUMsQ0FBQyxDQUFDO2dCQUMvRixDQUFDO2dCQUVELElBQUksQ0FBQyxHQUFHLENBQUMsS0FBSyxDQUNaLGFBQWEsSUFBSSxDQUFDLEVBQUUsU0FBUyxrQkFBa0IsQ0FBQyxJQUFJLENBQUMsK0JBQStCLElBQUksQ0FBQyxLQUFLLENBQUMsTUFBTSxFQUFFLEVBQUUsQ0FDMUcsQ0FBQztnQkFFRixJQUFJLENBQUMsSUFBSSxDQUFDLEtBQUssQ0FBQyxHQUFHLENBQUMsSUFBb0MsQ0FBQyxFQUFFLENBQUM7b0JBQzFELE1BQU0sSUFBSSxLQUFLLEVBQUUsQ0FBQztnQkFDcEIsQ0FBQztnQkFFRCxPQUFPLE9BQU8sQ0FBQyxJQUFJLENBQUMsQ0FBQyxFQUFFLE1BQU0sRUFBRSxFQUFFLEVBQUUsQ0FBQyxNQUFNLENBQUMsQ0FBQztZQUM5QyxDQUFDO1lBRUQsWUFBWSxDQUNWLE1BQWtCLEVBQ2xCLE1BQW9CLEVBQ3BCLFdBQW9CO2dCQUVwQixPQUFPLElBQUksQ0FBQyxPQUFPLENBQUMsa0JBQWtCLENBQUMsVUFBVSxFQUFFLE1BQU0sRUFBRSxNQUFNLEVBQUUsV0FBVyxDQUFDLENBQUM7WUFDbEYsQ0FBQztZQUVEOzs7ZUFHRztZQUNILGtCQUFrQixDQUNoQixNQUF3QixFQUN4QixNQUFvQixFQUNwQixXQUFvQjtnQkFFcEIsT0FBTyxJQUFJLENBQUMsT0FBTyxDQUFDLGtCQUFrQixDQUFDLFdBQVcsRUFBRSxNQUFNLEVBQUUsTUFBTSxFQUFFLFdBQVcsQ0FBQyxDQUFDO1lBQ25GLENBQUM7WUFFRDs7O2VBR0c7WUFDSCxrQkFBa0IsQ0FDaEIsTUFBd0IsRUFDeEIsTUFBb0IsRUFDcEIsV0FBb0I7Z0JBRXBCLE9BQU8sSUFBSSxDQUFDLE9BQU8sQ0FBQyxrQkFBa0IsQ0FBQyxXQUFXLEVBQUUsTUFBTSxFQUFFLE1BQU0sRUFBRSxXQUFXLENBQUMsQ0FBQztZQUNuRixDQUFDO1lBRUQseUJBQXlCLENBQ3ZCLE1BQStCLEVBQy9CLE1BQW9CLEVBQ3BCLFdBQW9CO2dCQUlwQixPQUFPLElBQUksQ0FBQyxPQUFPLENBQUMsa0JBQWtCLENBQUMsbUJBQW1CLEVBQUUsTUFBTSxFQUFFLE1BQU0sRUFBRSxXQUFXLENBQUMsQ0FBQztZQUMzRixDQUFDO1lBRUQsd0JBQXdCLENBQ3RCLE1BQThCLEVBQzlCLE1BQW9CLEVBQ3BCLFdBQW9CO2dCQUlwQixPQUFPLElBQUksQ0FBQyxPQUFPLENBQUMsa0JBQWtCLENBQUMsa0JBQWtCLEVBQUUsTUFBTSxFQUFFLE1BQU0sRUFBRSxXQUFXLENBQUMsQ0FBQztZQUMxRixDQUFDO1lBRUQ7OztlQUdHO1lBQ0gsbUJBQW1CLENBQ2pCLE1BQXlCLEVBQ3pCLE1BQW9CLEVBQ3BCLFdBQW9CO2dCQUlwQixPQUFPLElBQUksQ0FBQyxPQUFPLENBQUMsa0JBQWtCLENBQUMsWUFBWSxFQUFFLE1BQU0sRUFBRSxNQUFNLEVBQUUsV0FBVyxDQUFDLENBQUM7WUFDcEYsQ0FBQztZQUVEOzs7ZUFHRztZQUNILHVCQUF1QixDQUNyQixNQUE2QixFQUM3QixNQUFvQixFQUNwQixXQUFvQjtnQkFJcEIsT0FBTyxJQUFJLENBQUMsT0FBTyxDQUFDLGtCQUFrQixDQUFDLGlCQUFpQixFQUFFLE1BQU0sRUFBRSxNQUFNLEVBQUUsV0FBVyxDQUFDLENBQUM7WUFDekYsQ0FBQztZQUVELCtCQUErQixDQUM3QixNQUFxQyxFQUNyQyxNQUFvQixFQUNwQixXQUFvQjtnQkFJcEIsT0FBTyxJQUFJLENBQUMsT0FBTyxDQUFDLGtCQUFrQixDQUFDLDJCQUEyQixFQUFFLE1BQU0sRUFBRSxNQUFNLEVBQUUsV0FBVyxDQUFDLENBQUM7WUFDbkcsQ0FBQztZQUVELDRCQUE0QixDQUMxQixNQUFrQyxFQUNsQyxNQUFvQixFQUNwQixXQUFvQjtnQkFJcEIsT0FBTyxJQUFJLENBQUMsT0FBTyxDQUFDLGtCQUFrQixDQUFDLHVCQUF1QixFQUFFLE1BQU0sRUFBRSxNQUFNLEVBQUUsV0FBVyxDQUFDLENBQUM7WUFDL0YsQ0FBQztZQUVEOzs7ZUFHRztZQUNILHdCQUF3QixDQUN0QixNQUE4QixFQUM5QixNQUFvQixFQUNwQixXQUFvQjtnQkFJcEIsT0FBTyxJQUFJLENBQUMsT0FBTyxDQUFDLGtCQUFrQixDQUFDLGtCQUFrQixFQUFFLE1BQU0sRUFBRSxNQUFNLEVBQUUsV0FBVyxDQUFDLENBQUM7WUFDMUYsQ0FBQztZQUVEOzs7ZUFHRztZQUNILGtCQUFrQixDQUNoQixNQUF3QixFQUN4QixNQUFvQixFQUNwQixXQUFvQjtnQkFFcEIsT0FBTyxJQUFJLENBQUMsT0FBTyxDQUFDLGtCQUFrQixDQUFDLFdBQVcsRUFBRSxNQUFNLEVBQUUsTUFBTSxFQUFFLFdBQVcsQ0FBQyxDQUFDO1lBQ25GLENBQUM7WUFFRDs7ZUFFRztZQUNILFdBQVcsQ0FDVCxNQUF3QixFQUN4QixNQUFvQixFQUNwQixXQUFvQjtnQkFFcEIsT0FBTyxJQUFJLENBQUMsT0FBTyxDQUFDLGtCQUFrQixDQUFDLFNBQVMsRUFBRSxNQUFNLEVBQUUsTUFBTSxFQUFFLFdBQVcsQ0FBQyxDQUFDO1lBQ2pGLENBQUM7WUFFRDs7ZUFFRztZQUNILFdBQVc7Z0JBQ1QsT0FBTyxPQUFPLENBQUMsTUFBTSxDQUFDLGlCQUFpQixDQUFDLENBQUM7WUFDM0MsQ0FBQzs7OztnQ0E1TUEsU0FBUyxDQUFDLHlCQUF5QixDQUFDO1lBQ3JDLDZKQUFRLElBQUksNkRBa0JYOzs7OztTQXhLVSxrQkFBa0IifQ==
+ */ export class MemoryProvingQueue {
+    jobTimeoutMs;
+    generateId;
+    timeSource;
+    proofStore;
+    log;
+    queue;
+    jobsInProgress;
+    runningPromise;
+    metrics;
+    tracer;
+    constructor(client, /** Timeout the job if an agent doesn't report back in this time */ jobTimeoutMs = 60 * 1000, /** How often to check for timed out jobs */ pollingIntervalMs = 1000, generateId = defaultIdGenerator, timeSource = defaultTimeSource, proofStore = new InlineProofStore()){
+        this.jobTimeoutMs = jobTimeoutMs;
+        this.generateId = generateId;
+        this.timeSource = timeSource;
+        this.proofStore = proofStore;
+        this.log = createLogger('prover-client:prover-pool:queue');
+        this.queue = new PriorityMemoryQueue((a, b)=>(a.epochNumber ?? 0) - (b.epochNumber ?? 0));
+        this.jobsInProgress = new Map();
+        this.tracer = client.getTracer('MemoryProvingQueue');
+        this.metrics = new ProvingQueueMetrics(client, 'MemoryProvingQueue');
+        this.runningPromise = new RunningPromise(this.poll.bind(this), this.log, pollingIntervalMs);
+    }
+    start() {
+        if (this.runningPromise.isRunning()) {
+            this.log.warn('Proving queue is already running');
+            return;
+        }
+        this.runningPromise.start();
+        this.log.info('Proving queue started');
+    }
+    async stop() {
+        if (!this.runningPromise.isRunning()) {
+            this.log.warn('Proving queue is already stopped');
+            return;
+        }
+        await this.runningPromise.stop();
+        this.log.info('Proving queue stopped');
+    }
+    async getProvingJob({ timeoutSec = 1 } = {}) {
+        if (!this.runningPromise.isRunning()) {
+            throw new Error('Proving queue is not running. Start the queue before getting jobs.');
+        }
+        try {
+            const job = await this.queue.get(timeoutSec);
+            if (!job) {
+                return undefined;
+            }
+            if (job.signal?.aborted) {
+                return undefined;
+            }
+            job.heartbeat = this.timeSource();
+            this.jobsInProgress.set(job.id, job);
+            return {
+                id: job.id,
+                type: job.type,
+                inputsUri: job.inputsUri,
+                epochNumber: job.epochNumber
+            };
+        } catch (err) {
+            if (err instanceof TimeoutError) {
+                return undefined;
+            }
+            throw err;
+        }
+    }
+    resolveProvingJob(jobId, result) {
+        if (!this.runningPromise.isRunning()) {
+            throw new Error('Proving queue is not running.');
+        }
+        const job = this.jobsInProgress.get(jobId);
+        if (!job) {
+            this.log.warn(`Job id=${jobId} not found. Can't resolve`);
+            return Promise.resolve();
+        }
+        this.jobsInProgress.delete(jobId);
+        if (!job.signal?.aborted) {
+            job.resolve(result);
+        }
+        return Promise.resolve();
+    }
+    rejectProvingJob(jobId, reason) {
+        if (!this.runningPromise.isRunning()) {
+            throw new Error('Proving queue is not running.');
+        }
+        const job = this.jobsInProgress.get(jobId);
+        if (!job) {
+            this.log.warn(`Job id=${jobId} not found. Can't reject`);
+            return Promise.resolve();
+        }
+        this.jobsInProgress.delete(jobId);
+        if (job.signal?.aborted) {
+            return Promise.resolve();
+        }
+        // every job should be retried with the exception of the public VM since its in development and can fail
+        if (job.attempts < MAX_RETRIES && job.type !== ProvingRequestType.PUBLIC_VM) {
+            job.attempts++;
+            this.log.warn(`Job id=${job.id} type=${ProvingRequestType[job.type]} failed with error: ${reason}. Retry ${job.attempts}/${MAX_RETRIES}`);
+            this.queue.put(job);
+        } else {
+            const logFn = job.type === ProvingRequestType.PUBLIC_VM && !process.env.AVM_PROVING_STRICT ? this.log.warn : this.log.error;
+            logFn(`Job id=${job.id} type=${ProvingRequestType[job.type]} failed with error: ${reason}`);
+            job.reject(new Error(reason));
+        }
+        return Promise.resolve();
+    }
+    heartbeat(jobId) {
+        if (!this.runningPromise.isRunning()) {
+            throw new Error('Proving queue is not running.');
+        }
+        const job = this.jobsInProgress.get(jobId);
+        if (job) {
+            job.heartbeat = this.timeSource();
+        }
+        return Promise.resolve();
+    }
+    isJobRunning(jobId) {
+        return this.jobsInProgress.has(jobId);
+    }
+    poll() {
+        const now = this.timeSource();
+        this.metrics.recordQueueSize(this.queue.length());
+        for (const job of this.jobsInProgress.values()){
+            if (job.signal?.aborted) {
+                this.jobsInProgress.delete(job.id);
+                continue;
+            }
+            if (job.heartbeat + this.jobTimeoutMs < now) {
+                this.log.warn(`Job ${job.id} type=${ProvingRequestType[job.type]} has timed out`);
+                this.jobsInProgress.delete(job.id);
+                job.heartbeat = 0;
+                this.queue.put(job);
+            }
+        }
+    }
+    async enqueue(type, inputs, signal, epochNumber) {
+        if (!this.runningPromise.isRunning()) {
+            return Promise.reject(new Error('Proving queue is not running.'));
+        }
+        const { promise, resolve, reject } = promiseWithResolvers();
+        const id = this.generateId();
+        const inputsUri = await this.proofStore.saveProofInput(id, type, inputs);
+        const item = {
+            id,
+            type,
+            inputsUri,
+            signal,
+            promise,
+            resolve,
+            reject,
+            attempts: 1,
+            heartbeat: 0,
+            epochNumber: epochNumber ?? 0
+        };
+        if (signal) {
+            signal.addEventListener('abort', ()=>reject(new AbortError('Operation has been aborted')));
+        }
+        this.log.debug(`Adding id=${item.id} type=${ProvingRequestType[type]} proving job to queue depth=${this.queue.length()}`);
+        if (!this.queue.put(item)) {
+            throw new Error();
+        }
+        return promise.then(({ result })=>result);
+    }
+    getTubeProof(inputs, signal, epochNumber) {
+        return this.enqueue(ProvingRequestType.TUBE_PROOF, inputs, signal, epochNumber);
+    }
+    /**
+   * Creates a proof for the given input.
+   * @param input - Input to the circuit.
+   */ getBaseParityProof(inputs, signal, epochNumber) {
+        return this.enqueue(ProvingRequestType.BASE_PARITY, inputs, signal, epochNumber);
+    }
+    /**
+   * Creates a proof for the given input.
+   * @param input - Input to the circuit.
+   */ getRootParityProof(inputs, signal, epochNumber) {
+        return this.enqueue(ProvingRequestType.ROOT_PARITY, inputs, signal, epochNumber);
+    }
+    getPrivateBaseRollupProof(inputs, signal, epochNumber) {
+        return this.enqueue(ProvingRequestType.PRIVATE_BASE_ROLLUP, inputs, signal, epochNumber);
+    }
+    getPublicBaseRollupProof(inputs, signal, epochNumber) {
+        return this.enqueue(ProvingRequestType.PUBLIC_BASE_ROLLUP, inputs, signal, epochNumber);
+    }
+    /**
+   * Creates a proof for the given input.
+   * @param input - Input to the circuit.
+   */ getMergeRollupProof(inputs, signal, epochNumber) {
+        return this.enqueue(ProvingRequestType.MERGE_ROLLUP, inputs, signal, epochNumber);
+    }
+    /**
+   * Creates a proof for the given input.
+   * @param input - Input to the circuit.
+   */ getBlockRootRollupProof(inputs, signal, epochNumber) {
+        return this.enqueue(ProvingRequestType.BLOCK_ROOT_ROLLUP, inputs, signal, epochNumber);
+    }
+    getSingleTxBlockRootRollupProof(inputs, signal, epochNumber) {
+        return this.enqueue(ProvingRequestType.SINGLE_TX_BLOCK_ROOT_ROLLUP, inputs, signal, epochNumber);
+    }
+    getEmptyBlockRootRollupProof(inputs, signal, epochNumber) {
+        return this.enqueue(ProvingRequestType.EMPTY_BLOCK_ROOT_ROLLUP, inputs, signal, epochNumber);
+    }
+    /**
+   * Creates a proof for the given input.
+   * @param input - Input to the circuit.
+   */ getBlockMergeRollupProof(inputs, signal, epochNumber) {
+        return this.enqueue(ProvingRequestType.BLOCK_MERGE_ROLLUP, inputs, signal, epochNumber);
+    }
+    /**
+   * Creates a proof for the given input.
+   * @param input - Input to the circuit.
+   */ getRootRollupProof(inputs, signal, epochNumber) {
+        return this.enqueue(ProvingRequestType.ROOT_ROLLUP, inputs, signal, epochNumber);
+    }
+    /**
+   * Creates an AVM proof.
+   */ getAvmProof(inputs, signal, epochNumber) {
+        return this.enqueue(ProvingRequestType.PUBLIC_VM, inputs, signal, epochNumber);
+    }
+    /**
+   * Verifies a circuit proof
+   */ verifyProof() {
+        return Promise.reject('not implemented');
+    }
+}
+_ts_decorate([
+    trackSpan('MemoryProvingQueue.poll')
+], MemoryProvingQueue.prototype, "poll", null);