o

;j
�@svdZd
dl
mZ
d
dlmZmZ
d
dlmZmZ
d
dl	m
Z
mZ
d
dlm
Z
mZ
d
dlmZ
dZGd	d
�d
ee�ZdS)z*Contains the MemoryDatabase implementation�)
�
LooseObjectDB)�	ObjectDBR�	ObjectDBW)�OStream�IStream)�	BadObject�UnsupportedOperation)�ZippedStoreShaWriter�DecompressMemMapReader)
�BytesIO)
�MemoryDBcs`eZ
dZd
Z�f
dd�Zdd�Zdd�Zdd	�Zd
d�Zdd
�Z	dd�Z
dd�Zdd�Z�Z
S)ra/
A memory database stores everything to memory, providing fast IO and object
    retrieval. It should be used to buffer results and obtain SHAs before writing
    it to the actual physical storage, as it allows to query whether object already
    exists in the target storage before introducing actual IOc

s t��
�
td
�
|_t�|_dS)Nzpath/doesnt/matter)�super�__init__r�_db�dict�_cache�
�self�
�	__class__��@/opt/saltstack/salt/lib/python3.10/site-packages/gitdb/db/mem.pyr's



zMemoryDB.__init__cCstd
�
�
)Nz$MemoryDB's always stream into memory)
r)r�streamrrr�set_ostream.s
zMemoryDB.set_ostreamcCs`t�}|j
�|�

|j
�|
�
}
|��
|�d
�

t|��dd�}t|
j	|
j
|
j|�|j|
j	<|
S)Nr
F)
Zclose_on_deletion)
r	rr�store�close�seekr
�getvaluerZbinsha�type�sizer)r�istreamZzstreamZ
decomp_streamrrrr1s





zMemoryDB.storecCs
|
|jvS�
N)
r�r�sharrr�
has_object@�

zMemoryDB.has_objectcCs
|�|
�
Sr!)
rr"rrr�infoCs
z
MemoryDB.infoc
Cs@z|j|
}|j
�d
�

|WSty
}
zt|
�
|�d}~w
w)Nr
)rrr�KeyErrorr)rr#�ostream�
errrrGs






��zMemoryDB.streamc

Cs
t|j
�
Sr!)�lenrrrrrrQr%z
MemoryDB.sizec

Cs
|j�
�Sr!)r�keysrrrr�sha_iterTr%zMemoryDB.sha_itercCsZd
}|
D]&}|�|�
rq|�
|�
}t|j
���
}t|j|j||�}|�|�

|d7}q|S)ar
Copy the streams as identified by sha's yielded by sha_iter into the given odb
        The streams will be copied directly
        **Note:** the object will only be written if it did not exist in the target db

        :return: amount of streams actually copied into odb. If smaller than the amount
            of input shas, one or more objects did already exist in odbr
�
)r$rr�datarrrr)rr,Zodb�countr#r(�sior rrr�stream_copyXs








zMemoryDB.stream_copy)�__name__�
__module__�__qualname__�__doc__rrrr$r&rrr,r1�
__classcell__rrrrr s
rN)r5Zgitdb.db.looserZ
gitdb.db.baserrZ
gitdb.baserrZ	gitdb.excrrZgitdb.streamr	r
�ior�__all__rrrrr�<module>
s