添加 UnionQuertSet (#5578)

* 添加 UnionQuertSet

* 跑通了

* 改变了 count 这类方法的代理模式

* 使用了老广的

Co-authored-by: xinwen <coderWen@126.com>

apps/perms/api/asset/user_permission/user_permission_assets/mixin.py

@@ -34,12 +34,12 @@ class UserAllGrantedAssetsQuerysetMixin:
     pagination_class = AllGrantedAssetPagination
     user: User
 
-    def get_union_queryset(self, qs_stage: QuerySetStage):
+    def get_queryset(self):
         if getattr(self, 'swagger_fake_view', False):
             return Asset.objects.none()
-        qs_stage.prefetch_related('platform').only(*self.only_fields)
         queryset = UserGrantedAssetsQueryUtils(self.user) \
-            .get_all_granted_assets(qs_stage)
+            .get_all_granted_assets()
+        queryset = queryset.prefetch_related('platform').only(*self.only_fields)
         return queryset
 
 
@@ -47,13 +47,13 @@ class UserFavoriteGrantedAssetsMixin:
     only_fields = serializers.AssetGrantedSerializer.Meta.only_fields
     user: User
 
-    def get_union_queryset(self, qs_stage: QuerySetStage):
+    def get_queryset(self):
         if getattr(self, 'swagger_fake_view', False):
             return Asset.objects.none()
         user = self.user
-        qs_stage.prefetch_related('platform').only(*self.only_fields)
         utils = UserGrantedAssetsQueryUtils(user)
-        assets = utils.get_favorite_assets(qs_stage=qs_stage)
+        assets = utils.get_favorite_assets()
+        assets = assets.prefetch_related('platform').only(*self.only_fields)
         return assets
 
 
@@ -63,58 +63,35 @@ class UserGrantedNodeAssetsMixin:
     pagination_node: Node
     user: User
 
-    def get_union_queryset(self, qs_stage: QuerySetStage):
+    def get_queryset(self):
         if getattr(self, 'swagger_fake_view', False):
             return Asset.objects.none()
         node_id = self.kwargs.get("node_id")
-        qs_stage.prefetch_related('platform').only(*self.only_fields)
+
         node, assets = UserGrantedAssetsQueryUtils(self.user).get_node_all_assets(
-            node_id, qs_stage=qs_stage
+            node_id
         )
+        assets = assets.prefetch_related('platform').only(*self.only_fields)
         self.pagination_node = node
         return assets
 
 
 # 控制格式的 ----------------------------------------------------
 
-class AssetsUnionQuerysetMixin:
-    def get_queryset_union_prefer(self):
-        if hasattr(self, 'get_union_queryset'):
-            # 为了支持 union 查询
-            queryset = Asset.objects.all().distinct()
-            queryset = self.filter_queryset(queryset)
-            qs_stage = QuerySetStage()
-            qs_stage.and_with_queryset(queryset)
-            queryset = self.get_union_queryset(qs_stage)
-        else:
-            queryset = self.filter_queryset(self.get_queryset())
-        return queryset
 
-
-class AssetsSerializerFormatMixin(AssetsUnionQuerysetMixin):
+class AssetsSerializerFormatMixin:
     serializer_class = serializers.AssetGrantedSerializer
     filterset_fields = ['hostname', 'ip', 'id', 'comment']
     search_fields = ['hostname', 'ip', 'comment']
 
-    def list(self, request, *args, **kwargs):
-        queryset = self.get_queryset_union_prefer()
 
-        page = self.paginate_queryset(queryset)
-        if page is not None:
-            serializer = self.get_serializer(page, many=True)
-            return self.get_paginated_response(serializer.data)
-
-        serializer = self.get_serializer(queryset, many=True)
-        return Response(serializer.data)
-
-
-class AssetsTreeFormatMixin(AssetsUnionQuerysetMixin, SerializeToTreeNodeMixin):
+class AssetsTreeFormatMixin(SerializeToTreeNodeMixin):
     """
     将 资产 序列化成树的结构返回
     """
 
     def list(self, request: Request, *args, **kwargs):
-        queryset = self.get_queryset_union_prefer()
+        queryset = self.filter_queryset(self.get_queryset())
 
         if request.query_params.get('search'):
             # 如果用户搜索的条件不精准，会导致返回大量的无意义数据。