perf: 优化授权资产 api，很多资产也不怕

apps/perms/utils/user_perm.py

@@ -24,7 +24,6 @@ class AssetPermissionPermAssetUtil:
         self.perm_ids = perm_ids
 
     def get_all_assets(self):
-        """ 获取所有授权的资产 """
         node_assets = self.get_perm_nodes_assets()
         direct_assets = self.get_direct_assets()
         # 比原来的查到所有 asset id 再搜索块很多，因为当资产量大的时候，搜索会很慢
@@ -34,10 +33,11 @@ class AssetPermissionPermAssetUtil:
     def get_perm_nodes_assets(self, flat=False):
         """ 获取所有授权节点下的资产 """
         from assets.models import Node
-        nodes = Node.objects \
-            .prefetch_related('granted_by_permissions') \
-            .filter(granted_by_permissions__in=self.perm_ids) \
-            .only('id', 'key')
+        from ..models import AssetPermission
+        nodes_ids = AssetPermission.objects \
+            .filter(id__in=self.perm_ids) \
+            .values_list('nodes', flat=True)
+        nodes = Node.objects.filter(id__in=nodes_ids).only('id', 'key')
         assets = PermNode.get_nodes_all_assets(*nodes)
         if flat:
             return set(assets.values_list('id', flat=True))
@@ -46,9 +46,11 @@ class AssetPermissionPermAssetUtil:
     @timeit
     def get_direct_assets(self, flat=False):
         """ 获取直接授权的资产 """
-        assets = Asset.objects.order_by() \
-            .filter(granted_by_permissions__id__in=self.perm_ids) \
-            .distinct()
+        from ..models import AssetPermission
+        asset_ids = AssetPermission.objects \
+            .filter(id__in=self.perm_ids) \
+            .values_list('assets', flat=True)
+        assets = Asset.objects.filter(id__in=asset_ids).distinct()
         if flat:
             return set(assets.values_list('id', flat=True))
         return assets
@@ -152,6 +154,7 @@ class UserPermAssetUtil(AssetPermissionPermAssetUtil):
         assets = assets.filter(nodes__id=node.id).order_by().distinct()
         return assets
 
+    @timeit
     def _get_indirect_perm_node_all_assets(self, node):
         """  获取间接授权节点下的所有资产
         此算法依据 `UserAssetGrantedTreeNodeRelation` 的数据查询