One Hat Cyber Team

Your IP: 216.73.216.79
Server IP: 97.74.87.16
Server: Linux 16.87.74.97.host.secureserver.net 5.14.0-503.38.1.el9_5.x86_64 #1 SMP PREEMPT_DYNAMIC Fri Apr 18 08:52:10 EDT 2025 x86_64
Server Software: Apache
PHP Version: 8.2.28
Buat File | Buat Folder

Dir : ~/var/opt/nydus/ops/customer_local_ops/util/

View File Name : retry.py

# -*- coding: utf-8 -*-

"""\
© Copyright. All rights reserved.

"""

from __future__ import unicode_literals
from datetime import datetime, timedelta
from functools import wraps
import logging

LOG = logging.getLogger(__name__)


class Retry:
    """Ops decorated with @retry can return a Retry instance to request retry
    with the interval and timeout as specified in the decorator.  See also the
    RETRY sentinel if there is no intermediate_result.
    """

    def __init__(self, intermediate_result=None):
        if not (intermediate_result is None or
                isinstance(intermediate_result, dict)):
            raise TypeError('Retry intermediate_result must be dictionary '
                            'but got %s' % type(intermediate_result))
        self.intermediate_result = intermediate_result

    def __str__(self):
        return 'Retry ' + str(self.__dict__)


RETRY = Retry()  # A Retry that can be returned from ops decorated with @retry
# when there is no intermediate_result.


def retry(interval=2, timeout=60):
    """Op decorator that returns a retry request to Nydus when the op returns
    a Retry instance.  Op will be retried by Nydus after `interval` seconds.
    Returns failure at the specified timeout (seconds).
    """

    def deco(original):
        @wraps(original)
        def wrapper(*args, **kw):
            intermediate_result = kw.get('intermediate_result', None)

            now = datetime.utcnow()
            timeout_abs = (intermediate_result and
                           intermediate_result.get('timeout') or
                           now + timedelta(seconds=timeout))

            if now >= timeout_abs:
                msg = 'Timeout waiting for {} ({}s)'.format(getattr(original, '__name__', '?'), timeout)
                LOG.error(msg)
                return False, msg

            result = original(*args, **kw)

            if isinstance(result, Retry):
                ir = result.intermediate_result or {}
                ir.setdefault('timeout', timeout_abs)
                return False, ir, interval
            return result

        return wrapper
    return deco