在Django Rest Framework中如何限制关联序列化器的查询字段？

这个问题我碰到过好多次了——DRF默认处理关联字段时确实会拉取整个对象的所有列，哪怕只需要主键来生成链接，确实有点浪费性能。这里有两种靠谱的解决办法：

方法一：在视图层通过预查询优化（推荐） #

视图层是处理数据查询逻辑的最佳位置，我们可以用Prefetch配合only来精准控制关联对象的查询字段：

from django.db.models import Prefetch
from rest_framework import generics

class AlbumDetailView(generics.RetrieveAPIView):
    serializer_class = AlbumSerializer
    # 预取tracks时只获取pk字段，刚好满足生成链接的需求
    queryset = Album.objects.prefetch_related(
        Prefetch('tracks', queryset=Track.objects.only('pk'))
    )

这样DRF在查询专辑时，预取关联的track数据只会拉取pk列，避免了不必要的数据库开销。

方法二：自定义关联字段类 #

如果想把这个限制逻辑封装在序列化器里，可以自定义HyperlinkedRelatedField的子类，重写get_queryset方法来限制字段：

from rest_framework import serializers

class OptimizedHyperlinkedRelatedField(serializers.HyperlinkedRelatedField):
    def get_queryset(self):
        # 调用父类的queryset后，追加only('pk')只保留主键字段
        return super().get_queryset().only('pk')

class AlbumSerializer(serializers.ModelSerializer):
    tracks = OptimizedHyperlinkedRelatedField(
        many=True,
        read_only=True,
        view_name='track-detail'
    )
    class Meta:
        model = Album
        fields = ('album_name', 'artist', 'tracks')

这种方式会让每次通过这个序列化器查询tracks时，自动只获取pk字段，适合需要在多个视图中复用这个序列化器的场景。

补充说明 #

为什么默认会查询所有列？因为HyperlinkedRelatedField需要获取关联对象的主键来生成对应的URL，但DRF默认的查询集没有做字段限制，所以会拉取整个对象的所有字段。上面两种方法都能精准解决这个问题，个人更推荐第一种视图层优化的方式，符合关注点分离的原则。

内容的提问来源于stack exchange，提问作者Phil Gyford