本次是WebSockets的第六次分享了，我們這次只是對于之前的功能做下優化，順便利用下之前的操作資料的介面，使用下資料庫的練習，
我們都知道，在聊天室里面會有一個上線的概念，上線了要通知大家，然后下線了要告訴大家誰離開了，發小小大家要都能可以收到，那么我們基于這個去完善下我們的功能，
首先，我們的登錄用戶還是用了之前分享中分享的資料庫相關呢，我們現在直接來呼叫，我們之前有從資料庫獲取用戶，然后校驗密碼，密碼成功我們才讓登錄成功，這個FastAPI 學習之路（四十六）WebSockets（三）登錄后才可以聊天就已經分享了，
其次，我們去建立了鏈接，但是我們現在的鏈接都是沒有針對發訊息做管理呢，我們去擴展下我們之前的發訊息的類，
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}) @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 connection in self.active_connections: if connection["user"] == user: await connection['ws'].send_json(message) async def broadcast(self, data: str): # 廣播訊息 for connection in self.active_connections: await connection['ws'].send_text(data)
增加了廣播訊息，和增加了發所有訊息和針對個人發訊息，我們現在實作的是針對所有人和廣播，那么我們看下，我們的管理聊天鏈接的已經完善，
接下來我們就是去實作如何接受訊息，
@app.websocket("/items/ws/{user}") async def websocket_endpoint( websocket: WebSocket, user:str, cookie_or_token: str = Depends(get_cookie_or_token), ): await manager.connect(user, websocket) await manager.broadcast(user + "進入聊天室") try: while True: data = await websocket.receive_text() await manager.send_personal_message(f"你說了: {data}", websocket) await manager.broadcast(f"用戶:{user} 說: {data}") except WebSocketDisconnect as e: manager.disconnect(user, websocket) await manager.broadcast("用戶{}離開".format(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="https://www.cnblogs.com/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.valuehttps://www.cnblogs.com/leiziv5/p/= ''
event.preventDefault()
}
function logout() {
window.localStorage.removeItem("token")
window.location.href='https://www.cnblogs.com/login'
}
</script>
</body>
</html>
其實我們就是在連接的時候增加了對應人的連接，那么我們看下，我們應該測驗下效果，
首先我們看下，我們第一個用戶已經進來了，我們看下第二個用戶的登錄，登錄后，我們發現用戶二進來了，我們看下用戶1是否接受到用戶二上線的訊息了？
我們可以看到，當二完成了登錄之后呢，用戶一也接受到了用戶二的訊息
我們可以試下發送訊息，
這樣我們的聊天更加完善了，我們從簡單的聊天室入手，慢慢去改造我們的代碼，測驗我們的代碼，逐步去完善，慢慢去改造實作，
文章首發在公眾號，歡迎關注，
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/335094.html
標籤：其他