Merge pull request #3696 from bingo-soft/master

Add support for DISTINCT clause

docs/en/reference/query-builder.rst

@@ -87,6 +87,21 @@ and ``delete($tableName)``:
 You can convert a query builder to its SQL string representation
 by calling ``$queryBuilder->getSQL()`` or casting the object to string.
 
+DISTINCT-Clause
+~~~~~~~~~~~~~~~
+
+The ``SELECT`` statement can be specified with a ``DISTINCT`` clause:
+
+.. code-block:: php
+
+    <?php
+
+    $queryBuilder
+        ->select('name')
+        ->distinct()
+        ->from('users')
+    ;
+
 WHERE-Clause
 ~~~~~~~~~~~~
 

lib/Doctrine/DBAL/Query/QueryBuilder.php

@@ -52,13 +52,12 @@ class QueryBuilder
      */
     private $connection;
 
-    /**
-     * The array of SQL parts collected.
-     *
-     * @var mixed[]
+    /*
+     * The default values of SQL parts collection
      */
-    private $sqlParts = [
+    private const SQL_PARTS_DEFAULTS = [
         'select'   => [],
+        'distinct' => false,
         'from'     => [],
         'join'     => [],
         'set'      => [],
@@ -69,6 +68,13 @@ class QueryBuilder
         'values'   => [],
     ];
 
+    /**
+     * The array of SQL parts collected.
+     *
+     * @var mixed[]
+     */
+    private $sqlParts = self::SQL_PARTS_DEFAULTS;
+
     /**
      * The complete SQL string for this query.
      *
@@ -469,6 +475,25 @@ class QueryBuilder
         return $this->add('select', $selects);
     }
 
+    /**
+     * Adds DISTINCT to the query.
+     *
+     * <code>
+     *     $qb = $conn->createQueryBuilder()
+     *         ->select('u.id')
+     *         ->distinct()
+     *         ->from('users', 'u')
+     * </code>
+     *
+     * @return $this This QueryBuilder instance.
+     */
+    public function distinct() : self
+    {
+        $this->sqlParts['distinct'] = true;
+
+        return $this;
+    }
+
     /**
      * Adds an item that is to be returned in the query result.
      *
@@ -1083,8 +1108,7 @@ class QueryBuilder
      */
     public function resetQueryPart($queryPartName)
     {
-        $this->sqlParts[$queryPartName] = is_array($this->sqlParts[$queryPartName])
-            ? [] : null;
+        $this->sqlParts[$queryPartName] = self::SQL_PARTS_DEFAULTS[$queryPartName];
 
         $this->state = self::STATE_DIRTY;
 
@@ -1098,7 +1122,8 @@ class QueryBuilder
      */
     private function getSQLForSelect()
     {
-        $query = 'SELECT ' . implode(', ', $this->sqlParts['select']);
+        $query = 'SELECT ' . ($this->sqlParts['distinct'] ? 'DISTINCT ' : '') .
+                  implode(', ', $this->sqlParts['select']);
 
         $query .= ($this->sqlParts['from'] ? ' FROM ' . implode(', ', $this->getFromClauses()) : '')
             . ($this->sqlParts['where'] !== null ? ' WHERE ' . ((string) $this->sqlParts['where']) : '')

tests/Doctrine/Tests/DBAL/Query/QueryBuilderTest.php

@@ -50,6 +50,17 @@ class QueryBuilderTest extends DbalTestCase
         self::assertEquals('SELECT u.id FROM users u', (string) $qb);
     }
 
+    public function testSimpleSelectWithDistinct() : void
+    {
+        $qb = new QueryBuilder($this->conn);
+
+        $qb->select('u.id')
+           ->distinct()
+           ->from('users', 'u');
+
+        self::assertEquals('SELECT DISTINCT u.id FROM users u', (string) $qb);
+    }
+
     public function testSelectWithSimpleWhere() : void
     {
         $qb   = new QueryBuilder($this->conn);