Gevent-Websocket检测关闭连接

foo*_*bar 6 python websocket gevent bottle

我正在使用带有bottle.py的gevent-websocket来提供日志文件.如何从客户端检测到websocket连接是否已关闭?

目前我正在写作,直到我收到管道损坏错误:

return sock.send(data, flags)
error: [Errno 32] Broken pipe
Run Code Online (Sandbox Code Playgroud)

但我想在服务器上正确检测客户端是否关闭了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()
Run Code Online (Sandbox Code Playgroud)

和相应的客户端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()
      };
});
Run Code Online (Sandbox Code Playgroud)

欢迎对我的代码或解决方案进行任何其他改进.

Ivo*_*Ivo 7

if ws.socket is not None:在套接字上发送数据之前尝试测试.

  • 最好是要求宽恕而不是许可.仍有竞争条件. (4认同)