本文旨在帮助开发者解决在使用 Jupyter Notebook API 通过 WebSocket 连接执行代码时遇到的 “socket is already closed” 错误。我们将分析错误原因，并提供通过重新连接 WebSocket 并确保消息格式正确来解决此问题的方案，确保代码能够顺利执行并接收到服务器的响应。
在使用 Jupyter Notebook API 通过 WebSocket 连接执行代码时，遇到 “socket is already closed” 错误通常表明 WebSocket 连接在接收到服务器响应之前意外关闭。这可能是由于多种原因造成的，例如网络问题、服务器端错误或客户端代码中的逻辑错误。以下提供一种解决方案，通过重新建立连接以及确保消息格式正确来解决此问题。
解决方案：重新连接 WebSocket 并修正消息格式
以下步骤描述了如何重新连接 WebSocket，并确保发送到 Jupyter Notebook 服务器的消息格式正确，从而解决 “socket is already closed” 错误。
错误分析：
从原始问题描述可以看出，错误发生在 ws.recv() 尝试接收服务器响应时。这暗示着 WebSocket 连接可能在发送执行请求后，但在收到响应前关闭了。Jupyter Notebook 服务器的日志也显示了一些警告信息，例如 “No session ID specified” 和 “No channel specified”，这表明客户端发送的请求可能缺少必要的参数。
重新连接 WebSocket：
在循环接收消息之前，如果检测到连接关闭，应该重新建立 WebSocket 连接。这可以通过将 create_connection 放在一个 try…except 块中，并在捕获到 WebSocketConnectionClosedException 异常时重新调用它来实现。
import jsonfrom websocket import create_connection, WebSocketConnectionClosedExceptionimport timedef execute_code(kernel_id, session_id, code, headers): ws_url = f"ws://127.0.0.1:8888/api/kernels/{kernel_id}/channels?session_id={session_id}" ws = create_connection(ws_url, header=headers) ws.send(json.dumps(send_execute_request(code))) try: while True: rsp = json.loads(ws.recv()) msg_type = rsp["msg_type"] # 处理不同类型的消息，例如 'execute_result', 'stream', 'error' 等 if msg_type == 'execute_result': # 处理执行结果 print("Execute Result:", rsp["content"]["data"]) break # 结束循环，因为我们已经得到了执行结果 elif msg_type == 'stream': # 处理输出流（stdout/stderr） print("Stream Output:", rsp["content"]["text"]) elif msg_type == 'error': # 处理错误信息 print("Error:", rsp["content"]["ename"], rsp["content"]["evalue"]) break # 结束循环，因为发生了错误 except WebSocketConnectionClosedException as e: print(f"WebSocket connection closed: {e}") # 在这里可以选择重新连接，或者抛出异常，取决于你的应用逻辑 # 例如： # ws = create_connection(ws_url, header=headers) # 尝试重新连接 raise # 抛出异常，向上层处理 finally: ws.close()
修正消息格式：
Jupyter Notebook 服务器的日志表明，请求可能缺少 session ID 和 channel 信息。确保在创建 WebSocket 连接时，URL 中包含了正确的 session_id 参数。此外，检查发送到 WebSocket 的消息格式是否符合 Jupyter Notebook API 的要求。 特别注意时间戳格式，需要包含时区信息。
import datetimeimport uuiddef send_execute_request(code): msg_id = str(uuid.uuid1()) session_id = str(uuid.uuid1()) # You can generate a new session ID for each request now = datetime.datetime.now(datetime.timezone.utc).isoformat() # Include timezone information msg = { "header": { "msg_id": msg_id, "username": "test", "session": session_id, "data": now, "msg_type": "execute_request", "version": "5.0" }, "parent_header": { "msg_id": msg_id, "username": "test", "session": session_id, "data": now, "msg_type": "execute_request", "version": "5.0" }, "metadata": {}, "content": { "code": code, "silent": False, "store_history": True, "user_expressions": {}, "allow_stdin": False }, "buffers": [], "channel": "shell" # Explicitly specify the channel } return msg
处理服务器响应：
修改后的代码示例中，execute_code 函数现在会处理不同类型的服务器响应（execute_result，stream，error）。请根据你的应用需求修改这些处理逻辑。
完整示例
import requests import json from websocket import create_connection, WebSocketConnectionClosedException import datetime import uuid base = "http://127.0.0.1:8888" # 替换为你的 Jupyter Notebook 地址 headers = {"Authorization": "Token your_token"} # 替换为你的 token def create_session(file_name): url = base + '/api/sessions' params = '{"path":"%s","type":"notebook","name":"","kernel":{"id":null,"name":"env37"}}' % file_name response = requests.post(url, headers=headers, data=params) session = json.loads(response.text) return session def get_notebook_content(notebook_path): url = base + '/api/contents' + notebook_path response = requests.get(url, headers=headers) file = json.loads(response.text) code = [c['source'] for c in file['content']['cells'] if len(c['source']) > 0] return code def send_execute_request(code): msg_id = str(uuid.uuid1()) session_id = str(uuid.uuid1()) # You can generate a new session ID for each request now = datetime.datetime.now(datetime.timezone.utc).isoformat() # Include timezone information msg = { "header": { "msg_id": msg_id, "username": "test", "session": session_id, "data": now, "msg_type": "execute_request", "version": "5.0" }, "parent_header": { "msg_id": msg_id, "username": "test", "session": session_id, "data": now, "msg_type": "execute_request", "version": "5.0" }, "metadata": {}, "content": { "code": code, "silent": False, "store_history": True, "user_expressions": {}, "allow_stdin": False }, "buffers": [], "channel": "shell" # Explicitly specify the channel } return msg def execute_code(kernel_id, session_id, code, headers): ws_url = f"ws://127.0.0.1:8888/api/kernels/{kernel_id}/channels?session_id={session_id}" ws = create_connection(ws_url, header=headers) ws.send(json.dumps(send_execute_request(code))) try: while True: rsp = json.loads(ws.recv()) msg_type = rsp["msg_type"] # 处理不同类型的消息，例如 'execute_result', 'stream', 'error' 等 if msg_type == 'execute_result': # 处理执行结果 print("Execute Result:", rsp["content"]["data"]) break # 结束循环，因为我们已经得到了执行结果 elif msg_type == 'stream': # 处理输出流（stdout/stderr） print("Stream Output:", rsp["content"]["text"]) elif msg_type == 'error': # 处理错误信息 print("Error:", rsp["content"]["ename"], rsp["content"]["evalue"]) break # 结束循环，因为发生了错误 except WebSocketConnectionClosedException as e: print(f"WebSocket connection closed: {e}") # 在这里可以选择重新连接，或者抛出异常，取决于你的应用逻辑 # 例如： # ws = create_connection(ws_url, header=headers) # 尝试重新连接 raise # 抛出异常，向上层处理 finally: ws.close() # Example usage: file_name = "example2.ipynb" # 替换为你的 notebook 文件名 notebook_path = "/" + file_name session = create_session(file_name) kernel = session["kernel"] kernel_id = kernel["id"] session_id = session["id"] code = get_notebook_content(notebook_path) for c in code: try: execute_code(kernel_id, session_id, c, headers) except WebSocketConnectionClosedException: print(f"Failed to execute code: {c}") # Handle reconnection or error as needed
注意事项
身份验证： 确保在请求头中包含正确的身份验证信息（例如，Token）。错误处理： 在实际应用中，需要更完善的错误处理机制，例如重试连接、记录错误日志等。依赖项： 确保已安装必要的 Python 库，例如 requests 和 websocket-client。Jupyter Notebook 配置： 检查 Jupyter Notebook 服务器的配置，确保 WebSocket 连接已启用，并且没有防火墙或代理阻止连接。
总结
通过重新连接 WebSocket 并确保消息格式正确，可以有效地解决 Jupyter Notebook API 中的 “socket is already closed” 错误。 此外，需要完善错误处理机制，以便在出现问题时能够及时发现并解决。
以上就是解决 Jupyter Notebook WebSocket 连接关闭错误的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1375937.html
微信扫一扫 
支付宝扫一扫 
