|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/mercurial/__pycache__/ |
Upload File : |
o
�]LbR~ � @ sV d dl mZ d dlZddlmZmZmZ ddlmZm Z m
Z
m Z m Z m
Z
ddlmZmZ e jr[d dlmZmZmZmZmZmZmZmZmZ ddlm Z e eeeeeeeeee f
�s[J �ej Z ej Z ej Z ej!Z!G d d
� d
e"�Z#d d � Z$d
d� Z%G dd� de"�Z&G dd� de&�Z'dZ(de( Z)de( Z*dZ+e e+�Z,de, Z-dZ.dZ/dZ0G dd � d e"�Z1dS ) � )�absolute_importN� )�bin�hex�nullrev)�encoding�error�obsolete�pycompat�scmutil�util)� repoviewutil�
stringutil) �Any�Callable�Dict�Iterable�List�Optional�Set�Tuple�Union)� localrepoc @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�BranchMapCachez8mapping of filtered views of repo with their branchcachec C s
i | _ d S �N)� _per_filter��self� r �5/usr/lib/python3/dist-packages/mercurial/branchmap.py�__init__E �
zBranchMapCache.__init__c C s | � |� | j|j S r )� updatecacher �
filtername�r �repor r r � __getitem__H s
zBranchMapCache.__getitem__c s� |j }|j}| j�|�� � du s� �|�st�|�� g }� du rLt�|�}|durH|�|�}| | � � � |j j
|j
}|� � fdd�|D �� nt|�� |� |j � j
d d�� |r`� �||� � �|�siJ |��� | j|j< dS )z<Update the cache for the given filtered view on a repositoryNc 3 s � | ]
}|� j kr|V qd S r )�tiprev)�.0�r��bcacher r � <genexpr>c � � z-BranchMapCache.updatecache.<locals>.<genexpr>r )�start)� changelogr# r �get�validfor� branchcache�fromfile� subsettable�filtered�copy� filteredrevs�extend�revsr' �update)r r% �clr# r9 �
subsetname�subset� extrarevsr r* r r" L s&
zBranchMapCache.updatecachec s� |j }|j� |j}g }t� }t�|�D ] }||7 }|D ]}� |�} || �\}
} | r.|�|� qq|rft� fdd�|D ��} t|||| � � | |d�}
dD ]}|�
|�}|
� |�re|
| j |< |
�
|� dS qLdS dS )z�Replace the branchmap cache for a repo with a branch mapping.
This is likely only called during clone with a branch map from a
remote.
c 3 s � | ] }t � |��V qd S r )�int�r( �node��clrevr r r, � s � z)BranchMapCache.replace.<locals>.<genexpr>)� closednodes)s bases immutables servedN)r/ �rev�
branchinfo�setr
�
itervalues�add�maxr2 rA r5 r1 r �write)r r% �remotebranchmapr; � clbranchinfo�rbheads�closed�bheads�hr) �b�c�rtiprev�cache� candidate�rviewr rB r �replaceo s@
��
�
�� zBranchMapCache.replacec C s | j �� d S r )r �clearr r r r rY � � zBranchMapCache.clearc C s: |� � }| j�� D ]\}}|jr|�|�}|�|� q d S r )�
unfilteredr �items�_delayedr5 rK )r r% �unfir# rU r r r �
write_delayed� s
��z BranchMapCache.write_delayedN)
�__name__�
__module__� __qualname__�__doc__r r& r" rX rY r_ r r r r r B s #' r c C s t dt�t| �� ��)zEraises ValueError when branchcache found a node which does not existsznode %s does not exist)�
ValueErrorr
�sysstrr �rA r r r � _unknownnode� s rg c C s | j d ur
d| j S dS )Ns branch cache (%s)s branch cache�r# )r% r r r �_branchcachedesc� s
ri c @ s� e Zd ZdZddedddfdd�Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
e
Zdd� Zedd� �Zdd� Zed d � �Zd d � Zd d!� Zd"d#� Zd$d%� Zd3d'd(�Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z dS )4r2 a< A dict like object that hold branches heads cache.
This cache is used to avoid costly computations to determine all the
branch heads of a repo.
The cache is serialized on disk in the following format:
<tip hex node> <tip rev number> [optional filtered repo hex hash]
<branch head hex node> <open/closed state> <branch name>
<branch head hex node> <open/closed state> <branch name>
...
The first line is used to check if the cache is still valid. If the
branch cache is for a filtered repo view, an optional third hash is
included that hashes the hashes of all filtered and obsolete revisions.
The open/closed state is represented by a single letter 'o' or 'c'.
This field can be used to avoid changelog reads when determining if a
branch head closes a branch or not.
r Nc C s� || _ d| _|du r|j| _n|| _|| _|| _|du r!t� | _n|| _t|�| _ d| _
t� | _ || _ | j du r?dd� | _ dS dS )z�hasnode is a function which can be used to verify whether changelog
has a given node or not. If it's not provided, we assume that every node
we have exists in changelogFNc S s dS )NTr )�xr r r �<lambda>� s z&branchcache.__init__.<locals>.<lambda>)
�_repor] �nullid�tipnoder' � filteredhashrG � _closednodes�dict�_entries�_closedverified�_verifiedbranches�_hasnode)r r% �entriesrn r' ro rD �hasnoder r r r � s"
�zbranchcache.__init__c C s2 | j rdS | jD ] }| �|�st|� qd| _ dS )z verify the closed nodes we haveNT)rs rp ru rg )r rA r r r �
_verifyclosed� s
�
zbranchcache._verifyclosedc C sJ || j vs
|| jv r dS | j | D ] }| �|�s t|� q| j�|� dS )z'verify head nodes for the given branch.N)rr rt ru rg rI )r �branch�nr r r �
_verifybranch� s
�zbranchcache._verifybranchc C s, t | j�� �| j }|D ]}| �|� q dS )z"verifies nodes of all the branchesN)rG rr �keysrt r{ )r �needverificationrR r r r �
_verifyall� s �zbranchcache._verifyallc C s
t | j�S r )�iterrr r r r r �__iter__ r! zbranchcache.__iter__c C s || j |< d S r )rr )r �key�valuer r r � __setitem__ rZ zbranchcache.__setitem__c C s | � |� | j| S r �r{ rr �r r� r r r r& �
zbranchcache.__getitem__c C � | � |� || jv S r r� r� r r r � __contains__
r� zbranchcache.__contains__c c s0 � t �| j�D ]\}}| �|� ||fV qd S r )r
� iteritemsrr r{ )r �k�vr r r r� s
�
�zbranchcache.iteritemsc C r� )z2checks whether a branch of this name exists or notr� )r �labelr r r � hasbranch s
zbranchcache.hasbranchc
C sL d }z�zT|� | �|��}t|�}t|��d��dd�}|d d� \}}t|�t|�}}d }|jj }t
|�dkr>t|d �}| |||||d�} | � |�sPt d��| �
||� W n= ttfyl Y W |rj|�� d S d S ty� }
z |jjr�d} |j�| t|�t�|
�f � d } W Y d }
~
nd }
~
ww W |r�|�� | S |r�|�� w w )N�
� � )rn r' ro rw z tip differss invalid %s: %s
)�cachevfs� _filenamer �next�rstrip�splitr r? r/ rw �lenr1 rd �load�IOError�OSError�close� Exception�ui� debugflag�debugri r � forcebytestr) �clsr% �f�lineiter�cachekey�last�lrevro rw r+ �inst�msgr r r r3 s\ �
��������
�
�zbranchcache.fromfilec C s~ |D ]:}|� d�}|s q|�dd�\}}}|dvr td��t�|�� �}t|�}| j�|g �� |� |dkr<| j
� |� qdS )z[fully loads the branchcache by reading from the file using the line
iterator passedr� r� r� s oczinvalid branch state� cN) r� r� rd r �tolocal�stripr rr �
setdefault�appendrp rI )r r% r� �linerA �stater� r r r r� J s
��zbranchcache.loadc C s d}| j r d|| j f }|S )z7name of a branchcache file for a given repo or repoviews branch2s %s-%srh )r% �filenamer r r r� Z s zbranchcache._filenamec C sR z |j �| j�}W n
ty Y dS w | j|krdS tj|| jdd�}| j|kS )z�check that cache contents are valid for (a subset of) this repo
- False when the order of changesets changed or if we detect a strip.
- True when cache is up-to-date for the current repo or its subset.FT�� needobsolete)r/ rA r' �
IndexErrorrn r ro )r r% rA �tiphashr r r r1 b s �
zbranchcache.validforc C s> |d }d}t |�D ]}|| jvr|}d} ||fS q
||fS )zhReturn tuple with last open head in heads and false,
otherwise return last closed head and true.���TF)�reversedrp )r �heads�tiprO rQ r r r �
_branchtipw s
�zbranchcache._branchtipc C s | � | | �d S )z�Return the tipmost open head on branch head, otherwise return the
tipmost closed head on branch.
Raise KeyError for unknown branch.r )r� )r ry r r r � branchtip� s zbranchcache.branchtipc s � fdd�|D �S )Nc 3 s � | ]
}|� j vr|V qd S r )rp )r( rz r r r r, � r- z'branchcache.iteropen.<locals>.<genexpr>r )r �nodesr r r �iteropen� s zbranchcache.iteropenFc C s* | � |� | j| }|st| �|��}|S r )r{ rr �listr� )r ry rO r� r r r � branchheads� s
zbranchcache.branchheadsc c s. � t �| �D ]\}}||f| �|� V qd S r )r
r� r� )r �bnr� r r r � iterbranches� s ��zbranchcache.iterbranchesc C s | � � t�| j�S )zreturns all the heads)r~ r
rH rr r r r r � iterheads� s zbranchcache.iterheadsc C s" t | �| j| j| j| j| j| j�S )z-return an deep copy of the branchcache object)�typerl rr rn r' ro rp r r r r r6 � s �zbranchcache.copyc
C sT |� � }t|dd�sd| _d S zx|j| �|�ddd�}t| j�d| j g}| jd ur2|� t| j�� |�
d� |�d � d}t t
�| j��D ](\}}t�|�}|D ] }|d 7 }|| jv r_d
} nd } |�
d t|�| |f � qQqF|�� |j�d
dt|�t| j�|� d| _W d S tttjfy� }
z|j� dt � |
� � W Y d }
~
d S d }
~
ww )N� finalizedT� w)�
atomictemps %dr� r� r r r� � os %s %s %s
� branchcaches% wrote %s with %d labels and %d nodes
Fs couldn't write branch cache: %s
) �currenttransaction�getattrr] r� r� r rn r' ro r� rK �join�sortedr
r� rr r � fromlocalrp r� r� �logri r� r� r� r �Abortr� r r� ) r r% �trr� r� � nodecountr� r� rA r� r� r r r rK � sJ
�� ����zbranchcache.writec s� t �� }|j� i }|�� j}|D ] }||�\}}|�|g ��|� |r,| j�� � |�� q| j
} d}
|� � jj } t
�|d�} t�|�D ]�\}}
| j�|g �}� fdd�|D �}t� }t|
�D ]j}|| v req^|sm|�|� q^dd� | |�D �}t� }t� }t� }|D ]#}|| v r�|�|� q�||v s�||�d |kr�|�|� q�|�|� q�t|�t|� kr�dks�n |�|� |�|� |�|� |�|� q^|r�|
du r�t� �� �}
||
r�t|�}|t|�kr�t� �||��}||8 }|r�� fd d�t|�D �| |< t|
�}|| k�r |} qC| | j
k�r| | _
� � | �| _ | � |��sM|j | _ t | _
| � � D ]"}|�s1�q*t� fd
d �|D ��}|| j
k�rK� � |�| _ || _
�q*t!j"|| j
d d
�| _"t �� | }|j#�$ddt%|�|� | �&|� dS )z�Given a branchhead cache, self, that may have extra nodes or be
missing heads, and a generator of nodes that are strictly a superset of
heads missing, this function updates self to be correct.
Ns obsoletec s h | ]}� � |��qS r �rE r@ �r; r r � <setcomp> � z%branchcache.update.<locals>.<setcomp>c S s g | ]}|t kr|�qS r )r )r( �pr r r �
<listcomp> s z&branchcache.update.<locals>.<listcomp>r r c s g | ]}� � |��qS r rf )r( rE r� r r r� > r� c 3 s � | ]}� � |�V qd S r r� r@ r� r r r, S s � z%branchcache.update.<locals>.<genexpr>Tr� r� s updated %s in %.4f seconds
)'r �timerr/ �revbranchcacherF r� r� rp rI rA r' r[ �
parentrevsr �getrevsr
r� rr r0 rG r� r� r: �difference_update�headrevs�minrJ � ancestorsrn r1 rm r r� r ro r� r� ri rK ) r r% �revgen� starttime� newbranches�
getbranchinfor) ry � closesbranch�ntiprev� topoheadsr� �obsrevs� newheadrevsrP �bheadset� uncertain�newrev�parents�
samebranch� otherbranch�
obsparentsr� �floorrevr� r' r� �durationr r� r r: � s�
� $
� �� �zbranchcache.update)F) r` ra rb rc r r rx r{ r~ r� r� r& r� r� r\ r� � classmethodr3 r� � staticmethodr� r1 r� r� r� r� r� r� r6 rK r: r r r r r2 � sD
�&
,
(r2 c @ s e Zd ZdZdd� ZdS )�remotebranchcachez@Branchmap info for a remote connection, should not write locallyc C s d S r r r$ r r r rK i s zremotebranchcache.writeN)r` ra rb rc rK r r r r r� f s r� s -v1s rbc-namess rbc-revss >4sI�@ � i���l c @ sj e Zd ZdZddd�Zdd� Zejdd� �Zd d
� Z d d � Z
d
d� Z dd� Z ddd�Z
dd� Zdd� ZdS )r� a� Persistent cache, mapping from revision number to branch name and close.
This is a low level cache, independent of filtering.
Branch names are stored in rbc-names in internal encoding separated by 0.
rbc-names is append-only, and each branch name is only stored once and will
thus have a unique index.
The branch info for each revision is stored in rbc-revs as constant size
records. The whole file is read into memory, but it is only 'parsed' on
demand. The file is usually append-only but will be truncated if repo
modification is detected.
The record for each revision contains the first 4 bytes of the
corresponding node hash, and the record is only used if it still matches.
Even a completely trashed rbc-revs fill thus still give the right result
while converging towards full recovery ... assuming no incorrectly matching
node hashes.
The record also contains 4 bytes where 31 bits contains the index of the
branch and the last bit indicate that it is a branch close commit.
The usage pattern for rbc-revs is thus somewhat similar to 00changelog.i
and will grow with it but be 1/8th of its size.
Tc
C s |j d u sJ �|| _g | _t� | _d| _z|j�t�}t |�| _|r-dd� |�
d�D �| _W n t t fy? |r=| j
| _Y nw | jrrz|j�t�}|| jd d �< W n t t fyq } z|j�dt�|� � W Y d }~nd }~ww tt | j�t t |j��| _| jdkr�g | _t | j�| _d S )Nr c S s g | ]}t �|��qS r )r r� )r( r� r r r r� � s
�z+revbranchcache.__init__.<locals>.<listcomp>� s( couldn't read revision branch cache: %s
)r# rl �_names� bytearray�_rbcrevs�
_rbcsnameslenr� �read� _rbcnamesr� r� r� r� � _branchinforF r� r� r r� r� � _rbcrecsizer/ � _rbcrevslen�_rbcnamescount)r r% �readonly�bndata�datar� r r r r � sH
���� �����
zrevbranchcache.__init__c C sF d| _ | jd d �= d| _t| jj�| _t| jt �| _ t
� | d� d S )Nr s
_namesreverse) r� r� r� r� rl r/ r� r� r� r� r �clearcachedpropertyr r r r �_clear� s zrevbranchcache._clearc C s dd� t | j�D �S )Nc S s i | ]\}}||�qS r r )r( r) rR r r r �
<dictcomp>� r� z0revbranchcache._namesreverse.<locals>.<dictcomp>)� enumerater� r r r r �
_namesreverse� s z revbranchcache._namesreversec C s
| j j}|t }|tkr|�|�S t| j�|t k r | �|�S |�|�dt � }t
t t �
| j�|�\}}t|t@ �}|r@|tM }|dkrEn;||krfz| j| |fW S tye | j j�d� | �� Y nw | j j�d| � |t }| j|d�= t| j|�| _| �|�S )zWReturn branch name and close flag for rev, using and updating
persistent cache.Ns sR referenced branch names not found - rebuilding revision branch cache from scratch
sP history modification detected - truncating revision branch cache to revision %d
)rl r/ r� r rF r� r� r� rA � _rbcnodelen� unpack_from�
_rbcrecfmtr �buffer�bool�
_rbccloseflag�_rbcbranchidxmaskr� r� r� r� r r� r� ) r rE r/ � rbcrevidx�reponode� cachenode� branchidxr� �truncater r r rF � sB
� � � ��
zrevbranchcache.branchinfoc C sx | j j}|�|�\}}|| jv r| j| }nt| j�}| j�|� || j|< |�|�}|r1|tO }| � |||� ||fS )z7Retrieve branch info from changelog and update _rbcrevs)
rl r/ rF r r� r� r� rA r
�
_setcachedata)r rE r/ rR r� r r r r r r� � s
zrevbranchcache._branchinfoc C s| |j \}}|| jv r| j| }nt| j�}| j�|� || j|< |r&|tO }| �|| jj� |�|� dt
| �v r<| ` dS dS )z%add new data information to the cacherF N) rF r r� r� r� r
r rl r/ rA �vars)r rE �changelogrevisionry r� r r r r �setdata s
�zrevbranchcache.setdatac C s� |t krdS |t }|t }t| j�}||k r"| j�dtt|� � tt| j|||� t | j
|�| _
| j � � }|rB|�
d| j� dS dS )z:Writes the node's branch data to the in-memory cache data.Nr� s write-revbranchcache)r r� r� r� r8 rJ �
_rbcmininc� pack_intor
r� r� rl r� � addfinalizerK )r rE rA r r � requiredsize�rbccurr� r r r r s
�z revbranchcache._setcachedataNc C s� | j }d}d}zhz5| jt| j�k r |jdd�}d}| �|� | jt }|t| j�kr<d}|du r6|jdd�}| � ||� W n% t
t t j
t jfyb } z|j�d|t�|�f � W Y d}~nd}~ww W |durn|�� dS dS |dury|�� w w )z!Save branch cache if it is dirty.N� F)�waits namess+ couldn't write revision branch cache%s: %s
)rl r� r� r� �wlock� _writenamesr� r� r� �
_writerevsr� r� r r� � LockErrorr� r� r r� �release)r r� r% r �stepr. r� r r r rK , s:
� ����� �
�zrevbranchcache.writec C s� | j dkr+|j�td�}|�� | jkr|�d� n|�� |j� dt � d| _ d| _
| j dkr?|jj t dd� |j�td�}|�d�
dd � | j| j d
� D ��� |�� | _|�� t| j�| _ d
S ) z,write the new branch names to revbranchcacher � abr� s %s changed - rewriting it
T)�
ignoremissings wbc s s � | ]}t �|�V qd S r )r r� )r( rR r r r r, Y s
� �
�z-revbranchcache._writenames.<locals>.<genexpr>N)r� r� �openr� �tellr� rK r� r� r� r� �
unlinkpathr� r� r� r� )r r% r� r r r r I s&
��
zrevbranchcache._writenamesc C s� t t|j�t| j�t �}|j�td��<}|�� |kr;|j� dt|f � |�
|� |�� |kr7d}|�
|� |� � |t }|� | j||� � W d � n1 sSw Y || _
dS )z$write the new revs to revbranchcacher% s truncating cache/%s to %d
r N)r� r� r/ r� r� r� r' r( r� r� �seekr rK r� )r r% r. r9 r� �endr r r r! b s
�
�
zrevbranchcache._writerevs)Tr )r` ra rb rc r r r �
propertycacher rF r� r r rK r r! r r r r r� { s
$
/
r� )2�
__future__r �structrA r r r � r r r r
r r �utilsr
r �
TYPE_CHECKING�typingr r r r r r r r r r �anyr4 �calcsizer r �objectr rg ri r2 r� � _rbcversionr� r� r
r� r r r r
r� r r r r �<module> sV , ��_ <