File: //usr/lib64/python3.8/multiprocessing/__pycache__/resource_sharer.cpython-38.pyc
U
e5dè ã @ 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ÚstopZwin32Ú 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 ©Znew_sock© ú7/usr/lib64/python3.8/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 s6 t | j¡ }| ¡ }t |¡W 5 Q R £ 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 r )r Z send_handle)r
r ©Znew_fdr
r r 1 s z DupFd.__init__.<locals>.sendc s t ˆ ¡ d S r )Ú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 |¡W 5 Q R £ 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.detachNr 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 sZ | j J | jdkr| ¡ | jd7 _||f| j| j< | j| jfW 5 Q R £ S 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)Ú
connectionr0 r Úcurrent_processr2 r r! Úgetpid)Zidentr0 Ú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| _| j ¡ D ]\}\}}|ƒ qŠ| j ¡ W 5 Q R X dS )z:Stop the background thread and clear registered resources.r r/ Nr1 z._ResourceSharer thread did not stop when asked)r3 r0 r) r+ r r4 r2 r r r, ÚjoinZis_aliver Z sub_warningr* r% ÚitemsÚclear)r Ztimeoutr0 r8 r7 r r r
r
r r [ s$
ÿ
z_ResourceSharer.stopc C sj | j ¡ D ]\}\}}|ƒ q
| j ¡ | j | j¡ t ¡ | _| jd k rT| j ¡ d | _d | _
d | _ d S r ) r% r: r; r& Úappendr) r' r( r* r r+ r, )r r7 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 handlesr1 )ÚtargetT)r3 r= r* ÚAssertionErrorr Údebugr r4 r2 r6 r+ r' ZThreadÚ_serveZdaemonÚstartr, )r r= Útr
r
r r. ~ s
z_ResourceSharer._startc C s® t tdƒr t tjt ¡ ¡ zh| j ¡ T}| ¡ }|d krHW 5 Q R £ W qª|\}}| j |¡\}}z|||ƒ W 5 |ƒ X W 5 Q R X W q t
¡ s¢t j
t ¡ Ž Y q X q d S )NÚpthread_sigmask)ÚhasattrÚsignalrD Ú SIG_BLOCKÚ
valid_signalsr* ZacceptZrecvr% Úpopr Z
is_exitingÚsysÚ
excepthookÚexc_info)r r
Úmsgr7 Zdestination_pidr r r
r
r rA ‰ s
z_ResourceSharer._serve)N) r r r r r r Ú staticmethodr r r- r. rA r
r
r
r r# = s
r# )r! rF r rJ r' Ú r Úcontextr r Ú__all__ÚplatformÚobjectr r r# r r r
r
r
r Ú<module> s
`