如何在Symfony 6.2中结合Doctrine使用PostGIS的<->距离运算符？

问题背景 #

需求为查找距离指定坐标最近的Poi卡片，计算距离并将结果赋值给Poi实体中未映射ORM的distance字段。尝试用Doctrine QueryBuilder执行DQL时，因DQL不支持PostGIS的<->距离运算符报错；使用原生SQL可得到数组结果，但无法自动映射到Poi实体，需要更优雅的实现方案。

解决方案1：NativeQuery + ResultSetMapping（推荐） #

该方案既支持PostGIS的原生运算符，又能将查询结果自动映射到Poi实体，同时完成distance字段的赋值。

步骤1：完善Poi实体 #

添加setDistance方法，用于手动赋值非映射的distance字段：

class Poi
{
    #[ORM\Column(nullable: true)]
    private ?float $latitude = null;

    #[ORM\Column(nullable: true)]
    private ?float $longitude = null;

    #[ORM\Column(type: PostGISType::GEOGRAPHY, nullable: true)]
    private string $geog;
   
    public ?float $distance = null;

    // 添加该方法
    public function setDistance(?float $distance): void
    {
        $this->distance = $distance;
    }
}

步骤2：构建NativeQuery与映射 #

use Doctrine\ORM\Query\ResultSetMapping;

// 构建结果集映射，关联实体字段与SQL查询字段
$rsm = new ResultSetMapping();
$rsm->addEntityResult(Poi::class, 'p');
$rsm->addFieldResult('p', 'id', 'id'); // 替换为你的Poi实体主键字段名
$rsm->addFieldResult('p', 'latitude', 'latitude');
$rsm->addFieldResult('p', 'longitude', 'longitude');
$rsm->addFieldResult('p', 'geog', 'geog');
// 映射计算出的distance字段为标量结果
$rsm->addScalarResult('distance', 'distance');

// 原生SQL查询，使用PostGIS的<->运算符计算距离并排序
$sql = '
    SELECT 
        p.id,
        p.latitude,
        p.longitude,
        p.geog,
        (p.geog <-> ST_Point(:lng, :lat)) AS distance
    FROM fiche p
    ORDER BY distance ASC
    LIMIT 9
';

// 创建并执行NativeQuery
$query = $this->getEntityManager()->createNativeQuery($sql, $rsm);
$query->setParameter('lng', $lng);
$query->setParameter('lat', $lat);

// 获取结果并赋值distance字段
/** @var array<int, array{Poi, float}> $results */
$results = $query->getResult();
$pois = [];
foreach ($results as $result) {
    $poi = $result[0];
    $poi->setDistance($result['distance']);
    $pois[] = $poi;
}

解决方案2：QueryBuilder结合原生SQL片段 #

如果偏好使用QueryBuilder，可嵌入原生SQL片段支持<->运算符，再手动处理实体映射与距离赋值：

$qb = $this->createQueryBuilder('p')
    // 添加距离计算为隐藏字段，避免干扰实体属性
    ->addSelect('(p.geog <-> ST_Point(:lng, :lat)) AS HIDDEN distance')
    // 按距离升序排序
    ->orderBy('distance', 'ASC')
    ->setParameter('lng', $lng)
    ->setParameter('lat', $lat)
    ->setMaxResults(9);

// 获取包含实体数据与距离的数组结果
$arrayResults = $qb->getQuery()->getArrayResult();

/** @var Poi[] $pois */
$pois = [];
foreach ($arrayResults as $item) {
    $poi = $this->find($item['id']);
    if ($poi) {
        $poi->setDistance($item['distance']);
        $pois[] = $poi;
    }
}

原DQL报错原因 #

Doctrine的DQL有独立的语法规则与运算符集，无法识别PostGIS的自定义<->距离运算符，因此直接在DQL中使用会触发语法错误，必须通过原生SQL相关方案绕过DQL的语法检查。

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