解决 WebSocket 连接中 ConnectionClosedOK 错误-Python教程-PHP中文网

解决 WebSocket 连接中 ConnectionClosedOK 错误

花韻仙語

发布： 2025-08-07 17:50:15

原创

786人浏览过

解决 websocket 连接中 connectionclosedok 错误

本文将帮助开发者理解和解决在使用 Python websockets 库时遇到的 ConnectionClosedOK 错误。该错误通常发生在客户端发送一次请求后，服务端关闭连接的情况下。本文将介绍两种解决方案：第一种方案，客户端在每次请求时建立新的 WebSocket 连接，适用于客户端请求次数较少的情况。第二种方案，服务端通过循环保持 WebSocket 连接的活跃状态，允许客户端多次请求，并探讨了如何优雅地关闭连接。

在 websockets 库中，服务端在处理完一次客户端请求后默认会关闭连接，这会导致客户端在尝试后续请求时遇到 ConnectionClosedOK 错误。要解决这个问题，我们需要根据实际需求选择合适的方案。

方案一：客户端每次请求建立新连接

这种方案适用于客户端请求次数较少，或者每次请求都需要独立连接的场景。核心思想是移除客户端代码中的 while 循环，利用 async for 循环的特性，每次迭代都建立一个新的连接。

修改后的客户端代码如下：

import time
import json
import asyncio
import websockets
import nest_asyncio
nest_asyncio.apply()


async def get_data(websocket, key):
    order = {'key': key}
    json_data = json.dumps(order)
    await websocket.send(json_data)
    res = await asyncio.wait_for(websocket.recv(), timeout=10)
    res = json.loads(res)
    return res


async def main():
    i = 0
    uri = "ws://localhost:8765"
    async for websocket in websockets.connect(uri, timeout=15, ping_timeout=None, ping_interval=None):
        if i < 10:
            i = i + 1
            print(f'counter {i}')
            res = await get_data(websocket, 'key')
            print(f"res: {res}")
            time.sleep(3)  # slow down for tests
        else:
            break

    print('main exit')


if __name__ == "__main__":
    asyncio.get_event_loop().run_until_complete(main())

登录后复制

注意事项：

当退出 async for websocket 循环时，可能会抛出 GeneratorExit 异常。可以使用 try/except 块捕获该异常，进行必要的清理工作。

方案二：服务端保持连接活跃

如果需要客户端和服务端保持长时间的连接，以便进行多次数据交换，则需要在服务端代码中添加一个循环，使得 handle_request 函数持续运行。

挖错网

一款支持文本、图片、视频纠错和AIGC检测的内容审核校对平台。

查看详情

修改后的服务端代码如下：

import json
import asyncio
import websockets
import nest_asyncio
nest_asyncio.apply()


async def handle_request(websocket):
    while True:
        try:
            message = await websocket.recv()
            data = json.loads(message)

            order = {'key': data['key'], 'val1': 1.0, 'val2': 2.0}
            json_data = json.dumps(order)
            await websocket.send(json_data)
        except websockets.ConnectionClosedOK:
            break
    print("handle_request")


async def main():
    async with websockets.serve(handle_request, "localhost", 8765, ping_interval=None):
        print('world running...')
        await asyncio.Future()  # run forever
    print('world done')


if __name__ == "__main__":
    asyncio.run(main())

登录后复制

代码解释：