FastAPI 学习之路(五十)WebSockets(六)聊天室完善

我们这次只是对于之前的功能做下优化,顺便利用下之前的操作数据的接口,使用下数据库的练习。

在聊天里会有一个上线的概念。上线要通知大家,下线也要通知大家谁离开了,基于此功能我们完善下代码。

首先,我们的登录用户还是用了之前分享中的数据库相关用户,直接调用,之前也是从数据库获取用户,校验密码,密码成功后才让登录。这个在FastApi学习之路(四十七)WebSockets(三)登录后才可以聊天中就已经提到过了。

其次,我们建立了链接管理类,但是我们现在的链接都是没有针对发消息做管理的,我们去扩展下我们之前的发消息的类。

复制代码
"""
websocket 链接管理
"""


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)

增加了广播消息,和增加了发所有消息和针对个人发消息。我们现在实现的是针对所有人和广播。那么我们看下,我们的管理聊天链接的已经完善。

接下来我们就是去实现如何接受消息。

复制代码
@app.websocket("/items/ws/{user}")
async def websocket_chat_method(
        websocket: WebSocket,
        user: str,
        cookie_or_token: str = Depends(get_cookie_or_token)
):
    await ws_manager.connect(user, websocket)
    await ws_manager.broadcast(user + "进入聊天室")
    try:
        while True:
            data = await websocket.receive_text()
            await ws_manager.send_personal_message(f"你说:{data}", websocket)
            await ws_manager.broadcast(f"用户{user}说:{data}")
    except WebSocketDisconnect as e:
        await ws_manager.disconnect(user, websocket)
        await ws_manager.broadcast(f"用户{user}离开")

其实很简单,当你登录的时候,就广播一下,某某进入了聊天室,然后就是等消息,那么我们看下前端需要如何改造:

复制代码
<!DOCTYPE html>
<html>
<head>
    <title>Chat</title>
</head>
<body>
<h1>WebSocket 聊天</h1>
<form action="" onsubmit="sendMessage(event)">
    <input type="text" id="messageText" 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/items/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")

        ws.send(input.value)

        input.value = ''

        event.preventDefault()

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

</body>

</html>

其实就是在登录的时候增加了对应人的链接,接下来看下效果:

登录A:

首先,第一个用户已经登录进来了,接下来登录第用户B

登录B:

看下用户A能否接收到用户B的上线消息:

如上图,我们可以看到,当用户B登录后,用户A也收到了B上线的消息,我们试着发送消息:

用户A发送消息

用户B收到消息:

用户B发送消息:

用户A收到消息:

当用户离开时:

这样聊天室就相对完善了。我们从简单的聊天室入手,慢慢改造我们的代码,慢慢完善

相关推荐
东方佑2 天前
基于FastAPI与LangChain的Excel智能数据分析API开发实践
langchain·excel·fastapi
杰瑞学AI5 天前
我的全栈学习之旅:FastAPI (持续更新!!!)
后端·python·websocket·学习·http·restful·fastapi
Python私教9 天前
浏览器不再拦请求:FastAPI 跨域(CORS)配置全解析
fastapi
带娃的IT创业者11 天前
第2集:技术选型的智慧:Flask vs FastAPI,GLM-4 vs GPT
python·gpt·flask·fastapi·glm·技术选型
万粉变现经纪人14 天前
如何解决 pip install 安装报错 ImportError: cannot import name ‘xxx’ from ‘yyy’ 问题
python·selenium·测试工具·flask·scikit-learn·fastapi·pip
闲人编程16 天前
2025年,如何选择Python Web框架:Django, Flask还是FastAPI?
前端·后端·python·django·flask·fastapi·web
chao_78916 天前
Union 和 Optional 区别
开发语言·数据结构·python·fastapi
BTU_YC16 天前
Django+FastAPI+Vue微服务架构指南
架构·django·fastapi
Gerlat小智16 天前
【Python精讲 16】实战项目演练(二):用Flask/FastAPI发布你的第一个Web API
python·flask·fastapi
onelafite17 天前
淘宝开放平台拍立淘接口返回参数及调用操作指南
api·fastapi