Python BaseHTTPServer, como faço para capturar/capturar erros de “tubo quebrado”?

Question

Eu construo um mecanismo de tradução de URL curto em Python e estou vendo uma tonelada de erros de "tubo quebrado" e estou curioso para saber como capturá-lo melhor ao usar as classes BaseHTTPServer.Este não é o código inteiro, mas dá uma ideia do que estou fazendo até agora:

    from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
    import memcache

    class clientThread(BaseHTTPRequestHandler):

            def do_GET(self):
                    content = None
                    http_code,response_txt,long_url = \
                            self.ag_trans_url(self.path,content,'GET')
                    self.http_output( http_code, response_txt, long_url )
                    return

            def http_output(self,http_code,response_txt,long_url):
                    self.send_response(http_code)
                    self.send_header('Content-type','text/plain')
                    if long_url:
                            self.send_header('Location', long_url)
                    self.end_headers()
                    if response_txt:
                            self.wfile.write(response_txt)
                    return

        def ag_trans_url(self, orig_short_url, post_action, getpost):
                short_url = 'http://foo.co' + orig_short_url

                # fetch it from memcache
                long_url = mc.get(short_url)

                # other magic happens to look it up from db if there was nothing
                # in memcache, etc
                return (302, None, log_url)

def populate_memcache()
        # connect to db, do lots of mc.set() calls

def main():
        populate_memcache()
        try:
                port = 8001
                if len(sys.argv) > 1:
                        port = int(sys.argv[1])
                server = HTTPServer(('',port), clientThread)
                #server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
                print '[',str(datetime.datetime.now()),'] short url processing has begun'

                server.serve_forever()
        except KeyboardInterrupt,SystemExit:
                print '^C received, shutting down server'
                server.socket.close()

O código em si funciona muito bem, mas começou a gerar erros quase imediatamente durante a produção:

Traceback (most recent call last):
  File "/usr/lib/python2.5/SocketServer.py", line 222, in handle_request
    self.process_request(request, client_address)
  File "/usr/lib/python2.5/SocketServer.py", line 241, in process_request
    self.finish_request(request, client_address)
  File "/usr/lib/python2.5/SocketServer.py", line 254, in finish_request
    self.RequestHandlerClass(request, client_address, self)
  File "/usr/lib/python2.5/SocketServer.py", line 522, in __init__
    self.handle()
  File "/usr/lib/python2.5/BaseHTTPServer.py", line 316, in handle
    self.handle_one_request()
  File "/usr/lib/python2.5/BaseHTTPServer.py", line 310, in handle_one_request
    method()
  File "/opt/short_url_redirector/shorturl.py", line 38, in do_GET
    self.http_output( http_code, response_txt, long_url )
  File "/opt/short_url_redirector/shorturl.py", line 52, in http_output
    self.send_response(http_code)
  File "/usr/lib/python2.5/BaseHTTPServer.py", line 370, in send_response
    self.send_header('Server', self.version_string())
  File "/usr/lib/python2.5/BaseHTTPServer.py", line 376, in send_header
    self.wfile.write("%s: %s\r\n" % (keyword, value))
  File "/usr/lib/python2.5/socket.py", line 274, in write
    self.flush()
  File "/usr/lib/python2.5/socket.py", line 261, in flush
    self._sock.sendall(buffer)
error: (32, 'Broken pipe')

A maior parte desses erros parece resultar de um problema ao chamar o método send_header() onde tudo que estou escrevendo é isto:

self.send_header('Location', long_url)

Então, estou curioso para saber onde tentar capturar essa exceção de IO no meu código...escrevo chamadas try/except em torno de cada uma das chamadas self.send_header/self.end_headers/self.wfile.write?O outro erro que vejo de vez em quando é este, mas não tenho certeza de qual exceção observar para capturar isso:

Traceback (most recent call last):
  File "/usr/lib/python2.5/SocketServer.py", line 222, in handle_request
    self.process_request(request, client_address)
  File "/usr/lib/python2.5/SocketServer.py", line 241, in process_request
    self.finish_request(request, client_address)
  File "/usr/lib/python2.5/SocketServer.py", line 254, in finish_request
    self.RequestHandlerClass(request, client_address, self)
  File "/usr/lib/python2.5/SocketServer.py", line 522, in __init__
    self.handle()
  File "/usr/lib/python2.5/BaseHTTPServer.py", line 316, in handle
    self.handle_one_request()
  File "/usr/lib/python2.5/BaseHTTPServer.py", line 299, in handle_one_request
    self.raw_requestline = self.rfile.readline()
  File "/usr/lib/python2.5/socket.py", line 381, in readline
    data = self._sock.recv(self._rbufsize)
error: (104, 'Connection reset by peer')

Answer 1

A exceção "tubo quebrado" significa que seu código tentou gravar em um soquete/tubo que a outra extremidade fechou.Se a outra extremidade for um navegador da Web, o usuário poderá ter interrompido a solicitação.Você pode ignorar o rastreamento;isso não indica um problema sério.Se quiser suprimir a mensagem, você pode tentar ...exceto bloqueie todo o código em sua função http_output e registre a exceção, se desejar.

Além disso, se desejar que seu servidor HTTP processe mais de uma solicitação por vez, você precisará que sua classe de servidor use uma das classes SocketServer.ForkingMixIn e SocketServer.ThreadingMixIn.Verifique a documentação do módulo SocketServer para obter detalhes.

Adicionar:A exceção "conexão redefinida por peer" significa que seu código tentou ler de um soquete inativo.Se quiser suprimir o traceback, você precisará estender a classe BaseHTTPServer e substituir o método handle_one_request para adicionar um try ...exceto bloco.De qualquer forma, você precisará de uma nova classe de servidor para implementar a sugestão anterior sobre o processamento de mais de uma solicitação por vez.

Answer 2

Isto parece ser um bug no SocketServer, veja este link Bug do Python:14574

Uma correção (funciona para mim no Python 2.7) é substituir o método SocketServer.StreamRequestHandler finish(), algo assim:

...
def finish(self,*args,**kw):
  try:
    if not self.wfile.closed:
      self.wfile.flush()
      self.wfile.close()
  except socket.error:
    pass
  self.rfile.close()

  #Don't call the base class finish() method as it does the above
  #return SocketServer.StreamRequestHandler.finish(self)

Answer 3

Na minha aplicação, o erro não ocorreu em finish(), ocorreu em handle().Esta correção captura o broken pipe erros:

class MyHTTPRequestHandler(BaseHTTPServer.BaseHTTPRequestHandler):

    ...

    def handle(self):
        try:
            BaseHTTPServer.BaseHTTPRequestHandler.handle(self)
        except socket.error:
            pass