新功能(celery 任务): 是否保存成功的日志(适用于频率高的celery任务,成功不需要保存日志,则传False)
parent
38a2351b78
commit
d479697594
|
@ -31,7 +31,7 @@ def getIP():
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
@BaseCeleryApp(name='apps.vadmin.monitor.tasks.get_monitor_info')
|
@BaseCeleryApp(name='apps.vadmin.monitor.tasks.get_monitor_info', save_success_logs=False)
|
||||||
def get_monitor_info():
|
def get_monitor_info():
|
||||||
"""
|
"""
|
||||||
定时获取系统监控信息
|
定时获取系统监控信息
|
||||||
|
|
|
@ -27,7 +27,13 @@ def get_cache(alias=None):
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
def BaseCeleryApp(name):
|
def BaseCeleryApp(name, save_success_logs=True):
|
||||||
|
"""
|
||||||
|
celery 保存日志基础类
|
||||||
|
:param name: celery任务名字
|
||||||
|
:param save_success_logs: 是否保存成功的日志(适用于频率高的celery任务,成功不需要保存日志,则传False)
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
def wraps(func):
|
def wraps(func):
|
||||||
@app.task(name=name)
|
@app.task(name=name)
|
||||||
@functools.wraps(func)
|
@functools.wraps(func)
|
||||||
|
@ -40,6 +46,8 @@ def BaseCeleryApp(name):
|
||||||
res = None
|
res = None
|
||||||
try:
|
try:
|
||||||
res = func(*args, **kwargs)
|
res = func(*args, **kwargs)
|
||||||
|
if not save_success_logs:
|
||||||
|
return res
|
||||||
obj.result = str(res)
|
obj.result = str(res)
|
||||||
obj.status = True
|
obj.status = True
|
||||||
except Exception as exc:
|
except Exception as exc:
|
||||||
|
|
Loading…
Reference in New Issue