Yf @ 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)utilstopwin32 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 D/opt/alt/python35/lib64/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)r0 r. r( r* r r1 r/ r r r+ joinZis_aliver Z sub_warningr) r$ itemsclear)r Ztimeoutr. r5 r4 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$ r7 r8 r% appendr( r& r' r) r r* r+ )r r4 r r r
r
r r, p s "
z_ResourceSharer._afterforkc C s| d d l m } 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)r0 r: r debugr r1 r/ r) r3 r* r& ZThread_serveZdaemonstartr+ )r r: 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
msgr4 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
`