관리-도구
편집 파일: resource_sharer.cpython-310.pyc
o 6��f � @ s� d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZ ddlm Z dgZ ej dkr<e dg7 Z G d d� de �Z n e d g7 Z G d d � d e �ZG d d � d e �Ze� ZejZdS )� N� )�process)� reduction)�util�stopZwin32� DupSocketc @ � 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� �F/opt/alt/python310/lib64/python3.10/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 sB t �| j��}|�� }t�|�W d � S 1 sw Y 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 @ r )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 d � S 1 sw Y 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 | _t�� | _d | _d | _d | _t� | t j � d S )Nr ) �_key�_cache� threadingZLock�_lock� _listener�_address�_threadr Zregister_after_forkr$ � _afterfork)r r r r r ? s z_ResourceSharer.__init__c C sf | j �& | jdu r | �� | jd7 _||f| j| j< | j| jfW d � S 1 s,w Y dS )z+Register resource, returning an identifier.Nr )r( r* �_startr% r& )r r r r r r r H 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)� connectionr/ r �current_processr1 r r" �getpid)Zidentr/ �address�key�cr r r r Q s z _ResourceSharer.get_connectionNc C s� ddl m} | j�[ | jdur\|| jt�� jd�}|�d� |�� | j � |� | j � � r2t � d� | j�� d| _ d| _d| _| j�� D ] \}\}}|� qE| j�� W d � dS W d � dS 1 sgw Y dS )z:Stop the background thread and clear registered resources.r r. Nr0 z._ResourceSharer thread did not stop when asked)r2 r/ r( r* r r3 r1 r r r+ �joinZis_aliver Z sub_warningr) r& �items�clear)r Ztimeoutr/ r7 r6 r r r r r r Z s* � �"�z_ResourceSharer.stopc C s\ | j �� D ] \}\}}|� q| j �� | j�� | jd ur#| j�� d | _d | _d | _d S r ) r& r9 r: r( �_at_fork_reinitr) r r* r+ )r r6 r r r r r r, o s z_ResourceSharer._afterforkc C sj ddl m} | jd u sJ d��t�d� |t�� jd�| _| jj| _ t j | j d�}d|_ |�� || _d S )Nr )�ListenerzAlready have Listenerz0starting listener and thread for sending handlesr0 )�targetT)r2 r< r) r �debugr r3 r1 r5 r* r' ZThread�_serveZdaemon�startr+ )r r< �tr r r r- z s z_ResourceSharer._startc C s� t td�rt�tjt�� � zD| j�� �5}|�� }|d u r( W d � W d S |\}}| j� |�\}}z |||� W |� n|� w W d � n1 sMw Y W n t � � sbt j t �� � Y q)N�pthread_sigmask)�hasattr�signalrB � SIG_BLOCK� valid_signalsr) ZacceptZrecvr& �popr Z is_exiting�sys� excepthook�exc_info)r r �msgr6 Zdestination_pidr r r r r r? � s* � �� ��z_ResourceSharer._server ) r r r r r r � staticmethodr r r, r- r? r r r r r$ = s r$ )r" rD r rH r' � r �contextr r �__all__�platform�objectr r r$ r r r r r r �<module> s" \