Django一对多关联中分组统计未包含无帖子用户的问题解决

问题背景 #

现有Django User和Post一对多关联模型：

class User(models.Model):
    first_name = models.CharField("First name", max_length=150)
    last_name = models.CharField("Last name", max_length=150)


class Post(models.Model):
    created = models.DateTimeField(auto_now_add=True)
    user = models.ForeignKey(User, on_delete=models.CASCADE, related_name="posts")
    content = models.CharField(max_length=300)

示例数据：

User表

id	first_name	last_name
1	Jon	Skeet
2	Gordon	Linoff
3	Another	Soul

Post表

user	content
1	Jon #1
1	Jon #2
1	Jon #3
2	Gordon #1
2	Gordon #2

需要开发API返回所有用户的两个信息：

• 帖子数量
• 最近3篇帖子

现有实现存在两个问题：

• 使用.filter(posts__isnull=False)过滤后，无帖子的用户不会出现在结果中；移除该过滤条件后，所有用户的posts_count均为1
• 通过循环逐个查询用户的帖子，存在冗余查询，性能低下

解决方案 #

1. 修正帖子计数逻辑 #

之前使用Count("id")统计的是用户自身的数量，正确做法是统计关联的posts数量，即Count("posts")。该方法会自动将无帖子用户的计数设为0，无需额外过滤。

2. 用子查询避免循环查询 #

利用Django的Subquery和OuterRef，在一次主查询中获取每个用户的最近3篇帖子，彻底解决N+1查询问题。

3. 调整序列化器结构 #

让recent_posts字段直接关联PostsSerializer，无需手动处理序列化逻辑。

完整代码实现 #

序列化器代码

from rest_framework import serializers
from .models import User, Post

class PostsSerializer(serializers.ModelSerializer):
    class Meta:
        model = Post
        fields = ("content", "created")

class UserSerializer(serializers.ModelSerializer):
    posts_count = serializers.IntegerField(read_only=True)
    recent_posts = PostsSerializer(many=True, read_only=True)

    class Meta:
        model = User
        fields = ("first_name", "last_name", "posts_count", "recent_posts")

视图函数代码

from django.db.models import Count, Subquery, OuterRef
from rest_framework.decorators import api_view
from rest_framework.response import Response
from .models import User, Post
from .serializers import UserSerializer

@api_view(["GET"])
def get_user(request):
    # 定义子查询：获取当前用户的最近3篇帖子
    recent_posts_subquery = Post.objects.filter(
        user=OuterRef('pk')
    ).order_by("-created")[:3]

    # 查询所有用户，同时标注帖子数量和最近3篇帖子
    users = User.objects.annotate(
        posts_count=Count("posts"),
        recent_posts=Subquery(recent_posts_subquery.values('content', 'created'))
    )

    serializer = UserSerializer(users, many=True)
    return Response(serializer.data)

效果说明 #

• 所有用户（包括无帖子的用户）都会被返回
• posts_count正确显示每个用户的帖子数量，无帖子用户为0
• recent_posts返回每个用户最近3篇帖子（按创建时间倒序），无帖子用户返回空数组
• 仅执行2次数据库查询（一次查询用户，一次子查询帖子），性能大幅提升

内容的提问来源于stack exchange，提问作者Amir Shabani