3
\ @ 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)utilstopZwin32 DupSocketc @ s e Zd ZdZdd Zdd ZdS )r z Picklable wrapper for a socket.c s( |j fdd}tj| j| _d S )Nc s j |}| j| d S )N)shareZ
send_bytes)connpidr )new_sock 7/usr/lib64/python3.6/multiprocessing/resource_sharer.pysend s
z DupSocket.__init__..send)dup_resource_sharerregisterclose_id)selfZsockr r )r r
__init__ s zDupSocket.__init__c C s* t j| j}|j }tj|S Q R X dS )z1Get the socket. This should only be called once.N)r get_connectionr Z
recv_bytessocketZ 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 j| fdd} fdd}tj||| _d S )Nc s t j| | d S )N)r Z send_handle)r r
)new_fdr r
r 1 s z DupFd.__init__..sendc s t j d S )N)osr r )r r r
r 3 s z DupFd.__init__..close)r r r r r )r fdr r r )r r
r / s
zDupFd.__init__c C s" t j| j }tj|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 resouces using background thread.c C s@ d| _ i | _g | _tj | _d | _d | _d | _t j
| t j d S )Nr )
_key_cache
_old_locks threadingLock_lock _listener_address_threadr Zregister_after_forkr!
_afterfork)r r r r
r ? s
z_ResourceSharer.__init__c
C sN | j > | jdkr| j | 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} | \}}||tj jd}|j|tj f |S )z s
`