与管理器相关的 Django 重复查询

Question

我正在努力优化我的 ORM 查询。我有两个应用程序，“app1”和“app2”。一个类“app2”具有一个类 app1 的外键，如下所示:

#app1/models.py
class C1App1(WithDateAndOwner):
  def get_c2_app2(self):
    res = self.c2app2_set.all()
    if res.count() > 0:
      return res[0]
    else:
      return None

#app2/models.py
class C2App2(WithDateAndOwner):
  c1app1 = models.ForeignKey("app1.C1App1")
  is_ok = models.BooleanField(default=False)

现在我在管理页面中显示 C1App1 的所有实例的 C2App2:

#app1/admin.py
@admin.register(C1App1)
class C1App1Admin(admin.MyAdmin):
  list_display = ("get_c2_app2")
  list_select_related = ()
  list_prefetch_related = ("c2app2_set",)
  list_per_page = 10

prefetch_related 减少了这个查询:

SELECT ••• FROM `app2_c2app2` WHERE `app2_c2app2`.`c1app1_id` = 711
  Duplicated 19 times.

到:

SELECT ••• FROM `app2_c2app2` WHERE `app2_c2app2`.`c1app1_id` IN (704, 705, 706, 707, 708, 709, 710, 711, 702, 703) ORDER BY `app2_c2app2`.`id` DESC

而且很好。现在，如果我想过滤对 C2App2 的属性“is_ok”的查询:

#app1/models.py
class C1App1(WithDateAndOwner):
  def get_c2_app2(self):
    res = self.c2app2_set.filter(is_ok=False)
    if res.count() > 0:
      return res[0]
    else:
      return None

我还有这个预取查询:

SELECT ••• FROM `c2app2_set` WHERE `app2_c2app2`.`c1app1_id` IN (704, 705, 706, 707, 708, 709, 710, 711, 702, 703) ORDER BY `app2_c2app2`.`id` DESC

但是对于 C1App1 (10) 的每个显示实例都复制了那个。 :

SELECT ••• FROM `app2_c2app2` WHERE (`app2_c2app2`.`c1app1_id` = 711 AND `app2_c2app2`.`is_ok` = 1)
  Duplicated 13 times.

实际上，对于显示的 10 个 id 中的 3 个，查询也会再次重复，这会导致这 13 个重复的查询。我该怎么做才能不重复这些查询？似乎 prefetch_related 在这里不再有帮助。

Answer 1

prefetch_related 仅在您使用 .all() 时有效。如果您应用任何其他转换，如 .filter()，将进行新的数据库查询。这是因为 prefetch_related 简单地缓存了列表中的所有相关实例，所以 Django 无法对列表执行 filter()。要解决您的问题，您应该使用 Prefetch目的。您可以将 queryset 参数传递给它。因此，不要使用 list_prefetch_related，而是重写管理类中的 get_queryset 方法。

def get_queryset(*args, **kwargs):
     qs = super().get_queryset(*args, **kwargs)
     qs = qs.prefetch_related(Prefetch('c2app2_set', queryset=C2App2.objects.filter(is_ok=False)))
     return qs

和

class C1App1(WithDateAndOwner):
   def get_c2_app2(self):
      res = self.c2app2_set.all()
      if res.count() > 0:
         return res[0]
      else:
         return None