V->SrSSKJr SSKJr SSKJr SSKrSSKrSSKrSSKrSSK r SSK r SSK r SSK J r \RrSr"SS \5r"S S \5r"S S \5r"SS\5r"SS\5rSSSSS\SS4SjrSrSrg!\a \rNWf=f)z!Implementation of retrying logic.)absolute_import)division)unicode_literalsN) exceptionsc\rSrSrSrSrSrg) RetryerState*z+Object that holds the state of the retryer.c(XlX lX0lg)ahInitializer 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. Nretrialtime_passed_mstime_to_wait_ms)selfr rrs %lib/googlecloudsdk/core/util/retry.py__init__RetryerState.__init__-sL(*r N)__name__ __module__ __qualname____firstlineno____doc__r__static_attributes__rrr r *s 3 +rr c2^\rSrSrSrU4SjrSrSrU=r$)RetryException<z#Raised to stop retrials on failure.cP>XlX lX0l[[U]U5 gN)message last_resultstatesuperrr)rr!r"r# __class__s rrRetryException.__init__?s"L"J .$(1rcSRURURRURRURR S9$)Nzvlast_result={last_result}, last_retrial={last_retrial}, time_passed_ms={time_passed_ms},time_to_wait={time_to_wait_ms})r" last_retrialrr)formatr"r#r rr)rs r__str__RetryException.__str__EsK --3V ,,!ZZ//#zz88 $ : : .4.<=r)r"r!r#) rrrrrrr*r __classcell__)r%s@rrr<s+2 ==rrc\rSrSrSrSrg) WaitExceptionOz Raised when timeout was reached.rNrrrrrrrrrr.r.Os(rr.c\rSrSrSrSrg)MaxRetrialsExceptionSz&Raised when too many retrials reached.rNr0rrrr2r2Ss.rr2cV\rSrSrSrSSS\SS4SjrSrSrS Sjr S Sjr S r g) RetryerWz5Retries a function based on specified retry strategy.NcLXlX lX0lX@lXPlX`lg)aInitializer 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 rrRetryer.__init__Zs' &#)E&O1+rcURb&URUR::a [SX5eURb4URUR -UR:a [ SX5egg)NReachedTimeout)r8r r2r9rrr.)rresultr#s r _RaiseIfStopRetryer._RaiseIfStopqso %$*<*< *M F :: $    5 5 58I8I IIv55 J%rcUnU(aUR(aMSnURS:a*U[R"XC- UR5:aUnOX0RU--nUR(a%U[R"5UR-- nUR (a[ X0R 5nU$g)aqGet 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:mathlogr;randomr=min)rr(sleep_ms wait_time_mshundred_years_mss r_GetTimeToWaitRetryer._GetTimeToWaitxsL + +.  - - 3 txx  +  . .H090*, << L L,  $//99  <)>)>?  rc^^^^TbTOSmTbTO0mUUU4SjnTcSnOU4SjnUnURXgUS9upU (a[R"U SU SS9 U $) a"Retries 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. rcV>T"T0TD6S4$! S[R"54s$=fr )sysexc_info)argsfunckwargssrTryFunc)Retryer.RetryOnException..TryFuncs5$T$V$d**$S\\^##s (cUSSL$)Nr)xss r*Retryer.RetryOnException..s !A$d"2rc>>USnUcgT"USUSUSU5$)Nr_Frr)try_func_resultr#rXshould_retry_ifs rShouldRetryFunc1Retryer.RetryOnException..ShouldRetryFuncs3"1%  x{HQK!eLLr)rgrPr_retb) RetryOnResultrreraise) rrZrYr[rgrPr\ should_retryrhrHrXs ```` rRetryOnExceptionRetryer.RetryOnExceptionsz(#4D)VrF$ 2lM %l))*BF!!5 MrcZ^UbUOSnUbUO0n[5nSn[T5(aTnOU4Sjn[U[R5(a [ U5n O[ R"U5n U"U0UD6n [5U- n [U 5n URX|5n [X{U 5nU"X5(dU $U S:Xa [SX5eUR(aURX5 URX5 [U 5 US- nM![a Sn Nf=f)a;Retries 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>UT:H$r r)r`rargs rrb'Retryer.RetryOnResult..s !"6rzSleep iteration stopr_)_GetCurrentTimeMscallable isinstancecollections_abcIterableiter itertoolsrepeatnextrS StopIterationr r2r<rI_SleepMs)rrZrYr[rgrP start_time_msr rn sleep_genrHrsleep_from_genrr#s ` rrlRetryer.RetryOnResults.&#4D)VrF%'MG  $l6l(O4455x.i""8,i T$V$f(*]:nGi--gF7ODe & ( ( B "#96II ! !   / & lg)  s D D*)D*)r:r;r8r9r<r=)NNNN) rrrrr_DEFAULT_JITTER_MSrrIrSrorlrrrrr5r5Ws@="&D,0r?rgrPrBc>[TT TTT S9nURTXT T S9$![a4nURSn[R "USUSS9 SnAgSnAff=f)N)r>r?r@rArB)rYr[rgrPr_rerj)r5ror2r"rrm) rYr[retryermre to_reraiser@frAr>r?rgrPrBs rDecoratedFunction+RetryOnException..DecoratedFunction*s{!%A- /G :  % %ad6E/7&99 :??1%jA:a=9:s" A *AA ) functoolspartialrowraps) rr>r?rPr@rArBrgrs ```````` rrorosa0Y   %A!'- // ??1 : : : rcD[[R"5S-5$)Nr)inttimerrrruru=s TYY[4   rc6[R"US- 5 g)Ng@@)rsleep)rs rrrAs**_v %&r)r __future__rrr collectionsrr{rLrNrWrgooglecloudsdk.corerabcrxAttributeErrorrobjectr Exceptionrr.r2r5rorurrrrrs (&' * OO/ +6+$=Y=&)N)/>/kfk\$D"1(,%) 5p!'} / s BBB