首页 > 解决方案 > 在 Django 视图中使用 Websocket 不起作用

问题描述

问题总结

我正在使用 Django 和 web-sockets 将数据发送到前端(React 组件)。当我运行应用程序并从控制台发送数据时,一切正常。当我使用前端的按钮触发运行相同功能的 Django 视图时,它不起作用并生成令人困惑的错误消息。

我希望能够单击开始将数据发送到 websocket 的前端按钮。

我是 Django、websockets 和 React 的新手,因此请您耐心等待。

概述

  1. Django 后端和 React 前端使用 Django 通道(网络套接字)连接。
  2. 用户单击前端的按钮,这fetch()在 Django REST API 端点上进行。
  3. [不工作]上述端点的视图开始通过 web-socket 发送数据。
  4. 使用此值更新前端。

简短的错误描述

错误 Traceback 很长,因此在本文末尾包含。它开始于:

内部服务器错误:/api/run-create

并以:

ConnectionResetError: [WinError 10054] 现有连接被远程主机强行关闭

我试过的

在Django 视图之外发送数据

    import json
    import time
    
    import numpy as np
    import websocket
    
    
    def gen_fake_path(num_cities):
        path = list(np.random.choice(num_cities, num_cities, replace=False))
        path = [int(num) for num in path]
        return json.dumps({"path": path})
    
    
    def fake_run(num_cities, limit=1000):
        ws = websocket.WebSocket()
        ws.connect("ws://localhost:8000/ws/canvas_data")
        while limit:
            path_json = gen_fake_path(num_cities)
            print(f"Sending {path_json} (limit: {limit})")
            ws.send(path_json)
            time.sleep(3)
            limit -= 1
        print("Sending complete!")
        ws.close()
        return

附加细节

相关文件和配置

消费者.py

    class AsyncCanvasConsumer(AsyncWebsocketConsumer):
        async def connect(self):
            self.group_name = "dashboard"
            await self.channel_layer.group_add(self.group_name, self.channel_name)
            await self.accept()
    
        async def disconnect(self, close_code):
            await self.channel_layer.group_discard(self.group_name, self.channel_name)
    
        async def receive(self, text_data=None, bytes_data=None):
            print(f"Received: {text_data}")
            data = json.loads(text_data)
            to_send = {"type": "prep", "path": data["path"]}
            await self.channel_layer.group_send(self.group_name, to_send)
    
        async def prep(self, event):
            send_json = json.dumps({"path": event["path"]})
            await self.send(text_data=send_json)

相关视图.py

    @api_view(["POST", "GET"])
    def run_create(request):
        serializer = RunSerializer(data=request.data)
        if not serializer.is_valid():
            return Response({"Bad Request": "Invalid data..."}, status=status.HTTP_400_BAD_REQUEST)
        # TODO: Do run here.
        serializer.save()
        fake_run(num_cities, limit=1000)
        return Response(serializer.data, status=status.HTTP_200_OK)

相关设置.py

    WSGI_APPLICATION = 'evolving_salesman.wsgi.application'
    ASGI_APPLICATION = 'evolving_salesman.asgi.application'
    
    CHANNEL_LAYERS = {
        "default": {
            "BACKEND": "channels.layers.InMemoryChannelLayer"
        }
    }

相关路由.py

    websocket_url_pattern = [
        path("ws/canvas_data", AsyncCanvasConsumer.as_asgi()),
    ]

完全错误

https://pastebin.com/rnGhrgUw

编辑:解决方案

Kunal Solanke 的建议解决了这个问题。而不是使用fake_run()我使用以下内容:

    layer = get_channel_layer()
    for i in range(10):
        path = list(np.random.choice(4, 4, replace=False))
        path = [int(num) for num in path]
        async_to_sync(layer.group_send)("dashboard", {"type": "prep", "path": path})
        time.sleep(3)

标签: djangowebsocketdjango-channels

解决方案


我建议您使用 get_channel_layer 实用程序,而不是创建从同一服务器到自身的新连接。因为您最终会通过打开这么多连接来增加服务器负载。获得通道层后,您可以像通常发送 evnet 一样简单地进行组发送。你可以在这里阅读更多

from channels.layers import get_channel_layer
from asgiref.sync import async_to_sync
def media_image(request,chat_id) :
    if request.method == "POST" :
        data = {}
        if request.FILES["media_image"] is not None :
            item = Image.objects.create(owner = request.user,file=request.FILES["media_image"])
            message=Message.objects.create(item =item,user=request.user )
            chat = Chat.objects.get(id=chat_id)
            chat.messages.add(message)
            layer = get_channel_layer()
            item = {
               "media_type": "image",
                "url" : item.file.url,
                "user" : request.user.username,
                'caption':item.title
            }
            async_to_sync(layer.group_send)(
                'chat_%s'%str(chat_id),
#this is the channel group name,which is defined inside your consumer"
                {
                    "type":"send_media",
                    "item" : item
                    
                }
            )

        return HttpResponse("media sent")

在错误日志中,我可以看到第一次迭代握手成功,第二次握手失败。您可以通过在 for 循环中打印一些内容来检查。如果是这种情况,握手很可能由于多个连接而失败。我不知道 Inmemrorycache 支持多少个来自同一来源的连接,但这可能是第二个连接断开的原因。您可以在频道文档中获得一些想法。如果您不想更改代码,请尝试使用 redis,如果您使用的是 linux,这很容易。


推荐阅读