VSfFjUddlZddlmZmZddlZdZdZeed<dZ eed<dZ eed <Gd d Z y) N)NoReturnOptionalcrtjdttjj S)z google.cloud.firestore_v1.rate_limiter.utcnow() is deprecated. Use datetime.datetime.now(datetime.timezone.utc) instead. zxgoogle.cloud.firestore_v1.rate_limiter.utcnow() is deprecated. Use datetime.datetime.now(datetime.timezone.utc) instead.)warningswarnDeprecationWarningdatetimeutcnoweC:\Users\user\Documents\project_loop\venv\Lib\site-packages\google/cloud/firestore_v1/rate_limiter.pyr r s0  MM D    # # %%r idefault_initial_tokensi,default_phase_lengthi@Bmicroseconds_per_secondcreZdZdZedefdedeedefdZdZ ddeed e d efd Z d Z d e fd Zd e fdZy) RateLimitera Implements 5/5/5 ramp-up via Token Bucket algorithm. 5/5/5 is a ramp up strategy that starts with a budget of 500 operations per second. Additionally, every 5 minutes, the maximum budget can increase by 50%. Thus, at 5:01 into a long bulk-writing process, the maximum budget becomes 750 operations per second. At 10:01, the budget becomes 1,125 operations per second. The Token Bucket algorithm uses the metaphor of a bucket, or pile, or really any container, if we're being honest, of tokens from which a user is able to draw. If there are tokens available, you can do the thing. If there are not, you can not do the thing. Additionally, tokens replenish at a fixed rate. Usage: rate_limiter = RateLimiter() tokens = rate_limiter.take_tokens(20) if not tokens: queue_retry() else: for _ in range(tokens): my_operation() Args: initial_tokens (Optional[int]): Starting size of the budget. Defaults to 500. phase_length (Optional[int]): Number of seconds, after which, the size of the budget can increase by 50%. Such an increase will happen every [phase_length] seconds if operation requests continue consistently. Ninitial_tokensglobal_max_tokens phase_lengthc4d|_||_d|_d|_||_|j|_|jJt |j|j|_t |j |j|_||_d|_y)Nr) _operations_this_phase_global_max_tokens_start _last_refill_available_tokens_maximum_tokensmin _phase_length_phase)selfrrrs r __init__zRateLimiter.__init__Gs,-# #437 9='5%)$:$:  " " .%(&&(?(?&D "$'t';';T=T=T#UD #/ r ctjjtjj}|jxs||_|j xs||_y)N)r nowtimezoneutcrr)r r s r _start_clockzRateLimiter._start_clocklsG""&&x'8'8'<'<=kk+V  --7r num allow_lessreturnc|j|j|j|rdn|}|j|k\rBt |j|}|xj|zc_|xj |z c_|Sy)zCReturns the number of available tokens, up to the amount requested.r)r& _check_phase_refillrrr)r r'r(minimum_tokens _num_to_takes r take_tokenszRateLimiter.take_tokensqsw   (c  ! !^ 3t55s;L  " "l 2 "  ' '< 7 ' r cptjjtjj|jz }|j |j z}||jk(ry|j}d|_|j}||_|r!|j|kDr|jyyy)acIncrements or decrements [_phase] depending on traffic. Every [_phase_length] seconds, if > 50% of available traffic was used during the window, increases [_phase], otherwise, decreases [_phase]. This is a no-op unless a new [_phase_length] number of seconds since the start was crossed since it was last called. Nr) r r#r$r%rsecondsrrr_increase_maximum_tokens)r ageexpected_phaseoperations_last_phaseprevious_phases r r,zRateLimiter._check_phases    ! !("3"3"7"7 84;; F "kkT-?-?? T[[ ( %)%@%@&'#"kk$  !T[[>%A  ) ) +&B r ct|jdz|_|j&t|j|j|_yy)Ng?)roundrrr)r s r r3z$RateLimiter._increase_maximum_tokenssF$T%9%9C%?@  " " .#&t';';T=T=T#UD  /r ctjjtjj}||jz }|r|||_|j dk\r|j |_y|jtz }t||j z}t|j |j|z|_yy)zUReplenishes any tokens that should have regenerated since the last operation.r+N) r r#r$r%rr2rr microsecondsrr9r)r r#time_since_last_refill_percent_of_max new_tokenss r r-zRateLimiter._refills"*!2!2!6!6x7H7H7L7L!M584;L;L5L ! #D &--2)-)=)=& +77:QQ #($:N:N(N"O *-((**Z7*&# "r )r+F)__name__ __module__ __qualname____doc__rrintrr!r&boolr0r,rr3r-r r r rr&s|D5+/0 ##$C=# #J8 x} d s ,@V(V r r) r typingrrrr rrC__annotations__rrrr r r rGsD% &"!"c"&&YYr