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

2021-10-24 21:04:20 阅读：279 来源： 互联网

标签：websocket FastAPI token ws var 五十一 message document 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。

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

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

标签：websocket,FastAPI,token,ws,var,五十一,message,document,WebSockets
来源： https://www.cnblogs.com/leiziv5/p/15416901.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

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