U ð¤ùdÅã@sxUdZddlmZddlZddlZddlmZddlmZm Z ddl mZmZm Z mZmZmZmZmZddlmZeddd Zd ed<eddd Zd ed<ededƒd Zd ed<ddœdd„Zddœdd„Zddœdd„Zdddœdd„Zddœdd„Zdddœd d!„ZGd"d#„d#ƒZ ed$ed%efd&Z!d$d$d'œd(d)„Z"d*d+dd,œd-d.„Z#d/Z$d0ed1<d2Z%d0ed3<Gd4d5„d5ƒZ&dS)6zInternal helpers for CSOT.é)ÚannotationsN)Údeque)Ú ContextVarÚToken)ÚAnyÚCallableÚDequeÚMutableMappingÚOptionalÚTupleÚTypeVarÚcast)ÚWriteConcernÚTIMEOUT)ÚdefaultzContextVar[Optional[float]]ÚRTTçzContextVar[float]ÚDEADLINEÚinfúOptional[float]©ÚreturncCs t d¡S©N)rÚget©rrú1/tmp/pip-unpacked-wheel-oblwsawz/pymongo/_csot.pyÚget_timeout srÚfloatcCst ¡Sr)rrrrrrÚget_rtt$srcCst ¡Sr)rrrrrrÚget_deadline(srÚNone)ÚrttrcCst |¡dSr)rÚset)r!rrrÚset_rtt,sr#cCstƒs dSt ¡t ¡Sr)rrrÚtimeÚ monotonicrrrrÚ remaining0sr&)Úmax_timeoutrcCstƒ}|dkr|St||ƒS)z4Return the remaining timeout clamped to a max value.N)r&Úmin)r'ÚtimeoutrrrÚclamp_remaining6sr*c@sDeZdZdZdZddœdd„Zddœdd „Zd d d ddœd d„ZdS)Ú_TimeoutContextzInternal timeout context manager. Use :func:`pymongo.timeout` instead:: with pymongo.timeout(0.5): client.test.test.insert_one({}) ©Ú_timeoutÚ_tokensr)r)cCs||_d|_dSrr,)Úselfr)rrrÚ__init__Isz_TimeoutContext.__init__rcCsZt |j¡}t ¡}|jr(t ¡|jntdƒ}t t||ƒ¡}t d¡}|||f|_ |S)Nrr)rr"r-rrr$r%rr(rr.)r/Ú timeout_tokenZ prev_deadlineZ next_deadlineÚdeadline_tokenÚ rtt_tokenrrrÚ __enter__Ms z_TimeoutContext.__enter__rr )Úexc_typeÚexc_valÚexc_tbrcCs4|jr0|j\}}}t |¡t |¡t |¡dSr)r.rÚresetrr)r/r5r6r7r1r2r3rrrÚ__exit__Vs z_TimeoutContext.__exit__N)Ú__name__Ú __module__Ú__qualname__Ú__doc__Ú __slots__r0r4r9rrrrr+>s r+ÚF.)Úbound)Úfuncrcs,t ˆ¡dddddœ‡fdd„ƒ}tt|ƒS)z/Apply the client's timeoutMS to this operation.rr?)r/ÚargsÚkwargsrc sTtƒdkrD|j}|dk rDt|ƒˆ|f|ž|ŽW5QR£SQRXˆ|f|ž|ŽSr)rr-r+)r/rBrCr)©rArrÚcsot_wrapperes "zapply..csot_wrapper)Ú functoolsÚwrapsr r?)rArErrDrÚapplybsrHr zOptional[WriteConcern])ÚcmdÚ write_concernrcCs:|r |jrdS|j}tƒdk r*| dd¡|r6||d<dS)z+Apply the given write concern to a command.NZwtimeoutZwriteConcern)Zis_server_defaultÚdocumentrÚpop)rIrJZwcrrrÚapply_write_concernqs rMé ÚintÚ_MAX_RTT_SAMPLESéÚ_MIN_RTT_SAMPLESc@sTeZdZUdZded<ddœdd„Zddd œd d„Zddœdd „Zddœdd„ZdS)Ú MovingMinimumz4Tracks a minimum RTT within the last 10 RTT samples.zDeque[float]Úsamplesr rcCsttd|_dS)N)Úmaxlen)rrPrT©r/rrrr0…szMovingMinimum.__init__r)ÚsamplercCs|dkrdS|j |¡dS)Nr)rTÚappend)r/rWrrrÚ add_sampleˆszMovingMinimum.add_samplecCst|jƒtkrt|jƒSdS)z7Get the min, or 0.0 if there aren't enough samples yet.r)ÚlenrTrRr(rVrrrrs zMovingMinimum.getcCs|j ¡dSr)rTÚclearrVrrrr8–szMovingMinimum.resetN) r:r;r<r=Ú__annotations__r0rYrr8rrrrrS€s rS)'r=Ú __future__rrFr$ÚcollectionsrZcontextvarsrrÚtypingrrrr r rrr Zpymongo.write_concernrrr\rrrrrrr#r&r*r+r?rHrMrPrRrSrrrrÚs.(!