File: //lib64/python3.9/site-packages/borg/__pycache__/locking.cpython-39.opt-1.pyc
a
HZ�h1@ � @ s d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZm Z ddl
m Z d\Z Z
Zd\ZZe e�Z
G dd � d �ZG d
d � d e�ZG d d
� d
e �ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� d�ZG dd� d�ZG dd� d�ZdS ) � N� )�platform)�Error�ErrorWithTraceback)�
create_logger)�add�removeZremove2)Zshared� exclusivec @ sB e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� TimeoutTimerz�
A timer for timeout checks (can also deal with "never timeout").
It can also compute and optionally execute a reasonable sleep time (e.g. to avoid
polling too often or to support thread/process rescheduling).
Nc C sX |dur|dk rt d��|| _|du rB|du r4d}ntd|d �}|| _d| _d| _dS )ag
Initialize a timer.
:param timeout: time out interval [s] or None (never timeout, wait forever) [default]
:param sleep: sleep interval [s] (>= 0: do sleep call, <0: don't call sleep)
or None (autocompute: use 10% of timeout [but not more than 60s],
or 1s for "never timeout" mode)
Nr ztimeout must be >= 0g �?g N@g $@)�
ValueError�timeout_interval�min�sleep_interval�
start_time�end_time)�self�timeout�sleep� r �2/usr/lib64/python3.9/site-packages/borg/locking.py�__init__ s zTimeoutTimer.__init__c C s d� | jj| j| j| j| j�S )Nz1<{}: start={!r} end={!r} timeout={!r} sleep={!r}>)�format� __class__�__name__r r r r �r r r r �__repr__, s �zTimeoutTimer.__repr__c C s& t � � | _| jd ur"| j| j | _| S �N)�timer r r r r r r �start1 s
zTimeoutTimer.startc C s | j dkrt�| j � d S �Nr )r r r r r r r r 7 s
zTimeoutTimer.sleepc C s | j d uot�� | j kS r )r r r r r r � timed_out; s zTimeoutTimer.timed_outc C s | � � r dS | �� dS d S )NTF)r r r r r r �timed_out_or_sleep> s z TimeoutTimer.timed_out_or_sleep)NN)
r �
__module__� __qualname__�__doc__r r r r r r! r r r r r
s
r
c @ s e Zd ZdZdS )� LockError� Failed to acquire the lock {}.N�r r" r# r$ r r r r r% F s r% c @ s e Zd ZdZdS )�
LockErrorTr& Nr' r r r r r( J s r( c @ s e Zd ZdZdS )� LockTimeoutz/Failed to create/acquire the lock {} (timeout).Nr' r r r r r) N s r) c @ s e Zd ZdZdS )�
LockFailedz*Failed to create/acquire the lock {} ({}).Nr' r r r r r* R s r* c @ s e Zd ZdZdS )� NotLockedz/Failed to release the lock {} (was not locked).Nr' r r r r r+ V s r+ c @ s e Zd ZdZdS )� NotMyLockz=Failed to release the lock {} (was/is locked, but not by me).Nr' r r r r r, Z s r, c @ sl e Zd ZdZddd�Zdd� Zdd� Zd d
� Zdd d �Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�
ExclusiveLocka/ An exclusive Lock based on mkdir fs operation being atomic.
If possible, try to use the contextmanager here like::
with ExclusiveLock(...) as lock:
...
This makes sure the lock is released again if the block is left, no
matter how (e.g. if an exception occurred).
Nc C sP || _ || _tj�|�| _|p$t�� | _tj�| jd| j �| _ d| _
d| _ d S )N�%s.%d-%xTF) r r �os�path�abspathr �get_process_id�id�join� unique_name�kill_stale_locks�stale_warning_printed)r r0 r r r3 r r r r i s zExclusiveLock.__init__c C s | � � S r ��acquirer r r r � __enter__r s zExclusiveLock.__enter__c G s | � � d S r ��release�r �excr r r �__exit__u s zExclusiveLock.__exit__c C s d| j j� d| j�d�S �N�<z: �>)r r r5 r r r r r x s zExclusiveLock.__repr__c
C sR |d u r| j }|d u r | j}tj�| j�\}}tj�| j�}d }�z�zNt�d|d |�}tj� ||�}t
|d�� W d � n1 s�0 Y W n8 t y� } z t | jt
|��d �W Y d }~�nd }~0 0 t||��� } zt�|| j� W n� t �yt | �� �rR| Y W |d u�rPzt�|� W n Y n0 zt�|� W S Y S 0 S | �� | �� �rpt| j�d �Y q�0 d }| W |d u�r�zt�|� W n Y n0 zt�|� W S Y S 0 S q�W |d u�rNzt�|� W n Y n0 zt�|� W n Y n0 nD|d u�rLzt�|� W n Y n0 zt�|� W n Y n0 0 d S )Nz.tmp�.�wb)r r r/ r0 �split�basenamer5 �tempfileZmkdtempr4 �open�OSErrorr* �strr
r �replace�by_me�unlink�rmdir�kill_stale_lockr! r) )
r r r � parent_path� base_nameZunique_base_nameZ temp_pathZtemp_unique_name�err�timerr r r r9 { s~ "*
�
�
�
zExclusiveLock.acquirec
C s� | � � st| j��| �� s$t| j��t�| j� zt�| j� W n> t y~ } z&|j
t
j t
j t
j
fvrj|�W Y d }~n
d }~0 0 d S r )� is_lockedr+ r0 rL r, r/ rM r5 rN rI �errno� ENOTEMPTY�EEXIST�ENOENT)r rR r r r r< � s
zExclusiveLock.releasec C s t j�| j�S r )r/ r0 �existsr r r r rT � s zExclusiveLock.is_lockedc C s t j�| j�S r )r/ r0 rY r5 r r r r rL � s zExclusiveLock.by_mec
C s� zt �| j�}W n ty$ Y �n0 |D �] }z6|�dd�\}}|�dd�\}}t|�}t|d�}W n& ty� t�d|| j� Y dS 0 t �
|||�r� dS | j s�| j s�t�
d|� d| _ dS z&t �t j�| j|�� t�
d |� W q* t�y6 } z0| j �s t�d
|t| �� d| _ W Y d } ~ dS d } ~ 0 0 q*zt �| j� W n^ t�y� } zD| jtjtjtjfv �r�W Y d } ~ dS t�d t| �� W Y d } ~ dS d } ~ 0 0 dS ) N�-r rC � z9Found malformed lock %s in %s. Please check/fix manually.FzLFound stale lock %s, but not deleting because self.kill_stale_locks = False.TzKilled stale lock %s.z0Found stale lock %s, but cannot delete due to %sz Failed to remove lock dir: %s)r/ �listdirr0 �FileNotFoundError�rsplit�intr �logger�errorr �
process_aliver6 r7 �warningrM r4 rI rJ rN rU rV rW rX )
r �names�nameZhost_pidZ
thread_str�hostZpid_str�pid�threadrR r r r rO � sH
z ExclusiveLock.kill_stale_lockc C s@ | � � r<t�| j�D ]}t�tj�| j|�� qt�| j� d S r )rT r/ r\ r0 rM r4 rN )r re r r r �
break_lock� s zExclusiveLock.break_lockc C sl t j�| jd| �}| �� rZ| �� rZt|d�� W d � n1 sD0 Y t �| j� || | _| _dS )�0migrate the lock ownership from old_id to new_idr. rD N) r/ r0 r4 rT rL rH rM r5 r3 )r �old_id�new_idZnew_unique_namer r r � migrate_lock� s zExclusiveLock.migrate_lock)NNN)NN)r r" r# r$ r r: r? r r9 r< rT rL rO ri rm r r r r r- ^ s
*1r- c @ sR e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�
LockRosterz�
A Lock Roster to track shared/exclusive lockers.
Note: you usually should call the methods with an exclusive lock held,
to avoid conflicting access by multiple threads/processes/machines.
Nc C s || _ |pt�� | _d| _d S )NT)r0 r r2 r3 r6 )r r0 r3 r r r r s zLockRoster.__init__c C s� z�t | j��}t�|�}W d � n1 s,0 Y | jr�ttfD ]|}z || }W n tyj Y qDY n0 t� }|D ]<\}}}t �
|||�r�|� |||f� qvt �
d||||� qvt|�||< qDW n ttfy� i }Y n0 |S )Nz:Removed stale %s roster lock for host %s pid %d thread %d.)rH r0 �json�loadr6 �SHARED� EXCLUSIVE�KeyError�setr rb r r` rc �listr] r ) r �f�data�key�entries�elementsrf rg rh r r r rp s( (
�
zLockRoster.loadc C s< t | jd�� }t�||� W d � n1 s.0 Y d S )N�w)rH r0 ro �dump)r rw rv r r r �save" s zLockRoster.savec C s( zt �| j� W n ty" Y n0 d S r )r/ rM r0 r] r r r r r & s zLockRoster.removec C s | � � }dd� |�|g �D �S )Nc S s h | ] }t |��qS r ��tuple��.0�er r r � <setcomp>. � z!LockRoster.get.<locals>.<setcomp>)rp �get)r rx �rosterr r r r� , s zLockRoster.getc s t � fdd�|D ��S )Nc 3 s | ]}� � |� V qd S r )r� )r� rx r r r � <genexpr>1 r� z#LockRoster.empty.<locals>.<genexpr>)�all)r �keysr r r �empty0 s zLockRoster.emptyc C s� | � � }zdd� || D �}W n ty6 t� }Y n0 |tkrN|�| j� n8|tkrd|�| j� n"|tkrz|� | j� n t
d| ��t dd� |D ��||< | � |� d S )Nc S s h | ] }t |��qS r r~ r� r r r r� 6 r� z$LockRoster.modify.<locals>.<setcomp>zUnknown LockRoster op %rc s s | ]}t |�V qd S r )ru r� r r r r� D r� z$LockRoster.modify.<locals>.<genexpr>)
rp rs rt �ADDr r3 �REMOVE�discard�REMOVE2r r ru r} )r rx �opr� rz r r r �modify3 s zLockRoster.modifyc C sb | j d }| _ zFz| �|t� W n ty8 || _Y n0 || _| �|t� W || _ n|| _ 0 dS )rj FN)r6 r� r� rs r3 r� )r rx rk rl Zkillingr r r rm G s zLockRoster.migrate_lock)N) r r" r# r$ r rp r} r r� r� r� rm r r r r rn � s
rn c @ st e Zd ZdZd dd�Zdd� Zdd � Zd
d � Zd d d
�Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) �Locka�
A Lock for a resource that can be accessed in a shared or exclusive way.
Typically, write access to a resource needs an exclusive lock (1 writer,
no one is allowed reading) and read access to a resource needs a shared
lock (multiple readers are allowed).
If possible, try to use the contextmanager here like::
with Lock(...) as lock:
...
This makes sure the lock is released again if the block is left, no
matter how (e.g. if an exception occurred).
FNc C sP || _ || _|| _|| _|p"t�� | _t|d |d�| _t |d ||d�| _
d S )Nz.roster)r3 z
.exclusive)r3 r ) r0 � is_exclusiver r r r2 r3 rn �_rosterr- �_lock)r r0 r r r r3 r r r r j s z
Lock.__init__c C s | � � S r r8 r r r r r: w s zLock.__enter__c G s | � � d S r r; r= r r r r? z s z
Lock.__exit__c C s d| j j� d| j�d�S r@ )r r r3 r r r r r } s z
Lock.__repr__c C s� |d u r| j }|p| jpd}|r<| �||� | j�tt� nJ| j�4 |d urZ| j�|t� | j�t t� W d � n1 s|0 Y || _ | S )Ng�������?)
r� r �_wait_for_readers_finishingr� r� rr r� r� r� rq )r r r r r r r r9 � s ,z Lock.acquirec C s� t | j|��� }| j�� zJ|d ur2| j�|t� t| j� t
��dkrLW d S |d urb| j�|t � W n | j� � � Y n 0 | j� � |�
� rt| j��qd S r )r
r r r� r9 r� r� r� �lenr� rq r� r< r! r) r0 )r r r rS r r r r� � s
z Lock._wait_for_readers_finishingc C s� | j r8| j�tt� | j�tt�r,| j�� | j� � nL| j�6 | j�tt� | j�tt�rf| j�� W d � n1 sz0 Y d S r )
r� r� r� rr r� r� rq r r� r< r r r r r< � s
z Lock.releasec C s | j s| jdtd� d S )NT�r r )r� r9 rq r r r r �upgrade� s z Lock.upgradec C s | j r| jdtd� d S )NFr� )r� r9 rr r r r r � downgrade� s zLock.downgradec C s | j o| j�� o| j�� S r )r� r� rT rL r r r r �got_exclusive_lock� s zLock.got_exclusive_lockc C s | j �� | j�� d S r )r� r r� ri r r r r ri � s
zLock.break_lockc C st || _ | jr,| j�||� | j�t||� nD| j�. | j�||� | j�t||� W d � n1 sf0 Y d S r )r3 r� r� rm r� rr rq )r rk rl r r r rm � s zLock.migrate_lock)FNNN)NNN)r r" r# r$ r r: r? r r9 r� r< r� r� r� ri rm r r r r r� [ s
r� ) rU ro r/ rG r � r Zhelpersr r r` r r� r� r� rq rr r r
r% r( r) r* r+ r, r- rn r� r r r r �<module> s*
5 ^