转载:https://www.vimiix.com/post/2018/07/26/channels2-tutorial/
认识 Channels 之前,需要先了解一下 asgi ,全名:Asynchronous Server Gateway Interface
。它是区别于 wsgi 的一种异步服务网关接口,不仅仅只是通过 asyncio
以异步方式运行,还支持多种协议。完整的文档戳这里
关联的几个项目:
- https://github.com/django/asgiref ASGI内存中的通道层,函数的同步异步之间相互转化需要
- https://github.com/django/daphne 支持HTTP,HTTP2和WebSocket协议服务器,启动 Channels 的项目需要
- https://github.com/django/channels_redis Channels专属的通道层,使用Redis作为其后备存储,并支持单服务器和分片配置以及群组支持。(这个项目是 Channels 的一个附属项目,配置的时候作为可选项使用,该软件包的早期版本被称为
asgi_redis
,如果你在使用 Channels 1.x项目,它仍可在PyPI下通过这个名称使用。但channels_redis
仅适用于 Channels 2 项目。)
备注:之前体验使用过一个基于 asgi 的web 框架 uvicon,也是同类项目。
流程图
实践
还是和官方同步,以一个聊天室的例子说起,Channels 提供从 PYPI 直接pip下载安装,
pip install -U channels==2.0.2 Django==2.0.4 channels_redis==2.1.1
这里,我们不仅安装了 Channels 2.0,还安装了Django 2.0 作为我们web项目框架,安装了channels_redis作为 Channels的后端存储通道层。
下载好包以后,把 channels 作为 Django 的一个应用(Django新建项目就不赘述了,假设我们已经建立好另一个项目叫 my_project
),添加到配置文件的 INSTALLED_APPS
中:
- my_project/settings.py
INSTALLED_APPS = ( 'django.contrib.auth', 'django.contrib.contenttypes', 'django.contrib.sessions', 'django.contrib.sites', ... 'channels', 'chat' )
chat
是我们准备建立的聊天室应用,接着就建立我们的 asgi 应用,并指定其要使用的路由。在 settings 同级目录下新建一个routing.py
的文件: - my_project/routing.py
from channels.auth import AuthMiddlewareStack from channels.routing import ProtocolTypeRouter, URLRouter import chat.routing application = ProtocolTypeRouter({ # (http->django views is added by default) # 普通的HTTP请求不需要我们手动在这里添加,框架会自动加载过来 'websocket': AuthMiddlewareStack( URLRouter( chat.routing.websocket_urlpatterns ) ), })
chat.routing
以及chat.routing.websocket_urlpatterns
是我们后面会自己建立的模块。紧接着,我们需要在 Django 的配置文件中继续配置 Channels 的 asgi 应用和通道层的信息:
- my_project/settings.py
ASGI_APPLICATION = "my_project.routing.application" # 上面新建的 asgi 应用 CHANNEL_LAYERS = { 'default': { # 这里用到了 channels_redis 'BACKEND': 'channels_redis.core.RedisChannelLayer', 'CONFIG': { 'hosts': [('127.0.0.1', 6379)], # 配置你自己的 redis 服务信息 }, } }
到这里,Channels 的基本配置就差不多了。
我们说以官方给的聊天室的例子来说明,所以最开始的时候,在
INSTALLED_APPS
中,我们添加了一个chat
的应用。下面就来创建这个应用。channels的应用和Django的还是不太一样,所以就不使用
python manage.py startapp
命令创建应用了。我们手动创建几个文件。 - chat/routing.py
from django.urls import path from . import consumers websocket_urlpatterns = [ # 路由,指定 websocket 链接对应的 consumer path('ws/chat/<str:room_name>/', consumers.ChatConsumer), ]
在 routing 文件中,我们定义了一个
websocket_urlpatterns
的路由列表,并设定了ChatCunsumer
这个消费类(理解成Django的view
)。这个 routing 文件就是上面my_project/routing.py
中引用到的路由。下面来看看consumer的实现:
import json from asgiref.sync import async_to_sync from channels.generic.websocket import WebsocketConsumer #customer相当于django中的view class ChatConsumer(WebsocketConsumer): def connect(self): # 当 websocket 一链接上以后触发该甘薯 self.room_name = self.scope['url_route']['kwargs']['room_name'] self.room_group_name = 'chat_%s' % self.room_name # 把当前链接添加到聊天室 # 注意 `group_add` 只支持异步调用,所以这里需要使用`async_to_sync`转换为同步调用 async_to_sync(self.channel_layer.group_add)( self.room_group_name, self.channel_name ) # 接受该链接 self.accept() def disconnect(self, close_code): # 断开链接是触发该函数 # 将该链接移出聊天室 async_to_sync(self.channel_layer.group_discard)( self.room_group_name, self.channel_name ) def receive(self, text_data): # 前端发送来消息时,通过这个接口传递 text_data_json = json.loads(text_data) message = text_data_json['message'] # 发送消息到当前聊天室 async_to_sync(self.channel_layer.group_send)( self.room_group_name, { # 这里的type要在当前类中实现一个相应的函数, # 下划线或者'.'的都会被Channels转换为下划线处理, # 所以这里写 'chat.message'也没问题 'type': 'chat_message', 'message': message } ) # 从聊天室拿到消息,后直接将消息返回回去 def chat_message(self, event): message = event['message'] # Send message to WebSocket self.send(text_data=json.dumps({ 'message': message }))
以上就是以同步的简单的实现了聊天室的收发功能,我们来看看这个互动过程是怎样进行的。
请求链接
当前端发起一个 websocket
的请求过来的时候,会自动触发 connect()
函数事件。
前端发起请求的实现方式:
- chat/templates/chat/room.html
<script> var roomName = {{ room_name_json }}; var chatSocket = new WebSocket( 'ws://' + window.location.host + '/ws/chat/' + roomName + '/'); ... </script>
触发
connect
事件以后,进入函数,self.scope
可以类比的理解为django中的self.request
,从url中取出room_name
字段备用,这里的变量名是在路由中设置的。chat/urls.py
from django.urls import path from . import views urlpatterns = [ path('', views.index, name='index'), path('<str:room_name>/', views.room, name='room'), ]
self.channel_name
是每一个websocket请求过来时候, Channels 自动帮我们生成的一个个性化名称,实现代码就是用随机函数组装: - channels/layers.py : L243
async def new_channel(self, prefix="specific."): """ Returns a new channel name that can be used by something in our process as a specific channel. """ return "%s.inmemory!%s" % ( prefix, "".join(random.choice(string.ascii_letters) for i in range(12)), )
然后我们将新的
channel_name
和room_name
绑定,也相当于,将当前链接加入到指定聊天室。端口链接
端口链接
前端实现方式:
- chat/templates/chat/room.html
<script> .... chatSocket.onclose = function(e) { console.error('Chat socket closed unexpectedly'); }; .... </script>
当 server 端的 WebSocket 调用
self.close()
关闭时,前端的chatSocket.onclose
会自动触发。如果前端页面关闭,导致端口链接时,后端的disconnect()
函数会自动触发。
接收信息
前端实现方式:
- chat/templates/chat/room.html
<script> .... // 点击按键,发送信息是很常见的逻辑 document.querySelector('#chat-message-submit').onclick = function(e) { var messageInputDom = document.querySelector('#chat-message-input'); var message = messageInputDom.value; chatSocket.send(JSON.stringify({ 'message': message })); messageInputDom.value = ''; }; </script>
chatSocket.send
会将信息发送到服务端,服务端的receive()
事件将会自动触发,将收到的message
送到对应的 group 中,type
字段是一个可以被调用的对象名,Channels 会根据这个名称调度相应的方法来处理逻辑。当在相应的函数中处理结束以后,调用
send()
方法将消息传递回前端。前端接收信息的方式:
<script> .... chatSocket.onmessage = function(e) { var data = JSON.parse(e.data); var message = data['message']; document.querySelector('#chat-log').value += (message + '\n'); }; .... </script>
chatSocket.onmessage
会接收到后端发来的信息,前端再将该信息渲染到页面上。
异步写法
上面是同步方式的写法,官方还给出了异步的写法,在我们的项目中,也采用了异步的写法,Channels 对于异步的支持是非常好的。
import json from channels.generic.websocket import AsyncWebsocketConsumer class ChatConsumer(AsyncWebsocketConsumer): async def connect(self): self.room_name = self.scope['url_route']['kwargs']['room_name'] self.room_group_name = 'chat_%s' % self.room_name await self.channel_layer.group_add( self.room_group_name, self.channel_name ) await self.accept() async def disconnect(self, close_code): await self.channel_layer.group_discard( self.room_group_name, self.channel_name ) async def receive(self, text_data): text_data_json = json.loads(text_data) message = text_data_json['message'] await self.channel_layer.group_send( self.room_group_name, { 'type': 'chat_message', 'message': message } ) async def chat_message(self, event): message = event['message'] await self.send(text_data=json.dumps({ 'message': message }))
从任意地方推送消息
仅仅只了解单次的收发,可能在项目中不能满足真正的需求。假如我们在项目中,其他的地方也需要向前端推送消息,怎么办?
这里就体现出了使用 Channel Layers 的好处。Channel Layers 因为是使用 redis 作为消息通道层,这使得一个应用中两个实例之间可以实现通信。如果你不想通过数据库来传递所有消息或事件,那么 Channel Layers 是开发分布式实时应用程序的非常好的实践。
有了这个,我们就可以项目的任何地方(脱离了 consumer
以外),给通道内广播信息。我们从上面可以知道,要想调用 group_send
需要有 self.channel_layer
的实例。但在脱离了 consumer 的情况下, Channels 提供了 get_channel_layer
函数接口来获取它。
from channels.layers import get_channel_layer channel_layer = get_channel_layer()
然后,一旦你获取到了它,你就可以调用它上面的方法。但请记住,channel layers仅支持异步方法,因此你可以从自己的异步上下文中调用它:
for chat_name in chats: await channel_layer.group_send( chat_name, {"type": "chat.system_message", "text": announcement_text}, )
或者,如果你的上下文是同步环境,可以使用 async_to_sync
来转换调用:
from asgiref.sync import async_to_sync async_to_sync(channel_layer.group_send)("chat", {"type": "chat.force_disconnect"}