github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-12-24 13:02:37 +00:00
Code Issues Projects Releases Wiki Activity

perf: AssetTree if with_assets pre fetch asset attrs

Browse Source
This commit is contained in:
Bai
2025-12-22 16:51:08 +08:00
parent 22a27946a7
commit a88e35156a
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
apps/assets/tree/asset_tree.py
View File

@@ -51,7 +51,7 @@ class AssetTree(Tree):
self._category_platform_ids = set()
# 节点下是否包含资产
self._with_assets = with_assets
self._node_assets_mapper = defaultdict(set)
self._node_assets_mapper = defaultdict(dict)
# 是否构建完整树，包含所有节点，否则只包含有资产的节点
self._full_tree = full_tree
@@ -107,11 +107,16 @@ class AssetTree(Tree):
@timeit
def _load_nodes_assets_and_count(self):
q = self._make_assets_q_object()
node_asset_pairs = Asset.objects.filter(q).values_list('node_id', 'id')
for nid, aid in list(node_asset_pairs):
nid = str(nid)
aid = str(aid)
self._node_assets_mapper[nid].add(aid)
assets = Asset.objects.filter(q).values(
'node_id', 'id', 'platform_id', 'name', 'address', 'is_active', 'comment', 'org_id'
)
for asset in list(assets):
asset['id'] = str(asset['id'])
asset['platform_id'] = str(asset['platform_id'])
asset['node_id'] = str(asset['node_id'])
nid = str(asset['node_id'])
aid = str(asset['id'])
self._node_assets_mapper[nid][aid] = asset
for nid, assets in self._node_assets_mapper.items():
self._nodes_assets_count_mapper[nid] = len(assets)