如何在Doctrine DQL中添加原生SQL？适配复杂计算字段需求

我懂你现在的困境——要让Doctrine DQL生成和遗留系统完全一模一样的复杂计算字段，尤其是这种嵌套了好几层IF的表达式，确实头疼。不过别慌，这里有几个实用的解决方案帮你搞定：

方案1：直接嵌入原生SQL片段（使用RAW()函数） #

Doctrine允许在DQL中用RAW()包裹原生SQL代码片段，这样你可以直接把遗留系统里的复杂计算逻辑原封不动地放进去，不需要做任何转换。

举个例子，在QueryBuilder里这么写：

$qb->addSelect('RAW(
    IF( wo.date_approved = 0, 0, 
        IF( ship.date_shipped > wo.date_approved, 
            ROUND( 
                IF( (ship.date_shipped - wo.date_approved) > wo.time_inactive, 
                    (ship.date_shipped - wo.date_approved) - wo.time_inactive, 
                    ship.date_shipped - wo.date_approved 
                ) / 86400, 2
            ), 
            0 
        ) 
    ) AS T
)');

⚠️ 注意事项：

• 确保DQL里使用的实体别名（wo、ship）和原生片段里的完全一致，避免别名冲突
• 这种方式依赖你当前使用的数据库方言（比如MySQL），如果后续切换数据库可能需要调整，但既然是适配遗留系统，数据库应该是固定的，所以问题不大

方案2：自定义DQL函数（优雅的长期复用方案） #

如果这个复杂计算逻辑会在多个查询里用到，不如把它封装成一个自定义DQL函数，这样代码更整洁，也方便后续维护。

第一步：创建自定义函数类 #

use Doctrine\ORM\Query\AST\Functions\FunctionNode;
use Doctrine\ORM\Query\Lexer;
use Doctrine\ORM\Query\Parser;
use Doctrine\ORM\Query\SqlWalker;

class CalculateTValue extends FunctionNode
{
    public $woDateApproved;
    public $shipDateShipped;
    public $woTimeInactive;

    public function parse(Parser $parser)
    {
        $parser->match(Lexer::T_IDENTIFIER);
        $parser->match(Lexer::T_OPEN_PARENTHESIS);
        $this->woDateApproved = $parser->ArithmeticPrimary();
        $parser->match(Lexer::T_COMMA);
        $this->shipDateShipped = $parser->ArithmeticPrimary();
        $parser->match(Lexer::T_COMMA);
        $this->woTimeInactive = $parser->ArithmeticPrimary();
        $parser->match(Lexer::T_CLOSE_PARENTHESIS);
    }

    public function getSql(SqlWalker $sqlWalker)
    {
        return sprintf(
            'IF( %s = 0, 0, IF( %s > %s, ROUND( IF( (%s - %s) > %s, (%s - %s) - %s, %s - %s ) / 86400, 2), 0 ) )',
            $this->woDateApproved->dispatch($sqlWalker),
            $this->shipDateShipped->dispatch($sqlWalker),
            $this->woDateApproved->dispatch($sqlWalker),
            $this->shipDateShipped->dispatch($sqlWalker),
            $this->woDateApproved->dispatch($sqlWalker),
            $this->woTimeInactive->dispatch($sqlWalker),
            $this->shipDateShipped->dispatch($sqlWalker),
            $this->woDateApproved->dispatch($sqlWalker),
            $this->woTimeInactive->dispatch($sqlWalker),
            $this->shipDateShipped->dispatch($sqlWalker),
            $this->woDateApproved->dispatch($sqlWalker)
        );
    }
}

第二步：注册自定义函数 #

在Doctrine的配置文件里（比如Symfony项目的config/packages/doctrine.yaml）添加这个函数的注册：

doctrine:
    orm:
        dql:
            string_functions:
                CalculateTValue: App\Doctrine\DQL\CalculateTValue

第三步：在DQL中使用 #

之后你就可以像调用普通DQL函数一样使用它了：

$qb->addSelect('CalculateTValue(wo.date_approved, ship.date_shipped, wo.time_inactive) AS T');

这种方式的好处是代码复用性高，而且更符合DQL的语法风格，后续修改计算逻辑只需要改一次函数类就行。

方案3：直接使用原生SQL查询 #

如果你的整个查询逻辑都高度依赖遗留系统的原生SQL，也可以直接用Doctrine的原生查询功能，这样能100%保证SQL和遗留系统一致。

示例代码：

use Doctrine\ORM\Query\ResultSetMappingBuilder;

// 原封不动复制遗留系统的SQL
$sql = 'SELECT 
            wo.*, ship.*,
            IF( wo.date_approved = 0, 0, 
                IF( ship.date_shipped > wo.date_approved, 
                    ROUND( 
                        IF( (ship.date_shipped - wo.date_approved) > wo.time_inactive, 
                            (ship.date_shipped - wo.date_approved) - wo.time_inactive, 
                            ship.date_shipped - wo.date_approved 
                        ) / 86400, 2
                    ), 
                    0 
                ) 
            ) AS T
        FROM work_order wo
        JOIN shipment ship ON wo.id = ship.work_order_id
        -- 这里可以添加原有的WHERE/ORDER BY等条件
        ';

// 创建结果集映射，把查询结果映射到你的实体类
$rsm = new ResultSetMappingBuilder($entityManager);
$rsm->addRootEntityFromClassMetadata(WorkOrder::class, 'wo');
$rsm->addJoinedEntityFromClassMetadata(Shipment::class, 'ship', 'wo', 'shipment'); // 假设WorkOrder有一个shipment关联
$rsm->addScalarResult('T', 'T'); // 映射计算出来的T字段

$query = $entityManager->createNativeQuery($sql, $rsm);
$results = $query->getResult();

这种方式适合完全照搬遗留SQL的场景，不需要将SQL转换为DQL，但需要手动配置结果集映射，稍微繁琐一点，但能确保输出和原系统完全一致。

内容的提问来源于stack exchange，提问作者Alex.Barylski