Granian项目中ASGI WebSocket连接处理的Bug分析与修复

背景介绍

Granian是一个高性能的Python ASGI服务器实现。在ASGI(异步服务器网关接口)规范中，WebSocket连接的处理流程有着严格的状态机要求。最近在Granian项目中发现了一个关于WebSocket连接处理的Bug，这个Bug会导致在特定场景下出现ASGI流程错误。

问题现象

当开发者编写一个ASGI应用，在WebSocket连接处理中使用异步任务来响应连接请求时，Granian会出现异常行为。具体表现为：

1. WebSocket连接事件(websocket.connect)会被多次触发
2. 最终导致RuntimeError: ASGI flow error异常
3. 相同代码在其他ASGI服务器(如Uvicorn、Daphne)上运行正常

技术分析

ASGI WebSocket规范要求

根据ASGI规范，WebSocket连接的处理应该遵循以下流程：

1. 服务器接收到WebSocket连接请求后，向应用发送websocket.connect事件
2. 应用必须响应websocket.accept消息才能建立完整连接
3. 在这期间，服务器应该阻塞等待应用的响应

Granian的实现问题

Granian的原始实现中存在以下问题：

1. 重复发送connect事件：每当应用调用receive()时，Granian都会无条件返回websocket.connect事件，而不是等待accept响应
2. 状态机不完整：没有正确维护WebSocket连接的状态转换，导致在应用尚未accept连接时就允许后续操作

问题复现场景

当开发者使用如下模式时特别容易触发此问题：

async def app(scope, receive, send):
    if scope["type"] == "websocket":
        tasks = []
        try:
            while True:
                incoming_message = await receive()
                if incoming_message["type"] == "websocket.disconnect":
                    return
                # 将消息处理放入后台任务
                tasks.append(asyncio.create_task(dispatch(incoming_message, send)))

这种模式在Django Channels等框架中很常见，主循环可以继续接收新消息，而将消息处理交给后台任务。

解决方案

Granian项目团队已经确认了这个问题，并在2.3.2版本中修复。修复的核心点是：

1. 阻塞等待accept：在收到websocket.connect事件后，服务器会阻塞等待应用发送websocket.accept响应
2. 完善状态机：正确维护WebSocket连接的状态，确保只有在accept之后才允许后续操作

开发者建议

对于使用Granian的开发者，建议：

1. 升级到2.3.2或更高版本
2. 在编写WebSocket处理逻辑时，仍然应该尽快发送accept响应，避免长时间阻塞
3. 复杂的消息处理逻辑可以考虑使用后台任务，但要确保accept在主协程中完成

总结

这个Bug的修复体现了ASGI实现中状态机管理的重要性。Granian团队快速响应并修复了这个问题，使得它能够更好地兼容各种ASGI应用模式，特别是像Django Channels这样广泛使用的框架。对于高性能Python Web开发来说，这类底层实现的完善至关重要。