/Jf=ddlZddlZGddeZGddZGddZGdd ZGd d ZGd d ZGddZ GddZ dS)NceZdZfdZxZS)RequestExceededExceptionc||_||_d||}t|dS)aError when requested amount exceeds what is allowed The request that raised this error should be retried after waiting the time specified by ``retry_time``. :type requested_amt: int :param requested_amt: The originally requested byte amount :type retry_time: float :param retry_time: The length in time to wait to retry for the requested amount z*@*@D ' ' ' ) (rc|j5|j}|j|r#||||cdddS|||r||||n"|||cdddS ddddS#1swxYwYdS)acConsume an a requested amount :type amt: int :param amt: The amount of bytes to request to consume :type request_token: RequestToken :param request_token: The token associated to the consumption request that is used to identify the request. So if a RequestExceededException is raised the token should be used in subsequent retry consume() request. :raises RequestExceededException: If the consumption amount would exceed the maximum allocated bandwidth :rtype: int :returns: The amount consumed N) r_r$rrb is_scheduled,_release_requested_amt_for_scheduled_request_projected_to_exceed_max_rate!_raise_request_exceeded_exception_release_requested_amtr amt request_tokentime_nows rrDzLeakyBucket.consumes$Z B B',,..H*77 FF BHH B B B B B B B B 33CBB B66223AA B B B B B B B B B B B B B B B B B B B B B B B B B BsA B=AB==CCcN|j||}||jkSr/)r`get_projected_rater\)r rnrpprojected_rates rrjz)LeakyBucket._projected_to_exceed_max_rates'+>>sHMM..rcb|j||||Sr/)rbprocess_scheduled_consumptionrlrms rriz8LeakyBucket._release_requested_amt_for_scheduled_requests: #AA    **3999rc|t|jz }|j|||}t ||)N)rr)r[r\rbschedule_consumptionr)r rnrorpallocated_timers rrkz-LeakyBucket._raise_request_exceeded_exception%sSuT^4440EE   '*    rc<|j|||Sr/)r`record_consumption_rate)r rnrps rrlz"LeakyBucket._release_requested_amt.s  223AAA r)NNN) rrrr rDrjrirkrlrrrrYrYs" !A!A!A!AFBBB>///:::   rrYc&eZdZdZdZdZdZdS)rcc"i|_d|_dS)z*Schedules when to consume a desired amountrN) _tokens_to_scheduled_consumption _total_waitrs rr zConsumptionScheduler.__init__4s02-rc||jvS)zIndicates if a consumption request has been scheduled :type token: RequestToken :param token: The token associated to the consumption request that is used to identify the request. )r})r tokens rrhz!ConsumptionScheduler.is_scheduled9s===rcT|xj|z c_|j|d|j|<|jS)aSchedules a wait time to be able to consume an amount :type amt: int :param amt: The amount of bytes scheduled to be consumed :type token: RequestToken :param token: The token associated to the consumption request that is used to identify the request. :type time_to_consume: float :param time_to_consume: The desired time it should take for that specific request amount to be consumed in regardless of previously scheduled consumption requests :rtype: float :returns: The amount of time to wait for the specific request before actually consuming the specified amount. ) wait_durationtime_to_consume)r~r})r rnrrs rrwz)ConsumptionScheduler.schedule_consumptionBs@& O+!-.8 8 -e4rc|j|}t|j|dz d|_dS)zProcesses a scheduled consumption request that has completed :type token: RequestToken :param token: The token associated to the consumption request that is used to identify the request. rrN)r}popmaxr~)r rscheduled_retrys rruz2ConsumptionScheduler.process_scheduled_consumption\sE?CCEJJ  /@A A1  rN)rrrr rhrwrurrrrcrc3sP >>>   4      rrccDeZdZd dZedZdZdZdZdZ dS) ra皙?c0||_d|_d|_dS)aTracks the rate of bandwidth consumption :type a: float :param a: The constant to use in calculating the exponentional moving average of the bandwidth rate. Specifically it is used in the following calculation: current_rate = alpha * new_rate + (1 - alpha) * current_rate This value of this constant should be between 0 and 1. N)_alpha _last_time _current_rate)r alphas rr zBandwidthRateTracker.__init__js  !rc"|jdS|jS)zmThe current transfer rate :rtype: float :returns: The current tracked transfer rate N)rrrs r current_ratez!BandwidthRateTracker.current_ratezs ? "3!!rc@|jdS|||S)aZGet the projected rate using a provided amount and time :type amt: int :param amt: The proposed amount to consume :type time_at_consumption: float :param time_at_consumption: The proposed time to consume at :rtype: float :returns: The consumption rate if that amt and time were consumed Nr)r*_calculate_exponential_moving_average_rater rntime_at_consumptions rrrz'BandwidthRateTracker.get_projected_rates/ ? "3>> $   rcx|j||_d|_dS||||_||_dS)aRecord the consumption rate based off amount and time point :type amt: int :param amt: The amount that got consumed :type time_at_consumption: float :param time_at_consumption: The time at which the amount was consumed Nr)rrrrs rrzz,BandwidthRateTracker.record_consumption_ratesK ? "1DO!$D  F!LL $  .rcJ||jz }|dkrtdS||z S)Nrinf)rr[)r rnr time_deltas r_calculate_ratez$BandwidthRateTracker._calculate_rates0(4?: ?? << j!!rch|||}|j|zd|jz |jzzS)N)rrr)r rnrnew_rates rrz?BandwidthRateTracker._calculate_exponential_moving_average_rates:''-@AA{X%T[Drs y,        !!!!!!!!$########LnnnnnnnnbZZZZZZZZz3 3 3 3 3 3 3 3 lNONONONONONONONONONOr