perf: 兼用处理一下 tree

This commit is contained in:
ibuler
2024-01-10 13:37:55 +08:00
committed by Bryan
parent df655f304a
commit 2b15fc5e8b
2 changed files with 14 additions and 7 deletions

View File

@@ -196,10 +196,11 @@ class AssetPermissionListSerializer(AssetPermissionSerializer):
@classmethod
def setup_eager_loading(cls, queryset):
"""Perform necessary eager loading of data."""
queryset = queryset.annotate(
users_amount=Count("users"),
user_groups_amount=Count("user_groups"),
assets_amount=Count("assets"),
nodes_amount=Count("nodes"),
)
queryset = queryset \
.prefetch_related('labels', 'labels__label') \
.annotate(users_amount=Count("users"),
user_groups_amount=Count("user_groups"),
assets_amount=Count("assets"),
nodes_amount=Count("nodes"),
)
return queryset