File: //opt/saltstack/salt/lib/python3.10/site-packages/zmq/utils/__pycache__/garbage.cpython-310.pyc
o
�N�g� � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z m
Z
m Z ddl Z edddg�Z
G d d
� d
e �ZG d d � d �Ze� ZdS )
zcGarbage collection thread for representing zmq refcount of Python objects
used in zero-copy sends.
� N)�
namedtuple)�getpid)�Event�Lock�Thread�gcref�obj�eventc s( e Zd ZdZ� fdd�Zdd� Z� ZS )�GarbageCollectorThreadz4Thread in which garbage collection actually happens.c s* t � �� || _d| _t� | _t� | _d S )NT)�super�__init__�gc�daemonr �pidr �ready)�selfr
�� __class__� �E/opt/saltstack/salt/lib/python3.10/site-packages/zmq/utils/garbage.pyr s
z GarbageCollectorThread.__init__c C s� t d u s
t � | jkr| j�� d S z| jj�tj�}d|_ |�
| jj � W | j�� n| j�� w t d u s<t � | jkr>d S |� � }|dkrGn&t
|�dkrOdnd}t�||�d }| jj�|d �}|rk|jrk|j�� ~q2|�� d S )Nr T� DIE� �L�Q)r r r �setr
�context�socket�zmqZPULLZlinger�bind�url�recv�len�struct�unpack�refs�popr �close)r �s�msg�fmt�key�tupr r r �run s,
�
zGarbageCollectorThread.run)�__name__�
__module__� __qualname__�__doc__r r, �
__classcell__r r r r r
s r
c s� e Zd ZdZdZdZdZdZd� fdd� Ze dd� �Z
e
j dd� �Z
d d
� Z d d � Z
d
d� Zdd� Ze dd� �Zdd� Zdd� Zddd�Zdd� Z� ZS ) �GarbageCollectoraF PyZMQ Garbage Collector
Used for representing the reference held by libzmq during zero-copy sends.
This object holds a dictionary, keyed by Python id,
of the Python objects whose memory are currently in use by zeromq.
When zeromq is done with the memory, it sends a message on an inproc PUSH socket
containing the packed size_t (32 or 64-bit unsigned int),
which is the key in the dict.
When the PULL socket in the gc thread receives that message,
the reference is popped from the dict,
and any tracker events that should be signaled fire.
Nzinproc://pyzmq.gc.01c sL t � �� i | _d | _d | _|| _t� | _d| _d | _ d | _
t � | j
� d S )NF)r r r$ r �thread�_contextr �_lock�
_stay_down�_push� _push_mutex�atexit�register�_atexit)r r r r r r O s
zGarbageCollector.__init__c C sB | j d u r tj�d�rddlm} |�� | _ | j S t�� | _ | j S )NZgeventr )�green)r4 r r. �
startswithr r< ZContext)r r< r r r r [ s
�zGarbageCollector.contextc C s, | � � r| jr
t�dt� | �� || _d S )Nz(Replacing gc context while gc is running)�is_aliver$ �warnings�warn�RuntimeWarning�stopr4 )r �ctxr r r r g s �
c C s d| _ | �� dS )zxatexit callback
sets _stay_down flag so that gc doesn't try to start up again in other atexit handlers
TN)r6 rB �r r r r r; q s zGarbageCollector._atexitc C s | � � sdS | �� dS )z"stop the garbage-collection threadN)r> �_stoprD r r r rB y s zGarbageCollector.stopc C s& d| _ d| _d| _| j�� d| _dS )zSClear state
called after stop or when setting up a new subprocess
N)r7 r8 r3 r$ �clearr rD r r r �_clear s
zGarbageCollector._clearc C s\ | j �tj�}|�| j� |�d� |�� | jr | j�� | j �
� | j � � | � � d S )Nr )
r r r �PUSH�connectr �sendr& r7 r3 �joinZtermrG )r �pushr r r rE � s
zGarbageCollector._stopc C sJ t | dd�r
td��| �� r| jdu r"| j�tj�| _| j�| j � | jS )z?The PUSH socket for use in the zmq message destructor callback.r6 Fz'zmq gc socket requested during shutdownN)
�getattr� RuntimeErrorr> r7 r r r rH rI r rD r r r � _push_socket� s z GarbageCollector._push_socketc C sP | j dur| jt� kr| �� t� | _i | _t| �| _ | j �� | j j�� dS )z�Start a new garbage collection thread.
Creates a new zmq Context used for garbage collection.
Under most circumstances, this will only be called once per process.
N) r3 r r rG r$ r
�startr �waitrD r r r rP � s
zGarbageCollector.startc C s0 t du st � | jks| jdu s| j�� sdS dS )zsIs the garbage collection thread currently running?
Includes checks for process shutdown or fork.
NFT)r r r3 r> rD r r r r> � s
�zGarbageCollector.is_alivec C sh | � � s$| jr dS | j� | � � s| �� W d � n1 s w Y t||�}t|�}|| j|< |S )z4store an object and (optionally) event for zero-copyr N)r> r6 r5 rP r �idr$ )r r r r+ Ztheidr r r �store� s � �
zGarbageCollector.storec
C s8 | � � sd S z| �� W d S ty } z|�d }~ww �N)r> rB � Exception)r �er r r �__del__� s ��zGarbageCollector.__del__rT )r- r. r/ r0 r$ r4 r5 r r �propertyr �setterr; rB rG rE rO rP r> rS rW r1 r r r r r2 ; s*
r2 )r0 r9 r" r? � collectionsr �osr � threadingr r r r r r
r2 r
r r r r �<module> s &