在Django REST框架中优化数据库查询

在Django REST框架中优化数据库查询,第1张

在Django REST框架中优化数据库查询

Django REST
framework无法像Django本身一样为您自动优化查询。您可以在许多地方找到技巧,包括Django文档。它已经提到的是Django的REST框架应该自动,虽然有与之相关的一些挑战。

这个问题是非常特定于您的情况的,您正在使用一个自定义项

SerializerMethodField
,该自定义项要求返回的每个对象。由于您正在使用
Friends.objects
管理器发出新请求,因此优化查询非常困难。

但是,您可以通过不创建新的查询集,而从其他位置获取好友计数来使问题更好。这将需要在

Friendship
模型上创建向后关系,很可能是通过
related_name
字段上的参数,因此您可以预取所有
Friendship
对象。但这仅在需要完整对象而不仅仅是对象数量时才有用。

这将导致视图和序列化器类似于以下内容:

class Friendship(models.Model):    from_friend = models.ForeignKey(User, related_name="friends")    to_friend = models.ForeignKey(User)class GetAllUsers(generics.ListAPIView):    ...    def get_queryset(self):        return User.objects.all().prefetch_related("friends")class GetAllUsersSerializer(serializers.ModelSerializer):    ...    def get_is_friend_already(self, obj):        request = self.context.get('request', None)        friends = set(friend.from_friend_id for friend in obj.friends)        if request.user != obj and request.user.id in friends: return True        else: return False

如果只需要计数对象(类似于使用

queryset.count()
queryset.exists()
),则可以在查询集中对行添加反向关系计数。这可以通过在您的
get_queryset
方法中添加
.annotate(friends_count=Count("friends"))
到末尾(如果为
related_name
was
friends
)来完成,这会将
friends_count
每个对象的属性设置为好友数。

这将导致视图和序列化器类似于以下内容:

class Friendship(models.Model):    from_friend = models.ForeignKey(User, related_name="friends")    to_friend = models.ForeignKey(User)class GetAllUsers(generics.ListAPIView):    ...    def get_queryset(self):        from django.db.models import Count        return User.objects.all().annotate(friends_count=Count("friends"))class GetAllUsersSerializer(serializers.ModelSerializer):    ...    def get_is_friend_already(self, obj):        request = self.context.get('request', None)        if request.user != obj and obj.friends_count > 0: return True        else: return False

这两种解决方案都将避免N + 1个查询,但是您选择的查询取决于您要实现的目标。



欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5668146.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-17
下一篇 2022-12-16

发表评论

登录后才能评论

评论列表(0条)

保存