o OZh @sddlZddlZddlZddlZddlZddlmZddlmZddlm Z dgZ ej dkr.send)dup_resource_sharerregisterclose_id)selfsockrrrr__init__s zDupSocket.__init__cCsBt|j}|}t|WdS1swYdS)z1Get the socket. This should only be called once.N)rget_connectionrZ recv_bytessocketZ fromshare)rr r rrrdetach$s$zDupSocket.detachN__name__ __module__ __qualname____doc__rrrrrrrs DupFdc@r )r"z-Wrapper for fd which can be used at any time.cs4t|fdd}fdd}t|||_dS)Ncst||dSr )rZ send_handle)r r Znew_fdrrr1szDupFd.__init__..sendcstdSr )osrrr#rrr3szDupFd.__init__..close)r$rrrr)rfdrrrr#rr/s   zDupFd.__init__cCs:t|j }t|WdS1swYdS)z-Get the fd. This should only be called once.N)rrrrZ recv_handle)rr rrrr7s$z DupFd.detachNrrrrrr"-s c@sNeZdZdZddZddZeddZdd d Zd d Z d dZ ddZ dS)_ResourceSharerz.Manager for resources using background thread.cCs:d|_i|_t|_d|_d|_d|_t |t j dS)Nr) _key_cache threadingLock_lock _listener_address_threadrZregister_after_forkr& _afterfork)rrrrr?s z_ResourceSharer.__init__cCsf|j&|jdur ||jd7_||f|j|j<|j|jfWdS1s,wYdS)z+Register resource, returning an identifier.Nr)r+r-_startr'r()rrrrrrrHs  $z_ResourceSharer.registercCs<ddlm}|\}}||tjd}||tf|S)zr?r+_at_fork_reinitr,rr-r.)rr:rrrrrr/os     z_ResourceSharer._afterforkcCsjddlm}|jdusJdtd|tjd|_|jj|_ t j |j d}d|_ |||_dS)Nr)ListenerzAlready have Listenerz0starting listener and thread for sending handlesr3)targetT)r5rBr,rdebugrr6r4r9r-r)Thread_servedaemonstartr.)rrBtrrrr0zs    z_ResourceSharer._startcCsttdrttjt zD|j5}|}|dur( WdWdS|\}}|j |\}}z |||W|n|wWdn1sMwYWnt sbt j t Yq)Npthread_sigmask)hasattrsignalrJ SIG_BLOCK valid_signalsr,acceptrecvr(poprZ is_exitingsys excepthookexc_info)rr msgr:Zdestination_pidrrrrrrFs*    z_ResourceSharer._server ) rrr r!rr staticmethodrrr/r0rFrrrrr&=s    r&)r$rLrrRr)rcontextrr__all__platformobjectrr"r&rrrrrrs"       \