U ÷Õ/e» ã@s<ddlmZddlmZmZddlmZGdd„deƒZdS)é)Ú annotations)ÚCounterMetricFamilyÚGaugeMetricFamily)ÚPrometheusCollectorcs$eZdZ‡fdd„Zdd„Z‡ZS)ÚSemaphoreMetricCollectorcstƒ |¡d|_dS)NZ semaphore)ÚsuperÚ__init__Z subsystem)ÚselfÚserver©Ú __class__©úS/tmp/pip-unpacked-wheel-g426oqom/distributed/http/scheduler/prometheus/semaphore.pyr s z!SemaphoreMetricCollector.__init__c cspz|jjd}Wntk r&YdSXt| d¡ddgd}t| d¡ddgd}t| d¡d dgd}t| d ¡d dgd}t| d ¡d dgd}t| d¡ddgdd}|j ¡D]Ž\}} | |g| ¡| |gt |j |ƒ¡| |g|j d|¡| |g|j d |¡| |g|j d |¡| |g|j d|¡q¸|V|V|V|V|V|VdS)NZ semaphoresÚ max_leaseszhMaximum leases allowed per semaphore Note: This will be constant for each semaphore during its lifetime.Úname)ÚlabelsZ active_leasesz/Amount of currently active leases per semaphoreZpending_leasesz0Amount of currently pending leases per semaphoreZ acquire_totalz-Total number of leases acquired per semaphoreZ release_totalz·Total number of leases released per semaphore Note: If a semaphore is closed while there are still leases active, this count will not equal `semaphore_acquired_total` after execution.Zaverage_pending_lease_timeaExponential moving average of the time it took to acquire a lease per semaphore Note: This only includes time spent on scheduler side, it does not include time spent on communication. Note: This average is calculated based on order of leases instead of time of lease acquisition.Ús)rÚunitÚpending) r Ú extensionsÚKeyErrorrZ build_namerrÚitemsZ add_metricÚlenZleasesZmetrics) r Zsem_extZsemaphore_max_leases_familyZsemaphore_active_leases_familyZsemaphore_pending_leasesZsemaphore_acquire_totalZsemaphore_release_totalZ$semaphore_average_pending_lease_timeZsemaphore_nameZsemaphore_max_leasesr r rÚcollect s„ûýýýû÷ ÿ ÿ ÿ ÿ ÿ þz SemaphoreMetricCollector.collect)Ú__name__Ú __module__Ú __qualname__rrÚ __classcell__r r r rrs rN)Ú __future__rZprometheus_client.corerrZdistributed.http.prometheusrrr r r rÚs