�
_�*j� � � � d dl Z d dlZd dlZd dlZd dlmZmZ d dlmZmZm Z d dl
m Z m Z m
Z
d dlmZ er
d dlmZmZmZ e d� Z G d� d ee � Zy)
� N)�datetime�timezone)�
TYPE_CHECKING�Generic�TypeVar)�Envelope�Item�
PayloadRef)�format_timestamp)�Any�Callable�Optional�Tc � � e Zd ZdZdZdZdZdZ dd�Zdd�Z de
fd �Z dd
�Z dd �Z
dd �Zdd
�Zdd�Zdd�Zdd�Zedd�� Zy)�Batcher�d i� g @� �returnNc � �� g | _ || _ || _ d| _ t j
� | _ t j � | _ t j � | _
d | _ d | _ t t d� r<t j | j" � �d�fd� }t j$ |�� y y )NT�register_at_forkc �( �� �� } | � | � y y �N� )�method�
weak_resets ��B/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/_batcher.py�_reset_in_childz)Batcher.__init__.<locals>._reset_in_child. s �� �#� ���%��H� &� )�after_in_child�r N)�_buffer�
_capture_func�_record_lost_func�_running� threading�Lock�_lock�local�_active�Event� _flush_event�_flusher� _flusher_pid�hasattr�os�weakref�
WeakMethod�_reset_thread_stater )�self� capture_func�record_lost_funcr r s @r �__init__zBatcher.__init__ s� �� �
#%�� � )���!1�� � ��
� �^�^�%��
�*3�/�/�*;�� �/8���/@���6:��
�-1��� �2�)� *� �+�+�D�,D�,D�E�J�
�
� � �� ?� +r c �� � g | _ d| _ t j � | _ t j
� | _ t j � | _ d | _ d | _
y �NT) r! r$ r% r& r' r( r) r* r+ r, r- �r3 s r r2 zBatcher._reset_thread_state5 sJ � ��� � ��
� �^�^�%��
� ���(�� � %�O�O� -��� ��
� ��r c �� � | j syt j � }| j |k( ry| j 5 | j |k( r
ddd� y|| _ t j | j �� | _ d| j _ | j j � ddd� y# t $ r d| _ Y ddd� yw xY w# 1 sw Y yxY w)z�For forking processes we might need to restart this thread.
This ensures that our process actually has that thread running.
FTN)�target) r$ r/ �getpidr- r' r% �Thread� _flush_loopr, �daemon�start� RuntimeError)r3 �pids r �_ensure_threadzBatcher._ensure_thread> s� � � �}�}���i�i�k�� � � �� #��
�Z�Z� � � �C�'�� �Z� !$�D� � %� ,� ,�D�4D�4D� E�D�M�#'�D�M�M� �
��
�
�#�#�%� �& ��
�
� !&��
� �# �Z�
�� �& �s/ �C�=C�B:�:C�
C�C�C�C!c �" � d| j _ | j rr| j j | j
t
j � z � | j j � | j � | j r�qy y r8 ) r) �flagr$ r+ �wait�FLUSH_WAIT_TIME�random�clear�_flushr9 s r r> zBatcher._flush_loop^ s` � �
!�� � ���m�m� � � � "� "�4�#7�#7�&�-�-�/�#I� J� � � � #� #� %� �K�K�M� �m�mr c � � t | j dd� ry d| j _ | j � r | j � d| j _ y | j
5 t
| j � | j k\ r,| j |� d d d � d| j _ y | j j |� t
| j � | j k\ r| j j � d d d � d| j _ y # 1 sw Y �xY w# d| j _ w xY w�NrE FT)�getattrr) rE rC r, r' �lenr! �MAX_BEFORE_DROP� _record_lost�append�MAX_BEFORE_FLUSHr+ �set�r3 �items r �addz Batcher.addi s� � � �4�<�<��� /�� �� � ��
&��&�&�(�D�M�M�,A�� !&�D�L�L� � ����t�|�|�$��(<�(<�<��%�%�d�+� � � !&�D�L�L� � � � �#�#�D�)��t�|�|�$��(=�(=�=��%�%�)�)�+� � !&�D�L�L� � ��� !&�D�L�L� �s6 � D2 � D2 �&5D&�D2 �5AD&� D2 �&D/�+D2 �2Ec �n � | j �y d| _ | j j � d | _ y )NF)r, r$ r+ rS r9 s r �killz Batcher.kill� s/ � � �=�=� � � ��
� ��� � � � ��
r c �� � t | j dd� }d| j _ | j � || j _ y # || j _ w xY wrL )rM r) rE rJ )r3 �
was_actives r �flushz
Batcher.flush� sG � � �T�\�\�6�5�9�
� �� � �� +� �K�K�M� *�D�L�L� ��
�D�L�L� �s �A � A c � � |j t | j | j dt | j
� it
d| j
D �cg c] }| j |� �� c}d��� �� � y c c}w )N�
item_count� )�version�items)�json)�type� content_type�headers�payload)�add_itemr �TYPE� CONTENT_TYPErN r! r
�_to_transport_format)r3 �enveloperU s r �_add_to_envelopezBatcher._add_to_envelope� sx � ���� ��Y�Y� !� .� .� �#�d�l�l�"3�� #�#$�HL� � �"�HT��D� 5� 5�d� ;� �"���
�
��"s �A=c �t � t dt t j t j
� � i�� }| j 5 t | j � dk( r
d d d � y | j |� | j j � d d d � | j |� |S # 1 sw Y � xY w)N�sent_at)rd r ) r r r �nowr �utcr' rN r! rk rI r" )r3 rj s r rJ zBatcher._flush� s� � �� � 0�� � �h�l�l�1K� L�M�
�� �Z�Z��4�<�<� �A�%�� �Z�
� !� !�(� +� �L�L� � � � �
���8�$��� �Z�s �B.�(+B.�.B7c � � y r r rT s r rP zBatcher._record_lost� s � � r c � � y r r )rU s r ri z Batcher._to_transport_format� s � � r )r4 zCallable[[Envelope], None]r5 zCallable[..., None]r Nr )rU r r N)rj r r N)r zOptional[Envelope])rU r r r )�__name__�
__module__� __qualname__rR rO rG rg rh r6 r2 �boolrC r> rV rX r[ rk rJ rP � staticmethodri r r r r r s� � ����O��O�
�D��L�@�2�@� 0�@�
� @�8!� �� �@ �&�2 �+�
�& �
� �
� �
r r )r/ rH r% r0 r r �typingr r r �sentry_sdk.enveloper r r
�sentry_sdk.utilsr r r
r r r r r r �<module>rz sG �� �
� � � '� 2� 2� :� :� -��.�.� �C�L��g
�g�a�j� g
r