Adding a setOutputDialect method

This method allows to set the platform used to quote identifiers and strings (a different one from the one used in the connection)

Author: moufmouf

src/Mouf/Database/MagicQuery.php

@@ -3,6 +3,7 @@
 namespace Mouf\Database;
 
 use Doctrine\DBAL\Connection;
+use Doctrine\DBAL\Platforms\AbstractPlatform;
 use Doctrine\DBAL\Platforms\MySqlPlatform;
 use function array_filter;
 use function array_keys;
@@ -34,6 +35,10 @@ class MagicQuery
     private $connection;
     private $cache;
     private $schemaAnalyzer;
+    /**
+     * @var AbstractPlatform
+     */
+    private $platform;
     /**
      * @var \Twig_Environment
      */
@@ -48,6 +53,7 @@ class MagicQuery
     public function __construct($connection = null, $cache = null, SchemaAnalyzer $schemaAnalyzer = null)
     {
         $this->connection = $connection;
+        $this->platform = $connection ? $connection->getDatabasePlatform() : new MySqlPlatform();
         if ($cache) {
             $this->cache = $cache;
         } else {
@@ -73,6 +79,19 @@ public function setEnableTwig($enableTwig = true)
         return $this;
     }
 
+    /**
+     * Overrides the output dialect used to generate SQL. By default, the dialect of the connection is used.
+     * If no connection is used, MySQL platform is used by default.
+     *
+     * @param AbstractPlatform $platform
+     */
+    public function setOutputDialect(AbstractPlatform $platform): self
+    {
+        $this->platform = $platform;
+
+        return $this;
+    }
+
     /**
      * Returns merged SQL from $sql and $parameters. Any parameters not available will be striped down
      * from the SQL.
@@ -175,8 +194,7 @@ public function parse($sql)
      */
     public function toSql(NodeInterface $sqlNode, array $parameters = array(), bool $extrapolateParameters = true)
     {
-        $platform = $this->connection ? $this->connection->getDatabasePlatform() : new MySqlPlatform();
-        return (string) $sqlNode->toSql($parameters, $platform, 0, SqlRenderInterface::CONDITION_GUESS, $extrapolateParameters);
+        return (string) $sqlNode->toSql($parameters, $this->platform, 0, SqlRenderInterface::CONDITION_GUESS, $extrapolateParameters);
     }
 
     /**

tests/Mouf/Database/MagicQueryTest.php

@@ -3,6 +3,7 @@
 namespace Mouf\Database;
 
 use Doctrine\Common\Cache\ArrayCache;
+use Doctrine\DBAL\Platforms\PostgreSqlPlatform;
 use Doctrine\DBAL\Schema\Schema;
 use Mouf\Database\SchemaAnalyzer\SchemaAnalyzer;
 use PHPUnit\Framework\TestCase;
@@ -434,4 +435,13 @@ public function testBuildPreparedStatement()
         $sql = 'SELECT id FROM users WHERE status IN :status';
         $this->assertEquals("SELECT id FROM users WHERE status IN (:status)", self::simplifySql($magicQuery->buildPreparedStatement($sql, ['status' => [1,2]])));
     }
+
+    public function testSetOutputDialect()
+    {
+        $magicQuery = new MagicQuery(null, new ArrayCache());
+        $magicQuery->setOutputDialect(new PostgreSqlPlatform());
+
+        $sql = 'SELECT id FROM users';
+        $this->assertEquals('SELECT "id" FROM "users"', self::simplifySql($magicQuery->buildPreparedStatement($sql)));
+    }
 }