Flask中使用websocket

因為業務需求需要使用websocket，所以在flask中使用websocket。在網上很多的資料，本文只記錄我使用的方法。

首先是需要導入包

pip install gevent-websocket

導入包之后直接在項目中使用

from flask import Flask, request
from geventwebsocket.handler import WebSocketHandler         #提供WS（websocket）協議處理
from geventwebsocket.server import WSGIServer                #websocket服務承載
#WSGIServer導入的就是gevent.pywsgi中的類
from gevent.pywsgi import WSGIServer
from geventwebsocket.websocket import WebSocket

app = Flask(__name__)

clientList = []

@app.route("/")
def func():
    client_socket = request.environ.get('wsgi.websocket')
    if not client_socket:
        return "error"
    clientList.append(client_socket)
    while 1:
        for i in clientList:
            try:
                i.send(json.dumps(“111111”))
            except Exception as e:
                continue


if __name__ == '__main__':
    http_server = WSGIServer(('127.0.0.1', 8888), application=app, handler_class=WebSocketHandler)
    http_server.serve_forever()

其實使用比較簡單，這里還是說一下我踩得幾個坑吧：

1.在這個使用之后運行的時候尤其是pycharm運行的時候不要直接拿IDE的運行，而是使用命令行python app.py來運行，否則是直接運行flask項目，而不是使用下面的這種方式去啟動項目。

2.因為在Gin框架中可以將get請求升級成websocket，這里也是同理。但是我之前想使用flask restful這個包，直接把get請求升級成websocket，但是那種方式是行不通的。現在還沒有解決，如果有人成功了，可以在本文下留言到自己的博客地址供大家學習。非常感謝！

3.因為代碼中有一個while 1，這導致在運行代碼的時候會一直在這個請求里阻塞住，這個也是由於個人能力有限，只能把websocket這個接口單獨拆出來做一個服務。當然這個在和很多業內的朋友討論的時候也一致認為應該如此。但是如果有人解決，也請參考上一條，不勝感激！