feat: 添加websocket模块

apps/ws/admin.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

apps/ws/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class WsConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'ws'

apps/ws/consumers.py

@@ -0,0 +1,74 @@
+from channels.generic.websocket import AsyncWebsocketConsumer
+import json
+
+
+class RoomConsumer(AsyncWebsocketConsumer):
+    async def connect(self):
+        self.room_name = self.scope['url_route']['kwargs']['room_name']
+        self.room_group_name = 'chat_%s' % self.room_name
+
+        # Join room group
+        await self.channel_layer.group_add(
+            self.room_group_name,
+            self.channel_name
+        )
+        username = self.scope['user'].username
+        await self.channel_layer.group_send(
+            self.room_group_name,
+            {
+                'type': 'chat',
+                'msg': f'你好,{username}, 欢迎进入{self.room_name}房间' ,
+                'from': '系统',
+                'to': username
+            }
+        )
+        await self.accept()
+    
+    async def disconnect(self, close_code):
+        # Leave room group
+        await self.channel_layer.group_discard(
+            self.room_group_name,
+            self.channel_name
+        )
+
+    async def receive_json(self, content, **kwargs):
+        sender_user = self.scope["user"]
+        if content['type'] == 'chat':
+            content['from'] = sender_user.username
+            await self.channel_layer.group_send(
+                self.room_group_name,
+                content
+            )
+
+    async def chat(self, content):
+        await self.send(json.dumps(content, ensure_ascii=False))
+
+
+class MyConsumer(AsyncWebsocketConsumer):
+    async def connect(self):
+        user_id = self.scope['user'].id
+        self.room_group_name = f'user_{user_id}'
+        # Join room group
+        await self.channel_layer.group_add(
+            self.room_group_name,
+            self.channel_name
+        )
+        await self.channel_layer.group_send(
+            self.room_group_name,
+            {
+                'type': 'event',
+                'msg': '你好,' + self.scope['user'].username,
+                'from': '系统'
+            }
+        )
+        await self.accept()
+
+    async def disconnect(self, close_code):
+        # Leave room group
+        await self.channel_layer.group_discard(
+            self.room_group_name,
+            self.channel_name
+        )
+
+    async def event(self, content):
+        await self.send(json.dumps(content, ensure_ascii=False))

apps/ws/models.py

@@ -0,0 +1,3 @@
+from django.db import models
+
+# Create your models here.

apps/ws/routing.py

@@ -0,0 +1,9 @@
+from django.urls import path
+from apps.ws.consumers import MyConsumer, RoomConsumer
+
+WS_BASE_URL = 'ws/'
+
+websocket_urlpatterns = [
+    path(f'{WS_BASE_URL}my/', MyConsumer.as_asgi()),
+    path(WS_BASE_URL + '<str:room_name>/', RoomConsumer.as_asgi()),
+]

apps/ws/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

apps/ws/views.py

@@ -0,0 +1,3 @@
+from django.shortcuts import render
+
+# Create your views here.

server/asgi.py

@@ -10,16 +10,16 @@ https://docs.djangoproject.com/en/3.0/howto/deployment/asgi/
 import os
 from channels.routing import ProtocolTypeRouter, URLRouter
 from django.core.asgi import get_asgi_application
-from channels.auth import AuthMiddlewareStack
+from apps.utils.middlewares import TokenAuthMiddleware
-import apps.monitor.routing
+import apps.ws.routing
 
 os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'server.settings')
 
 application = ProtocolTypeRouter({
     "http": get_asgi_application(),
-    "websocket": AuthMiddlewareStack(
+    "websocket": TokenAuthMiddleware(
         URLRouter(
-            apps.monitor.routing.websocket_urlpatterns
+            apps.ws.routing.websocket_urlpatterns
         )
     )
 })