websocket 介绍
介绍引自 https://segmentfault.com/a/1190000012709475
群聊
from flask import Flask, request, render_template from geventwebsocket.handler import WebSocketHandler from gevent.pywsgi import WSGIServer from geventwebsocket.websocket import WebSocket app = Flask(__name__) user_socket_list = [] @app.route(‘/conn_ws‘) #http 协议, def ws_app(): user_socket = request.environ.get(‘wsgi.websocket‘) #type:WebSocket user_socket_list.append(user_socket) print(len(user_socket_list),user_socket_list) while True: # user_socket 是一个内存地址 msg = user_socket.receive() # hang 住了 print(msg) for usocket in user_socket_list: usocket.send(msg) # print(request.environ) # print(request.environ.get(‘wsgi.websocket‘))# # --> <geventwebsocket.websocket.WebSocket object at 0x0379A9D0> ws对象 # return ‘555‘ # 遇到retur就断了 # 状态1 已经开启了 # 3 开启了,断开了 @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()
<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>群聊</title> </head> <body> <input type="text" id="send_str"> <button id="send_btn" onclick="send()">发消息</button> <div id="chat_list"> </div> </p> </body> <script type="text/javascript"> var ws = new WebSocket(‘ws://192.168.11.124:9009/conn_ws‘); <!-- 创建连接 --> ws.onmessage = function (messageEvent) { console.log(messageEvent.data); var ptag = document.createElement(‘p‘); ptag.innerText = messageEvent.data; document.getElementById(‘chat_list‘).appendChild(ptag); }; function send() { var send_str = document.getElementById(‘send_str‘).value; ws.send(send_str); } </script> </html>
单聊
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‘) to_user_socket = user_socket_dict.get(to_user) to_user_socket.send(msg) print(msg) # user_socket.send(msg) # print(request.environ.get(‘wsgi.websocket‘))# # --> <geventwebsocket.websocket.WebSocket object at 0x0379A9D0> ws对象 # return ‘555‘ # 遇到retur就断了 # 状态1 已经开启了 # 3 开启了,断开了 @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()
<!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"> 消息: <input type="text" id="send_str"> <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.124: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, message:message }; var json_str = JSON.stringify(send_str); ws.send(json_str); } // js代码是异步代码; </script> </html>
原文：https://www.cnblogs.com/zhangchen-sx/p/10588052.html