FastAPI 學習之路（五十二）WebSockets(八)接受/傳送json格式訊息

阿新 • • 發佈：2021-10-24

相對於之前的text的訊息來說，我們之前傳送的訊息都是text的居多，那麼對於text格式的訊息來說，我們處理起來，尤其是後端要麻煩的多，那麼我們可不可以傳遞json格式的，對於前後端來說都好處理的一種格式的訊息，那麼答案來說是可以的。我們需要做下處理。

我們在之前的websockets管理中的的訊息的處理全部改造成json格式的。我們看下如何處理呢

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})

    def disconnect(self, user: str, ws: WebSocket):
        # 關閉時 移除ws物件
        self.active_connections.remove({"user": user, "ws": ws})


    async  
def send_other_message_json(self, message: dict, user: str):
        # 傳送個人訊息
        for connection in self.active_connections:
            if connection["user"] == user:
                await connection['ws'].send_json(message)

    async def broadcast_json(self, data: dict):
        # 廣播訊息
        for 
 connection in self.active_connections:
            await connection['ws'].send_json(data)

我們現在所有的訊息，我們只支援的json格式呢，我們可以對的處理訊息，

manager = ConnectionManager()


@app.websocket("/ws/{user}/")
async  def websocket_many_point(
        websocket: WebSocket,
        user:str,
        cookie_or_token: str = Depends(get_cookie_or_token),
        ):
    print(user)
    await manager.connect(user, websocket)
    try:
        while True:
            data = await websocket.receive_json()
            senduser=data['username']
            if senduser:
                await manager.send_other_message_json(data,senduser)
            else:
                await  manager.broadcast_json(data)
    except WebSocketDisconnect as e:
        manager.disconnect(user, websocket)

修改完之後，我們針對前段也需要修改，修改成支援我們的json格式資料

<!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/"+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>

　　我們部署下，重啟就可以完成我們的介面了。

所有的都接受到了，當我只對一個人說是，只有這個人接收到了這條訊息。

文章首發在公眾號，歡迎關注。