mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-10-22 08:19:04 +00:00
a7fa2331bd
* feat: 添加缓存模块，添加组织资源统计 * refactor * recover .gitkeep * refactor * 合并信号处理 * 修复组织添加用户没有发信号 * 修改了一个log级别 Co-authored-by: xinwen <coderWen@126.com>
35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
from django.db.transaction import on_commit
|
|
|
|
from common.cache import *
|
|
from .utils import current_org, tmp_to_org
|
|
from .tasks import refresh_org_cache_task
|
|
from orgs.models import Organization
|
|
|
|
|
|
class OrgRelatedCache(Cache):
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.current_org = Organization.get_instance(current_org.id)
|
|
|
|
def get_current_org(self):
|
|
"""
|
|
暴露给子类控制组织的回调
|
|
1. 在交互式环境下能控制组织
|
|
2. 在 celery 任务下能控制组织
|
|
"""
|
|
return self.current_org
|
|
|
|
def refresh(self, *fields):
|
|
with tmp_to_org(self.get_current_org()):
|
|
return super().refresh(*fields)
|
|
|
|
def refresh_async(self, *fields):
|
|
"""
|
|
在事务提交之后再发送信号，防止因事务的隔离性导致未获得最新的数据
|
|
"""
|
|
def func():
|
|
logger.info(f'CACHE: Send refresh task {self}.{fields}')
|
|
refresh_org_cache_task.delay(self, *fields)
|
|
on_commit(func)
|