perf: 接口sql优化 /api/v1/perms/asset-permissions/<uuid:pk>/assets/all/

This commit is contained in:
fangfang.dong
2023-07-03 16:33:58 +08:00
committed by Bryan
parent dd846d4183
commit 785e4cc3e4
2 changed files with 6 additions and 6 deletions

View File

@@ -24,12 +24,9 @@ class AssetPermissionPermAssetUtil:
def get_perm_nodes_assets(self, flat=False):
""" 获取所有授权节点下的资产 """
node_ids = AssetPermission.nodes.through.objects \
.filter(assetpermission_id__in=self.perm_ids) \
.values_list('node_id', flat=True) \
.distinct()
node_ids = list(node_ids)
nodes = PermNode.objects.filter(id__in=node_ids).only('id', 'key')
from assets.models import Node
nodes = Node.objects.prefetch_related('granted_by_permissions').filter(
granted_by_permissions__in=self.perm_ids).only('id', 'key')
assets = PermNode.get_nodes_all_assets(*nodes)
if flat:
return assets.values_list('id', flat=True)