wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-10 01:02:56 +08:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
qwertzman-png 2023-04-21 01:41:18 +02:00 committed by GitHub
parent a2010185a0
commit 292bfb82b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 29 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
web/scripts/api.js
View File

@ -21,9 +21,37 @@ class ComfyApi extends EventTarget {
this.dispatchEvent(new CustomEvent("status", { detail: status }));
} catch (error) {
this.dispatchEvent(new CustomEvent("status", { detail: null }));
}
}, 1000);
}
#pollMessageQueue() {
let lastMessage = -1;
setInterval(async () => {
const poll_messages = await (await fetch("/poll_messages")).json();
if(lastMessage === -1) lastMessage = poll_messages.length -1;
for(let i = lastMessage; i<poll_messages.length; i++) {
const msg = poll_messages[i];
console.log('MSG', msg);
switch (msg.type) {
case "status":
this.dispatchEvent(new CustomEvent("status", { detail: msg.data.status }));
break;
case "progress":
this.dispatchEvent(new CustomEvent("progress", { detail: msg.data }));
break;
case "executing":
this.dispatchEvent(new CustomEvent("executing", { detail: msg.data.node }));
break;
case "executed":
this.dispatchEvent(new CustomEvent("executed", { detail: msg.data }));
break;
}
lastMessage = i;
}
}, 3000)
}
/**
* Creates and connects a WebSocket for realtime updates
@ -106,6 +134,7 @@ class ComfyApi extends EventTarget {
*/
init() {
this.#createSocket();
}
/**