File: //usr/lib/python3.5/multiprocessing/__pycache__/resource_sharer.cpython-35.pyc
Fa @ s d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z d g Z e j
d k r e d g 7Z Gd d d e Z n# e d
g 7Z Gd d
d
e Z
Gd d
d
e Z e Z e j Z d S) N )process) reduction)utilstopZwin32 DupSocketc @ s. e Z d Z d Z d d Z d d Z d S)r z Picklable wrapper for a socket.c s: | j f d d } t j | j | _ d S)Nc s j | } | j | d S)N)shareZ
send_bytes)connpidr )new_sock 5/usr/lib/python3.5/multiprocessing/resource_sharer.pysend s z DupSocket.__init__.<locals>.send)dup_resource_sharerregisterclose_id)selfZsockr r )r r
__init__ s zDupSocket.__init__c C s9 t j | j } | j } t j | SWd QRXd S)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 Z d Z d Z d d Z d d Z d S)r z-Wrapper for fd which can be used at any time.c sL t j | f d d } f d d } t j | | | _ d S)Nc s t j | | d S)N)r Z send_handle)r r
)new_fdr r
r 1 s z DupFd.__init__.<locals>.sendc s t j 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 | j } t j | SWd QRXd S)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 @ ss e Z d Z d Z d d Z d d Z e d d Z d d d
Z d d Z d
d Z
d d Z d S)_ResourceSharerz-Manager for resouces using background thread.c C s\ d | _ i | _ g | _ t j | _ 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 sc | j S | j d k r# | j | j d 7_ | | f | j | j <| j | j f SWd QRXd S)z+Register resource, returning an identifier.Nr )r' r) _startr" r# )r r r r r r
r I s
z_ResourceSharer.registerc C sT d d l m } | \ } } | | d t j j } | j | t j f | S)z<Return connection from which to receive identified resource.r )Clientauthkey)
connectionr- r current_processr. r r getpid)Zidentr- addresskeycr r r
r R s
z _ResourceSharer.get_connectionNc C s d d l m } | j | j d k r | | j d t j j } | j d | j | j j
| | j j r t j
d | j j d | _ d | _ d | _ x* | j j D] \ } \ } } | q W| j j Wd QRXd S)z:Stop the background thread and clear registered resources.r )r- Nr. z._ResourceSharer thread did not stop when asked)r/ r- r' r) r r0 r. r r r* joinZis_aliver Z sub_warningr( r# itemsclear)r Ztimeoutr- r4 r3 r r r r r
r [ s"
" z_ResourceSharer.stopc C s x* | j j D] \ } \ } } | q W| j j | j j | j t j | _ | j d k rx | j j d | _ d | _
d | _ d S)N) r# r6 r7 r$ appendr' r% r& r( r r) r* )r r3 r r r r r
r+ p s "
z_ResourceSharer._afterforkc C s d d l m } | j d k s% t t j d | d t j j | _ | j j | _
t j d | j
} d | _ | j | | _ d S)Nr )Listenerz0starting listener and thread for sending handlesr. targetT)r/ r9 r( AssertionErrorr debugr r0 r. r2 r) r% ZThread_serveZdaemonstartr* )r r9 tr r r
r, ~ s
z_ResourceSharer._startc C s t t d r. t j t j t d t j x yv | j j ` } | j } | d k r_ P| \ } } | j j
| \ } } z | | | Wd | XWd QRXWq1 t j s t
j t
j Yq1 Xq1 Wd S)Npthread_sigmaskr )hasattrsignalr@ SIG_BLOCKrangeNSIGr( ZacceptZrecvr# popr Z
is_exitingsys
excepthookexc_info)r r msgr3 Zdestination_pidr r r r r
r= s z_ResourceSharer._serve) r r r r r r staticmethodr r r+ r, r= r r r r
r! = s
r! )r rB r rG r% r r r __all__platformobjectr r r! r r r r r r
<module> s
`