flask python geventwebsocket实现websocket

Posted 2020-12-28 knowledgesky

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了flask python geventwebsocket实现websocket相关的知识，希望对你有一定的参考价值。

前端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()

以上是关于flask python geventwebsocket实现websocket的主要内容，如果未能解决你的问题，请参考以下文章

python中flask框架怎么选择文件

python flask怎么指定端口启动

python Python Flask WSGI路由技术#python #flask #wsgi

python常用框架之一 Flask 初识Flask

python flask 基础入门