Doctrine嵌套实体Eager Loading问题：setFetchMode无法多级加载？

问题背景 #

现有实体结构：

class A {
    #[OneToMany(mappedBy: 'source', targetEntity: B::class)]
    private Collection $bCollection;
}

class B {
    private A $source;

    #[ManyToOne(targetEntity: C::class)]
    private C $c;
}

class C {
}

使用Query::setFetchMode()尝试加载A->B->C全量实体时，仅能正确加载A的bCollection，但B的$c始终是未初始化的代理对象，且要求不能使用JOIN或partial语法（后者将在Doctrine 3中被禁用）。

解决方案：基于批量加载的多级水合实现 #

核心思路是分阶段批量加载关联实体，避免N+1查询的同时完成嵌套关联的初始化，具体步骤：

• 加载根实体及直接关联集合
  首先查询A实体并加载其关联的B集合：

  use Doctrine\ORM\Mapping\ClassMetadataInfo;
  
  // 查询A实体并加载B集合
  $aEntities = $em->createQueryBuilder()
      ->select('entity')
      ->from(A::class, 'entity')
      ->getQuery()
      ->setFetchMode(A::class, 'bCollection', ClassMetadataInfo::FETCH_EAGER)
      ->getResult();
  

• 收集嵌套关联的外键ID
  从已加载的B集合中提取所有C的外键ID，过滤重复ID以减少查询量：

  $cIds = [];
  /** @var A $a */
  foreach ($aEntities as $a) {
      /** @var B $b */
      foreach ($a->getBCollection() as $b) {
          $cId = $b->getCId(); // 假设B类有获取C外键的方法，或直接通过代理对象获取ID
          if ($cId && !in_array($cId, $cIds)) {
              $cIds[] = $cId;
          }
      }
  }
  

• 批量加载嵌套关联实体
  批量查询所有需要的C实体，并将其映射到对应的B实例：

  if (!empty($cIds)) {
      /** @var C[] $cEntities */
      $cEntities = $em->getRepository(C::class)->findBy(['id' => $cIds]);
      // 将C实体按ID分组，方便快速查找
      $cMap = [];
      foreach ($cEntities as $c) {
          $cMap[$c->getId()] = $c;
      }
  
      // 关联C到对应的B实例
      foreach ($aEntities as $a) {
          /** @var B $b */
          foreach ($a->getBCollection() as $b) {
              $cId = $b->getCId();
              if (isset($cMap[$cId])) {
                  // 假设B类有setC方法
                  $b->setC($cMap[$cId]);
                  // 初始化代理对象（如果需要）
                  $em->getUnitOfWork()->initializeObject($b->getC());
              }
          }
      }
  }
  

关键说明 #

• 这种方式完全避免了JOIN和partial语法，符合要求。
• 通过批量查询将N+1问题转化为2次额外查询，性能更优。
• 需确保实体类提供获取外键ID和设置关联实体的方法，如getCId()、setC()。
• 若需要通用实现，可封装成自定义Repository或Hydrator，递归处理任意层级的关联加载。

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