如何在Doctrine2中优化N+1查询且不破坏数据完整性

问题根源 #

你遇到的问题本质是笛卡尔积导致的Doctrine结果集还原失败：当同时LEFT JOIN两个一对多关联（签证、使馆）时，SQL会返回笛卡尔积结果（比如1个国家×3个签证×5个使馆=15条记录），Doctrine在将这些记录合并为Country实体对象时，无法正确收集所有关联的签证和使馆数据，最终只保留了部分子数据。

三种可行解决方案 #

方案1：手动批量查询关联数据（最稳妥，推荐） #

放弃一次JOIN所有关联，改为分三次查询：先拿分页后的国家列表，再批量查询所有对应国家的签证和使馆，最后手动把关联数据映射到国家对象上。总查询次数仅3次，完全解决N+1，且不会丢失数据。

代码示例：

// 1. 查询分页后的国家列表
$countries = $this->entityManager
    ->createQueryBuilder()
    ->select('c')
    ->from(Country::class, 'c')
    ->setMaxResults($condition->first)
    ->setFirstResult($condition->offset)
    ->getQuery()
    ->getResult();

// 2. 提取所有国家ID，用于批量查询关联数据
$countryIds = array_map(fn($country) => $country->getId(), $countries);

// 3. 批量查询所有对应国家的签证
$visas = $this->entityManager
    ->createQueryBuilder()
    ->select('cv')
    ->from(Visa::class, 'cv')
    ->where('cv.country IN (:countryIds)')
    ->setParameter('countryIds', $countryIds)
    ->getQuery()
    ->getResult();

// 4. 批量查询所有对应国家的使馆
$embassies = $this->entityManager
    ->createQueryBuilder()
    ->select('ce')
    ->from(Embassy::class, 'ce')
    ->where('ce.country IN (:countryIds)')
    ->setParameter('countryIds', $countryIds)
    ->getQuery()
    ->getResult();

// 5. 把签证按国家ID分组，方便映射
$visasMap = [];
foreach ($visas as $visa) {
    $key = $visa->getCountry()->getId();
    $visasMap[$key][] = $visa;
}

// 6. 使馆同理分组
$embassiesMap = [];
foreach ($embassies as $embassy) {
    $key = $embassy->getCountry()->getId();
    $embassiesMap[$key][] = $embassy;
}

// 7. 把关联数据绑定到对应的国家对象上
foreach ($countries as $country) {
    $id = $country->getId();
    if (isset($visasMap[$id])) {
        // 假设Country类有setVisas方法，或用addVisa逐个添加（更符合实体设计）
        $country->setVisas($visasMap[$id]);
    }
    if (isset($embassiesMap[$id])) {
        $country->setEmbassies($embassiesMap[$id]);
    }
}

方案2：用EXTRA_LAZY关联+批量加载（代码改动最小） #

在实体的关联注解中配置fetch="EXTRA_LAZY"和batchSize，让Doctrine自动批量加载关联数据，避免N+1。这种方式不需要改查询代码，只需要调整实体映射：

在Country实体的关联属性上添加配置：

// 签证关联
/**
 * @ORM\OneToMany(targetEntity=Visa::class, mappedBy="country", fetch="EXTRA_LAZY", batchSize=20)
 */
private $visas;

// 使馆关联
/**
 * @ORM\OneToMany(targetEntity=Embassy::class, mappedBy="country", fetch="EXTRA_LAZY", batchSize=20)
 */
private $embassies;

配置后，当你循环访问每个国家的$country->getVisas()或$country->getEmbassies()时，Doctrine会每20个国家批量查询一次关联数据，原来的200+查询会缩减为10次左右（200/20），兼顾性能和代码简洁性。

方案3：分两次使用FETCH JOIN（适合对关联数据有特殊查询条件的场景） #

如果需要对签证或使馆添加额外查询条件，可以分两次用FETCH JOIN获取关联数据：先查国家+签证的FETCH JOIN（分页），再用这些国家ID查国家+使馆的FETCH JOIN，最后合并数据。

代码示例：

// 1. 第一次查询：国家+签证（FETCH JOIN），带分页
$countries = $this->entityManager
    ->createQueryBuilder()
    ->select('c, cv')
    ->from(Country::class, 'c')
    ->leftJoin('c.visas', 'cv')
    ->setMaxResults($condition->first)
    ->setFirstResult($condition->offset)
    ->getQuery()
    ->getResult();

// 2. 提取国家ID
$countryIds = array_map(fn($country) => $country->getId(), $countries);

// 3. 第二次查询：国家+使馆（FETCH JOIN）
$countriesWithEmbassies = $this->entityManager
    ->createQueryBuilder()
    ->select('c, ce')
    ->from(Country::class, 'c')
    ->leftJoin('c.embassies', 'ce')
    ->where('c.id IN (:countryIds)')
    ->setParameter('countryIds', $countryIds)
    ->getQuery()
    ->getResult();

// 4. 把使馆数据映射到原国家对象
$embassiesMap = [];
foreach ($countriesWithEmbassies as $c) {
    $embassiesMap[$c->getId()] = $c->getEmbassies();
}

foreach ($countries as $country) {
    $id = $country->getId();
    if (isset($embassiesMap[$id])) {
        $country->setEmbassies($embassiesMap[$id]);
    }
}

注意：Doctrine不允许同时对两个集合关联使用FETCH JOIN，所以必须分开两次查询。

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