support for bit comparison
This commit is contained in:
Fabio B. Silva
parent
568698e321
commit
5c89d7ffcb
5 changed files with 199 additions and 0 deletions
63
lib/Doctrine/ORM/Query/AST/Functions/BitAndFunction.php
Normal file
63
lib/Doctrine/ORM/Query/AST/Functions/BitAndFunction.php
Normal file
|
|
@ -0,0 +1,63 @@
|
||||||
|
<?php
|
||||||
|
/*
|
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||||
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||||
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||||
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||||
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||||
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||||
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||||
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||||
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||||
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
|
*
|
||||||
|
* This software consists of voluntary contributions made by many individuals
|
||||||
|
* and is licensed under the LGPL. For more information, see
|
||||||
|
* <http://www.doctrine-project.org>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
namespace Doctrine\ORM\Query\AST\Functions;
|
||||||
|
|
||||||
|
use Doctrine\ORM\Query\Lexer;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* "BIT_AND" "(" ArithmeticPrimary "," ArithmeticPrimary ")"
|
||||||
|
*
|
||||||
|
* @license http://www.opensource.org/licenses/lgpl-license.php LGPL
|
||||||
|
* @link www.doctrine-project.org
|
||||||
|
* @since 2.2
|
||||||
|
* @author Fabio B. Silva <fabio.bat.silva@gmail.com>
|
||||||
|
*/
|
||||||
|
class BitAndFunction extends FunctionNode
|
||||||
|
{
|
||||||
|
public $firstArithmetic;
|
||||||
|
public $secondArithmetic;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @override
|
||||||
|
*/
|
||||||
|
public function getSql(\Doctrine\ORM\Query\SqlWalker $sqlWalker)
|
||||||
|
{
|
||||||
|
$platform = $sqlWalker->getConnection()->getDatabasePlatform();
|
||||||
|
return $platform->getBitAndComparisonExpression(
|
||||||
|
$this->firstArithmetic->dispatch($sqlWalker),
|
||||||
|
$this->secondArithmetic->dispatch($sqlWalker)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @override
|
||||||
|
*/
|
||||||
|
public function parse(\Doctrine\ORM\Query\Parser $parser)
|
||||||
|
{
|
||||||
|
$parser->match(Lexer::T_IDENTIFIER);
|
||||||
|
$parser->match(Lexer::T_OPEN_PARENTHESIS);
|
||||||
|
|
||||||
|
$this->firstArithmetic = $parser->ArithmeticPrimary();
|
||||||
|
$parser->match(Lexer::T_COMMA);
|
||||||
|
$this->secondArithmetic = $parser->ArithmeticPrimary();
|
||||||
|
|
||||||
|
$parser->match(Lexer::T_CLOSE_PARENTHESIS);
|
||||||
|
}
|
||||||
|
}
|
||||||
63
lib/Doctrine/ORM/Query/AST/Functions/BitOrFunction.php
Normal file
63
lib/Doctrine/ORM/Query/AST/Functions/BitOrFunction.php
Normal file
|
|
@ -0,0 +1,63 @@
|
||||||
|
<?php
|
||||||
|
/*
|
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||||
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||||
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||||
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||||
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||||
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||||
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||||
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||||
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||||
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
|
*
|
||||||
|
* This software consists of voluntary contributions made by many individuals
|
||||||
|
* and is licensed under the LGPL. For more information, see
|
||||||
|
* <http://www.doctrine-project.org>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
namespace Doctrine\ORM\Query\AST\Functions;
|
||||||
|
|
||||||
|
use Doctrine\ORM\Query\Lexer;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* "BIT_OR" "(" ArithmeticPrimary "," ArithmeticPrimary ")"
|
||||||
|
*
|
||||||
|
* @license http://www.opensource.org/licenses/lgpl-license.php LGPL
|
||||||
|
* @link www.doctrine-project.org
|
||||||
|
* @since 2.2
|
||||||
|
* @author Fabio B. Silva <fabio.bat.silva@gmail.com>
|
||||||
|
*/
|
||||||
|
class BitOrFunction extends FunctionNode
|
||||||
|
{
|
||||||
|
public $firstArithmetic;
|
||||||
|
public $secondArithmetic;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @override
|
||||||
|
*/
|
||||||
|
public function getSql(\Doctrine\ORM\Query\SqlWalker $sqlWalker)
|
||||||
|
{
|
||||||
|
$platform = $sqlWalker->getConnection()->getDatabasePlatform();
|
||||||
|
return $platform->getBitOrComparisonExpression(
|
||||||
|
$this->firstArithmetic->dispatch($sqlWalker),
|
||||||
|
$this->secondArithmetic->dispatch($sqlWalker)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @override
|
||||||
|
*/
|
||||||
|
public function parse(\Doctrine\ORM\Query\Parser $parser)
|
||||||
|
{
|
||||||
|
$parser->match(Lexer::T_IDENTIFIER);
|
||||||
|
$parser->match(Lexer::T_OPEN_PARENTHESIS);
|
||||||
|
|
||||||
|
$this->firstArithmetic = $parser->ArithmeticPrimary();
|
||||||
|
$parser->match(Lexer::T_COMMA);
|
||||||
|
$this->secondArithmetic = $parser->ArithmeticPrimary();
|
||||||
|
|
||||||
|
$parser->match(Lexer::T_CLOSE_PARENTHESIS);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -53,6 +53,8 @@ class Parser
|
||||||
'mod' => 'Doctrine\ORM\Query\AST\Functions\ModFunction',
|
'mod' => 'Doctrine\ORM\Query\AST\Functions\ModFunction',
|
||||||
'size' => 'Doctrine\ORM\Query\AST\Functions\SizeFunction',
|
'size' => 'Doctrine\ORM\Query\AST\Functions\SizeFunction',
|
||||||
'date_diff' => 'Doctrine\ORM\Query\AST\Functions\DateDiffFunction',
|
'date_diff' => 'Doctrine\ORM\Query\AST\Functions\DateDiffFunction',
|
||||||
|
'bit_and' => 'Doctrine\ORM\Query\AST\Functions\BitAndFunction',
|
||||||
|
'bit_or' => 'Doctrine\ORM\Query\AST\Functions\BitOrFunction',
|
||||||
);
|
);
|
||||||
|
|
||||||
/** READ-ONLY: Maps BUILT-IN datetime function names to AST class names. */
|
/** READ-ONLY: Maps BUILT-IN datetime function names to AST class names. */
|
||||||
|
|
|
||||||
|
|
@ -317,6 +317,54 @@ class QueryDqlFunctionTest extends \Doctrine\Tests\OrmFunctionalTestCase
|
||||||
$this->assertTrue(strtotime($arg[0]['add']) > 0);
|
$this->assertTrue(strtotime($arg[0]['add']) > 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @group DDC-1213
|
||||||
|
*/
|
||||||
|
public function testBitOrComparison()
|
||||||
|
{
|
||||||
|
$dql = 'SELECT m, ' .
|
||||||
|
'BIT_OR(4, 2) AS bit_or,' .
|
||||||
|
'BIT_OR( (m.salary/100000) , 2 ) AS salary_bit_or ' .
|
||||||
|
'FROM Doctrine\Tests\Models\Company\CompanyManager m ' .
|
||||||
|
'ORDER BY ' .
|
||||||
|
'm.id ' ;
|
||||||
|
$result = $this->_em->createQuery($dql)->getArrayResult();
|
||||||
|
|
||||||
|
$this->assertEquals(4 | 2, $result[0]['bit_or']);
|
||||||
|
$this->assertEquals(4 | 2, $result[1]['bit_or']);
|
||||||
|
$this->assertEquals(4 | 2, $result[2]['bit_or']);
|
||||||
|
$this->assertEquals(4 | 2, $result[3]['bit_or']);
|
||||||
|
|
||||||
|
$this->assertEquals(($result[0][0]['salary']/100000) | 2, $result[0]['salary_bit_or']);
|
||||||
|
$this->assertEquals(($result[1][0]['salary']/100000) | 2, $result[1]['salary_bit_or']);
|
||||||
|
$this->assertEquals(($result[2][0]['salary']/100000) | 2, $result[2]['salary_bit_or']);
|
||||||
|
$this->assertEquals(($result[3][0]['salary']/100000) | 2, $result[3]['salary_bit_or']);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @group DDC-1213
|
||||||
|
*/
|
||||||
|
public function testBitAndComparison()
|
||||||
|
{
|
||||||
|
$dql = 'SELECT m, ' .
|
||||||
|
'BIT_AND(4, 2) AS bit_and,' .
|
||||||
|
'BIT_AND( (m.salary/100000) , 2 ) AS salary_bit_and ' .
|
||||||
|
'FROM Doctrine\Tests\Models\Company\CompanyManager m ' .
|
||||||
|
'ORDER BY ' .
|
||||||
|
'm.id ' ;
|
||||||
|
$result = $this->_em->createQuery($dql)->getArrayResult();
|
||||||
|
|
||||||
|
$this->assertEquals(4 & 2, $result[0]['bit_and']);
|
||||||
|
$this->assertEquals(4 & 2, $result[1]['bit_and']);
|
||||||
|
$this->assertEquals(4 & 2, $result[2]['bit_and']);
|
||||||
|
$this->assertEquals(4 & 2, $result[3]['bit_and']);
|
||||||
|
|
||||||
|
$this->assertEquals(($result[0][0]['salary']/100000) & 2, $result[0]['salary_bit_and']);
|
||||||
|
$this->assertEquals(($result[1][0]['salary']/100000) & 2, $result[1]['salary_bit_and']);
|
||||||
|
$this->assertEquals(($result[2][0]['salary']/100000) & 2, $result[2]['salary_bit_and']);
|
||||||
|
$this->assertEquals(($result[3][0]['salary']/100000) & 2, $result[3]['salary_bit_and']);
|
||||||
|
}
|
||||||
|
|
||||||
protected function generateFixture()
|
protected function generateFixture()
|
||||||
{
|
{
|
||||||
$manager1 = new CompanyManager();
|
$manager1 = new CompanyManager();
|
||||||
|
|
|
||||||
|
|
@ -1392,6 +1392,29 @@ class SelectSqlGenerationTest extends \Doctrine\Tests\OrmTestCase
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @group DDC-1213
|
||||||
|
*/
|
||||||
|
public function testSupportsBitComparison()
|
||||||
|
{
|
||||||
|
$this->assertSqlGeneration(
|
||||||
|
'SELECT BIT_OR(4,2), BIT_AND(4,2), u FROM Doctrine\Tests\Models\CMS\CmsUser u',
|
||||||
|
'SELECT (4 | 2) AS sclr0, (4 & 2) AS sclr1, c0_.id AS id2, c0_.status AS status3, c0_.username AS username4, c0_.name AS name5 FROM cms_users c0_'
|
||||||
|
);
|
||||||
|
$this->assertSqlGeneration(
|
||||||
|
'SELECT BIT_OR(u.id,2), BIT_AND(u.id,2) FROM Doctrine\Tests\Models\CMS\CmsUser u WHERE BIT_OR(u.id,2) > 0',
|
||||||
|
'SELECT (c0_.id | 2) AS sclr0, (c0_.id & 2) AS sclr1 FROM cms_users c0_ WHERE (c0_.id | 2) > 0'
|
||||||
|
);
|
||||||
|
$this->assertSqlGeneration(
|
||||||
|
'SELECT BIT_OR(u.id,2), BIT_AND(u.id,2) FROM Doctrine\Tests\Models\CMS\CmsUser u WHERE BIT_AND(u.id , 4) > 0',
|
||||||
|
'SELECT (c0_.id | 2) AS sclr0, (c0_.id & 2) AS sclr1 FROM cms_users c0_ WHERE (c0_.id & 4) > 0'
|
||||||
|
);
|
||||||
|
$this->assertSqlGeneration(
|
||||||
|
'SELECT BIT_OR(u.id,2), BIT_AND(u.id,2) FROM Doctrine\Tests\Models\CMS\CmsUser u WHERE BIT_OR(u.id , 2) > 0 OR BIT_AND(u.id , 4) > 0',
|
||||||
|
'SELECT (c0_.id | 2) AS sclr0, (c0_.id & 2) AS sclr1 FROM cms_users c0_ WHERE (c0_.id | 2) > 0 OR (c0_.id & 4) > 0'
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
public function testCustomTypeValueSql()
|
public function testCustomTypeValueSql()
|
||||||
{
|
{
|
||||||
if (DBALType::hasType('negative_to_positive')) {
|
if (DBALType::hasType('negative_to_positive')) {
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue