Comment puis-je déclencher une exception de délai d'attente Tornado dans un ContextDecorator?

Question

Je cherche à:

• Faire une enveloppe de minuteur personnalisé, pour l'enregistrement/d'autres fins de synchronisation
• Inclure préemptive sortant si l'opération enveloppé (s) dépasse une durée prédéfinie

voilà ce que j'ai jusqu'à présent:

from contextlib import ContextDecorator 
import datetime 

from tornado import gen, ioloop 
from tornado.concurrent import Future 


class timing_logger(ContextDecorator): 
    def __init__(self, allowed_ms): 
     self.allowed_ms = allowed_ms 
     self.f = Future() 
     # this exception is not bubbled up by Tornado but fires 
     gen.with_timeout(datetime.timedelta(seconds=1+allowed_ms/1000), self.f) 

    def __enter__(self): 
     self.start_time = datetime.datetime.now() 
     return self 

    def __exit__(self, exc_type, exc_val, traceback): 
     self.f.set_result(True) 
     elapsed_time_ms = (datetime.datetime.now() - self.start_time).total_seconds() * 1000 

     if exc_type == gen.TimeoutError: 
      raise TimeoutError('ruh oh, this is reeeally bad') 

     if elapsed_time_ms > self.allowed_ms: 
      raise TimeoutError('took {actual} ms, but was only allowed {allowed}.'.format(
       actual=elapsed_time_ms, allowed=self.allowed_ms)) 

     else: 
      print('worked. nothing to see here...') 
     return False 


@gen.coroutine 
def main(): 

    with timing_logger(1000): 
     # real application may be a variety of coroutines 
     # and regular function calls, some of which may hang 
     # for a long time 
     for i in range(25): 
      yield gen.sleep(0.1) 


if __name__ == "__main__": 
    ioloop.IOLoop.current().run_sync(
     lambda: main()) 

le problème que j'ai ici est que parce que je ne suis pas ce qui donne le gen.with_timeout avenir, dans la pile je vois:

$python test.py 
ERROR:tornado.application:Future <tornado.concurrent.Future object at 0x10c7cb668> exception was never retrieved: tornado.gen.TimeoutError: Timeout 
Traceback (most recent call last): 
    File "test.py", line 48, in <module> 
    lambda: main()) 
<snip> 
    yielded = self.gen.send(value) 
    File "test.py", line 43, in main 
    yield gen.sleep(0.1) 
    File "test.py", line 28, in __exit__ 
    actual=elapsed_time_ms, allowed=self.allowed_ms)) 
TimeoutError: took 2606.2940000000003 ms, but was only allowed 1000. 

Le délai d'attente de tornade est pas être « barboter » (faute de meilleur mot).

Je voudrais avoir la __exit__ capture l'exception, pour que je puisse le manipuler et ouvrir une session appropriée, tout-relançant comme un autre type d'exception.

Je ne sais pas si je dois:

• Ne pas utiliser un ContextDecorator tout
• Faites quelque chose de différent avec comment/où je l'appel tornade
• ????

Je sais que dans cet exemple, je peux envelopper tout le code d'appel dans un coroutine et ajouter un délai d'attente à la fonction main autour de l'enregistreur de synchronisation wrap, comme ceci:

@gen.coroutine 
def main(): 


    @gen.coroutine 
    def f(): 
     with timing_logger(1000): 
      # real application may be a variety of coroutines 
      # and regular function calls, some of which may hang 
      # for a long time 
      for i in range(25): 
       yield gen.sleep(0.1) 

    future = f() 
    yield gen.with_timeout(datetime.timedelta(seconds=1), future) 

Mais je suis cherche à inclure le ci-dessus dans mon ContextDecorator parce que d'avoir à copier cela autour de tout ce que je veux utiliser timing_logger devient fastidieux ainsi que sujettes aux erreurs.

Comment puis-je obtenir les fonctionnalités voulues pour permettre à mon ContextDecorator d'inclure les délais d'attente dans le cadre de la façon dont il fonctionne?

En utilisant Python 3.6.1 et la dernière Tornado (4.5.1).

Answer 1

Plutôt que d'utiliser un délai d'attente de tornade, vous pouvez utiliser un signal pour interrompre et de déclencher cette exception (si alarm only allows integer, second inputs):

def timeout_handler(signum, frame): 
     raise gen.TimeoutError() 

    self.signal = signal.signal(signal.SIGALRM, timeout_handler) 
    signal.alarm(1 + self.allowed_ms // 1000) 

Cela soulèvera l'exception appropriée, entraînant une ContextDecorator complète qui ressemble à:

from contextlib import ContextDecorator 
import datetime 

from tornado import gen, ioloop 
from tornado.concurrent import Future 


class timing_logger(ContextDecorator): 
    def __init__(self, allowed_ms): 
     self.allowed_ms = allowed_ms 
     self.f = Future() 
     # this exception is not bubbled up by Tornado but fires 
     gen.with_timeout(datetime.timedelta(seconds=1+allowed_ms/1000), self.f) 

    def __enter__(self): 
     self.start_time = datetime.datetime.now() 
     def timeout_handler(signum, frame): 
      raise gen.TimeoutError() # could be any type of exception 

     self.signal = signal.signal(signal.SIGALRM, timeout_handler) 
     signal.alarm(1 + self.allowed_ms // 1000) 

     return self 

    def __exit__(self, exc_type, exc_val, traceback): 
     signal.alarm(0) 
     self.f.set_result(True) 
     elapsed_time_ms = (datetime.datetime.now() - self.start_time).total_seconds() * 1000 

     if exc_type == gen.TimeoutError: 
      raise TimeoutError('ruh oh, this is reeeally bad') 

     if elapsed_time_ms > self.allowed_ms: 
      raise TimeoutError('took {actual} ms, but was only allowed {allowed}.'.format(
       actual=elapsed_time_ms, allowed=self.allowed_ms)) 

     else: 
      print('worked. nothing to see here...') 
     return False 

Notez que vous devez réinitialiser l'alarme dans le __exit__ ou il plus tard le feu dans votre code.