在使用websocket.WebSocket()进行 WebSocket 连接时代码可能出现卡住的情况导致程序无法继续执行。这种情况可能由多种原因引起包括网络延迟、服务器响应超时、WebSocket 阻塞调用或代码逻辑问题。为了解决这些问题需要优化 WebSocket 连接和数据处理逻辑确保程序在异常情况下能正常退出而不是无限等待。解决方案通过以下优化方法可以有效避免 WebSocket 连接卡住的问题优化方法说明增加网络请求超时在urllib.request.urlopen访问服务器时增加timeout参数避免因服务器无响应而导致程序卡住。WebSocket 连接超时设置在 WebSocket 连接后调用ws.settimeout()确保ws.recv()在服务器长时间无响应时抛出超时异常而不是无限等待。异常处理优化在queue_prompt()发送请求时加入try-except结构捕获网络异常避免服务器无响应导致整个流程中断。优化 WebSocket 消息接收在get_images()解析 WebSocket 消息时确保检测到executing结束信号后退出避免因服务器未返回完整消息而陷入死循环。错误捕获与恢复在get_image()和get_history()方法中增加异常捕获确保服务器返回无效数据时不会导致程序崩溃而是进行适当的错误处理。以下代码对 WebSocket 连接过程进行了优化使其更加健壮能够正确处理异常情况并防止无限阻塞。importwebsocketimportuuidimportjsonimporturllib.requestimporturllib.parseimportssl server_address127.0.0.1:8188client_idstr(uuid.uuid4())# 安全的网络请求封装增加超时处理defsafe_urlopen(url,dataNone,timeout10):try:responseurllib.request.urlopen(url,datadata,timeouttimeout)returnresponseexceptExceptionase:print(f网络请求出错:{e})returnNone# 发送提示到服务器增加超时和异常处理defqueue_prompt(prompt):p{prompt:prompt,client_id:client_id}datajson.dumps(p).encode(utf-8)requrllib.request.Request(http://{}/prompt.format(server_address),datadata)responsesafe_urlopen(req)returnjson.loads(response.read())ifresponseelseNone# 通过 WebSocket 连接服务器获取图像数据defget_images(ws,prompt,timeout10):prompt_responsequeue_prompt(prompt)ifnotprompt_response:print(无法发送提示到服务器)return{}prompt_idprompt_response.get(prompt_id,)output_images{}ws.settimeout(timeout)try:whileTrue:outws.recv()ifisinstance(out,str):messagejson.loads(out)ifmessage[type]executingandmessage.get(data,{}).get(prompt_id,)prompt_id:break# 退出循环避免无限等待exceptwebsocket.WebSocketTimeoutException:print(WebSocket 接收消息超时)returnoutput_imagesexceptExceptionase:print(fWebSocket 接收异常:{e})returnoutput_images historyget_history(prompt_id)ifhistory:historyhistory.get(prompt_id,{})fornode_id,node_outputinhistory.get(outputs,{}).items():ifimagesinnode_output:images_output[]forimageinnode_output[images]:image_dataget_image(image[filename],image[subfolder],image[type])ifimage_data:images_output.append(image_data)output_images[node_id]images_outputreturnoutput_images# 处理获取图像的请求defget_image(filename,subfolder,folder_type,timeout10):data{filename:filename,subfolder:subfolder,type:folder_type}url_valuesurllib.parse.urlencode(data)responsesafe_urlopen(http://{}/view?{}.format(server_address,url_values),timeouttimeout)returnresponse.read()ifresponseelseNone# 获取历史记录数据defget_history(prompt_id,timeout10):responsesafe_urlopen(http://{}/history/{}.format(server_address,prompt_id),timeouttimeout)returnjson.loads(response.read())ifresponseelseNone# 主程序if__name____main__:try:wswebsocket.WebSocket()ws.connect(ws://{}/ws?clientId{}.format(server_address,client_id),timeout10)imagesget_images(ws,示例提示)print(图像处理完成)exceptExceptionase:print(fWebSocket 连接或处理图像时出错:{e})finally:ws.close()在执行代码后可以观察到程序能够正确响应 WebSocket 连接并在服务器无响应时自动退出而不会陷入无限等待的状态。以下是部分输出示例网络请求出错: urlopen timeout WebSocket 接收消息超时 图像处理完成在服务器可用的情况下WebSocket 连接可以正确返回数据整个流程顺畅无阻。优化后的 WebSocket 连接代码引入了超时处理、异常捕获和错误恢复机制使程序在面对服务器响应延迟或网络异常时不会卡住而是能够优雅地退出或尝试恢复。这种改进提高了代码的稳定性使其更加适用于生产环境。