U ~fh@sUdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z m Z ddl mZmZmZmZmZmZmZmZerddlmZe d dd Zd ed <e d d d Zded <e dedd Zded<ddddZddddZddddZdddddZ ddddZ!ddd d!d"Z"Gd#d$d$e Z#ed%ed&efd'Z$d%d%d(d)d*Z%d+d,dd-d.d/Z&d0Z'd1ed2<d3Z(d1ed4<Gd5d6d6Z)dS)7zInternal helpers for CSOT.) annotationsN)deque)AbstractContextManager) ContextVarToken) TYPE_CHECKINGAnyCallableDequeMutableMappingOptionalTypeVarcast) WriteConcernTIMEOUT)defaultzContextVar[Optional[float]]RTTzContextVar[float]DEADLINEinfOptional[float]returncCs tdSN)rgetrr1/tmp/pip-unpacked-wheel-36gvocj8/pymongo/_csot.py get_timeout#srfloatcCstSr)rrrrrrget_rtt'srcCstSr)rrrrrr get_deadline+sr None)rttrcCst|dSr)rset)r"rrrset_rtt/sr$cCsts dSttSr)rrrtime monotonicrrrr remaining3sr') max_timeoutrcCst}|dkr|St||S)z4Return the remaining timeout clamped to a max value.N)r'min)r(timeoutrrrclamp_remaining9sr+c@s@eZdZdZddddZdddd 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({}) r)r*cCs||_d|_dSr)_timeout_tokens)selfr*rrr__init__Jsz_TimeoutContext.__init__r!rcCsZt|j}t}|jr(t|jntd}tt||}t d}|||f|_ dS)Nrr) rr#r-rrr%r&rr)rr.)r/ timeout_tokenZ prev_deadlineZ next_deadlinedeadline_token rtt_tokenrrr __enter__Ns   z_TimeoutContext.__enter__r)exc_typeexc_valexc_tbrcCs4|jr0|j\}}}t|t|t|dSr)r.rresetrr)r/r5r6r7r1r2r3rrr__exit__Vs    z_TimeoutContext.__exit__N)__name__ __module__ __qualname____doc__r0r4r9rrrrr,Asr,F.)bound)funcrcsZtr.tdddddfdd }n"tdddddfdd }tt|S)zbApply the client's timeoutMS to this operation. Can wrap both asynchronous and synchronous methodsr)r/argskwargsrc s`tdkrJ|j}|dk rJt|$|f||IdHW5QRSQRX|f||IdHSrrr-r,r/rArBr*r@rr csot_wrapperfs   (zapply..csot_wrapperc sTtdkrD|j}|dk rDt||f||W5QRSQRX|f||SrrCrDrErrrFps   ")inspectiscoroutinefunction functoolswrapsrr>)r@rFrrErapplybs   rKzMutableMapping[str, Any]zOptional[WriteConcern])cmd write_concernrcCs:|r |jrdS|j}tdk r*|dd|r6||d<dS)z+Apply the given write concern to a command.NZwtimeoutZ writeConcern)Zis_server_defaultdocumentrpop)rLrMZwcrrrapply_write_concern|s   rP int_MAX_RTT_SAMPLES_MIN_RTT_SAMPLESc@sTeZdZUdZded<ddddZddd d d Zddd d ZddddZdS) MovingMinimumz4Tracks a minimum RTT within the last 10 RTT samples.z Deque[float]samplesr!rcCsttd|_dS)N)maxlen)rrSrWr/rrrr0szMovingMinimum.__init__r)samplercCs&|dkrtd||j|dS)Nrzduration cannot be negative ) ValueErrorrWappend)r/rZrrr add_sampleszMovingMinimum.add_samplecCst|jtkrt|jSdS)z7Get the min, or 0.0 if there aren't enough samples yet.r)lenrWrUr)rYrrrrs zMovingMinimum.getcCs|jdSr)rWclearrYrrrr8szMovingMinimum.resetN) r:r;r<r=__annotations__r0r]rr8rrrrrVs rV)*r= __future__rrIrGr% collectionsr contextlibrZ contextvarsrrtypingrrr r r r r rZpymongo.write_concernrrr`rrrrrr r$r'r+r,r>rKrPrSrUrVrrrrs4   (