U lHJe@@sddlZddlZddlZddlZddlmZmZmZddlm Z ddl m Z e e eZdaddZGdddZGd d d eZdS) N)config exceptionsutil)notices)NoticecCstr tdSN)clear_lock_filer r //usr/lib/python3/dist-packages/uaclient/lock.pyclear_lock_file_if_presentsr c@s2eZdZdZejedddZddZddZ d S) SingleAttemptLocka 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. :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. :raises: LockHeldError if lock is held. cfg lock_holdercGs||_||_dSrr )selfrr_argsr r r __init__$szSingleAttemptLock.__init__cCsj|j\}}|dkr(tj|j||d|jddt|jt j t j |jdt |jjdadS)Nr)Z lock_requestrpidlockz{}:{})Z operation)rZcheck_lock_infor LockHeldErrorrZ write_cacheformatosgetpidraddrZOPERATION_IN_PROGRESS functoolspartialdelete_cache_keyr)rZlock_pidZcur_lock_holderr r r __enter__(s zSingleAttemptLock.__enter__cCs|jddadS)Nr)rrr)rZ _exc_typeZ _exc_valueZ _tracebackr r r __exit__:s zSingleAttemptLock.__exit__N) __name__ __module__ __qualname____doc__rUAConfigstrrrrr r r r r s r csBeZdZdZdddejeeedfddZfdd Z Z S) SpinLocka Context manager for gaining exclusive access to the lock file. In contrast to the SingleAttemptLock, the SpinLock 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)rrr(r)cs tj||d||_||_dS)Nr )superrr(r))rrrr(r)r __class__r r rQszSpinLock.__init__c std|jd}ztWqWqtjk r}zs  (