← Back
Editing: resource_sharer.cpython-36.pyc
3 \� � @ 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( |j � � � fdd�}tj|� j�| _d S )Nc s � j |�}| j|� d S )N)�shareZ send_bytes)�conn�pidr )�new_sock� �7/usr/lib64/python3.6/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 s* t j| j��}|j� }tj|�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 j|�� � fdd�}� fdd�}tj||�| _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�� }tj|�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.detachN)r r r r r r 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 resouces using background thread.c C s@ d| _ i | _g | _tj� | _d | _d | _d | _t j | 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 sN | j �> | jdkr| j� | jd7 _||f| j| j< | j| jfS 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} | \}}||tj� jd�}|j|tj� f� |S )z<Return connection from which to receive identified resource.r )�Client)�authkey)� connectionr- r �current_processr. r r �getpid)Zidentr- �address�key�cr r r r R s z _ResourceSharer.get_connectionNc C s� ddl m} | j�� | jdk r�|| jtj� jd�}|jd� |j� | j j |� | j j � rdt j d� | jj� d| _ d| _d| _x | jj� D ]\}\}}|� q�W | jj� W dQ R X dS )z:Stop the background thread and clear registered resources.r )r- N)r. z._ResourceSharer thread did not stop when asked)r/ r- r' r) r r0 r. r r r* �joinZis_aliver Z sub_warningr( r# �items�clear)r Ztimeoutr- r4 r3 r r r r r r [ s" z_ResourceSharer.stopc C sn x | j j� D ]\}\}}|� q W | j j� | jj| j� tj� | _| jd k rX| jj � 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 sf ddl m} | jd kst�tjd� |tj� jd�| _| jj | _ t j | j d�}d|_|j� || _d S )Nr )�Listenerz0starting listener and thread for sending handles)r. )�targetT)r/ r9 r( �AssertionErrorr �debugr r0 r. r2 r) r% ZThread�_serveZdaemon�startr* )r r9 �tr r r r, ~ s z_ResourceSharer._startc C s� t td�r tjtjtdtj�� x�y\| jj� �H}|j� }|d krBP |\}}| j j |�\}}z|||� W d |� X W d Q R X W q" t j � s�t jt j� � Y q"X q"W d S )N�pthread_sigmaskr )�hasattr�signalr@ � SIG_BLOCK�range�NSIGr( ZacceptZrecvr# �popr Z is_exiting�sys� excepthook�exc_info)r r �msgr3 Zdestination_pidr r r r r r= � s z_ResourceSharer._serve)N) r r r r r r � staticmethodr r r+ r, r= r r r r r! = s r! )r rB r rG r% � r �contextr r �__all__�platform�objectr r r! r r r r r r �<module> s `
Save File
Cancel