FastAPI 学习之路(五十二)WebSockets(八)接受/发送json格式消息

前面我们发送的大多数都是text类型的消息,对于text消息来说,后端处理出来要麻烦的多,那么我们可以不可以传递json格式的数据,对于前后端来说都比较友好,答案是肯定的,我们需要做下处理。

首先,我们在websocket管理器中增加处理json格式消息的方法:

from typing import List, Dict

from starlette.websockets import WebSocket


class ConnectionManager:

    def __init__(self):
        """存放链接"""
        self.active_connections: List[Dict[str, WebSocket]] = []

    async def connect(self, user: str, ws: WebSocket):
        """链接"""
        await ws.accept()
        self.active_connections.append({"user": user, "ws": ws})

    async def disconnect(self, user: str, ws: WebSocket):
        """断开链接,移除"""
        self.active_connections.remove({"user": user, "ws": ws})

    @staticmethod
    async def send_personal_message(message: str, ws: WebSocket):
        """发送所有人消息"""
        await ws.send_text(message)

    async def send_other_message(self, message: dict, user: str):
        """发送个人消息"""
        for coon in self.active_connections:
            if coon["user"] == user:
                await coon["ws"].send_json(message)

    async def broadcast(self, data: str):
        """广播"""
        for conn in self.active_connections:
            await conn["ws"].send_text(data)

    async def broadcast_json(self, data: dict):
        """广播json数据"""
        for conn in self.active_connections:
            await conn["ws"].send_json(data)

接口中如何接收json数据呢,新增接口如下:

@app.websocket("/ws/json/{user}")
async def websocket_json_data(
        websocket: WebSocket,
        user: str,
        cookie_or_token: str = Depends(get_cookie_or_token)
):
    """发送接收json数据(前面一对一接口其实就是发送的json数据)"""
    await ws_manager.connect(user, websocket)
    try:
        while True:
            data = await websocket.receive_json()
            send_user = data["username"]
            if send_user:
                await ws_manager.send_other_message(data, send_user)
            else:
                await ws_manager.broadcast_json(data)
    except WebSocketDisconnect as e:
        await ws_manager.disconnect(user, websocket)

前端也要相对应的修改:
 

<!DOCTYPE html>
<html>
<head>
    <title>Chat</title>
</head>
<body>
<h1>WebSocket 聊天</h1>
<form action="" onsubmit="sendMessage(event)">
    <input type="text" id="messageText" autocomplete="off"/>
      <input type="text" id="username" autocomplete="off"/>
    <button>Send</button>
</form>
<button onclick="logout()">退出</button>
<ul id='messages'>
</ul>
<script>
    var  token=window.localStorage.getItem("token")
    if (token==null ){
        window.location.href="/login"
    }
    var ws = new WebSocket("ws://localhost:8000/ws/json/"+token+"?token="+token);

    ws.onmessage = function (event) {

        var messages = document.getElementById('messages')

        var message = document.createElement('li')
        let receiveJson = JSON.parse(event.data);
        console.log(receiveJson)
        var content = document.createTextNode(receiveJson.username+"说:"+receiveJson.messageText)

        message.appendChild(content)

        messages.appendChild(message)

    };

    function sendMessage(event) {

        var input = document.getElementById("messageText")
        var username = document.getElementById("username")
        let message = {messageText: input.value, username:username.value};
        let messageJson = JSON.stringify(message);
        ws.send(messageJson);
        // input.value = ''
        event.preventDefault()

    }
    function logout() {
        window.localStorage.removeItem("token")
        window.location.href='/login'
    }
</script>

</body>

</html>

部署看效果,因为发送给后端的数据格式是前端拼的json,所以页面上只需要填文本即可:

 

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-07-20 04:00:02       52 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-07-20 04:00:02       54 阅读
  3. 在Django里面运行非项目文件

    2024-07-20 04:00:02       45 阅读
  4. Python语言-面向对象

    2024-07-20 04:00:02       55 阅读

热门阅读

  1. PYQT按键长按机制

    2024-07-20 04:00:02       16 阅读
  2. godot使用ws

    2024-07-20 04:00:02       18 阅读
  3. mysql(四)

    2024-07-20 04:00:02       16 阅读
  4. Kubernetes Service 之 LoadBalancer

    2024-07-20 04:00:02       18 阅读
  5. 第五节 LLaVA模型lora推理模型解读(下篇)

    2024-07-20 04:00:02       16 阅读
  6. element ui 怎么调整table的行高

    2024-07-20 04:00:02       20 阅读
  7. vscode使用技巧及问题 VSCode/bugs/如何给VSCode降级

    2024-07-20 04:00:02       17 阅读
  8. python selenium4 EdgeDriver动态页面爬取

    2024-07-20 04:00:02       23 阅读
  9. 白骑士的C++教学高级篇 3.2 多线程与并发

    2024-07-20 04:00:02       23 阅读
  10. gpg-agent

    2024-07-20 04:00:02       19 阅读