a

HZ�h��@s<dd
lm
Z

Gdd�de�ZGdd�d�ZGdd�d�ZdS)	�
)
�Errorcs eZ
dZd
Z�f
dd�Z�ZS)�
StableDictz,A dict subclass with stable items() orderingc

stt
����
S�
N)�sorted�super�items�
�self�
�	__class__��=/usr/lib64/python3.9/site-packages/borg/helpers/datastruct.pyrs
zStableDict.items)�__name__�
__module__�__qualname__�__doc__r�
__classcell__rrr
r
rs

rc@sHeZ
dZd
ZGdd�dee�Zddd�
Zdd	�Zddd�
Z	dd
d�
Z
dS)�Bufferz/
    Provides a managed, resizable buffer.
    c
@seZ
dZd
ZdS)zBuffer.MemoryLimitExceededz2Requested buffer size {} is above the limit of {}.N�rrrrrrrr
�MemoryLimitExceededs
r�NcCs|
|_||_
|j|d
d�
dS)z�
        Initialize the buffer: use allocator(size) call to allocate a buffer.
        Optionally, set the upper <limit> for the buffer size.
        T)
�initN)�	allocator�limit�resize)r	r�sizerrrr
�__init__s

zBuffer.__init__c

Cs
t|j
�
Sr)�len�bufferrrrr
�__len__s
zBuffer.__len__FcCsJt|
�
}
|j
d
u
r*|
|j
kr*t�|
|j
��
|s:t|�
|
krF|�|
�
|_d
S)a$

        resize the buffer - to avoid frequent reallocation, we usually always grow (if needed).
        giving init=True it is possible to first-time initialize or shrink the buffer.
        if a buffer size beyond the limit is requested, raise Buffer.MemoryLimitExceeded (OSError).
        N)�intrrrrrr�r	rrrrr
r s




z
Buffer.resizecCs|
d
u
r|�|
|�
|j
S)z�
        return a buffer of at least the requested size (None: any current size).
        init=True can be given to trigger shrinking of the buffer to the given size.
        N)rrr!rrr
�get,s

z
Buffer.get)rN)
F)NF)rrrrr�OSErrorrrrrr"rrrr
r
s


rc@sPeZ
dZd
ZGdd�de�Zdd�Zdd�Zdd	�Zd
d�Z	dd
�Z
dd�ZdS)�EfficientCollectionQueuezL
    An efficient FIFO queue that splits received elements into chunks.
    c
@seZ
dZd
ZdS)z&EfficientCollectionQueue.SizeUnderflowzHCould not pop_front first {} elements, collection only has {} elements..Nrrrrr
�
SizeUnderflow;s
r%cCsg|_d
|_
|
|_||_dS)z�
        Initializes empty queue.
        Requires split_size to define maximum chunk size.
        Requires member_type to be type defining what base collection looks like.
        �N)�buffersr�
split_size�member_type)r	r(r)rrr
r>s


z!EfficientCollectionQueue.__init__c
Cs|js|�
�S|jd
}
|
S)z�
        Returns first chunk from queue without removing it.
        Returned collection will have between 1 and split_size length.
        Returns empty collection when nothing is queued.
        r&)r'r))r	rrrr
�
peek_frontIs



z#EfficientCollectionQueue.peek_frontcCsx|
|jkrt
�|
|j��
|
d
krt|jd
}t|
t|�
�}||d�}|rT||jd
<n|jd
=|
|8}
|j|8_qdS)z~
        Removes first size elements from queue.
        Throws if requested removal size is larger than whole queue.
        r&N)rr$r%r'�minr)r	rr�	to_removerrr
�	pop_frontTs










z"EfficientCollectionQueue.pop_frontcCs�|js|�
�g
|_|
r�|jd
}t|�
|jkrB|�
�}|j�|�

tt|
�
|jt|�
�}||
d|�7}|
|d�}
||jd
<|j|7_qdS)zm
        Adds data at end of queue.
        Takes care to chunk data into split_size sized elements.
        ���N)r'r)rr(�appendr+r)r	�datarZto_addrrr
�	push_backfs











z"EfficientCollectionQueue.push_backc


Cs|jS)
zF
        Current queue length for all elements in all chunks.
        �
rrrrr
rysz EfficientCollectionQueue.__len__c

Cs
|jd
kS)z4
        Returns true if queue isn't empty.
        r&r2rrrr
�__bool__sz!EfficientCollectionQueue.__bool__N)rrrrrr%rr*r-r1rr3rrrr
r$6s
r$N)�errorsr�dictrrr$rrrr
�<module>
s,