FastAPI 学习之路(五十一)WebSockets(七)实现一对一聊天

Python78

有了上一遍的基础,其实这个一对一很简单,我们在之前的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不一样,而且要发给的人我用了?拼接,实际中,可以作为参数带过来,这里只是讲路子,那么我们要发送给的人,需要在前端的页面,我们选择或者填写我们要发送的用户。

```html;gutter:true;
Chat

WebSocket 聊天

Send

退出

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

```

我才用了填写的方式,新增了一个input标签,用来让用户填写要发送给的用户,这样我们就完成了,一个简单的一个一聊天的demo。其实很简单。

我们来调试下,我们分别登陆两个账号,账号A 和账号B。

FastAPI 学习之路(五十一)WebSockets(七)实现一对一聊天

我们可以看到消息已经接受成功,反之发送也是可以的。为了验证效果,此时还可以增加第三个人,看能不能接受到消息即可。

文章首发在公众号,欢迎关注。

FastAPI 学习之路(五十一)WebSockets(七)实现一对一聊天

Original: https://www.cnblogs.com/leiziv5/p/15416901.html
Author: 北漂的雷子
Title: FastAPI 学习之路(五十一)WebSockets(七)实现一对一聊天