← Back
Editing: resource_sharer.cpython-35.pyc
��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)�util�stop�win32� 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)�conn�pidr )�new_sock� �D/opt/alt/python35/lib64/python3.5/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 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_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 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� threading�Lock�_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 )�Client�authkey)� connectionr. r �current_processr/ r r �getpid)Zidentr. �address�key�cr 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$ �items�clear)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 } | j d k s% t � 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) �AssertionErrorr �debugr r1 r/ r3 r* r& ZThread�_serveZdaemon�startr+ )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)N�pthread_sigmaskr )�hasattr�signalrA � SIG_BLOCK�range�NSIGr) ZacceptZrecvr$ �popr Z is_exiting�sys� excepthook�exc_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 rC r rH r& � r r r �__all__�platform�objectr r r" r r r r r r �<module> s `
Save File
Cancel