channels 2.x的使用

转载: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"}

 

posted @ 2018-12-28 20:11  小小财经  阅读(406)  评论(0编辑  收藏  举报