如何使用 ChannelNameRouter 在 Worker 和 Websocket(Django 和 Channels2.x)之间进行通信?

Zie*_*egl 7 django websocket python-3.x django-channels

我正在尝试设置一个使用 django2.0.2 和 channels2.1.1 的应用程序。我想要实现的是使用后台/工人任务来执行一些会产生数据的工作,这些数据应该动态地出现在网站上。我的问题主要与渠道有关:我如何正确地在工作人员和连接到 websocket 的消费者之间建立通信?

下面是一个突出问题的最小示例:这个想法是用户触发工作线程,工作线程生成一些数据并通过通道层将其发送到连接到 websocket 的消费者。

#routing.py
from channels.routing import ChannelNameRouter, ProtocolTypeRouter, URLRouter
from channels.auth import AuthMiddlewareStack
from django.urls import path
from testApp.consumers import *

application = ProtocolTypeRouter({
    "websocket":AuthMiddlewareStack(
        URLRouter([
            path("wspath",TestConsumer),
        ]),
    ),
    "channel":ChannelNameRouter({
        "test_worker": TestWorker,
    }),
})
Run Code Online (Sandbox Code Playgroud)

消费者:

#consumers.py
from channels.consumer import SyncConsumer
from channels.generic.websocket import WebsocketConsumer
from asgiref.sync import async_to_sync

class TestConsumer(WebsocketConsumer):
    def websocket_connect(self,message):
        async_to_sync(self.channel_layer.group_add)("testGroup",self.channel_name)
        self.connect()
        #I understand this next part is a bit weird, but I figured it 
        #is the most concise way to explain my problem
        async_to_sync(self.channel_layer.group_send)(
            "testGroup",
            {
                'type':"echo_msg",
                'msg':"sent from WebsocketConsumer",
            })

    def echo_msg(self, message):
        print("Message to WebsocketConsumer", message)

class TestWorker(SyncConsumer):
    def triggerWorker(self, message):
        async_to_sync(self.channel_layer.group_add)("testGroup",self.channel_name)
        async_to_sync(self.channel_layer.group_send)(
            "testGroup",
            {
                'type':"echo_msg",
                'msg':"sent from worker",
            })

    def echo_msg(self, message):
        print("Message to worker ", message)
Run Code Online (Sandbox Code Playgroud)

风景

#views.py
from django.shortcuts import render
import channels.layers
from asgiref.sync import async_to_sync

def index(request):
    if request.method == "POST":
        channel_layer = channels.layers.get_channel_layer()
        async_to_sync(channel_layer.send)('test_worker',{
            'type':'triggerWorker',
        })
    return render(
        request,
        "index.html",
        {})
Run Code Online (Sandbox Code Playgroud)

和 html:

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="utf-8">
    <script>
        console.log('ws://' + window.location.host)
        var socket = new WebSocket(
            'ws://' + window.location.host + "/wspath"
        );
    </script>
</head>
<div>Click to run worker</div>
<body>
    <form action="" method="POST">
        {% csrf_token %}
        <button type="submit">Start</button>   
    </form>
</body>
Run Code Online (Sandbox Code Playgroud)

现在,当我通过执行(在单独的控制台中)来运行它时

python3 manage.py runserver
Run Code Online (Sandbox Code Playgroud)

python3 manage.py runworker test_worker
Run Code Online (Sandbox Code Playgroud)

然后触发worker,runserver控制台输出:

给 WebsocketConsumer 的消息 {'type': 'echo_msg', 'msg': 'sent from WebsocketConsumer'}

其中 runworker 控制台输出:

给工人的消息 {'type': 'echo_msg', 'msg': 'sent from worker'}

给工人的消息 {'type': 'echo_msg', 'msg': 'sent from WebsocketConsumer'}

所以我可以发送worker -> worker, WebsocketConsumer -> WebsocketConsumer, WebsocketConsumer -> worker。

据我所知(这显然是错误的)应该还有一个消息工作者-> WebsocketConsumer,因为我将两者都添加到了“testGroup”中。

所以,我的问题是为什么 WebsocketConsumer 没有从工作人员那里收到任何东西(这是我感兴趣的,最终与 javaScript 建立通信)?或者,换句话说,为什么我只能从 WebsocketConsumer 向工作人员发送内容,反之亦然?

kag*_*ick 4

我正在运行你的代码。我可以看到一切正常。

您在 POST 上收到了初始消息 - 正在运行 - 您将其添加到组中。当 websocket 连接时,它会向工作线程发送一条消息。您可以在 runworker 终端中看到收到的信息。这会将其弹回给您的消费者,并在运行 runserver 的终端中打印出来。要将其返回到您的网络浏览器,您需要编写:

def echo_msg(self, message): print("Message to WebsocketConsumer", message) self.send(json.dumps(message))

在 Chrome 中打开开发者工具即可看到它回来了。转到网络 > 选择 websocket 连接 > 然后单击框架。

顺便说一句,您不需要一次又一次地将测试工作人员添加到同一组中。工人总是在运行。

另一个顺便说一句:如果您的组要为所有用户命名相同,则无需将工作人员添加到组中。您可以通过路由名称直接向您的工作人员发送消息(send 而不是 group_send)。Worker 可以将消息发送回组,而无需将其添加到组中。您只需将 websocket 消费者添加到组中即可。

另外,如果您不希望多个用户看到相同的消息,则根本不需要组。只需将消息发送给具有通道名称(self.channel_name)的工作人员即可将其发送回。

此外,您可能希望使用 json 消费者,而不是自己解析消息,但这取决于您。