参考文章: https://blog.csdn.net/xietansheng/article/details/115558069
服务器端
import asyncio
import websockets
import asyncio
import websockets
from websockets.legacy.server import WebSocketServerProtocol
import json
import time
import threading
clients={}
async def ws_handle(websocket: WebSocketServerProtocol, path: str):
# if(websocket in clients):
async for message in websocket:
print(message)
jsonmsg = json.loads(message)
if("deviceMac" in jsonmsg):
mac = jsonmsg["deviceMac"]
if(mac not in clients):
clients[mac] = websocket
await websocket.send(message)
async def sendmsg(mac,mssage):
# if(websocket in clients):
websocket = clients[mac]
await websocket.send(mssage)
async def main():
async with websockets.serve(ws_handle, "127.0.0.1", 8080):
await asyncio.Future() # run forever
def WebSocketServer():
asyncio.run(main())
thread = threading.Thread(target=WebSocketServer)
thread.start()
async def test():
flag=0
while(True):
flag = flag + 1
for client in clients:
await sendmsg(client, str(flag)+"haha") #依次给每个客户端都发一条信息
time.sleep(1)
def asyntest():
asyncio.run(test())
thread2 = threading.Thread(target=asyntest)
thread2.start()
客户端
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Title</title>
<script src="https://cdn.bootcss.com/jquery/3.2.0/jquery.js"></script>
</head>
<body>
<div id="time" style="width: 300px;height: 50px; text-align: center;line-height: 50px;margin-left: 40%;font-size: 20px"></div>
<input type="button" onclick="sayhello()" value="测试websocket"/>
<script>
window.ws = new WebSocket("ws://127.0.0.1:8080/"); //连接server
window.ws.onmessage = function (event) {
content = document.createTextNode(event.data); // 接收数据
$("#time").html(content);
};
function sayhello(params) {
console.log('{"deviceMac":"testdev"}');
window.ws.send('{"deviceMac":"testdev"}');
}
</script>
</body>
</html>