websocket实现简单的单聊 

 1 import json
 2 
 3 from flask import Flask, request, render_template
 4 
 5 from geventwebsocket.handler import WebSocketHandler
 6 from gevent.pywsgi import WSGIServer
 7 from geventwebsocket.websocket import WebSocket
 8 
 9 app = Flask(__name__)
10 
11 user_socket_dict = {}
12 
13 
14 @app.route('/conn_ws<suer>')
15 def ws_app(suer):
16     # print(request.environ)
17 
18     user_socket = request.environ.get('wsgi.websocket')  # type: WebSocket
19     user_socket_dict[suer] = user_socket
20     print(len(user_socket_dict), list(user_socket_dict))
21     while True:
22         msg = user_socket.receive()
23         msg_dict = json.loads(msg)
24         to_user = msg_dict.get('to_user')
25         to_user_socket = user_socket_dict.get(to_user)
26         to_user_socket.send(msg)
27 
28 
29 @app.route('/index')
30 def index():
31     return render_template('my_dws.html')
32 
33 
34 if __name__ == '__main__':
35     http_serv = WSGIServer(('0.0.0.0', 9527), app, handler_class=WebSocketHandler)
36 
37     http_serv.serve_forever()
后端
 1 <!DOCTYPE html>
 2 <html lang="en">
 3 <head>
 4     <meta charset="UTF-8">
 5     <title>Title</title>
 6 </head>
 7 <body>
 8 <p>
 9     <input type="text" id="nick">
10     <button onclick="login()">登录聊天</button>
11 </p>
12 
13 发送给：<input type="text" id="to_user">
14 消息：<input type="text" id="send_str">
15 <button id="send_btn" onclick="send()">发送消息</button>
16 
17 <p>
18 
19 <div id="chat_list"></div>
20 
21 </p>
22 </body>
23 <script type="application/javascript">
24     var ws = null;
25 
26     function login() {
27         var nick = document.getElementById('nick').value;
28         ws = new WebSocket('ws://192.168.11.67:9527/conn_ws' + nick);
29         ws.onmessage = function (messageEvent) {
30             console.log(messageEvent.data);
31             var ptag = document.createElement('p');
32             var message = JSON.parse(messageEvent.data);
33 
34             ptag.innerText = message.from_user + ':' + message.message;
35             document.getElementById('chat_list').appendChild(ptag);
36         };
37     }
38 
39     function send() {
40         var message = document.getElementById('send_str').value;
41         var send_str = {
42             from_user: document.getElementById('nick').value,
43             to_user: document.getElementById('to_user').value,
44             message: message
45         };
46         var json_send_str = JSON.stringify(send_str);
47         ws.send(json_send_str);
48     }
49 </script>
50 </html>
前端

 

posted @ 2019-03-24 19:47  過去  阅读(478)  评论(0)  编辑  收藏  举报