comfyui-node 1.6.2 → 1.6.4

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

@@ -1,43 +1,102 @@
-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;
-}
+import { JobQueueProcessor } from "./job-queue-processor.js";
+import { Workflow } from "./workflow.js";
+import { MultiWorkflowPoolOptions, PoolEvent, JobResults } 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("./interfaces.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;
+    /**
+     * Get a list of all registered clients with their current state
+     * @returns Array of client information objects
+     */
+    getClients(): Array<{
+        url: string;
+        nodeName: string;
+        state: "idle" | "busy" | "offline";
+        priority?: number;
+        workflowAffinityHashes?: string[];
+    }>;
+    /**
+     * Get information about a specific client by URL
+     * @param clientUrl - The URL of the client to query
+     * @returns Client information or null if not found
+     */
+    getClient(clientUrl: string): {
+        url: string;
+        nodeName: string;
+        state: "idle" | "busy" | "offline";
+        priority?: number;
+        workflowAffinityHashes?: string[];
+    } | null;
+    /**
+     * Get all clients that have affinity for a specific workflow
+     * @param workflow - The workflow to check affinity for
+     * @returns Array of client URLs that can handle this workflow
+     */
+    getClientsForWorkflow(workflow: Workflow<any>): string[];
+    /**
+     * Get all idle clients currently available for work
+     * @returns Array of idle client information
+     */
+    getIdleClients(): Array<{
+        url: string;
+        nodeName: string;
+        priority?: number;
+    }>;
+    /**
+     * Check if there are any clients available for a specific workflow
+     * @param workflow - The workflow to check
+     * @returns True if at least one client has affinity for this workflow
+     */
+    hasClientsForWorkflow(workflow: Workflow<any>): boolean;
+    /**
+     * Get statistics about the pool's current state
+     * @returns Pool statistics including client counts and queue depths
+     */
+    getPoolStats(): {
+        totalClients: number;
+        idleClients: number;
+        busyClients: number;
+        offlineClients: number;
+        totalQueues: number;
+        queues: Array<{
+            workflowHash: string;
+            pendingJobs: number;
+            type: "general" | "specific";
+        }>;
+    };
+}
 //# sourceMappingURL=multi-workflow-pool.d.ts.map

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

@@ -1 +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"}
+{"version":3,"file":"multi-workflow-pool.d.ts","sourceRoot":"","sources":["../../src/multipool/multi-workflow-pool.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAC7D,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,wBAAwB,EAAE,SAAS,EAAsC,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAGtH;;;;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;IAgH9B,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;IAKV;;;OAGG;IACH,UAAU,IAAI,KAAK,CAAC;QAClB,GAAG,EAAE,MAAM,CAAC;QACZ,QAAQ,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;QACnC,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,sBAAsB,CAAC,EAAE,MAAM,EAAE,CAAC;KACnC,CAAC;IAYF;;;;OAIG;IACH,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG;QAC5B,GAAG,EAAE,MAAM,CAAC;QACZ,QAAQ,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;QACnC,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,sBAAsB,CAAC,EAAE,MAAM,EAAE,CAAC;KACnC,GAAG,IAAI;IAgBR;;;;OAIG;IACH,qBAAqB,CAAC,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,GAAG,MAAM,EAAE;IAaxD;;;OAGG;IACH,cAAc,IAAI,KAAK,CAAC;QACtB,GAAG,EAAE,MAAM,CAAC;QACZ,QAAQ,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,CAAC;IAUF;;;;OAIG;IACH,qBAAqB,CAAC,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,GAAG,OAAO;IAYvD;;;OAGG;IACH,YAAY,IAAI;QACd,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC;QACpB,WAAW,EAAE,MAAM,CAAC;QACpB,cAAc,EAAE,MAAM,CAAC;QACvB,WAAW,EAAE,MAAM,CAAC;QACpB,MAAM,EAAE,KAAK,CAAC;YACZ,YAAY,EAAE,MAAM,CAAC;YACrB,WAAW,EAAE,MAAM,CAAC;YACpB,IAAI,EAAE,SAAS,GAAG,UAAU,CAAC;SAC9B,CAAC,CAAC;KACJ;CAeF"}