php - 在 Doctrine 2 中使用类表继承时 : how can one write a Native SQL Query that will return results from a child class?

Question

在类表继承层次结构上执行 native SQL 查询是 mentioned in the examples in the docs但我会在这里制作相关的例子:

<?php

use Doctrine\ORM\Query\ResultSetMapping;

// Equivalent DQL query: "select u from User u where u.name=?1"
// User is a mapped base class for other classes. User owns no associations.
$rsm = new ResultSetMapping;
$rsm->addEntityResult('User', 'u');
$rsm->addFieldResult('u', 'id', 'id');
$rsm->addFieldResult('u', 'name', 'name');
$rsm->addMetaResult('u', 'discr', 'discr'); // discriminator column
$rsm->setDiscriminatorColumn('u', 'discr');

$query = $this->_em->createNativeQuery('SELECT id, name, discr FROM users WHERE name = ?', $rsm);
$query->setParameter(1, 'romanb');

$users = $query->getResult();

来自示例:

Note that in the case of Class Table Inheritance, an example as above would result in partial objects if any objects in the result are actually a subtype of User. When using DQL, Doctrine automatically includes the necessary joins for this mapping strategy but with native SQL it is your responsibility.

上述查询的问题在于，虽然它会从父类 User 返回字段，但如果您希望从子类返回字段，那么上面的示例不会返回它们。我们如何做到这一点？

在继续之前，我已经花时间创建了一些示例类(一个是父类，另一个是子类)，它们应该适用于上述示例查询。让我为您提供它们:

/**
 * @ORM\Table(name="users",
 * @ORM\Entity
 * @ORM\InheritanceType("JOINED")
 * @ORM\DiscriminatorColumn(name="discr", type="string")
 * @ORM\DiscriminatorMap({
 *  "regular_user" = "RegularUser",
 *  "administrator" = "Administrator",
 * })
 */
class User
{
    /**
     * @ORM\Id
     * @ORM\GeneratedValue(strategy="AUTO")
     */
    protected $id;

    /**
     * @ORM\Column(name="name", type="string")
     */
    protected $name;
}

和:

/**
 * @ORM\Table(name="regular_users",
 * @ORM\Entity
 */
class RegularUser extends User
{
    /**
     * @ORM\Column(name="phone_number", type="string")
     */
    protected $phoneNumber;
}

所以我们有一个父类 User 和两个子类 RegularUser 和 Administrator (注意:我只提供了一个子类 RegularUser 但我确实为子类 Administrator 提供了鉴别器映射。

考虑一下:我们不想在数据库中查询名称为“romanb”的用户(如上面的示例)，而是想使用像 Solr 这样的全文搜索引擎来索引数据库。我们在 Solr 中查询 romanb，它会返回一个用户 ID 数组，这些用户的名称如 romanb 按相关性 排序。有了这些有序的 id，我们想查询数据库并返回一个常规用户的列表，这些用户完全水合(即包括他们的电话号码)并且按照与 Solr 返回的 id 相同的顺序排序。我们如何做到这一点？

Doctrine 不支持自定义排序依据，因为并非所有 SQL 引擎都支持它。 MySQL 支持它，它恰好是我们正在使用的数据库。

考虑以下 native SQL 查询:

$rsm = new ResultSetMapping;
$rsm->addEntityResult('User', 'u');
$rsm->addFieldResult('u', 'id', 'id');
$rsm->addFieldResult('u', 'name', 'name');
$rsm->addMetaResult('u', 'discr', 'discr');
$rsm->setDiscriminatorColumn('u', 'discr');
$rsm->addFieldResult('u', 'phone_number', 'phoneNumber'); // How do we map this result?

// Returned by Solr
$userIds = array(22, 3, 88, 109, 12);

$sql = <<<SQL
  SELECT u.id, u.name, u.discr, r.phone_number
  FROM users u
  INNER JOIN regular_users r ON users.id = regular_users.id
  WHERE u.id IN (?)
  ORDER BY FIELD(u.id, ?); # Custom SQL logic that will return an ordered set
SQL;

$query = $this->_em->createNativeQuery($sql, $rsm);
$query->setParameter(1, $userIds);
$query->setParameter(2, $userIds);

$users = $query->getResult();

虽然 SQL 没问题，但上面的查询将失败并出现错误。如果我们注释掉 $rsm->addFieldResult('u', 'phone_number', 'phoneNumber') 它将起作用并且 RegularUser 对象将在结果中返回(因为所有 ID 都属于普通用户并且我们在结果集映射中定义了鉴别器列)但是这些 RegularUser 对象都不包含电话号码。

是否有任何查询可以返回完全水合的 RegularUser 对象？

感谢阅读。

Answer 1

有时候，在 stackoverflow 上写下问题的好处在于，它可以帮助您更好地思考问题，并帮助您找到解决方案。

这是针对上述内容的有效 native SQL 查询:

<?php

$rsm = new ResultSetMapping;
$rsm->addEntityResult('RegularUser', 'r');
$rsm->addFieldResult('r', 'id', 'id');
$rsm->addFieldResult('r', 'name', 'name');
$rsm->addFieldResult('r', 'phone_number', 'phoneNumber');

$userIds = array(22, 3, 88, 109, 12);

$sql = <<<SQL
  SELECT r.id, u.name, r.phone_number
  FROM regular_users r
  INNER JOIN users u ON r.id = u.id
  WHERE u.id IN (?)
  ORDER BY FIELD(u.id, ?); # Custom SQL logic that will return an ordered set
SQL;

$query = $this->_em->createNativeQuery($sql, $rsm);
$query = $this->_em->createNativeQuery($sql, $rsm);
$query->setParameter(1, $userIds);
$query->setParameter(2, $userIds);

$users = $query->getResult();

假设您知道您要查询的 ID 属于 RegularUser 类，基本上没有必要涉及鉴别器映射。 p>

我希望这对其他人有帮助(我知道的极端情况)。