github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-09-17 15:59:04 +00:00
Code Issues Projects Releases Wiki Activity

perf: 优化 delay_run 执行

Browse Source
This commit is contained in:
Eric
2024-01-17 19:31:01 +08:00
committed by 老广
parent dc79346bdc
commit f74f8b7d8c
2 changed files with 17 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
apps/common/decorators.py
View File

@@ -101,7 +101,10 @@ def run_debouncer_func(cache_key, org, ttl, func, *args, **kwargs):
first_run_time = current
if current - first_run_time > ttl:
_loop_debouncer_func_args_cache.pop(cache_key, None)
_loop_debouncer_func_task_time_cache.pop(cache_key, None)
executor.submit(run_func_partial, *args, **kwargs)
logger.debug('executor submit run {}'.format(func.__name__,))
return
loop = _loop_thread.get_loop()
@@ -136,10 +139,12 @@ class Debouncer(object):
def _run_func_with_org(key, org, func, *args, **kwargs):
from orgs.utils import set_current_org
try:
set_current_org(org)
func(*args, **kwargs)
with transaction.atomic():
set_current_org(org)
func(*args, **kwargs)
except Exception as e:
logger.error('delay run error: %s' % e)
logger.error('thread {} delay run {} error: {}'.format(
threading.current_thread(), func.__name__, e))
_loop_debouncer_func_task_cache.pop(key, None)
_loop_debouncer_func_args_cache.pop(key, None)
_loop_debouncer_func_task_time_cache.pop(key, None)