【python flask】给多个webocket客户端推送信息

服务端

from flask import Flask,request
from geventwebsocket.handler import WebSocketHandler
from gevent.pywsgi import WSGIServer
import time,json


app = Flask(__name__)

websocketClient_list = []  # 用于存放所有的websocket客户端对象
@app.route("/")
def index():
    ws = request.environ.get("wsgi.websocket")
    if not ws:  # 使用的是http协议"
        pass
    else:  # 使用的是websocket协议
        websocketClient_list.append(ws)
        while True:
            # 监听连接的websocket客户端，如果message为none，就移除这个客户端
            message = ws.receive()
            if not message:
                websocketClient_list.remove(ws)
                ws.close()
                break
            else:
                # 给所有在线的websocket客户端发信息
                for client in websocketClient_list:
                    client.send(json.dumps({"time:":str(time.time())}, ensure_ascii=False))
                    time.sleep(1)

            print("当前在线人数：",len(websocketClient_list))  # 查看当前在线的websocket对象

    return "..."

if __name__ == '__main__':
    http_server = WSGIServer(("127.0.0.1", 5000), app, handler_class=WebSocketHandler)
    print("server start")
    http_server.serve_forever()

客户端

地址：http://www.websocket-test.com/

使用该在线工具作为websocket客户端

测试

我这里开了两个客户端：

当服务端收到任何一个客户端发送的消息时，就会给所有在线的客户端发送信息，并打印当前在线人数。当有客户端断开后，就清除这个客户端对象，然后再次打印在线人数。

改进

但是上述代码的问题是只有接收到客户端的信息时，服务端才会发送下一条信息。于是我将监听客户端的代码作为一个线程分离出去，就解决了这个问题，如下所示：

（该段代码不论有没有接收到客户端的信息，都会主动向所有在线的客户端发送信息，而且当有客户端断开时，服务端会将这个断开的客户端close）

from flask import Flask,request
from geventwebsocket.handler import WebSocketHandler
from gevent.pywsgi import WSGIServer
from gevent import monkey
import time,json,threading


app = Flask(__name__)

monkey.patch_all()

def thread_listenClient(ws):
    while True:
        # 监听连接的websocket客户端，如果message为none，就移除这个客户端
        message = ws.receive()
        if not message:
            websocketClient_list.remove(ws)
            ws.close()
            break

websocketClient_list = []  # 用于存放所有的websocket客户端对象
@app.route("/")
def index():
    ws = request.environ.get("wsgi.websocket")

    thread_obj = threading.Thread(target=thread_listenClient, args=(ws,))
    thread_obj.daemon = True
    thread_obj.start()

    if not ws:  # 使用的是http协议"
        pass

    else:  # 使用的是websocket协议
        websocketClient_list.append(ws)
        while True:
            # 给所有在线的websocket客户端发信息
            for client in websocketClient_list:
                client.send(json.dumps({"time:":str(time.time())}, ensure_ascii=False))
                time.sleep(1)

            print("当前在线人数：",len(websocketClient_list))  # 查看当前在线的websocket对象

    return "..."

if __name__ == '__main__':
    http_server = WSGIServer(("127.0.0.1", 5000), app, handler_class=WebSocketHandler)
    print("server start")
    http_server.serve_forever()