Edit File: resource_sharer.cpython-38.pyc
U @��d� � @ 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� �D/opt/alt/python38/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 `