问题描述
在 DRF 端点上工作时,我遇到了在预取集上调用 prefetch_related
和 first
的问题。让我们考虑两个模型:X
和 Y
; Y
包含 X
的外键。
然后我执行以下代码:
qs = X.objects.all().prefetch_related("y_set")
for x in qs:
for y in x.y_set.all():
print(e)
一切正常,django 按预期执行了 2 次查询。
然后我执行:
for x in qs:
for y in x.y_set.all():
print(e)
first = x.y_set.first()
在这个例子中,Django 执行了 n+2 次查询(至少对我而言)。
我找到了一种解决方法:
for x in qs:
for y in x.y_set.all():
print(e)
first = y_set.all()[0] if y_set.all() else None
但这对我来说并不令人满意 - 我想检查 qs
是否不为空然后取第一个元素有点乱,我肯定更喜欢使用 first
或其他一些函数隐藏了这个逻辑。
谁能解释一下为什么 first
不使用预取缓存,或者您能给我一个如何更清楚地处理它的提示吗? (我不想添加包装器来处理这个问题,我更喜欢原生 django orm 解决方案。另外我不能只从循环中取出第一个元素 - 我简化了这个例子)
提前致谢!
解决方法
.first()
基本上使用 SQL 中的 LIMIT
子句来获取查询的第一个对象。因此,当调用 queryset.first()
时,它自然会进行单独的查询。
您进一步问,既然查询集已经存在于内存中,为什么不.first()
简单地使用该求值查询集?好吧,让我这样说:
在查询集 .annotate(...).filter(...)
等上链接方法是很常见的,我们可以这样做:
queryset = SomeModel.objects.all()
for object in queryset:
print(object)
queryset2 = list(queryset.filter(a=1))
这里我们希望 queryset2
对数据库进行不同的查询,而不是在 python 级别过滤对象,因为由于某种原因数据库本身可能有新条目,或者我们可能甚至可能会进行一些注释而不是简单地调用 .filter()
,因此我们希望这是一个单独的查询。这本质上与 .first()
不会简单地使用预取对象的原因相同。