V-:dZddlmZddlmZddlmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z  ejZdZGdd eZGd d eZGd d eZGddeZGddeZdddddeddfdZdZdZy#e$reZYUwxYw)z!Implementation of retrying logic.)absolute_import)division)unicode_literalsN) exceptionsceZdZdZdZy) RetryerStatez+Object that holds the state of the retryer.c.||_||_||_y)aInitializer for RetryerState. Args: retrial: int, the retry attempt we are currently at. time_passed_ms: int, number of ms that passed since we started retryer. time_to_wait_ms: int, number of ms to wait for the until next trial. If this number is -1, it means the iterable item that specifies the next sleep value has raised StopIteration. N)retrialtime_passed_mstime_to_wait_ms)selfr r r s %lib/googlecloudsdk/core/util/retry.py__init__zRetryerState.__init__-sDL(D*DN)__name__ __module__ __qualname____doc__rrrr r *s 3 +rr c(eZdZdZfdZdZxZS)RetryExceptionz#Raised to stop retrials on failure.cV||_||_||_tt||yN)message last_resultstatesuperrr)rrrr __class__s rrzRetryException.__init__?s(DL"DDJ .$(1rcdj|j|jj|jj|jj S)Nzvlast_result={last_result}, last_retrial={last_retrial}, time_passed_ms={time_passed_ms},time_to_wait={time_to_wait_ms})r last_retrialr r )formatrrr r r )rs r__str__zRetryException.__str__EsK --3V ,,!ZZ//#zz88 $ : : .4.<=r)rrrrrr# __classcell__)rs@rrr<s+2 =rrceZdZdZy) WaitExceptionz Raised when timeout was reached.Nrrrrrrrr&r&Os(rr&ceZdZdZy)MaxRetrialsExceptionz&Raised when too many retrials reached.Nr'rrrr)r)Ss.rr)cHeZdZdZdddeddfdZdZdZ ddZ ddZ y) Retryerz5Retries a function based on specified retry strategy.NcX||_||_||_||_||_||_y)a>Initializer for Retryer. Args: max_retrials: int, max number of retrials before raising RetryException. max_wait_ms: int, number of ms to wait before raising exponential_sleep_multiplier: float, The exponential factor to use on subsequent retries. jitter_ms: int, random [0, jitter_ms] additional value to wait. status_update_func: func(result, state) called right after each trial. wait_ceiling_ms: int, maximum wait time between retries, regardless of modifiers added like exponential multiplier or jitter. N) _max_retrials _max_wait_ms_exponential_sleep_multiplier _jitter_ms_status_update_func_wait_ceiling_ms)r max_retrials max_wait_msexponential_sleep_multiplier jitter_msstatus_update_funcwait_ceiling_mss rrzRetryer.__init__Zs3 &D#D)ED&DO1D+Drc|j&|j|jkr td|||j4|j|j z|jkDr t d||yy)NReachedTimeout)r-r r)r.r r r&)rresultrs r _RaiseIfStopzRetryer._RaiseIfStopqss %$*<*< *M FE :: $    5 5 58I8I IIvu55 J%rcd|}|r|jrLd}|jdkDr)|tj||z |jkDr|}n||j|zz}|jr$|t j|jzz }|j rt ||j }|Sy)aGet time to wait after applying modifyers. Apply the exponential sleep multiplyer, jitter and ceiling limiting to the base sleep time. Args: last_retrial: int, which retry attempt we just tried. First try this is 0. sleep_ms: int, how long to wait between the current trials. Returns: int, ms to wait before trying next attempt with all waiting logic applied. l0jy g?r)r/mathlogr0randomr2min)rr!sleep_ms wait_time_mshundred_years_mss r_GetTimeToWaitzRetryer._GetTimeToWaitxsL + +.  - - 3 txx | +  . .H090*, $<< L L,  $//99  <)>)>?  rcndnifd}d}nfd}|}|j|||\} } | rtj| d| d| S) aVRetries the function if an exception occurs. Args: func: The function to call and retry. args: a sequence of positional arguments to be passed to func. kwargs: a dictionary of positional arguments to be passed to func. should_retry_if: func(exc_type, exc_value, exc_traceback, state) that returns True or False. sleep_ms: int or iterable for how long to wait between trials. Returns: Whatever the function returns. Raises: RetryException, WaitException: if function is retries too many times, or time limit is reached. rcT idfS#dtjfcYSxYwr)sysexc_info)argsfunckwargssrTryFuncz)Retryer.RetryOnException..TryFuncs5$T$V$d**$S\\^##s 'c|dduS)Nr)xss rz*Retryer.RetryOnException..s !A$d"2rc<|d}|y|d|d|d|S)NrPFrr)try_func_resultrrJshould_retry_ifs rShouldRetryFuncz1Retryer.RetryOnException..ShouldRetryFuncs3"1%  x{HQK!eLLr)rWrCrPrUtb) RetryOnResultrreraise) rrLrKrMrWrCrN should_retryrXr<rJs ```` rRetryOnExceptionzRetryer.RetryOnExceptions~(#4D)VrF$ 2lM %l))*BFH!!5 Mrc0||nd}||ni}t}d}tr}nfd}t|tjr t |} nt j|} ||i|} t|z } t| } |j|| } t|| | }|| |s| S| dk(r td| ||jr|j| ||j| |t| |dz }#t$rd} YxwxYw)aoRetries the function if the given condition is satisfied. Args: func: The function to call and retry. args: a sequence of arguments to be passed to func. kwargs: a dictionary of positional arguments to be passed to func. should_retry_if: result to retry on or func(result, RetryerState) that returns True or False if we should retry or not. sleep_ms: int or iterable, for how long to wait between trials. Returns: Whatever the function returns. Raises: MaxRetrialsException: function retried too many times. WaitException: time limit is reached. rrc|k(Srr)rQrRrWs rrSz'Retryer.RetryOnResult..s !"6rzSleep iteration stoprP)_GetCurrentTimeMscallable isinstancecollections_abcIterableiter itertoolsrepeatnextrF StopIterationr r)r1r=_SleepMs)rrLrKrMrWrC start_time_msr r] sleep_genr<r sleep_from_genr rs ` rr[zRetryer.RetryOnResults1&#4D)VrF%'MG $l6l(O445x.i""8,i T$V$f(*]:nGi--g~F7NODe &% ( B "#965II ! !   / & lg)  s? D DD)NNNN) rrrr_DEFAULT_JITTER_MSrr=rFr^r[rrrr+r+Ws@="&D,0.DecoratedFunction*s!%A- /G :  % %ad66E/7&99 :??1%jA:a=99:s) A(,A##A() functoolspartialr^wraps) rvr3r4rCr5r6r7rWrws ```````` rr^r^sa0Y   %A!'- // ??1 : : : rcBttjdzS)Nr)inttimerrrrbrb=s TYY[4   rc4tj|dz y)Ng@@)r}sleep)r s rrlrlAs**_v %&r)r __future__rrr collectionsrxrhr?rArIr}googlecloudsdk.corerabcreAttributeErrorrpobjectr Exceptionrr&r)r+r^rbrlrrrrs (&' * OO/ +6+$=Y=&)N)/>/kfk\$D"1(,%) 5p!'} / s BBB