基于上一篇内容,其实这个一对一也比较简单,我们在之前的websockets管理中已经实现了一对一发消息的内容,这次呢,我们只需要实现一对一如何处理消息即可。
import json
@app.websocket("/ws/{user}")
async def websocket_one_to_one(
websocket: WebSocket,
user: str,
cookie_or_token: str = Depends(get_cookie_or_token)
):
"""一对一聊天"""
await ws_manager.connect(user, websocket)
try:
while True:
data = await websocket.receive_text() # 前端发送的数据是:ws.send(input.value+"?"+username.value)
await ws_manager.send_other_message(message=json.loads(data.split("?")[0]), user=data.split("?")[1])
except WebSocketDisconnect as e:
await ws_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" value="要发送的消息" autocomplete="off"/>
<input type="text" id="username" value="要发送的用户" 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。
测试效果,分别登录两个用户,给对方发送消息:
可以看到消息已经接收成功,反之发送也是可以的,你也可以增加第三个人看是否能发送成功