Changeset 4684

Timestamp:

07/13/08 01:54:30 (5 months ago)

Author:

jwage

Message:

fixes #1195

Location:

branches/1.0

Files:

• lib/Doctrine/RawSql.php (modified) (4 diffs)
• tests/run.php (modified) (1 diff)

branches/1.0/lib/Doctrine/RawSql.php

@@ -73 +73 @@
             return $this;
         }
-        if ( ! isset($this->parts[$queryPartName])) {
+        if ( ! isset($this->_sqlParts[$queryPartName])) {
             $this->_sqlParts[$queryPartName] = array();
         }
@@ -221 +221 @@
         }
         
+        $q = 'SELECT ';
+        if($this->_sqlParts['distinct'] == true) $q .= 'DISTINCT ';
+
         // first add the fields of the root component
         reset($this->_queryComponents);
         $componentAlias = key($this->_queryComponents);
-
-        $q = 'SELECT ' . implode(', ', $select[$componentAlias]);
+        $q .= implode(', ', $select[$componentAlias]);
         unset($select[$componentAlias]);
 
@@ -238 +240 @@
             $this->_sqlParts['where'][] = $string;
         }
-        $copy = $this->_sqlParts;
-        unset($copy['select']);
 
         $q .= ( ! empty($this->_sqlParts['from']))?    ' FROM '     . implode(' ', $this->_sqlParts['from']) : '';
@@ -253 +253 @@
         }
         return $q;
+    }
+
+        /**
+     * getCountQuery
+     * builds the count query.
+     *
+     * @return string       the built sql query
+     */
+        public function getCountQuery($params = array())
+    {
+        //Doing COUNT( DISTINCT rootComponent.id )
+        //This is not correct, if the result is not hydrated by doctrine, but it mimics the behaviour of Doctrine_Query::getCountQuery
+        reset($this->_queryComponents);
+        $componentAlias = key($this->_queryComponents);
+        $tableAlias = $this->getSqlTableAlias($componentAlias);
+        $fields = array();
+
+        foreach ((array) $this->_queryComponents[$componentAlias]['table']->getIdentifierColumnNames() as $key) {
+                $fields[] = $tableAlias . '.' . $key;
+        }
+
+        $q = 'SELECT COUNT( DISTINCT '.implode(',',$fields).') as num_results';
+
+        $string = $this->getInheritanceCondition($this->getRootAlias());
+        if ( ! empty($string)) {
+            $this->_sqlParts['where'][] = $string;
+        }
+
+        $q .= ( ! empty($this->_sqlParts['from']))?    ' FROM '     . implode(' ', $this->_sqlParts['from']) : '';
+        $q .= ( ! empty($this->_sqlParts['where']))?   ' WHERE '    . implode(' AND ', $this->_sqlParts['where']) : '';
+        $q .= ( ! empty($this->_sqlParts['groupby']))? ' GROUP BY ' . implode(', ', $this->_sqlParts['groupby']) : '';
+        $q .= ( ! empty($this->_sqlParts['having']))?  ' HAVING '   . implode(' AND ', $this->_sqlParts['having']) : '';
+
+        if ( ! empty($string)) {
+            array_pop($this->_sqlParts['where']);
+        }
+
+        return $q;
+    }
+
+        /**
+     * count
+     * fetches the count of the query
+     *
+     * This method executes the main query without all the
+     * selected fields, ORDER BY part, LIMIT part and OFFSET part.
+     *
+     * This is an exact copy of the Dql Version
+     *
+     * @see Doctrine_Query::count()
+     * @param array $params        an array of prepared statement parameters
+     * @return integer             the count of this query
+     */
+    public function count($params = array())
+    {
+        $q = $this->getCountQuery();
+
+        if ( ! is_array($params)) {
+            $params = array($params);
+        }
+
+        $params = array_merge($this->_params['join'], $this->_params['where'], $this->_params['having'], $params);
+
+        $params = $this->convertEnums($params);
+
+        $results = $this->getConnection()->fetchAll($q, $params);
+
+        if (count($results) > 1) {
+            $count = count($results);
+        } else {
+            if (isset($results[0])) {
+                $results[0] = array_change_key_case($results[0], CASE_LOWER);
+                $count = $results[0]['num_results'];
+            } else {
+                $count = 0;
+            }
+        }
+
+        return (int) $count;
     }
 

branches/1.0/tests/run.php

@@ -87 +87 @@
 $tickets->addTestCase(new Doctrine_Ticket_1175_TestCase());
 $tickets->addTestCase(new Doctrine_Ticket_1192_TestCase());
+$tickets->addTestCase(new Doctrine_Ticket_1195_TestCase());
 $tickets->addTestCase(new Doctrine_Ticket_1205_TestCase());
 $tickets->addTestCase(new Doctrine_Ticket_1206_TestCase());