如何高效查询Django中反向关联的QuerySet数据？

我有两个通过ForeignKey建立一对多关联的模型：Car和Passenger。通过Car的search方法（基于django.contrib.postgres.search实现全文搜索）获取到符合条件的Car QuerySet后，希望高效获取所有关联的Passenger对象。

尝试直接调用cars.passenger_set.all()但QuerySet不支持该操作，目前采用的方式是先提取Car的id列表，再用__in过滤Passenger：

car_ids = Car.search('truck').values_list('id', flat=True)
passengers = Passenger.objects.filter(car_id__in=car_ids)

但当cars QuerySet包含数百条记录时，__in会生成WHERE id IN (...)语句而非SQL JOIN，在百万级数据场景下查询耗时超10秒，性能极差，求更优实现方案。

附Car.search对应的全文搜索实现代码：

from django.contrib.postgres.search import (
    SearchQuery,
    SearchRank,
    SearchVector,
)

RANK_THRESHOLD = 0.1


def keyword_search_cars(Model, query):
    vector = (
        SearchVector('name', weight='A')
        + SearchVector('description', weight='B')
    )
    query = SearchQuery(query)
    cars_results = (
        Model.objects.annotate(rank=SearchRank(vector, query))
        .filter(rank__gte=RANK_THRESHOLD)
        .order_by('-rank')
    )
    return cars_results

1. 子查询实现JOIN查询 #

无需提前提取ID列表，直接将Car的搜索QuerySet作为过滤条件传入Passenger查询，Django会自动生成JOIN语句，替代低效的IN (...)：

# 获取匹配的Car QuerySet
matched_cars = keyword_search_cars(Car, 'truck')
# 反向关联查询Passenger
passengers = Passenger.objects.filter(car__in=matched_cars)

或者显式使用Subquery控制逻辑：

from django.db.models import Subquery

matched_car_ids = keyword_search_cars(Car, 'truck').values('id')
passengers = Passenger.objects.filter(
    car_id__in=Subquery(matched_car_ids)
)

两种写法最终都会生成基于JOIN的SQL，避免大ID列表导致的性能问题。

2. 批量预取关联数据（适用于需同时处理Car和Passenger的场景） #

如果需要保留Car对象并获取其关联的Passenger，使用prefetch_related批量预取，避免N+1查询：

matched_cars = keyword_search_cars(Car, 'truck').prefetch_related('passenger_set')
# 遍历Car时直接访问passenger_set，无额外数据库请求
for car in matched_cars:
    car_passengers = car.passenger_set.all()

3. 数据库索引优化 #

• 为Car的全文搜索字段创建GIN索引，加速搜索过程：

  from django.contrib.postgres.indexes import GinIndex
  from django.contrib.postgres.search import SearchVectorField
  
  class Car(models.Model):
      name = models.CharField(max_length=100)
      description = models.TextField()
      # 可选：提前存储搜索向量，避免每次查询计算
      search_vector = SearchVectorField(null=True)
  
      class Meta:
          indexes = [
              GinIndex(fields=['search_vector']),
          ]
  

  可以通过信号或定时任务维护search_vector字段，比如在Car保存时更新：

  from django.db.models.signals import post_save
  from django.dispatch import receiver
  
  @receiver(post_save, sender=Car)
  def update_search_vector(sender, instance, **kwargs):
      instance.search_vector = (
          SearchVector('name', weight='A') + SearchVector('description', weight='B')
      )
      Car.objects.filter(pk=instance.pk).update(search_vector=instance.search_vector)
  

• 确认Passenger的car_id字段存在外键索引（Django默认会为ForeignKey自动创建，可通过数据库工具验证）。

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