Comment limiter le temps d'exécution d'un appel de fonction bloquante en Python ?

Limitation du temps d'exécution d'un appel de fonction

Vous avez rencontré un appel de fonction lié au socket dans votre code qui présente un blocage étendu occasionnel, atteignant parfois des heures. Cela pose une situation inacceptable. Pour atténuer ce problème, vous recherchez une solution dans votre code pour limiter le temps d'exécution de la fonction.

Comme suggéré dans la réponse acceptée, l'utilisation d'un autre thread peut fournir une solution. Cependant, une approche améliorée consiste à utiliser l'instruction with, offrant une commodité syntaxique pour la fonction timeout :

import signal
from contextlib import contextmanager

class TimeoutException(Exception): pass

@contextmanager
def time_limit(seconds):
    def signal_handler(signum, frame):
        raise TimeoutException("Timed out!")
    signal.signal(signal.SIGALRM, signal_handler)
    signal.alarm(seconds)
    try:
        yield
    finally:
        signal.alarm(0)

try:
    with time_limit(10):
        long_function_call()
except TimeoutException as e:
    print("Timed out!")

Cette approche vous permet de gérer le délai d'attente de manière plus élégante dans un bloc, en déclenchant une TimeoutException si la fonction dépasse le délai imparti.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!