__init__.py
71 lines
| 1.7 KiB
| text/x-python
|
PythonLexer
r547 | from rhodecode.lib.pidlock import DaemonLock, LockHeld | |||
from vcs.utils.lazy import LazyProperty | ||||
from decorator import decorator | ||||
import logging | ||||
import os | ||||
import sys | ||||
import traceback | ||||
from hashlib import md5 | ||||
r558 | import socket | |||
r547 | log = logging.getLogger(__name__) | |||
class ResultWrapper(object): | ||||
def __init__(self, task): | ||||
self.task = task | ||||
r679 | ||||
r547 | @LazyProperty | |||
def result(self): | ||||
return self.task | ||||
def run_task(task, *args, **kwargs): | ||||
try: | ||||
t = task.delay(*args, **kwargs) | ||||
log.info('running task %s', t.task_id) | ||||
return t | ||||
r558 | except socket.error, e: | |||
r679 | ||||
try: | ||||
conn_failed = e.errno == 111 | ||||
except AttributeError: | ||||
conn_failed = False | ||||
if conn_failed: | ||||
r558 | log.debug('Unable to connect to celeryd. Sync execution') | |||
else: | ||||
r679 | log.debug('Unable to connect to celeryd. Sync execution') | |||
r558 | except KeyError, e: | |||
log.debug('Unable to connect to celeryd. Sync execution') | ||||
r547 | except Exception, e: | |||
r558 | log.error(traceback.format_exc()) | |||
r679 | ||||
r558 | return ResultWrapper(task(*args, **kwargs)) | |||
r547 | ||||
def locked_task(func): | ||||
def __wrapper(func, *fargs, **fkwargs): | ||||
params = list(fargs) | ||||
params.extend(['%s-%s' % ar for ar in fkwargs.items()]) | ||||
r679 | ||||
r547 | lockkey = 'task_%s' % \ | |||
md5(str(func.__name__) + '-' + \ | ||||
'-'.join(map(str, params))).hexdigest() | ||||
log.info('running task with lockkey %s', lockkey) | ||||
try: | ||||
l = DaemonLock(lockkey) | ||||
ret = func(*fargs, **fkwargs) | ||||
l.release() | ||||
return ret | ||||
except LockHeld: | ||||
log.info('LockHeld') | ||||
r679 | return 'Task with key %s already running' % lockkey | |||
r547 | ||||
r679 | return decorator(__wrapper, func) | |||
r547 | ||||
r679 | ||||