o
    Á¿i  ã                   @   sò   d dl mZ d dlmZ d dlmZ d dlmZ eeƒZ	G dd„ de
eƒZG dd„ de
eƒZG d	d
„ d
e
eƒZdededdfdd„Zdeddfdd„Zdededdfdd„Zdededdfdd„Zdededdfdd„Zdededdfdd„ZdS )é    )ÚEnum)Ú
get_logger)Útelemetry_writer)ÚTELEMETRY_NAMESPACEc                   @   ó   e Zd ZdZdZdZdS )ÚENDPOINTÚ
test_cycleÚcode_coverageÚcoverage_reportN)Ú__name__Ú
__module__Ú__qualname__Ú
TEST_CYCLEÚCODE_COVERAGEÚCOVERAGE_REPORT© r   r   úd/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/internal/ci_visibility/telemetry/payload.pyr      ó    r   c                   @   s$   e Zd ZdZdZdZdZdZdZdS )ÚENDPOINT_PAYLOAD_TELEMETRYzendpoint_payload.byteszendpoint_payload.requestszendpoint_payload.requests_msz endpoint_payload.requests_errorszendpoint_payload.events_countz(endpoint_payload.events_serialization_msN)	r   r   r   ÚBYTESÚREQUESTS_COUNTÚREQUESTS_MSÚREQUESTS_ERRORSÚEVENTS_COUNTÚEVENTS_SERIALIZATION_MSr   r   r   r   r      s    r   c                   @   r   )ÚREQUEST_ERROR_TYPEÚtimeoutÚnetworkÚstatus_codeN)r   r   r   ÚTIMEOUTÚNETWORKÚSTATUS_CODEr   r   r   r   r      r   r   ÚendpointÚnbytesÚreturnNc                 C   ó4   t  d| |¡ d| jff}t tjtjj||¡ d S )Nz(Recording endpoint payload bytes: %s, %sr"   )	ÚlogÚdebugÚvaluer   Úadd_distribution_metricr   ÚCIVISIBILITYr   r   )r"   r#   Útagsr   r   r   Úrecord_endpoint_payload_bytes    ó
   ÿr,   c                 C   s2   t  d| ¡ d| jff}t tjtjjd|¡ d S )Nz&Recording endpoint payload request: %sr"   é   )	r&   r'   r(   r   Úadd_count_metricr   r*   r   r   )r"   r+   r   r   r   Úrecord_endpoint_payload_request(   s
   ÿr0   Úsecondsc                 C   ó8   t  d| |¡ d| jff}t tjtjj|d |¡ d S )Nz7Recording endpoint payload request time: %s, %s secondsr"   éè  )	r&   r'   r(   r   r)   r   r*   r   r   ©r"   r1   r+   r   r   r   Ú$record_endpoint_payload_request_time0   ó
   ÿr5   Ú
error_typec                 C   s:   t  d| |¡ d| jfd|ff}t tjtjjd|¡ d S )Nz0Recording endpoint payload request error: %s, %sr"   r7   r.   )	r&   r'   r(   r   r/   r   r*   r   r   )r"   r7   r+   r   r   r   Ú%record_endpoint_payload_request_error8   s
   ÿr8   Úcountc                 C   r%   )Nz/Recording endpoint payload events count: %s, %sr"   )	r&   r'   r(   r   r)   r   r*   r   r   )r"   r9   r+   r   r   r   Ú$record_endpoint_payload_events_count@   r-   r:   c                 C   r2   )Nz=Recording endpoint payload serialization time: %s, %s secondsr"   r3   )	r&   r'   r(   r   r)   r   r*   r   r   r4   r   r   r   Ú1record_endpoint_payload_events_serialization_timeH   r6   r;   )Úenumr   Úddtrace.internal.loggerr   Úddtrace.internal.telemetryr   Ú$ddtrace.internal.telemetry.constantsr   r   r&   Ústrr   r   r   Úintr,   r0   Úfloatr5   r8   r:   r;   r   r   r   r   Ú<module>   s    	