1楚昭、嘮嘮叨叨
最近又回顧了下Websocket拍顷，發(fā)現(xiàn)已經(jīng)忘的七七八八了。于是用js寫了客戶端尿贫，用python寫了服務(wù)端，來復(fù)習(xí)一下這方面的知識匆背。
2身冀、先看一下效果吧
2.1括享、效果1（一個(gè)客戶端連上服務(wù)的并發(fā)送消息）
2.2、效果2（另一個(gè)客戶端連上服務(wù)的并發(fā)送消息）
2.3剩愧、效果3（服務(wù)的收到客戶端的全部消息并返回消息）
2.4仁卷、效果4（一個(gè)客戶端掉線并不影響其它socket連接）
2.5锦积、效果5（列取全部連接客戶端對象和當(dāng)前發(fā)消息的客戶端對象）
3歉嗓、核心代碼
3.1、Python
交流群：1029344413 分享相關(guān)視頻資料
代碼：
#! -*- coding: utf-8 -*-"""Author: ZhenYuSha
Create Time: 2019-1-14
Info: Websocket 的使用示例"""import asyncioimport websockets
websocket_users = set()# 檢測客戶端權(quán)限哮幢，用戶名密碼通過才能退出循環(huán)asyncdef check_user_permit(websocket):
? ? print("new websocket_users:", websocket)
? ? websocket_users.add(websocket)
? ? print("websocket_users list:", websocket_users)
? ? while True:
? ? ? ? recv_str = await websocket.recv()
? ? ? ? cred_dict = recv_str.split(":")
? ? ? ? ifcred_dict[0] =="admin"andcred_dict[1] =="123456":
? ? ? ? ? ? response_str ="Congratulation, you have connect with server..."? ? ? ? ? ? await websocket.send(response_str)
? ? ? ? ? ? print("Password is ok...")
? ? ? ? ? ? return True
? ? ? ? else:
? ? ? ? ? ? response_str ="Sorry, please input the username or password..."print("Password is wrong...")
? ? ? ? ? ? await websocket.send(response_str)# 接收客戶端消息并處理志珍，這里只是簡單把客戶端發(fā)來的返回回去asyncdef recv_user_msg(websocket):
? ? while True:
? ? ? ? recv_text = await websocket.recv()
? ? ? ? print("recv_text:", websocket.pong, recv_text)
? ? ? ? response_text = f"Server return: {recv_text}"print("response_text:", response_text)
? ? ? ? await websocket.send(response_text)# 服務(wù)器端主邏輯asyncdef run(websocket, path):
? ? while True:
? ? ? ? try:
? ? ? ? ? ? await check_user_permit(websocket)
? ? ? ? ? ? await recv_user_msg(websocket)
? ? ? ? except websockets.ConnectionClosed:
? ? ? ? ? ? print("ConnectionClosed...", path)# 鏈接斷開print("websocket_users old:", websocket_users)
? ? ? ? ? ? websocket_users.remove(websocket)
? ? ? ? ? ? print("websocket_users new:", websocket_users)
? ? ? ? ? ? breakexcept websockets.InvalidState:
? ? ? ? ? ? print("InvalidState...")# 無效狀態(tài)breakexcept Exception as e:
? ? ? ? ? ? print("Exception:", e)if__name__=='__main__':
? ? print("127.0.0.1:8181 websocket...")
? ? asyncio.get_event_loop().run_until_complete(websockets.serve(run, "127.0.0.1", 8181))
? ? asyncio.get_event_loop().run_forever()
3.2伦糯、Html（JS）
代碼：
? ? ? ? var socket;
? ? ? ? if("WebSocket"in window) {
? ? ? ? ? ? var ws = new WebSocket("ws://127.0.0.1:8181/test");
? ? ? ? ? ? socket = ws;
? ? ? ? ? ? ws.onopen = function() {
? ? ? ? ? ? ? ? console.log('連接成功');
? ? ? ? ? ? ? ? alert("連接成功, 請輸入賬號和密碼");
? ? ? ? ? ? };
? ? ? ? ? ? ws.onmessage = function(evt) {
? ? ? ? ? ? ? ? var received_msg = evt.data;
? ? ? ? ? ? ? ? document.getElementById("showMes").value+=received_msg+"\n";
? ? ? ? ? ? };
? ? ? ? ? ? ws.onclose = function() {
? ? ? ? ? ? ? ? alert("斷開了連接");
? ? ? ? ? ? };
? ? ? ? } else {
? ? ? ? ? ? alert("瀏覽器不支持WebSocket");
? ? ? ? }
? ? ? ? function sendMeg(){
? ? ? ? ? ? var message=document.getElementById("name").value+":"+document.getElementById("mes").value;
? ? ? ? ? ? document.getElementById("showMes").value+=message+"\n\n";
? ? ? ? ? ? socket.send(message);
? ? ? ? }
