我正在尝试设置一个使用 django2.0.2 和 channels2.1.1 的应用程序。我想要实现的是使用后台/ worker 任务来执行一些会产生数据的工作,这些数据应该动态地出现在网站上。我的问题主要与 channel 有关:我如何正确地在工作人员和连接到 websocket 的消费者之间建立通信?
下面是一个突出问题的最小示例:这个想法是用户触发工作线程,工作线程生成一些数据并通过 channel 层将其发送到连接到 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,
}),
})
消费者:#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)
风景#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",
{})
和 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>
现在,当我通过执行(在单独的控制台中)来运行它时python3 manage.py runserver
和python3 manage.py runworker test_worker
然后触发worker,runserver控制台输出:Message to WebsocketConsumer {'type': 'echo_msg', 'msg': 'sent from WebsocketConsumer'}
其中 runworker 控制台输出:
Message to worker {'type': 'echo_msg', 'msg': 'sent from worker'}
Message to worker {'type': 'echo_msg', 'msg': 'sent from WebsocketConsumer'}
所以我可以发送worker -> worker, WebsocketConsumer -> WebsocketConsumer, WebsocketConsumer -> worker。
据我所知(这显然是错误的)应该还有一个消息 worker -> WebsocketConsumer,因为我将两者都添加到了“testGroup”中。
所以,我的问题是为什么 WebsocketConsumer 没有从工作人员那里收到任何东西(这是我感兴趣的,最终与 javaScript 建立通信)?或者,换句话说,为什么我只能从 WebsocketConsumer 向工作人员发送内容,反之亦然?
最佳答案
我正在运行你的代码。我可以看到一切正常。
您在 POST 上有该初始消息 - 正在运行 - 您将其添加到组中。当 websocket 连接时,它会向工作人员发送一条消息。您可以看到在您的 runworker 终端中收到。这会将它弹回给您的消费者,并在您运行 runserver 的终端中打印出来。要将其返回到您的网络浏览器,您需要编写:
def echo_msg(self, message):
print("Message to WebsocketConsumer", message)
self.send(json.dumps(message))
在 Chrome 中打开您的开发人员工具以查看它的恢复。转到网络 > 选择 websocket 连接 > 然后单击框架。
顺便说一句,您不需要一遍又一遍地将测试人员添加到同一个组中。 worker 总是在运行。
另一个顺便说一句:如果您的组将为所有用户命名为相同的名称,则您无需将您的工作人员添加到组中。您可以通过路由名称(发送而不是 group_send)直接向您的工作人员发送消息。工作人员可以将消息发送回组,而无需将其添加到组中。您只需要将 websocket 消费者添加到组中。
此外,如果您不希望多个用户看到相同的消息,则根本不需要组。只需将消息发送给具有 channel 名称(self.channel_name)的工作人员即可将其发回。
此外,您可能希望与 json 消费者一起工作,而不是自己解析消息,但这取决于您。
关于django - 如何使用 ChannelNameRouter 在 Worker 和 Websocket(Django 和 Channels2.x)之间进行通信?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50199118/