File: //proc/thread-self/root/lib64/python3.6/site-packages/zmq/utils/__pycache__/garbage.cpython-36.pyc
3
VS�_/ � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZm Z m
Z
ddl Z ddl Z edddg�Z
G d d
� d
e�ZG d d � d e�Ze� ZdS )
zcGarbage collection thread for representing zmq refcount of Python objects
used in zero-copy sends.
� N)�getpid)�
namedtuple)�Thread�Event�Lock�gcref�obj�eventc s( e Zd ZdZ� fdd�Zdd� Z� ZS )�GarbageCollectorThreadz4Thread in which garbage collection actually happens.c s. t t| �j� || _d| _t� | _t� | _d S )NT) �superr
�__init__�gcZdaemonr �pidr �ready)�selfr
)� __class__� � /usr/lib64/python3.6/garbage.pyr s
z GarbageCollectorThread.__init__c C s� t d kst � | jkr"| jj� d S z(| jjjtj�}d|_ |j
| jj � W d | jj� X xxt d ksnt � | jkrrd S |j � }|dkr�P t
|�dkr�dnd}tj||�d }| jjj|d �}|r�|jr�|jj� ~qZW |j� d S )Nr s DIE� �L�Q)r r r �setr
�context�socket�zmqZPULLZlingerZbind�urlZrecv�len�struct�unpack�refs�popr �close)r �s�msgZfmt�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� Ze dd� �Zdd� Zdd� Zddd�Zdd� Z� ZS )�GarbageCollectoraN 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 sP t t| �j� i | _d | _d | _|| _t� | _d| _ d | _
d | _ t j
| j� d S )NF)r 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 s> | j d kr8tjjd�r.ddlm} |j� | _ n
tj� | _ | j S )NZgeventr )�green)r. r r( �
startswithr r6 ZContext)r r6 r r r r [ s
zGarbageCollector.contextc C s, | j � r"| jrtjdt� | j� || _d S )Nz(Replacing gc context while gc is running)�is_aliver �warnings�warn�RuntimeWarning�stopr. )r Zctxr r r r f s
c C s d| _ | j� dS )z�atexit callback
sets _stay_down flag so that gc doesn't try to start up again in other atexit handlers
TN)r0 r< )r r r r r5 n s zGarbageCollector._atexitc C s | j � s dS | j� dS )z"stop the garbage-collection threadN)r8 �_stop)r r r r r< v s zGarbageCollector.stopc C sp | j jtj�}|j| j� |jd� |j� | jrB| jj� d | _d | _ | j
j � | j j � | j
j� d | _ d S )Ns DIE)r r r �PUSH�connectr �sendr! r1 r2 r- �joinZtermr �clear)r �pushr r r r= | s
zGarbageCollector._stopc C s8 | j � s| jdkr2| jjtj�| _| jj| j� | jS )zHThe PUSH socket for use in the zmq message destructor callback.
N)r8 r1 r r r r> r? r )r r r r � _push_socket� s z GarbageCollector._push_socketc C sP | j dk r | jt� kr | j� t� | _i | _t| �| _ | j j� | j jj� 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) r- r r r= r r
�startr �wait)r r r r rE � s
zGarbageCollector.startc C s2 t dks*t � | jks*| jdks*| jj� r.dS dS )z{Is the garbage collection thread currently running?
Includes checks for process shutdown or fork.
NFT)r r r- r8 )r r r r r8 � s
zGarbageCollector.is_alivec
C sT | j � s4| jrdS | j� | j � s*| j� W dQ R X t||�}t|�}|| j|< |S )z4store an object and (optionally) event for zero-copyr N)r8 r0 r/ rE r �idr )r r r r% Ztheidr r r �store� s
zGarbageCollector.storec C sD | j � s d S y | j� W n& tk
r> } z
|�W Y d d }~X nX d S )N)r8 r< � Exception)r �er r r �__del__� s zGarbageCollector.__del__)N)N)r' r( r) r* r r. r/ r r �propertyr �setterr5 r< r= rD rE r8 rH rK r+ r r )r r r, ; s
r, )r* r3 r �osr � collectionsr Z threadingr r r r9 r r r
�objectr, r
r r r r �<module> s %