comfyui-node 1.6.0 → 1.6.2

package/dist/multipool/job-state-registry.js

@@ -0,0 +1,283 @@
+import { randomUUID } from "node:crypto";
+import { JobProfiler } from "./job-profiler.js";
+export class JobStateRegistry {
+    pool;
+    clients;
+    // Map of jobId to JobState
+    jobs = new Map();
+    // Map of prompt_id to jobId
+    promptIdToJobId = new Map();
+    constructor(pool, clients) {
+        this.pool = pool;
+        this.clients = clients;
+    }
+    addJob(workflow) {
+        // Create new job id
+        const jobId = randomUUID();
+        let resolver = null;
+        const resultsPromise = new Promise((resolve) => {
+            resolver = resolve;
+        });
+        const jobState = {
+            jobId,
+            workflow,
+            status: "pending",
+            resolver,
+            resultsPromise
+        };
+        // Initialize profiler if enabled
+        if (this.pool.options.enableProfiling) {
+            jobState.profiler = new JobProfiler(Date.now(), workflow.toJSON());
+        }
+        this.jobs.set(jobId, jobState);
+        return jobId;
+    }
+    getJobStatus(jobId) {
+        const jobState = this.jobs.get(jobId);
+        if (!jobState) {
+            throw new Error(`Job with ID ${jobId} not found.`);
+        }
+        return jobState.status;
+    }
+    async cancelJob(jobId) {
+        const jobState = this.jobs.get(jobId);
+        if (!jobState) {
+            throw new Error(`Job with ID ${jobId} not found.`);
+        }
+        if (jobState.status === "completed" || jobState.status === "canceled") {
+            throw new Error(`Cannot cancel job ${jobId} with status ${jobState.status}.`);
+        }
+        if (jobState.status === "assigned" || jobState.status === "running") {
+            // Notify assigned client to cancel the job
+            if (jobState.assignedClientUrl) {
+                const client = this.clients.clients.get(jobState.assignedClientUrl);
+                if (client) {
+                    try {
+                        await client.api.ext.queue.interrupt(jobState.prompt_id);
+                        // Mark job as canceled
+                        jobState.status = "canceled";
+                        // Mark client as idle
+                        client.state = "idle";
+                        // Also resolve the promise to avoid hanging
+                        if (jobState.resolver) {
+                            const results = {
+                                status: "canceled",
+                                jobId: jobState.jobId,
+                                prompt_id: jobState.prompt_id,
+                                images: []
+                            };
+                            jobState.resolver(results);
+                            jobState.resolver = null;
+                        }
+                        // Process the queue to allow next job to proceed
+                        this.processQueue(jobState.workflow.structureHash);
+                    }
+                    catch (e) {
+                        console.error(`Failed to notify client ${jobState.assignedClientUrl} to cancel job ${jobId}:`, e);
+                    }
+                }
+            }
+        }
+        else {
+            // For pending or no_clients status, just mark as canceled
+            jobState.status = "canceled";
+            // Also resolve the promise to avoid hanging
+            if (jobState.resolver) {
+                const results = {
+                    status: "canceled",
+                    jobId: jobState.jobId,
+                    prompt_id: jobState.prompt_id,
+                    images: []
+                };
+                jobState.resolver(results);
+                jobState.resolver = null;
+            }
+            // Remove from queue if necessary
+            this.removeJobFromQueue(jobState);
+        }
+    }
+    setJobStatus(jobId, newStatus, assignedClientUrl) {
+        const jobState = this.jobs.get(jobId);
+        if (!jobState) {
+            throw new Error(`Job with ID ${jobId} not found.`);
+        }
+        jobState.status = newStatus;
+        if (assignedClientUrl) {
+            jobState.assignedClientUrl = assignedClientUrl;
+        }
+    }
+    updateJobAutoSeeds(jobId, autoSeeds) {
+        const jobState = this.jobs.get(jobId);
+        if (!jobState) {
+            throw new Error(`Job with ID ${jobId} not found.`);
+        }
+        jobState.autoSeeds = autoSeeds;
+    }
+    setPromptId(jobId, prompt_id) {
+        const jobState = this.jobs.get(jobId);
+        if (!jobState) {
+            throw new Error(`Job with ID ${jobId} not found.`);
+        }
+        jobState.prompt_id = prompt_id;
+        this.promptIdToJobId.set(prompt_id, jobId);
+        // Notify profiler of execution start
+        if (jobState.profiler) {
+            jobState.profiler.onExecutionStart(prompt_id);
+        }
+    }
+    completeJob(prompt_id) {
+        const jobState = this.jobs.get(this.promptIdToJobId.get(prompt_id) || "");
+        if (!jobState || !jobState.prompt_id) {
+            throw new Error(`No job state found for prompt_id ${prompt_id} when completing job.`);
+        }
+        if (jobState.prompt_id === prompt_id) {
+            jobState.status = "completed";
+            // Notify profiler of completion
+            if (jobState.profiler) {
+                jobState.profiler.onExecutionComplete();
+            }
+            if (jobState.resolver) {
+                const results = {
+                    status: "completed",
+                    jobId: jobState.jobId,
+                    prompt_id: jobState.prompt_id,
+                    images: []
+                };
+                // Prepare images
+                if (jobState.images && jobState.images.length > 0 && jobState.assignedClientUrl) {
+                    const client = this.clients.clients.get(jobState.assignedClientUrl);
+                    if (client) {
+                        for (let i = 0; i < jobState.images.length; i++) {
+                            const image = jobState.images[i];
+                            const imageUrl = client.api.ext.file.getPathImage(image);
+                            results.images.push(imageUrl);
+                        }
+                    }
+                }
+                // Add profiler stats if available
+                if (jobState.profiler) {
+                    results.profileStats = jobState.profiler.getStats();
+                }
+                jobState.resolver(results);
+                jobState.resolver = null;
+            }
+        }
+    }
+    processQueue(structureHash) {
+        let queue = this.pool.queues.get(structureHash || "general");
+        if (queue) {
+            queue.processQueue().catch(reason => {
+                console.error(`Error processing job queue for workflow hash ${structureHash}:`, reason);
+            });
+        }
+    }
+    async waitForResults(jobId) {
+        const jobState = this.jobs.get(jobId);
+        if (!jobState) {
+            throw new Error(`Job with ID ${jobId} not found.`);
+        }
+        if (!jobState.resultsPromise) {
+            throw new Error(`Job with ID ${jobId} does not have a results promise.`);
+        }
+        return jobState.resultsPromise;
+    }
+    addJobImages(prompt_id, images) {
+        const state = this.jobs.get(this.promptIdToJobId.get(prompt_id) || "");
+        if (!state) {
+            throw new Error(`No job state found for prompt_id ${prompt_id} when adding images.`);
+        }
+        if (state.prompt_id === prompt_id) {
+            state.images = [...images];
+            return;
+        }
+    }
+    removeJobFromQueue(jobState) {
+        let queue = this.pool.queues.get(jobState.workflow.structureHash || "general");
+        if (queue) {
+            queue.dequeueJob(jobState.jobId);
+        }
+    }
+    attachJobProgressListener(jobId, progressListener) {
+        const jobState = this.jobs.get(jobId);
+        if (!jobState) {
+            throw new Error(`Job with ID ${jobId} not found.`);
+        }
+        jobState.onProgress = progressListener;
+    }
+    attachJobPreviewListener(jobId, previewListener) {
+        const jobState = this.jobs.get(jobId);
+        if (!jobState) {
+            throw new Error(`Job with ID ${jobId} not found.`);
+        }
+        jobState.onPreview = previewListener;
+    }
+    updateJobProgress(prompt_id, value, max, nodeId) {
+        const state = this.jobs.get(this.promptIdToJobId.get(prompt_id) || "");
+        if (!state) {
+            console.warn(`No job state found for prompt_id ${prompt_id} when updating progress.`);
+            return;
+        }
+        if (state.onProgress && state.prompt_id === prompt_id) {
+            state.onProgress({ value, max });
+        }
+        // Notify profiler
+        if (state.profiler && nodeId !== undefined) {
+            state.profiler.onProgress(nodeId, value, max);
+        }
+    }
+    updateJobPreviewMetadata(prompt_id, metadata, blob) {
+        const state = this.jobs.get(this.promptIdToJobId.get(prompt_id) || "");
+        if (!state) {
+            console.warn(`No job state found for prompt_id ${prompt_id} when updating preview metadata.`);
+            return;
+        }
+        if (state.onPreview && state.prompt_id === prompt_id) {
+            state.onPreview({ metadata, blob });
+        }
+    }
+    setJobFailure(jobId, bodyJSON) {
+        const jobState = this.jobs.get(jobId);
+        if (!jobState) {
+            throw new Error(`Job with ID ${jobId} not found.`);
+        }
+        jobState.status = "failed";
+        // Notify profiler of completion (even on failure)
+        if (jobState.profiler) {
+            jobState.profiler.onExecutionComplete();
+        }
+        if (jobState.resolver) {
+            const results = {
+                status: "failed",
+                jobId: jobState.jobId,
+                prompt_id: jobState.prompt_id,
+                images: [],
+                error: bodyJSON
+            };
+            // Add profiler stats even on failure
+            if (jobState.profiler) {
+                results.profileStats = jobState.profiler.getStats();
+            }
+            jobState.resolver(results);
+            jobState.resolver = null;
+        }
+    }
+    /**
+     * Track node execution start for profiling
+     */
+    onNodeExecuting(prompt_id, nodeId) {
+        const state = this.jobs.get(this.promptIdToJobId.get(prompt_id) || "");
+        if (state?.profiler && state.prompt_id === prompt_id) {
+            state.profiler.onNodeExecuting(nodeId);
+        }
+    }
+    /**
+     * Track cached nodes for profiling
+     */
+    onCachedNodes(prompt_id, nodeIds) {
+        const state = this.jobs.get(this.promptIdToJobId.get(prompt_id) || "");
+        if (state?.profiler && state.prompt_id === prompt_id) {
+            state.profiler.onCachedNodes(nodeIds);
+        }
+    }
+}
+//# sourceMappingURL=job-state-registry.js.map

package/dist/multipool/job-state-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"job-state-registry.js","sourceRoot":"","sources":["../../src/multipool/job-state-registry.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzC,OAAO,EAAE,WAAW,EAAmB,MAAM,mBAAmB,CAAC;AA6BjE,MAAM,OAAO,gBAAgB;IAE3B,IAAI,CAAoB;IACxB,OAAO,CAAiB;IAExB,2BAA2B;IAC3B,IAAI,GAA0B,IAAI,GAAG,EAAE,CAAC;IAExC,4BAA4B;IAC5B,eAAe,GAAwB,IAAI,GAAG,EAAE,CAAC;IAEjD,YAAY,IAAuB,EAAE,OAAuB;QAC1D,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,MAAM,CAAC,QAAkB;QACvB,oBAAoB;QACpB,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;QAC3B,IAAI,QAAQ,GAA2C,IAAI,CAAC;QAC5D,MAAM,cAAc,GAAG,IAAI,OAAO,CAAa,CAAC,OAAO,EAAE,EAAE;YACzD,QAAQ,GAAG,OAAO,CAAC;QACrB,CAAC,CAAC,CAAC;QACH,MAAM,QAAQ,GAAa;YACzB,KAAK;YACL,QAAQ;YACR,MAAM,EAAE,SAAS;YACjB,QAAQ;YACR,cAAc;SACf,CAAC;QAEF,iCAAiC;QACjC,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC;YACtC,QAAQ,CAAC,QAAQ,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QACrE,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QAC/B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,YAAY,CAAC,KAAa;QACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,eAAe,KAAK,aAAa,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,QAAQ,CAAC,MAAM,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,KAAa;QAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,eAAe,KAAK,aAAa,CAAC,CAAC;QACrD,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,KAAK,WAAW,IAAI,QAAQ,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YACtE,MAAM,IAAI,KAAK,CAAC,qBAAqB,KAAK,gBAAgB,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;QAChF,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,KAAK,UAAU,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YACpE,2CAA2C;YAC3C,IAAI,QAAQ,CAAC,iBAAiB,EAAE,CAAC;gBAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;gBACpE,IAAI,MAAM,EAAE,CAAC;oBACX,IAAI,CAAC;wBAEH,MAAM,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;wBAEzD,uBAAuB;wBACvB,QAAQ,CAAC,MAAM,GAAG,UAAU,CAAC;wBAE7B,sBAAsB;wBACtB,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC;wBAEtB,4CAA4C;wBAC5C,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;4BACtB,MAAM,OAAO,GAAe;gCAC1B,MAAM,EAAE,UAAU;gCAClB,KAAK,EAAE,QAAQ,CAAC,KAAK;gCACrB,SAAS,EAAE,QAAQ,CAAC,SAAU;gCAC9B,MAAM,EAAE,EAAE;6BACX,CAAC;4BACF,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;4BAC3B,QAAQ,CAAC,QAAQ,GAAG,IAAI,CAAC;wBAC3B,CAAC;wBAED,iDAAiD;wBACjD,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;oBAErD,CAAC;oBAAC,OAAO,CAAC,EAAE,CAAC;wBACX,OAAO,CAAC,KAAK,CAAC,2BAA2B,QAAQ,CAAC,iBAAiB,kBAAkB,KAAK,GAAG,EAAE,CAAC,CAAC,CAAC;oBACpG,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;aAAM,CAAC;YACN,0DAA0D;YAC1D,QAAQ,CAAC,MAAM,GAAG,UAAU,CAAC;YAE7B,4CAA4C;YAC5C,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;gBACtB,MAAM,OAAO,GAAe;oBAC1B,MAAM,EAAE,UAAU;oBAClB,KAAK,EAAE,QAAQ,CAAC,KAAK;oBACrB,SAAS,EAAE,QAAQ,CAAC,SAAU;oBAC9B,MAAM,EAAE,EAAE;iBACX,CAAC;gBACF,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;gBAC3B,QAAQ,CAAC,QAAQ,GAAG,IAAI,CAAC;YAC3B,CAAC;YAED,iCAAiC;YACjC,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED,YAAY,CAAC,KAAa,EAAE,SAAoB,EAAE,iBAA0B;QAC1E,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,eAAe,KAAK,aAAa,CAAC,CAAC;QACrD,CAAC;QACD,QAAQ,CAAC,MAAM,GAAG,SAAsB,CAAC;QACzC,IAAI,iBAAiB,EAAE,CAAC;YACtB,QAAQ,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QACjD,CAAC;IACH,CAAC;IAED,kBAAkB,CAAC,KAAa,EAAE,SAAiC;QACjE,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,eAAe,KAAK,aAAa,CAAC,CAAC;QACrD,CAAC;QACD,QAAQ,CAAC,SAAS,GAAG,SAAS,CAAC;IACjC,CAAC;IAED,WAAW,CAAC,KAAa,EAAE,SAAiB;QAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,eAAe,KAAK,aAAa,CAAC,CAAC;QACrD,CAAC;QACD,QAAQ,CAAC,SAAS,GAAG,SAAS,CAAC;QAC/B,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAE3C,qCAAqC;QACrC,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACtB,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,WAAW,CAAC,SAAiB;QAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;QAC1E,IAAI,CAAC,QAAQ,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;YACrC,MAAM,IAAI,KAAK,CAAC,oCAAoC,SAAS,uBAAuB,CAAC,CAAC;QACxF,CAAC;QACD,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACrC,QAAQ,CAAC,MAAM,GAAG,WAAW,CAAC;YAE9B,gCAAgC;YAChC,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;gBACtB,QAAQ,CAAC,QAAQ,CAAC,mBAAmB,EAAE,CAAC;YAC1C,CAAC;YAED,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;gBACtB,MAAM,OAAO,GAAe;oBAC1B,MAAM,EAAE,WAAW;oBACnB,KAAK,EAAE,QAAQ,CAAC,KAAK;oBACrB,SAAS,EAAE,QAAQ,CAAC,SAAU;oBAC9B,MAAM,EAAE,EAAE;iBACX,CAAC;gBAEF,iBAAiB;gBACjB,IAAI,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,QAAQ,CAAC,iBAAiB,EAAE,CAAC;oBAChF,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;oBACpE,IAAI,MAAM,EAAE,CAAC;wBACX,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;4BAChD,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;4BACjC,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;4BACzD,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;wBAChC,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,kCAAkC;gBAClC,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;oBACtB,OAAO,CAAC,YAAY,GAAG,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;gBACtD,CAAC;gBAED,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;gBAC3B,QAAQ,CAAC,QAAQ,GAAG,IAAI,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAEO,YAAY,CAAC,aAAiC;QACpD,IAAI,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,aAAa,IAAI,SAAS,CAAC,CAAC;QAC7D,IAAI,KAAK,EAAE,CAAC;YACV,KAAK,CAAC,YAAY,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE;gBAClC,OAAO,CAAC,KAAK,CAAC,gDAAgD,aAAa,GAAG,EAAE,MAAM,CAAC,CAAC;YAC1F,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAAa;QAChC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,eAAe,KAAK,aAAa,CAAC,CAAC;QACrD,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,eAAe,KAAK,mCAAmC,CAAC,CAAC;QAC3E,CAAC;QACD,OAAO,QAAQ,CAAC,cAAc,CAAC;IACjC,CAAC;IAED,YAAY,CAAC,SAAiB,EAAE,MAAmB;QACjD,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;QACvE,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,oCAAoC,SAAS,sBAAsB,CAAC,CAAC;QACvF,CAAC;QACD,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YAClC,KAAK,CAAC,MAAM,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC;YAC3B,OAAO;QACT,CAAC;IACH,CAAC;IAEO,kBAAkB,CAAC,QAAkB;QAC3C,IAAI,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,aAAa,IAAI,SAAS,CAAC,CAAC;QAC/E,IAAI,KAAK,EAAE,CAAC;YACV,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAED,yBAAyB,CAAC,KAAa,EAAE,gBAG/B;QACR,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,eAAe,KAAK,aAAa,CAAC,CAAC;QACrD,CAAC;QACD,QAAQ,CAAC,UAAU,GAAG,gBAAgB,CAAC;IACzC,CAAC;IAED,wBAAwB,CAAC,KAAa,EAAE,eAG9B;QACR,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,eAAe,KAAK,aAAa,CAAC,CAAC;QACrD,CAAC;QACD,QAAQ,CAAC,SAAS,GAAG,eAAe,CAAC;IACvC,CAAC;IAED,iBAAiB,CAAC,SAAiB,EAAE,KAAa,EAAE,GAAW,EAAE,MAAwB;QACvF,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;QACvE,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,oCAAoC,SAAS,0BAA0B,CAAC,CAAC;YACtF,OAAO;QACT,CAAC;QACD,IAAI,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACtD,KAAK,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QACnC,CAAC;QAED,kBAAkB;QAClB,IAAI,KAAK,CAAC,QAAQ,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YAC3C,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,wBAAwB,CAAC,SAAc,EAAE,QAAa,EAAE,IAAU;QAChE,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;QACvE,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,oCAAoC,SAAS,kCAAkC,CAAC,CAAC;YAC9F,OAAO;QACT,CAAC;QACD,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACrD,KAAK,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED,aAAa,CAAC,KAAa,EAAE,QAAa;QACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,eAAe,KAAK,aAAa,CAAC,CAAC;QACrD,CAAC;QACD,QAAQ,CAAC,MAAM,GAAG,QAAQ,CAAC;QAE3B,kDAAkD;QAClD,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACtB,QAAQ,CAAC,QAAQ,CAAC,mBAAmB,EAAE,CAAC;QAC1C,CAAC;QAED,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACtB,MAAM,OAAO,GAAe;gBAC1B,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE,QAAQ,CAAC,KAAK;gBACrB,SAAS,EAAE,QAAQ,CAAC,SAAU;gBAC9B,MAAM,EAAE,EAAE;gBACV,KAAK,EAAE,QAAQ;aAChB,CAAC;YAEF,qCAAqC;YACrC,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;gBACtB,OAAO,CAAC,YAAY,GAAG,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACtD,CAAC;YAED,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAC3B,QAAQ,CAAC,QAAQ,GAAG,IAAI,CAAC;QAC3B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,SAAiB,EAAE,MAAc;QAC/C,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;QACvE,IAAI,KAAK,EAAE,QAAQ,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACrD,KAAK,CAAC,QAAQ,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,SAAiB,EAAE,OAAiB;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;QACvE,IAAI,KAAK,EAAE,QAAQ,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACrD,KAAK,CAAC,QAAQ,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;CACF"}

package/dist/multipool/logger.d.ts

@@ -0,0 +1,30 @@
+/**
+ * Logging utility for MultiWorkflowPool
+ * Provides structured logging with configurable log levels
+ */
+export type LogLevel = "debug" | "info" | "warn" | "error" | "silent";
+export declare class Logger {
+    private level;
+    private context;
+    constructor(context: string, level?: LogLevel);
+    setLevel(level: LogLevel): void;
+    private shouldLog;
+    private formatMessage;
+    debug(message: string, ...args: any[]): void;
+    info(message: string, ...args: any[]): void;
+    warn(message: string, ...args: any[]): void;
+    error(message: string, ...args: any[]): void;
+    /**
+     * Special log for queue operations (always uses timestamp format)
+     */
+    queue(workflowHash: string, message: string, ...args: any[]): void;
+    /**
+     * Special log for client events
+     */
+    client(clientName: string, event: string, message: string, ...args: any[]): void;
+}
+/**
+ * Create a logger instance for a specific context
+ */
+export declare function createLogger(context: string, level?: LogLevel): Logger;
+//# sourceMappingURL=logger.d.ts.map

package/dist/multipool/logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../src/multipool/logger.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,MAAM,MAAM,QAAQ,GAAG,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,QAAQ,CAAC;AAUtE,qBAAa,MAAM;IACjB,OAAO,CAAC,KAAK,CAAW;IACxB,OAAO,CAAC,OAAO,CAAS;gBAEZ,OAAO,EAAE,MAAM,EAAE,KAAK,GAAE,QAAiB;IAKrD,QAAQ,CAAC,KAAK,EAAE,QAAQ;IAIxB,OAAO,CAAC,SAAS;IAIjB,OAAO,CAAC,aAAa;IAMrB,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE;IAMrC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE;IAMpC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE;IAMpC,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE;IAMrC;;OAEG;IACH,KAAK,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE;IAQ3D;;OAEG;IACH,MAAM,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE;CAK1E;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,GAAE,QAAiB,GAAG,MAAM,CAE9E"}

package/dist/multipool/logger.js

@@ -0,0 +1,75 @@
+/**
+ * Logging utility for MultiWorkflowPool
+ * Provides structured logging with configurable log levels
+ */
+const LOG_LEVELS = {
+    debug: 0,
+    info: 1,
+    warn: 2,
+    error: 3,
+    silent: 4
+};
+export class Logger {
+    level;
+    context;
+    constructor(context, level = "info") {
+        this.context = context;
+        this.level = level;
+    }
+    setLevel(level) {
+        this.level = level;
+    }
+    shouldLog(level) {
+        return LOG_LEVELS[level] >= LOG_LEVELS[this.level];
+    }
+    formatMessage(level, message, ...args) {
+        const timestamp = new Date().toISOString();
+        const prefix = `[${timestamp}] [${level.toUpperCase()}] [${this.context}]`;
+        return args.length > 0 ? `${prefix} ${message}` : `${prefix} ${message}`;
+    }
+    debug(message, ...args) {
+        if (this.shouldLog("debug")) {
+            console.debug(this.formatMessage("debug", message), ...args);
+        }
+    }
+    info(message, ...args) {
+        if (this.shouldLog("info")) {
+            console.info(this.formatMessage("info", message), ...args);
+        }
+    }
+    warn(message, ...args) {
+        if (this.shouldLog("warn")) {
+            console.warn(this.formatMessage("warn", message), ...args);
+        }
+    }
+    error(message, ...args) {
+        if (this.shouldLog("error")) {
+            console.error(this.formatMessage("error", message), ...args);
+        }
+    }
+    /**
+     * Special log for queue operations (always uses timestamp format)
+     */
+    queue(workflowHash, message, ...args) {
+        if (this.shouldLog("debug")) {
+            const timestamp = new Date().toISOString();
+            const hashPrefix = workflowHash.substring(0, 16);
+            console.log(`[${timestamp}] [queue::${hashPrefix}] ${message}`, ...args);
+        }
+    }
+    /**
+     * Special log for client events
+     */
+    client(clientName, event, message, ...args) {
+        if (this.shouldLog("debug")) {
+            console.log(`[${event}@${clientName}] ${message}`, ...args);
+        }
+    }
+}
+/**
+ * Create a logger instance for a specific context
+ */
+export function createLogger(context, level = "info") {
+    return new Logger(context, level);
+}
+//# sourceMappingURL=logger.js.map

package/dist/multipool/logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.js","sourceRoot":"","sources":["../../src/multipool/logger.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAIH,MAAM,UAAU,GAA6B;IAC3C,KAAK,EAAE,CAAC;IACR,IAAI,EAAE,CAAC;IACP,IAAI,EAAE,CAAC;IACP,KAAK,EAAE,CAAC;IACR,MAAM,EAAE,CAAC;CACV,CAAC;AAEF,MAAM,OAAO,MAAM;IACT,KAAK,CAAW;IAChB,OAAO,CAAS;IAExB,YAAY,OAAe,EAAE,QAAkB,MAAM;QACnD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,QAAQ,CAAC,KAAe;QACtB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAEO,SAAS,CAAC,KAAe;QAC/B,OAAO,UAAU,CAAC,KAAK,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrD,CAAC;IAEO,aAAa,CAAC,KAAa,EAAE,OAAe,EAAE,GAAG,IAAW;QAClE,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,SAAS,MAAM,KAAK,CAAC,WAAW,EAAE,MAAM,IAAI,CAAC,OAAO,GAAG,CAAC;QAC3E,OAAO,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,OAAO,EAAE,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,OAAO,EAAE,CAAC;IAC3E,CAAC;IAED,KAAK,CAAC,OAAe,EAAE,GAAG,IAAW;QACnC,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5B,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,IAAI,CAAC,OAAe,EAAE,GAAG,IAAW;QAClC,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,IAAI,CAAC,OAAe,EAAE,GAAG,IAAW;QAClC,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAe,EAAE,GAAG,IAAW;QACnC,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5B,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAoB,EAAE,OAAe,EAAE,GAAG,IAAW;QACzD,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5B,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YAC3C,MAAM,UAAU,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACjD,OAAO,CAAC,GAAG,CAAC,IAAI,SAAS,aAAa,UAAU,KAAK,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,UAAkB,EAAE,KAAa,EAAE,OAAe,EAAE,GAAG,IAAW;QACvE,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5B,OAAO,CAAC,GAAG,CAAC,IAAI,KAAK,IAAI,UAAU,KAAK,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,YAAY,CAAC,OAAe,EAAE,QAAkB,MAAM;IACpE,OAAO,IAAI,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;AACpC,CAAC"}

package/dist/multipool/multi-workflow-pool.d.ts

@@ -0,0 +1,43 @@
+import { JobResults } from "./job-state-registry.js";
+import { JobQueueProcessor } from "./job-queue-processor.js";
+import { Workflow } from "./workflow.js";
+import { MultiWorkflowPoolOptions, PoolEvent } from "./interfaces.js";
+/**
+ * MultiWorkflowPool class to manage heterogeneous clusters of ComfyUI workers with different workflow capabilities.
+ * Using a fully event driven architecture to handle client connections, job submissions, and failover strategies.
+ * Zero polling is used; all operations are event driven. Maximizes responsiveness and scalability.
+ */
+export declare class MultiWorkflowPool {
+    private events;
+    private clientRegistry;
+    private jobRegistry;
+    queues: Map<string, JobQueueProcessor>;
+    options: Required<MultiWorkflowPoolOptions>;
+    private logger;
+    monitoringInterval?: Timer;
+    constructor(options?: MultiWorkflowPoolOptions);
+    init(): Promise<void>;
+    shutdown(): Promise<void>;
+    addClient(clientUrl: string, options?: {
+        workflowAffinity: Workflow<any>[];
+        priority?: number;
+    }): void;
+    removeClient(clientUrl: string): void;
+    submitJob(workflow: Workflow<any>): Promise<string>;
+    getJobStatus(jobId: string): import("./job-state-registry.js").JobStatus;
+    cancelJob(jobId: string): Promise<void>;
+    attachEventHook(event: string, listener: (e: PoolEvent) => void): void;
+    private assertQueue;
+    private attachHandlersToClient;
+    private printStatusSummary;
+    waitForJobCompletion(jobId: string): Promise<JobResults>;
+    attachJobProgressListener(jobId: string, progressListener: (progress: {
+        value: number;
+        max: number;
+    }) => void): void;
+    attachJobPreviewListener(jobId: string, previewListener: (preview: {
+        blob: Blob;
+        metadata: any;
+    }) => void): void;
+}
+//# sourceMappingURL=multi-workflow-pool.d.ts.map

package/dist/multipool/multi-workflow-pool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"multi-workflow-pool.d.ts","sourceRoot":"","sources":["../../src/multipool/multi-workflow-pool.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,UAAU,EAAoB,MAAM,yBAAyB,CAAC;AACvE,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAC7D,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,wBAAwB,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAGtE;;;;GAIG;AACH,qBAAa,iBAAiB;IAG5B,OAAO,CAAC,MAAM,CAAmB;IAGjC,OAAO,CAAC,cAAc,CAAiB;IAGvC,OAAO,CAAC,WAAW,CAAmB;IAGtC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAa;IAG5C,OAAO,EAAE,QAAQ,CAAC,wBAAwB,CAAC,CAAC;IAGnD,OAAO,CAAC,MAAM,CAAS;IAEvB,kBAAkB,CAAC,EAAE,KAAK,CAAC;gBAEf,OAAO,CAAC,EAAE,wBAAwB;IA4BxC,IAAI;IAyDJ,QAAQ;IA2Bd,SAAS,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QACrC,gBAAgB,EAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QAClC,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB;IAID,YAAY,CAAC,SAAS,EAAE,MAAM;IAIxB,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC;IA0BvC,YAAY,CAAC,KAAK,EAAE,MAAM;IAIpB,SAAS,CAAC,KAAK,EAAE,MAAM;IAI7B,eAAe,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC,EAAE,SAAS,KAAK,IAAI;IAO/D,OAAO,CAAC,WAAW;IAYnB,OAAO,CAAC,sBAAsB;IAsG9B,OAAO,CAAC,kBAAkB;IAmCpB,oBAAoB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC;IAI9D,yBAAyB,CAAC,KAAK,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,QAAQ,EAAE;QACpE,KAAK,EAAE,MAAM,CAAC;QACd,GAAG,EAAE,MAAM,CAAC;KACb,KAAK,IAAI;IAIV,wBAAwB,CAAC,KAAK,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC,OAAO,EAAE;QACjE,IAAI,EAAE,IAAI,CAAC;QACX,QAAQ,EAAE,GAAG,CAAC;KACf,KAAK,IAAI;CAGX"}