File: //lib/python3/dist-packages/uaclient/__pycache__/lock.cpython-38.pyc
U
��Jh� � @ s� d dl Z d dlZd dlZd dlmZ d dlmZmZmZ d dl m
Z
m Z m Z d dl
mZ d dlmZmZ d dlmZ d dlmZ e �e�e��ZG d d
� d
e
�Zeeed d d
�ej�Z ee e f d�dd�Z dd� Z G dd� d�Z!dS )� N)�Tuple)�
exceptions�system�util)�
DataObject�Field�StringDataValue)�notices)�DataObjectFile�DataObjectFileFormat)�UAFile)�Noticec @ s0 e Zd Zede�ede�gZeed�dd�ZdS )�LockData�lock_pid� lock_holder�r r c C s || _ || _d S �Nr )�selfr r � r �//usr/lib/python3/dist-packages/uaclient/lock.py�__init__ s zLockData.__init__N)�__name__�
__module__� __qualname__r r Zfields�strr r r r r r s �r �lockF)Zprivate)�returnc C s� z t �� } W n$ tjk
r0 tjt jd��Y nX d}| s>|S | j}| j}z t� d|g� t
|�|fW S tj k
r� t �
� s�t�d||� t
|�|f Y S t�d||� t�t j� | Y S X dS )a Return lock info if lock file is present the lock is active.
If process claiming the lock is no longer present, remove the lock file
and log a warning.
:return: A tuple (pid, string describing lock holder)
If no active lock, pid will be -1.
)Zlock_file_path)���� Zpsz.Found stale lock file previously held by %s:%sz1Removing stale lock file previously held by %s:%sN)�lock_data_file�readr ZInvalidFileFormatErrorZInvalidLockFile�pathr r r Zsubp�intZProcessExecutionErrorr Zwe_are_currently_root�LOG�debugZwarningZensure_file_absent)Z
lock_data_objZno_lockr r r r r �check_lock_info" s6
�� r% c C s t �� d S r )r �deleter r r r �clear_lock_file_if_presentL s r' c @ sB e Zd ZdZddd�eeed�dd�Zdd � Zd
d � Zd d
� Z dS )� RetryLockah
Context manager for gaining exclusive access to the lock file.
Create a lock file if absent. The lock file will contain a pid of the
running process, and a customer-visible description of the lock holder.
The RetryLock will try several times to acquire the lock before giving up.
The number of times to try and how long to sleep in between tries is
configurable.
:param lock_holder: String with the service name or command which is
holding the lock. This lock_holder string will be customer visible in
status.json.
:param sleep_time: Number of seconds to sleep before retrying if the lock
is already held.
:param max_retries: Maximum number of times to try to grab the lock before
giving up and raising a LockHeldError.
:raises: LockHeldError if lock is held after (sleep_time * max_retries)
�
� )�
sleep_time� max_retries�r r+ r, c G s || _ || _|| _d S r r- )r r r+ r, Z_argsr r r r e s zRetryLock.__init__c C sV t � \}}|dkr$tj| j||d��t�ttt� � �| jd�� t
j t j
| jd� d S )Nr )Z lock_requestr �pidr )Z operation)r% r �
LockHeldErrorr r �writer r �os�getpidr �addr
�OPERATION_IN_PROGRESS)r r Zcur_lock_holderr r r � grab_lockp s
���zRetryLock.grab_lockc
C s� t �d| j� d}z| �� W q�W q tjk
r~ } z<t �d|d |j� |d7 }|| jkrb|�n t� | j
� W 5 d }~X Y qX qd S )Nzspin lock starting for %sr z%RetryLock Attempt %d. %s. Spinning...� ) r# r$ r r5 r r/ �msgr, �time�sleepr+ )r Ztries�er r r � __enter__� s �
zRetryLock.__enter__c C s t �� t�tj� d S r )r r& r �remover
r4 )r Z _exc_typeZ
_exc_valueZ
_tracebackr r r �__exit__� s zRetryLock.__exit__N)
r r r �__doc__r r" r r5 r; r= r r r r r( P s �� r( )"Zloggingr1 r8 �typingr Zuaclientr r r Zuaclient.data_typesr r r Zuaclient.filesr Zuaclient.files.data_typesr
r Zuaclient.files.filesr Zuaclient.files.noticesr
Z getLoggerZ replace_top_level_logger_namer r# r ZJSONr r" r r% r' r( r r r r �<module> s&
�*