server
from flask import Flask, request, render_template from geventwebsocket.handler import WebSocketHandler from gevent.pywsgi import WSGIServer from geventwebsocket.websocket import WebSocket import json app = Flask(__name__) user_socket_dict = {} # 字典 @app.route('/conn_ws/<user_nick>') #http 协议, def ws_app(user_nick): user_socket = request.environ.get('wsgi.websocket') #type:WebSocket user_socket_dict[user_nick] = user_socket # { 昵称:user信息 } print(len(user_socket_dict), list(user_socket_dict.keys())) while True: # user_socket 是一个内存地址 msg = user_socket.receive() # hang 住了 msg_dict = json.loads(msg) to_user = msg_dict.get('to_user') if to_user == 'all': for ss in list(user_socket_dict.values()): ss.send(msg) else: to_user_socket = user_socket_dict.get(to_user) to_user_socket.send(msg) @app.route('/') def index(): return render_template('my_ws.html') if __name__ == '__main__': # app.run() http_ser = WSGIServer(('0.0.0.0',9009),app,handler_class=WebSocketHandler) # 应用程序网关接口 http_ser.serve_forever()
页面 html
<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>单聊</title> </head> <body> <p><input type="text" id="nick"> <button onclick="login()">登陆聊天室</button> </p> 发送给: <input type="text" id="to_user" placeholder="找谁聊?"> 消息: <input type="text" id="send_str"> <button id="sin_btn" onclick="sin_send()">发消息</button> <button id="send_btn" onclick="send()">群聊发消息</button> <p> <div id="chat_list"> </div> </p> </body> <script type="text/javascript"> var ws = null; //公共变量, function login() { var nick = document.getElementById('nick').value; ws = new WebSocket('ws://192.168.11.94:9009/conn_ws/' + nick); //登录时候创建连接 ws.onmessage = function (messageEvent) { //等待消息 console.log(messageEvent.data); var ptag = document.createElement('p'); var message = JSON.parse(messageEvent.data); ptag.innerText = message.from_user + ' : ' + message.message; document.getElementById('chat_list').appendChild(ptag); }; } function send() { var message = document.getElementById('send_str').value; var send_str = { from_user: document.getElementById('nick').value, // to_user:document.getElementById('to_user').value, to_user: 'all', // 群聊 message: message }; var json_str = JSON.stringify(send_str); ws.send(json_str); } function sin_send() { var message = document.getElementById('send_str').value; var send_str = { from_user: document.getElementById('nick').value, to_user:document.getElementById('to_user').value, message: message }; var json_str = JSON.stringify(send_str); ws.send(json_str); } // js代码是异步代码; </script> </html>