首页 > 解决方案 > Doctrine2:PostgreSQL 的 YEAR、MONTH、DAY 或 DATE_FORMAT

问题描述

由于 Doctrine2 中既没有DAY(), MONTH(), YEAR(), 也没有DATE_FORMAT(),当使用 PostgreSQL 数据库时,如何在 Query Builder 中使用这些函数之一?

我找到了几个教程,但它们都适用于 MySQL,没有适用于 PostgreSQL。

标签: postgresqldoctrine-ormsymfony4

解决方案


由于数据库供应商之间的 SQL 语法不同,因此不可能(或至少不是那么容易)创建独立于供应商的解决方案。所以这是 PostgreSQL 的一种方法。

我们要使用的 SQL 函数是to_char(),请参阅https://www.postgresql.org/docs/9.6/static/functions-formatting.html

首先我们需要创建一个自定义 DQL 函数,参见https://www.doctrine-project.org/projects/doctrine-orm/en/2.6/cookbook/dql-user-defined-functions.html

namespace App\DQL;

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

class ToChar extends FunctionNode
{
    public $timestamp = null;
    public $pattern = null;

    // This tells Doctrine's Lexer how to parse the expression:
    public function parse(Parser $parser)
    {
        $parser->match(Lexer::T_IDENTIFIER);
        $parser->match(Lexer::T_OPEN_PARENTHESIS);
        $this->timestamp = $parser->ArithmeticPrimary();
        $parser->match(Lexer::T_COMMA);
        $this->pattern = $parser->ArithmeticPrimary(); // I'm not sure about `ArithmeticPrimary()` but it works. Post a comment, if you know more details!
        $parser->match(Lexer::T_CLOSE_PARENTHESIS);
    }

    // This tells Doctrine how to create SQL from the expression - namely by (basically) keeping it as is:
    public function getSql(SqlWalker $sqlWalker)
    {
        return 'to_char('.$this->timestamp->dispatch($sqlWalker) . ', ' . $this->pattern->dispatch($sqlWalker) . ')';
    }
}

然后我们在 Symfony 4 中注册它,参见https://symfony.com/doc/current/doctrine/custom_dql_functions.html

# config/packages/doctrine.yaml

doctrine:
    orm:
        dql:
            string_functions:
                to_char: App\DQL\ToChar

现在我们可以在任何存储库中使用它:

return $this->createQueryBuilder('a')
    ->select("to_char(a.timestamp, 'YYYY') AS year")
    ->groupBy('year')
    ->orderBy('year', 'ASC')
    ->getQuery()
    ->getResult()
;

推荐阅读