35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
from __future__ import absolute_import
|
|
|
|
from functools import wraps
|
|
|
|
from . import logger
|
|
from . import Lock
|
|
from .exceptions import LockError
|
|
|
|
|
|
def simple_locking(lock_id, expiration=None):
|
|
"""
|
|
A decorator that wraps a function in a single lock getting algorithm
|
|
"""
|
|
def inner_decorator(function):
|
|
def wrapper(*args, **kwargs):
|
|
try:
|
|
# Trying to acquire lock
|
|
lock = Lock.acquire_lock(lock_id, expiration)
|
|
except LockError:
|
|
# Unable to acquire lock - non fatal
|
|
pass
|
|
else:
|
|
# Lock acquired, proceed normally, release lock afterwards
|
|
logger.debug('acquired lock: %s' % lock_id)
|
|
try:
|
|
return function(*args, **kwargs)
|
|
except:
|
|
# Re raise any exception that occured when calling wrapped
|
|
# function
|
|
raise
|
|
finally:
|
|
lock.release()
|
|
return wraps(function)(wrapper)
|
|
return inner_decorator
|