我正在使用带有 bottle.py 的 gevent-websocket 来提供日志文件。如何检测到 websocket 连接已从客户端关闭？
目前我只是在写，直到我得到一个损坏的管道错误：
return sock.send(data, flags)
error: [Errno 32] Broken pipe
但是我想在服务器上正确检测客户端是否关闭了 websocket 连接。
我的代码如下所示：
from geventwebsocket.handler import WebSocketHandler
from gevent.pywsgi import WSGIServer
import gevent.monkey
gevent.monkey.patch_all()
from bottle import route, Bottle, view, request, static_file
import json
import os
import time
app = Bottle()
# Other code
@app.route('/websocket/<filename>')
def ws_logfile(filename):
if request.environ.get('wsgi.websocket'):
ws = request.environ['wsgi.websocket']
try:
filename = os.path.join(os.getcwd(), "logfiles", filename)
logfile = file(filename)
lines = logfile.readlines()
for line in lines:
ws.send(json.dumps({'output': line}))
while True:
line = logfile.readline()
if line:
# Here detect if connection is closed
# form client then break out of the while loop
ws.send(json.dumps({'output': line}))
else:
time.sleep(1)
ws.close()
except geventwebsocket.WebSocketError, ex:
print "connection closed"
print '%s: %s' % (ex.__class__.__name__, ex)
if __name__ == '__main__':
http_server = WSGIServer(('127.0.0.1', 8000), app, handler_class=WebSocketHandler)
http_server.serve_forever()
和相应的客户端javascript代码：
jQuery(document).ready(function(){
ws = $.gracefulWebSocket("ws://" + document.location.host + "/websocket" + document.location.pathname);
ws.onmessage = function (msg) {
var message = JSON.parse(msg.data);
$("#log").append(message.output + "<br>" );
};
window.onbeforeunload = function() {
ws.onclose = function () {console.log('unlodad')};
ws.close()
};
});
欢迎对我的代码或解决方案进行任何其他改进。