如何处理Django Channels的DenyConnection错误并告知前端连接拒绝原因

当然可以实现！默认抛出channels.exceptions.DenyConnection确实只会触发前端的1006异常关闭事件——这个事件太通用，没法区分是连接被主动拒绝还是意外断开。这里给你两种实用的方案，让前端能明确知道连接被拒绝的具体原因：

方案一：手动发送自定义关闭帧（推荐） #

在消费者的connect方法里，先完成你的权限验证逻辑，若验证不通过，直接调用self.close()方法并传入自定义的关闭码和原因，不需要抛出DenyConnection。

后端代码示例 #

from channels.generic.websocket import AsyncWebsocketConsumer

class AuthRequiredConsumer(AsyncWebsocketConsumer):
    async def connect(self):
        # 模拟权限检查逻辑：比如验证用户是否登录、token是否有效等
        user_is_authenticated = await self._validate_user()
        
        if not user_is_authenticated:
            # 使用WebSocket私有自定义关闭码（4000-4999是标准预留的应用自定义区间）
            # 同时传递具体拒绝原因
            await self.close(code=4001, reason="用户未登录，无法建立WebSocket连接")
            return
        
        # 验证通过，正常接受连接
        await self.accept()
    
    async def _validate_user(self):
        # 这里实现你的实际验证逻辑
        # 比如从scope里获取user，检查是否认证
        return self.scope["user"].is_authenticated

前端处理代码 #

前端可以通过onclose事件的code和reason属性来区分不同的关闭原因：

const ws = new WebSocket('ws://your-domain/ws/your-endpoint/');

ws.onclose = (event) => {
    switch(event.code) {
        case 4001:
            alert(`连接被拒绝：${event.reason}`);
            // 引导用户登录等操作
            break;
        case 4002:
            alert(`连接被拒绝：${event.reason}`);
            // 处理其他拒绝场景，比如权限不足
            break;
        case 1006:
            console.log("连接意外断开，请重试");
            break;
        default:
            console.log(`连接关闭，代码：${event.code}，原因：${event.reason}`);
    }
};

方案二：自定义异常并重写关闭逻辑 #

如果你习惯用异常的方式处理，可以自定义一个带原因的异常，然后在中间件或者ASGI应用层捕获这个异常，转换成带原因的关闭帧。不过这种方式相对复杂一些，适合需要统一处理所有拒绝场景的项目。

自定义异常示例 #

class CustomDenyConnection(Exception):
    def __init__(self, reason, code=4000):
        self.reason = reason
        self.code = code
        super().__init__(reason)

在中间件中捕获并处理 #

from channels.middleware import BaseMiddleware

class DenyConnectionMiddleware(BaseMiddleware):
    async def __call__(self, scope, receive, send):
        try:
            return await super().__call__(scope, receive, send)
        except CustomDenyConnection as e:
            # 发送关闭帧
            await send({
                "type": "websocket.close",
                "code": e.code,
                "reason": e.reason.encode('utf-8'),
            })

然后在消费者里抛出这个自定义异常：

async def connect(self):
    if not await self._validate_user():
        raise CustomDenyConnection(reason="权限不足", code=4002)
    await self.accept()

注意事项 #

• WebSocket标准规定，自定义关闭码要在4000-4999区间内，避免和标准码冲突。
• 部分旧浏览器可能不支持传递reason字段，所以建议同时依靠自定义code来区分场景，双重保障。
• 不要在reason里传递敏感信息，避免安全风险。

内容的提问来源于stack exchange，提问作者Zuabi