瀏覽代碼

修复了websocket逻辑错误

Shellmiao 4 年之前
父節點
當前提交
75fb295edc
共有 2 個文件被更改,包括 7 次插入4 次删除
  1. 二進制
      db.sqlite3
  2. 7 4
      infrastructure/consumers.py

二進制
db.sqlite3


+ 7 - 4
infrastructure/consumers.py

@@ -16,7 +16,10 @@ class KnockConsumer(WebsocketConsumer):
 
 
 def knock_knock(username, message):
 def knock_knock(username, message):
     ws = user_dict.get(username)
     ws = user_dict.get(username)
-    message = json.dumps({'message': message})
-    ws.send(message)
-    print(message)
-    return True
+    if ws:
+        message = json.dumps({'message': message})
+        ws.send(message)
+        print(message)
+        return True
+    else:
+        return False