如何优化Django中基于Integer字段的关联查询性能？

问题背景 #

现有Django模型定义如下：

class Challenge(models.Model):
    name = models.CharField(...)
    version = models.IntegerField(...)

class Workout(models.Model):
    name = models.CharField(...)
    version = models.IntegerField(...) # 该字段可能不存在

# 关联中间表
class ChallengeWorkoutRelation(models.Model):
    challenge_id = models.IntegerField() # 因特殊原因使用IntegerField而非ForeignKey
    workout_id = models.IntegerField() # 因特殊原因使用IntegerField而非ForeignKey
    version = models.IntegerField(default=0)

需求是获取指定Challenge关联的所有Workout，当前实现代码如下：

for relation in ChallengeWorkoutRelation.objects.filter(challenge_id=challenge.pk, version=challenge.version):
    workout = Workout.objects.get(id=relation.workout_id)
    # 处理workout逻辑...
    
    # 另一种版本
    # workout = Workout.objects.get(id=relation.workout_id, version=challenge.version)

性能优化方案 #

当前代码存在N+1查询问题：每遍历一条关联记录就发起一次Workout查询，数据量越大性能越差，可通过以下方式优化：

方案1：批量查询（无需修改模型） #

先批量提取所有关联的workout_id，再一次性查询所有对应的Workout，将查询次数从N+1减少到2次：

# 第一步：获取所有符合条件的workout_id列表
workout_ids = ChallengeWorkoutRelation.objects.filter(
    challenge_id=challenge.pk,
    version=challenge.version
).values_list('workout_id', flat=True)

# 第二步：根据Workout是否有version字段，选择对应过滤条件
queryset = Workout.objects.filter(id__in=list(workout_ids))
if hasattr(Workout, 'version'):
    queryset = queryset.filter(version=challenge.version)

# 遍历处理所有Workout
for workout in queryset:
    # 处理workout逻辑...

方案2：调整模型为ForeignKey（推荐长期维护） #

如果业务允许修改模型，建议将中间表的challenge_id和workout_id改为ForeignKey，符合Django ORM设计规范，同时可通过prefetch_related实现预加载：

# 修改后的中间表模型
class ChallengeWorkoutRelation(models.Model):
    challenge = models.ForeignKey(Challenge, on_delete=models.CASCADE)
    workout = models.ForeignKey(Workout, on_delete=models.CASCADE)
    version = models.IntegerField(default=0)

# 查询时预加载关联的Workout
relations = ChallengeWorkoutRelation.objects.filter(
    challenge=challenge,
    version=challenge.version
).prefetch_related('workout')

for relation in relations:
    workout = relation.workout
    # 处理workout逻辑...

优化说明 #

• 方案1无需改动现有模型，快速解决N+1性能问题，适合无法修改模型的场景；
• 方案2更符合Django最佳实践，代码可读性、维护性更强，还能利用ORM的其他关联特性。

内容的提问来源于stack exchange，提问作者Tony.Luo