FastAPI 學習之路(五十一)WebSockets(七)實現一對一聊天


        有了上一遍的基礎,其實這個一對一很簡單,我們在之前的websockets管理中已經實現了一對一發消息的內容,這次呢,我們只需要實現一對一如何處理消息即可。

@app.websocket("/ws/{user}/")
async  def websocket_many_point(
        websocket: WebSocket,
        user:str,
        cookie_or_token: str = Depends(get_cookie_or_token),
        ):
    await manager.connect(user, websocket)
    try:
        while True:
            data = await websocket.receive_text()

            await  manager.send_other_message(message=data.split("?")[0],user=data.split("?")[1])
    except WebSocketDisconnect as e:
        manager.disconnect(user, websocket)

   其實很簡單,還是之前的方法,只是對應的path不一樣,而且要發給的人我用了?拼接,實際中,可以作爲參數帶過來,這裏只是講路子,那麼我們要發送給的人,需要在前端的頁面,我們選擇或者填寫我們要發送的用戶。

<!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')

        var content = document.createTextNode(event.data)

        message.appendChild(content)

        messages.appendChild(message)

    };

    function sendMessage(event) {

        var input = document.getElementById("messageText")
        var username = document.getElementById("username")

        ws.send(input.value+"?"+username.value)

        input.value = ''

        event.preventDefault()

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

</body>

</html>

  

  我才用了填寫的方式,新增了一個input標籤,用來讓用戶填寫要發送給的用戶,這樣我們就完成了,一個簡單的一個一聊天的demo。其實很簡單。

        我們來調試下,我們分別登陸兩個賬號,賬號A 和賬號B。

 

  我們可以看到消息已經接受成功,反之發送也是可以的。爲了驗證效果,此時還可以增加第三個人,看能不能接受到消息即可。

文章首發在公衆號,歡迎關注。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章