File: //usr/local/lib/python3.7/multiprocessing/__pycache__/resource_sharer.cpython-37.pyc
B
"��g� � @ s� d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZ ddlm Z dgZ
ej dkrxe
dg7 Z
G d d� de �Z
ne
d
g7 Z
G d d
� d
e �ZG d d
� d
e �Ze� ZejZdS )� N� )�process)� reduction)�util�stop�win32� DupSocketc @ s e Zd ZdZdd� Zdd� ZdS )r z Picklable wrapper for a socket.c s( |� � � � fdd�}t�|� j�| _d S )Nc s � � |�}| �|� d S )N)�shareZ
send_bytes)�conn�pidr )�new_sock� �;/usr/local/lib/python3.7/multiprocessing/resource_sharer.py�send s
z DupSocket.__init__.<locals>.send)�dup�_resource_sharer�register�close�_id)�selfZsockr r
)r r �__init__ s zDupSocket.__init__c C s* t �| j��}|�� }t�|�S Q R X dS )z1Get the socket. This should only be called once.N)r �get_connectionr Z
recv_bytes�socketZ fromshare)r r
r r
r
r �detach$ s zDupSocket.detachN)�__name__�
__module__� __qualname__�__doc__r r r
r
r
r r s �DupFdc @ s e Zd ZdZdd� Zdd� ZdS )r z-Wrapper for fd which can be used at any time.c s4 t �|�� � fdd�}� fdd�}t�||�| _d S )Nc s t �| � |� d S )N)r Z send_handle)r
r )�new_fdr
r r 1 s z DupFd.__init__.<locals>.sendc s t �� � d S )N)�osr r
)r r
r r 3 s z DupFd.__init__.<locals>.close)r r r r r )r �fdr r r
)r r r / s
zDupFd.__init__c C s" t �| j�� }t�|�S Q R X dS )z-Get the fd. This should only be called once.N)r r r r Z recv_handle)r r
r
r
r r 7 s z DupFd.detachN)r r r r r r r
r
r
r r - s c @ sN e Zd ZdZdd� Zdd� Zedd� �Zdd d
�Zd d � Z d
d� Z
dd� Z dS )�_ResourceSharerz.Manager for resources using background thread.c C s@ d| _ i | _g | _t�� | _d | _d | _d | _t �
| t j � d S )Nr )
�_key�_cache�
_old_locks� threading�Lock�_lock� _listener�_address�_threadr Zregister_after_forkr" �
_afterfork)r r
r
r r ? s
z_ResourceSharer.__init__c C sN | j �> | jdkr| �� | jd7 _||f| j| j< | j| jfS Q R X dS )z+Register resource, returning an identifier.Nr )r( r* �_startr# r$ )r r r r
r
r r I s
z_ResourceSharer.registerc C s< ddl m} | \}}||t�� jd�}|�|t�� f� |S )z<Return connection from which to receive identified resource.r )�Client)�authkey)�
connectionr. r �current_processr/ r r �getpid)Zidentr. �address�key�cr
r
r r R s
z _ResourceSharer.get_connectionNc C s� ddl m} | j�� | jdk r�|| jt�� jd�}|�d� |�� | j �
|� | j � � rdt �
d� | j�� d| _ d| _d| _x | j�� D ]\}\}}|� q�W | j�� W dQ R X dS )z:Stop the background thread and clear registered resources.r )r. N)r/ z._ResourceSharer thread did not stop when asked)r0 r. r( r* r r1 r/ r r r+ �joinZis_aliver Z sub_warningr) r$ �items�clear)r Ztimeoutr. r5 r4 r r r
r
r r [ s"
z_ResourceSharer.stopc C sn x | j �� D ]\}\}}|� q W | j �� | j�| j� t�� | _| jd k rX| j� � d | _d | _
d | _ d S )N) r$ r7 r8 r% �appendr( r& r' r) r r* r+ )r r4 r r r
r
r r, p s
z_ResourceSharer._afterforkc C sj ddl m} | jd ks td��t�d� |t�� jd�| _| jj | _
t j | j
d�}d|_|�� || _d S )Nr )�ListenerzAlready have Listenerz0starting listener and thread for sending handles)r/ )�targetT)r0 r: r) �AssertionErrorr �debugr r1 r/ r3 r* r& ZThread�_serveZdaemon�startr+ )r r: �tr
r
r r- ~ s
z_ResourceSharer._startc C s� t td�r t�tjtdtj�� x�y\| j�� �H}|�� }|d krBP |\}}| j �
|�\}}z|||� W d |� X W d Q R X W q" t � � s�t
jt
�� � Y q"X q"W d S )N�pthread_sigmaskr )�hasattr�signalrA � SIG_BLOCK�range�NSIGr) ZacceptZrecvr$ �popr Z
is_exiting�sys�
excepthook�exc_info)r r
�msgr4 Zdestination_pidr r r
r
r r> � s
z_ResourceSharer._serve)N) r r r r r r � staticmethodr r r, r- r> r
r
r
r r" = s
r" )r rC r rH r& � r �contextr r �__all__�platform�objectr r r" r r r
r
r
r �<module> s
`