FastAPI 學習之路（五十一）WebSockets(七)實現一對一聊天
阿新 • • 發佈：2021-10-24
有了上一遍的基礎，其實這個一對一很簡單，我們在之前的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。
我們可以看到訊息已經接受成功，反之傳送也是可以的。為了驗證效果，此時還可以增加第三個人，看能不能接受到訊息即可。
文章首發在公眾號，歡迎關注。