前端js逻辑：
<html>
<head></head>
<body>
<script type="text/javascript" src="http://libs.baidu.com/jquery/2.1.4/jquery.min.js"></script>
<script type="text/javascript">
var sock = null;
var wsuri = "ws://host:port/openapi/bigevent";
window.onload = function() {
console.log("onload");
sock = new WebSocket(wsuri);
sock.onopen = function() {
console.log("connected to " + wsuri);
}
sock.onclose = function(e) {
console.log("connection closed (" + e.code + ")");
}
sock.onerror = function(e) {
console.log("connection errored");
}
sock.onmessage = function(e) {
$("#message").append(‘<p>‘ + e.data + ‘</p>‘)
console.log(JSON.parse(e.data));
}}
</script>
<h1>WebSocket Echo Test</h1>
<div id="message" style="width:600px;height:300px;"></div>
</body>
</html>
后端逻辑
app.add_url_rule(‘/openapi/bigevent‘, view_func=big_event)
# route_url: ‘/openapi/bigevent‘
def big_event():
obj = {
‘recordsTotal‘: 0,
‘records‘: [],
‘errmsg‘: "",
‘errno‘: 0,
}
if request.environ.get(‘wsgi.websocket‘):
ws = request.environ[‘wsgi.websocket‘]
if ws is None:
abort(404)
else:
x = 0
while True:
if not ws.closed:
db = mongo.cx[Config.ALFRED_MONGO_DBNAME]
filter = {"$and": [{"ts": {"$gt": "1532535132"}}, {"ts": {"$lte": "1532585532"}}]}
docs = db.big_event.find(filter, {"_id": 0})
big_event_list = list()
for doc in docs:
big_event = dict()
big_event_list.append(big_event)
obj[‘records‘] = big_event_list
obj[‘recordsTotal‘] = len(big_event_list)
ws.send(json.dumps(obj))
sleep(Config.BIGEVENT_PUSH_CYCLE)
else:
break
# -*- coding: utf-8 -*
from gevent.pywsgi import WSGIServer
from geventwebsocket.handler import WebSocketHandler
from poseidon.app import create_app
from poseidon.settings import Config
app = Flask(__name__)
if __name__==‘__main__‘:
# app.run(‘0.0.0.0‘,port=8081)
http_server = WSGIServer((‘‘, 8081), app, handler_class=WebSocketHandler)
http_server.serve_forever()
原文地址：https://www.cnblogs.com/KnowledgeSky/p/9426710.html
时间： 2024-10-05 07:17:30