Merge pull request beberlei#234 from bubnovKelnik/master

Add Query/Sqlite/IfElse with tests

README.md

@@ -21,7 +21,7 @@ functions available in MySQL, Oracle, Sqlite and PostgreSQL.
 |:--:|:---------:|
 | MySQL | `ACOS, ASCII, ASIN, ATAN, ATAN2, BINARY, BIT_COUNT, BIT_XOR, CEIL, CHAR_LENGTH, CONCAT_WS, CONVERT_TZ, COS, COT, COUNTIF, CRC32, DATE, DATE_FORMAT, DATEADD, DATEDIFF, DATESUB, DAY, DAYNAME, DAYOFWEEK, DAYOFYEAR, DEGREES, DIV, FIELD, FIND_IN_SET, FLOOR, FROM_UNIXTIME, GREATEST, GROUP_CONCAT, HEX, HOUR, IFELSE, IFNULL, LAST_DAY, LEAST, LOG10, LPAD, MATCH, MD5, MINUTE, MONTH, MONTHNAME, NOW, NULLIF, PI, POWER, QUARTER, RADIANS, RAND, REGEXP, REPLACE, ROUND, RPAD, SECOND, SHA1, SHA2, SIN, SOUNDEX, STD, STDDEV, STRTODATE, STR_TO_DATE, SUBSTRING_INDEX, TAN, TIME, TIMEDIFF, TIMESTAMPADD, TIMESTAMPDIFF, TIMETOSEC, UNHEX, UNIX_TIMESTAMP, UUID_SHORT, VARIANCE, WEEK, WEEKDAY, YEAR, YEARWEEK` |
 | Oracle | `DAY, LISTAGG, MONTH, NVL, TO_CHAR, TO_DATE, TRUNC, YEAR` |
-| Sqlite | `DATE, MINUTE, HOUR, DAY, WEEK, WEEKDAY, MONTH, YEAR, STRFTIME, DATE_FORMAT*, IFNULL, REPLACE, ROUND` |
+| Sqlite | `DATE, MINUTE, HOUR, DAY, WEEK, WEEKDAY, MONTH, YEAR, STRFTIME, DATE_FORMAT*, CASE WHEN THEN ELSE END, IFNULL, REPLACE, ROUND` |
 | PostgreSQL | `TO_DATE, TO_CHAR, AT_TIME_ZONE, COUNT_FILTER` |
 
 > Note: Sqlite date functions are implemented as `strftime(format, value)`.

config/sqlite.yml

@@ -34,7 +34,7 @@ doctrine:
 #                 field: DoctrineExtensions\Query\Sqlite\Field
 #                 find_in_set: DoctrineExtensions\Query\Sqlite\FindInSet
 #                 group_concat: DoctrineExtensions\Query\Sqlite\GroupConcat
-#                 ifelse: DoctrineExtensions\Query\Sqlite\IfElse
+                 ifelse: DoctrineExtensions\Query\Sqlite\IfElse
                  ifnull: DoctrineExtensions\Query\Sqlite\IfNull
 #                 match: DoctrineExtensions\Query\Sqlite\MatchAgainst
 #                 md5: DoctrineExtensions\Query\Sqlite\Md5

src/Query/Sqlite/IfElse.php

@@ -0,0 +1,37 @@
+<?php
+
+namespace DoctrineExtensions\Query\Sqlite;
+
+use Doctrine\ORM\Query\AST\Functions\FunctionNode;
+use Doctrine\ORM\Query\Lexer;
+
+/**
+ * @author Mikhail Bubnov <[email protected]>
+ */
+class IfElse extends FunctionNode
+{
+    private $expr = array();
+
+    public function parse(\Doctrine\ORM\Query\Parser $parser)
+    {
+        $parser->match(Lexer::T_IDENTIFIER);
+        $parser->match(Lexer::T_OPEN_PARENTHESIS);
+        $this->expr[] = $parser->ConditionalExpression();
+
+        for ($i = 0; $i < 2; $i++)
+        {
+            $parser->match(Lexer::T_COMMA);
+            $this->expr[] = $parser->ArithmeticExpression();
+        }
+
+        $parser->match(Lexer::T_CLOSE_PARENTHESIS);
+    }
+
+    public function getSql(\Doctrine\ORM\Query\SqlWalker $sqlWalker)
+    {
+        return sprintf('CASE WHEN %s THEN %s ELSE %s END',
+            $sqlWalker->walkConditionalExpression($this->expr[0]),
+            $sqlWalker->walkArithmeticPrimary($this->expr[1]),
+            $sqlWalker->walkArithmeticPrimary($this->expr[2]));
+    }
+}

tests/Query/Sqlite/IfElseTest.php

@@ -0,0 +1,18 @@
+<?php
+
+namespace DoctrineExtensions\Tests\Query\Sqlite;
+
+use DoctrineExtensions\Tests\Query\SqliteTestCase;
+
+class IfElse extends SqliteTestCase
+{
+    public function testIfElse()
+    {
+        $dql = "SELECT ifelse(1 > 0, 1, 0) FROM DoctrineExtensions\\Tests\\Entities\\BlogPost b";
+
+        $this->assertEquals(
+            "SELECT CASE WHEN 1 > 0 THEN 1 ELSE 0 END AS {$this->columnAlias} FROM BlogPost b0_",
+            $this->entityManager->createQuery($dql)->getSql()
+        );
+    }
+}