メモ：Doctrine2でmysqlのYEAR,Monthを使えるようにするためのExtension

Month.php

<?php

namespace Nanaweb\Doctrine\DateFunctionExtension;


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

/**
 * MonthFunction ::=
 *     "MONTH" "(" ArithmeticPrimary ")"
 */
class Month extends FunctionNode
{
    public $dateExpression = null;

    public function parse(Parser $parser)
    {
        $parser->match(Lexer::T_IDENTIFIER);
        $parser->match(Lexer::T_OPEN_PARENTHESIS);

        $this->dateExpression = $parser->ArithmeticPrimary();

        $parser->match(Lexer::T_CLOSE_PARENTHESIS);
    }

    public function getSql(SqlWalker $sqlWalker)
    {
        return 'MONTH('.$this->dateExpression->dispatch($sqlWalker).')';
    }
}

Year.php

<?php

namespace Nanaweb\Doctrine\DateFunctionExtension;


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

/**
 * MonthFunction ::=
 *     "YEAR" "(" ArithmeticPrimary ")"
 */
class Year extends FunctionNode
{
    public $dateExpression = null;

    public function parse(Parser $parser)
    {
        $parser->match(Lexer::T_IDENTIFIER);
        $parser->match(Lexer::T_OPEN_PARENTHESIS);

        $this->dateExpression = $parser->ArithmeticPrimary();

        $parser->match(Lexer::T_CLOSE_PARENTHESIS);
    }

    public function getSql(SqlWalker $sqlWalker)
    {
        return 'YEAR('.$this->dateExpression->dispatch($sqlWalker).')';
    }
}