Fixing output dialect

Author: moufmouf

src/QueryFactory/SmartEagerLoad/Query/ManyToOnePartialQuery.php

@@ -5,6 +5,7 @@
 
 use Doctrine\DBAL\Connection;
 use Doctrine\DBAL\Platforms\MySqlPlatform;
+use Mouf\Database\MagicQuery;
 use TheCodingMachine\TDBM\QueryFactory\SmartEagerLoad\ManyToOneDataLoader;
 use TheCodingMachine\TDBM\QueryFactory\SmartEagerLoad\StorageNode;
 
@@ -13,59 +14,56 @@ class ManyToOnePartialQuery implements PartialQuery
     /**
      * @var string
      */
-    private $queryFrom;
+    private $mainTable;
     /**
      * @var string
      */
-    private $mainTable;
+    private $key;
     /**
-     * @var StorageNode
+     * @var string
      */
-    private $storageNode;
+    private $pk;
+    /**
+     * @var PartialQuery
+     */
+    private $partialQuery;
     /**
      * @var string
      */
-    private $key;
+    private $originTableName;
     /**
      * @var string
      */
-    private $pk;
+    private $columnName;
 
     public function __construct(PartialQuery $partialQuery, string $originTableName, string $tableName, string $pk, string $columnName)
     {
         // TODO: move this in a separate function. The constructor is called for every bean.
-        $mysqlPlatform = new MySqlPlatform();
-        $this->queryFrom = 'FROM ' .$mysqlPlatform->quoteIdentifier($tableName).
-            ' WHERE ' .$mysqlPlatform->quoteIdentifier($tableName).'.'.$mysqlPlatform->quoteIdentifier($pk).' IN '.
-            '(SELECT '.$mysqlPlatform->quoteIdentifier($originTableName).'.'.$mysqlPlatform->quoteIdentifier($columnName).' '.$partialQuery->getQueryFrom().')';
+        $this->partialQuery = $partialQuery;
         $this->mainTable = $tableName;
-        $this->storageNode = $partialQuery->getStorageNode();
         $this->key = $partialQuery->getKey().'__'.$columnName;
         $this->pk = $pk;
+        $this->originTableName = $originTableName;
+        $this->columnName = $columnName;
     }
 
     /**
      * Returns the SQL of the query, starting at the FROM keyword.
      */
     public function getQueryFrom(): string
     {
-        return $this->queryFrom;
-    }
-
-    /**
-     * Returns the name of the main table (main objects returned by this query)
-     */
-    public function getMainTable(): string
-    {
-        return $this->mainTable;
+        $mysqlPlatform = new MySqlPlatform();
+        return 'FROM ' .$mysqlPlatform->quoteIdentifier($this->mainTable).
+            ' WHERE ' .$mysqlPlatform->quoteIdentifier($this->mainTable).'.'.$mysqlPlatform->quoteIdentifier($this->pk).' IN '.
+            '(SELECT '.$mysqlPlatform->quoteIdentifier($this->originTableName).'.'.$mysqlPlatform->quoteIdentifier($this->columnName).' '.$this->partialQuery->getQueryFrom().')';
     }
 
     /**
      * Returns the object in charge of storing the dataloader associated to this query.
      */
     public function getStorageNode(): StorageNode
     {
-        return $this->storageNode;
+        return $this->partialQuery->getStorageNode();
     }
 
     /**
@@ -81,13 +79,24 @@ public function getKey(): string
      */
     public function registerDataLoader(Connection $connection): void
     {
-        if ($this->storageNode->hasManyToOneDataLoader($this->key)) {
+        $storageNode = $this->getStorageNode();
+        if ($storageNode->hasManyToOneDataLoader($this->key)) {
             return;
         }
 
         $mysqlPlatform = new MySqlPlatform();
-        $sql = 'SELECT DISTINCT ' .$mysqlPlatform->quoteIdentifier($this->mainTable).'.* '.$this->queryFrom;
+        $sql = 'SELECT DISTINCT ' .$mysqlPlatform->quoteIdentifier($this->mainTable).'.* '.$this->getQueryFrom();
 
-        $this->storageNode->setManyToOneDataLoader($this->key, new ManyToOneDataLoader($connection, $sql, $this->pk));
+        if (!$connection->getDatabasePlatform() instanceof MySqlPlatform) {
+            // We need to convert the query from MySQL dialect to something else
+            $sql = $this->getMagicQuery()->buildPreparedStatement($sql);
+        }
+
+        $storageNode->setManyToOneDataLoader($this->key, new ManyToOneDataLoader($connection, $sql, $this->pk));
+    }
+
+    public function getMagicQuery(): MagicQuery
+    {
+        return $this->partialQuery->getMagicQuery();
     }
 }

src/QueryFactory/SmartEagerLoad/Query/PartialQuery.php

@@ -4,6 +4,7 @@
 namespace TheCodingMachine\TDBM\QueryFactory\SmartEagerLoad\Query;
 
 use Doctrine\DBAL\Connection;
+use Mouf\Database\MagicQuery;
 use TheCodingMachine\TDBM\QueryFactory\SmartEagerLoad\StorageNode;
 
 /**
@@ -31,4 +32,6 @@ public function registerDataLoader(Connection $connection): void;
      * Returns the object in charge of storing the dataloader associated to this query.
      */
     public function getStorageNode(): StorageNode;
+
+    public function getMagicQuery(): MagicQuery;
 }

src/QueryFactory/SmartEagerLoad/Query/StaticPartialQuery.php

@@ -111,4 +111,9 @@ public function getParameters(): array
     {
         return $this->parameters;
     }
+
+    public function getMagicQuery(): MagicQuery
+    {
+        return $this->magicQuery;
+    }
 }