wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-29 21:13:33 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

fix: ensure runner cleanup on any unix socket startup failure

Browse Source
This commit is contained in:
hanli 2026-02-28 14:03:46 +08:00
parent 397bd47bc2
commit 807e9fbe88
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
server.py
View File

@ -1225,9 +1225,6 @@ class PromptServer():
if sys.platform == 'win32': if sys.platform == 'win32':
raise RuntimeError("Unix sockets are not supported on Windows. Please use --listen and --port instead.") raise RuntimeError("Unix sockets are not supported on Windows. Please use --listen and --port instead.")
runner = web.AppRunner(self.app, access_log=None)
await runner.setup()
if verbose: if verbose:
logging.info("Starting server\n") logging.info("Starting server\n")
@ -1236,14 +1233,19 @@ class PromptServer():
if not stat.S_ISSOCK(st_mode): if not stat.S_ISSOCK(st_mode):
raise RuntimeError(f"Refusing to remove non-socket path: {unix_socket}") raise RuntimeError(f"Refusing to remove non-socket path: {unix_socket}")
os.unlink(unix_socket) os.unlink(unix_socket)
site = web.UnixSite(runner, unix_socket)
await site.start() runner = web.AppRunner(self.app, access_log=None)
await runner.setup()
try: try:
site = web.UnixSite(runner, unix_socket)
await site.start()
os.chmod(unix_socket, 0o660) os.chmod(unix_socket, 0o660)
except OSError as e: except OSError as e:
await site.stop()
await runner.cleanup() await runner.cleanup()
raise RuntimeError(f"Failed to set socket permissions: {e}") raise RuntimeError(f"Failed to set socket permissions: {e}") from e
except Exception:
await runner.cleanup()
raise
self.address = unix_socket self.address = unix_socket
self.port = None self.port = None
self.unix_socket = unix_socket self.unix_socket = unix_socket