File: //lib/python3/dist-packages/hgext/remotefilelog/__pycache__/metadatastore.cpython-310.pyc
o
�]Lb� � @ sd d dl mZ d dlmZmZ ddlmZmZ G dd� dej�Z G dd� dej�Z
G d d
� d
e �Z d S ) � )�absolute_import)�hex�sha1nodeconstants� )� basestore� shallowutilc sj e Zd Z� fdd�Zddd�Zejjddd��Zejjdd � �Z d
d � Z
d d
� Z ddd�Z dd� Z
� ZS )�unionmetadatastorec s: t t| �j|i |�� || _|�d�| _|�dd�| _d S )N�
writestore�allowincompleteF)�superr �__init__�stores�getr r
)�self�args�kwargs�� __class__� �C/usr/lib/python3/dist-packages/hgext/remotefilelog/metadatastore.pyr s zunionmetadatastore.__init__Nc s� �du rt � �|�v r
g S i � � �fdd�}||fg}|rS|�� \}}z� �| j||�d�� |||�}|�|� W n tyP | jsE� ||f||fkrN� Y nw |s � S )��Returns as many ancestors as we're aware of.
return value: {
node: (p1, p2, linknode, copyfrom),
...
}
Nc s� | |fg}g }t � }|rY|�� \}}||f|v rq
|�||f� � �|�}|s/|�||f� q
|\}} }
} |tjkrG|�vrG|�| pC| |f� | tjkrW| �vrW|�| | f� |s |S �N)�set�pop�addr �appendr �nullid) �curname�curnode�queue�missing�seen�name�node�value�p1�p2�linknode�copyfrom�� ancestors�knownr r �traverse( s&
�z1unionmetadatastore.getancestors.<locals>.traverse�r+ )r r �update�_getpartialancestors�extend�KeyErrorr
) r r" r# r+ r, r r r �
newmissingr r) r � getancestors s0
�
���z unionmetadatastore.getancestorsc C sD | j D ]}z |j|||d�W S ty Y qw t|t|�f��)Nr- )r
r3 r1 r )r r" r# r+ �storer r r r/ S s
�z'unionmetadatastore._getpartialancestorsc C s@ | j D ]}z |�||�W S ty Y qw t|t|�f��r )r
� getnodeinfor1 r )r r" r# r4 r r r r5 ] s
�z unionmetadatastore.getnodeinfoc C � t d��)Ns5 cannot add content only to remotefilelog contentstore�� RuntimeError�r r" r# �datar r r r g � �zunionmetadatastore.addc C s" |}| j D ] }|r|�|�}q|S r )r
�
getmissing)r �keysr r4 r r r r< l s
�z unionmetadatastore.getmissingc C s | j D ]}|�||� qd S r )r
�
markledger)r �ledger�optionsr4 r r r r> s s
�z unionmetadatastore.markledgerc C s dd� | j D �}tj|� S )Nc S s g | ]}|� � �qS r )�
getmetrics)�.0�sr r r �
<listcomp>x s z1unionmetadatastore.getmetrics.<locals>.<listcomp>)r
r �sumdicts)r �metricsr r r rA w s
z unionmetadatastore.getmetricsr )�__name__�
__module__� __qualname__r r3 r �baseunionstore� retriabler/ r5 r r< r> rA �
__classcell__r r r r r
s
:
r c @ s& e Zd Zddd�Zdd� Zdd� ZdS ) �remotefilelogmetadatastoreNc C s | � ||�}t�|�}|S )r )�_getdatar � ancestormap)r r" r# r+ r: r* r r r r3 } s
z'remotefilelogmetadatastore.getancestorsc C � | � ||�| S r �r3 �r r" r# r r r r5 � � z&remotefilelogmetadatastore.getnodeinfoc C r6 )Ns7 cannot add metadata only to remotefilelog metadatastorer7 )r r" r# �parentsr' r r r r � r; z remotefilelogmetadatastore.addr )rG rH rI r3 r5 r r r r r rM | s
rM c @ s@ e Zd Zdd� Zddd�Zdd� Zdd � Zd
d � Zdd d
�ZdS )�remotemetadatastorec C s || _ || _d S r )� _fileservice�_shared)r �ui� fileservice�sharedr r r r � s
z remotemetadatastore.__init__Nc C s0 | j j|t|�fgdddd� | jj|||d�S )NTF)�force� fetchdata� fetchhistoryr- )rV �prefetchr rW r3 )r r" r# r+ r r r r3 � s �z remotemetadatastore.getancestorsc C rP r rQ rR r r r r5 � rS z remotemetadatastore.getnodeinfoc C r6 )Ns cannot add to a remote storer7 r9 r r r r � s zremotemetadatastore.addc C s |S r r )r r= r r r r< � � z remotemetadatastore.getmissingc C s d S r r )r r? r@ r r r r> � r_ z remotemetadatastore.markledgerr ) rG rH rI r r3 r5 r r< r> r r r r rU � s
rU N)
�
__future__r �mercurial.noder r � r r rJ r rM �objectrU r r r r �<module> s o