服务端
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()
相关文章
您阅读本篇文章共花了： 大家都在找：
flask：flask框架和django
python：python入门教程(非常详细)
后端：后端组
发表评论