Django APIView如何先返回旧通知对象再更新未读状态？

方案一：提前加载数据到内存，规避懒加载影响 #

Django的QuerySet是懒加载机制，只有实际访问数据时才会触发数据库查询。你可以先把需要返回的通知数据提前加载到内存，再执行更新操作，这样返回的就是更新前的原始数据：

class GetNotifications(ListAPIView):
    serializer_class = NotificationSerializer

    def get_queryset(self):
        # 转为列表触发数据库查询，将原始数据加载到内存
        notifications = list(Notification.objects.select_related().filter(user=self.request.user).order_by("-created_at"))
        # 更新数据库中未读通知的状态
        Notification.objects.select_for_update().filter(is_read=False, user=self.request.user).update(is_read=True)
        return notifications

此方法中，notifications列表存储的是更新前的原始对象，数据库内的未读通知会被标记为已读，完全匹配你的需求。

方案二：分阶段处理（适合大数量场景） #

如果通知数量极大，担心一次性加载到内存的性能问题，可以先获取未读通知的ID，完成更新后再查询所有通知。不过此方法返回的通知对象会是更新后的状态，仅当业务允许时适用：

class GetNotifications(ListAPIView):
    serializer_class = NotificationSerializer

    def get_queryset(self):
        # 获取当前用户未读通知的ID
        unread_ids = list(Notification.objects.filter(is_read=False, user=self.request.user).values_list('id', flat=True))
        # 更新未读状态
        if unread_ids:
            Notification.objects.select_for_update().filter(id__in=unread_ids).update(is_read=True)
        # 查询所有通知（此时已读状态已更新）
        return Notification.objects.select_related().filter(user=self.request.user).order_by("-created_at")

关键注意点 #

• select_for_update()用于避免并发场景下的竞态问题，建议保留该写法。
• 若你的NotificationSerializer需要序列化is_read字段，方案一能确保返回原始状态的字段值，更贴合需求。

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