主页 / 问题 / 1441377
Accepted
J. Doe
Asked:2022-08-21 23:04:38 +0000 UTC2022-08-21 23:04:38 +0000 UTC

告诉我在一定时间后更换处理程序的请求处理的实现？

  • 772

我有一个烧瓶应用程序，它接受一个请求，基于它执行一个函数，并将结果返回给用户。我需要确保如果函数在请求到达后五秒内没有返回结果，则将另一个函数的结果发送给用户。此外，即使在完成对请求的响应之后，原始功能仍继续工作并结束，这一点很重要。如果原始函数在 5 秒内返回结果，则需要在响应中返回其结果。

有什么功能可以做到这一点吗？

这是一个伪代码，大致描述了我需要什么：

@application.route("/", methods=['POST'])
def main():
    executor = SomeExecutor()
    # устанавливаем функцию, которая должна стриггериться по истечении 5 секунд, если main_function не успеет отработать
    executor.execute_on_timeout(alternative_function, args = [request], exec_after=5)
    # main_function продолжает работать даже по истечении 5 секунд
    executor.execute(main_function, args = [request]) 
    # получаем результат от одной из функций (ждем максимум 5 секунд + время выполнения alternative_function) 
    response = executor.result()
    return response
python python-3.x

1 个回答

  1. Best Answer

    这是使用concurrent.futures.wait的一种选择：

    from concurrent import futures
 
def with_delay(f, main_future, delay):
    @wraps(f)
    def delayed(*args, **kwargs):
        try:
             return main_future.result(timeout=delay)
        except futures.TimeoutError as e:
             return f(*args, **kwargs)
    return delayed

with futures.ThreadPoolExecutor(max_workers=2) as executor:
    main_future = executor.submit(main_function, request)
    alt_future = executor.submit(
         with_delay(alternative_function, main_future, delay=5),
         request
    )
    finished, pending = futures.wait(
         [main_future, alt_future], 
         return_when=futures.FIRST_COMPLETED
    )
    # в finished будут завершенная(-ые) задачи
    # можно реализовать логику, какой результат брать, если обе завершились
    # в коде ниже просто первый попавшийся
    print(finished[0].result())
    • 1

相关问题