a
Z��^�. c @ s� g d �Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
m Z ddl Z ddl
mZ ddl
mZ ejjZddlmZmZmZmZmZ G dd � d e�Ze� ZG d
d � d e�Z G d d
� d
e�Z dS ))�Queue� SimpleQueue�
JoinableQueue� N)�Empty�Full� )�
connection)�context)�debug�info�Finalize�register_after_fork�
is_exitingc @ s� e Zd Zd-dd�Zdd� Zdd� Zdd � Zd.d d �Zd/dd�Zd0dd�Z dd� Z
dd� Z dd� Z dd� Z
dd � Zd d � Zd d � Zd!d"� Zd#d$� Zed%d&� �Zed'd(� �Zed)d*� �Zed+d,� �ZdS )1r r c C s� |dkrddl m} || _tjdd�\| _| _|�� | _t �
� | _ t j
dkrTd | _n
|�� | _|�|�| _d| _| �� t j
dkr�t| tj� d S )Nr r )�
SEM_VALUE_MAXF�Zduplex�win32)Z synchronizer �_maxsizer �Pipe�_reader�_writer�Lock�_rlock�osZgetpid�_opid�sys�platform�_wlockZBoundedSemaphore�_sem�
_ignore_epipe�_resetr r � _after_fork��selfZmaxsize�ctx� r# �./usr/lib64/python3.9/multiprocessing/queues.py�__init__% s
zQueue.__init__c C s. t �| � | j| j| j| j| j| j| j| j fS �N)
r �assert_spawningr r r r r r r r �r! r# r# r$ � __getstate__9 s
�zQueue.__getstate__c C s0 |\| _ | _| _| _| _| _| _| _| �� d S r& ) r r r r r r r r r �r! Zstater# r# r$ � __setstate__> s �zQueue.__setstate__c C s t d� | jdd� d S )NzQueue._after_fork()T)�
after_fork)r r r( r# r# r$ r C s zQueue._after_forkFc C sj |r| j �� nt�t�� �| _ t�� | _d | _d | _ d| _
d| _ d | _ | j
j| _| jj| _| jj| _d S �NF)� _notemptyZ_at_fork_reinit� threading� Conditionr � collectionsZdeque�_buffer�_thread� _jointhread�_joincancelled�_closed�_closer �
send_bytes� _send_bytesr �
recv_bytes� _recv_bytes�poll�_poll)r! r, r# r# r$ r G s
z Queue._resetTNc C sz | j rtd| �d���| j�||�s(t�| j�8 | jd u rB| �� | j� |� | j�
� W d � n1 sl0 Y d S �NzQueue z
is closed) r6 �
ValueErrorr �acquirer r. r3 �
_start_threadr2 �append�notify�r! �obj�block�timeoutr# r# r$ �putV s
z Queue.putc C s� | j rtd| �d���|r\|d u r\| j� | �� }W d � n1 sF0 Y | j�� n||rlt�� | }| j�||�s~t �zL|r�|t�� }| �
|�s�t �n | �
� s�t �| �� }| j�� W | j�� n | j�� 0 t � |�S r> )
r6 r? r r; r �release�timeZ monotonicr@ r r= �_ForkingPickler�loads)r! rF rG �resZdeadliner# r# r$ �getb s* &
z Queue.getc C s | j | jj�� S r& )r r �_semlockZ
_get_valuer( r# r# r$ �qsize| s z Queue.qsizec C �
| � � S r& �r= r( r# r# r$ �empty� � z Queue.emptyc C s | j j�� S r& )r rO �_is_zeror( r# r# r$ �full� rT z
Queue.fullc C s
| � d�S r- )rN r( r# r# r$ �
get_nowait� rT zQueue.get_nowaitc C s | � |d�S r- )rH �r! rE r# r# r$ �
put_nowait� rT zQueue.put_nowaitc C sH d| _ z$| j�� W | j}|rDd | _|� n| j}|rBd | _|� 0 d S )NT)r6 r �closer7 )r! rZ r# r# r$ rZ � s �z Queue.closec C s t d� | jr| �� d S )NzQueue.join_thread())r r4 r( r# r# r$ � join_thread� s zQueue.join_threadc C s4 t d� d| _z| j�� W n ty. Y n0 d S )NzQueue.cancel_join_thread()T)r r5 r4 ZcancelZAttributeErrorr( r# r# r$ �cancel_join_thread� s zQueue.cancel_join_threadc
C s� t d� | j�� tjtj| j| j| j| j | j
j | j | j
| jfdd�| _d| j_t d� | j�� t d� | js�t| jtjt�| j�gdd�| _t| tj| j| jgd d�| _d S )
NzQueue._start_thread()ZQueueFeederThread)ZtargetZargsZnameTzdoing self._thread.start()z ... done self._thread.start()i����)Z exitpriorityi
)r r2 Zclearr/ ZThreadr �_feedr. r9 r r rZ r �_on_queue_feeder_errorr r3 ZdaemonZstartr5 r �_finalize_join�weakrefZrefr4 �_finalize_closer7 r( r# r# r$ rA � s2
��
�
�zQueue._start_threadc C s4 t d� | � }|d ur(|�� t d� nt d� d S )Nzjoining queue threadz... queue thread joinedz ... queue thread already dead)r �join)ZtwrZthreadr# r# r$ r_ � s
zQueue._finalize_joinc C sB t d� |�" | �t� |�� W d � n1 s40 Y d S )Nz telling queue thread to quit)r rB � _sentinelrC )�buffer�notemptyr# r# r$ ra � s
zQueue._finalize_closec
C sl t d� |j}|j} |j}
| j} t} tjdkr<|j}
|j}nd }
z�|� z| sT|
� W | � n| � 0 zh| � }|| u r�t d� |� W W d S t� |�}|
d u r�||� qh|
� z||� W |� qh|� 0 qhW n t
y� Y n0 W q@ t �yd } zf|�r t |dd�t
jk�r W Y d }~d S t� �r>td|� W Y d }~d S |�� |||� W Y d }~q@d }~0 0 q@d S )Nz$starting thread to feed data to piper z%feeder thread got sentinel -- exiting�errnor zerror in queue thread: %s)r r@ rI �waitZpopleftrc r r rK �dumpsZ
IndexErrorZ ExceptionZgetattrrf ZEPIPEr
r
)rd re r8 Z writelockrZ Z ignore_epipeZonerrorZ queue_semZnacquireZnreleaseZnwaitZbpopleftZsentinelZwacquireZwreleaserE �er# r# r$ r] � sN
z Queue._feedc C s dd l }|�� d S )Nr )� tracebackZ print_exc)ri rE rj r# r# r$ r^ s z Queue._on_queue_feeder_error�r )F�TNrl )�__name__�
__module__� __qualname__r% r) r+ r r rH rN rP rS rV rW rY rZ r[ r\ rA Z staticmethodr_ ra r] r^ r# r# r# r$ r # s0
=r c @ s@ e Zd Zddd�Zdd� Zdd� Zdd
d �Zd d
� Zdd� Zd S )r r c C s* t j| ||d� |�d�| _|�� | _d S )N)r" r )r r% Z Semaphore�_unfinished_tasksr0 �_condr r# r# r$ r% ) s zJoinableQueue.__init__c C s t �| �| j| jf S r& )r r) rq rp r( r# r# r$ r) . rT zJoinableQueue.__getstate__c C s, t �| |d d� � |dd � \| _| _d S )Ni����)r r+ rq rp r* r# r# r$ r+ 1 � zJoinableQueue.__setstate__TNc C s� | j rtd| �d���| j�||�s(t�| j�h | j�B | jd u rJ| �� | j �
|� | j � � | j�
� W d � n1 s~0 Y W d � n1 s�0 Y d S r> )r6 r? r r@ r r. rq r3 rA r2 rB rp rI rC rD r# r# r$ rH 5 s
zJoinableQueue.putc C sT | j �: | j�d�s td��| jj�� r2| j �� W d � n1 sF0 Y d S )NFz!task_done() called too many times)rq rp r@ r? rO rU Z
notify_allr( r# r# r$ � task_doneB s
zJoinableQueue.task_donec C s@ | j �& | jj�� s | j �� W d � n1 s20 Y d S r& )rq rp rO rU rg r( r# r# r$ rb I s zJoinableQueue.joinrk rl ) rm rn ro r% r) r+ rH rs rb r# r# r# r$ r ' s
r c @ sN e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
e j �Z
dS )r c C sH t jdd�\| _| _|�� | _| jj| _tj dkr:d | _
n
|�� | _
d S )NFr r ) r r r r r r r< r= r r r )r! r" r# r# r$ r% T s
zSimpleQueue.__init__c C s | j �� | j�� d S r& )r rZ r r( r# r# r$ rZ ] �
zSimpleQueue.closec C rQ r& rR r( r# r# r$ rS a rT zSimpleQueue.emptyc C s t �| � | j| j| j| jfS r& )r r' r r r r r( r# r# r$ r) d rt zSimpleQueue.__getstate__c C s" |\| _ | _| _| _| j j| _d S r& )r r r r r< r= r* r# r# r$ r+ h rr zSimpleQueue.__setstate__c C s: | j � | j�� }W d � n1 s&0 Y t�|�S r& )r r r: rK rL )r! rM r# r# r$ rN l s (zSimpleQueue.getc C sX t �|�}| jd u r"| j�|� n2| j� | j�|� W d � n1 sJ0 Y d S r& )rK rh r r r8 rX r# r# r$ rH r s
zSimpleQueue.putN)rm rn ro r% rZ rS r) r+ rN rH Z classmethod�typesZ GenericAliasZ__class_getitem__r# r# r# r$ r R s
r ) Z__all__r r r/ r1 rJ ru r` rf Zqueuer r Z_multiprocessingZ r r Z reductionZForkingPicklerrK Zutilr r
r r r
Zobjectr rc r r r# r# r# r$ Z<module>
s& {
+