Adding aggregation

README.md

@@ -89,16 +89,15 @@ Examples
             })
             ->get();
 
-**Distinct**
-
-Distinct is limited to a single column.
-
-    $names = Users::distinct('name')->get();
-
 **Group By**
 
-Grouping does not support sorting and limiting at this moment.
+    $users = Users::groupBy('title')->get(array('title', 'name'));
+
+**Aggregation**
 
-    $users = Users::groupBy('title')->get();
+    $price = DB::table('orders')->max('price');
+    $price = DB::table('orders')->min('price');
+    $price = DB::table('orders')->avg('price');
+    $total = DB::table('users')->sum('votes');
 
 All basis insert, update, delete and select methods should be implemented. Feel free to fork and help completing this library!
\ No newline at end of file

src/Jenssegers/Mongodb/Query.php

@@ -23,9 +23,6 @@ class Query extends \Illuminate\Database\Query\Builder {
         '<=' => '$lte',
         '>' => '$gt',
         '>=' => '$gte',
-        'in' => '$in',
-        'exists' => '$exists',
-        'or' => '$or',
     );
 
     /**
@@ -73,62 +70,78 @@ class Query extends \Illuminate\Database\Query\Builder {
             $this->columns = array();
         }
 
+        // Compile wheres
+        $wheres = $this->compileWheres();
 
-        if ($this->distinct)
+        // Use aggregation framework if needed
+        if ($this->groups || $this->aggregate)
         {
-            // We can only return an array of distinct values for a single column
-            return $this->collection->distinct($this->distinct, $this->compileWheres());
-        }
-        else if(count($this->groups))
-        {
-            // We can pass our wheres as a condition
-            $options = array();
-            $options['condition'] = $this->compileWheres();
-
-            // Initial value and reduce code
-            $initial = array("item" => "");
-            $reduce = "function (obj, prev) { prev.item = obj; }";
-            
-            // Get results
-            $result = $this->collection->group($this->groups, $initial, $reduce, $options);
-            $items = $result['retval'];
-
-            // Transform results to a nice array
-            $results = array();
-            foreach($items as $item)
+            $pipeline = array();
+
+            // Group
+            $group = array();
+            $group['_id'] = $this->groups ? $this->groups : 0;
+
+            // Columns
+            foreach ($this->columns as $column)
             {
-                $results[] = $item['item'];
+                $group[$column] = array('$last' => '$' . $column);
             }
 
-            // Return these results since we don't have a MongoCursor
-            return $results;
+            // Apply aggregation functions
+            if ($this->aggregate)
+            {
+                foreach ($this->aggregate['columns'] as $column)
+                {
+                    $group[$column] = array('$' . $this->aggregate['function'] => '$' . $column);
+                }
+            }
+
+            if ($wheres) $pipeline[] = array('$match' => $match);
+
+            $pipeline[] = array('$group' => $group);
+
+            // Apply order and limit
+            if ($this->orders) $pipeline[] = array('$sort' => $this->orders);
+            if ($this->limit) $pipeline[] = array('$limit' => $this->limit);
+
+            $results = $this->collection->aggregate($pipeline);
+
+            // Return results
+            return $results['result'];
         }
         else
         {
             // Get the MongoCursor
-            $cursor = $this->collection->find($this->compileWheres(), $this->columns);
-        }
+            $cursor = $this->collection->find($wheres, $this->columns);
 
-        // Apply order
-        if ($this->orders)
-        {
-            $cursor->sort($this->orders);
-        }
+            // Apply order, offset and limit
+            if ($this->orders) $cursor->sort($this->orders);
+            if ($this->offset) $cursor->skip($this->offset);
+            if ($this->limit) $cursor->limit($this->limit);
 
-        // Apply offset
-        if ($this->offset)
-        {
-            $cursor->skip($this->offset);
+            // Return results
+            return $cursor;
         }
+    }
 
-        // Apply limit
-        if ($this->limit)
+    /**
+     * Execute an aggregate function on the database.
+     *
+     * @param  string  $function
+     * @param  array   $columns
+     * @return mixed
+     */
+    public function aggregate($function, $columns = array('*'))
+    {
+        $this->aggregate = compact('function', 'columns');
+
+        $results = $this->get($columns);
+
+        if (isset($results[0]))
         {
-            $cursor->limit($this->limit);
+            return $results[0][$columns[0]];
         }
-
-        // Return results
-        return $cursor;
     }
 
     /**
@@ -157,7 +170,7 @@ class Query extends \Illuminate\Database\Query\Builder {
 
         foreach ($groups as $group)
         {
-            $this->groups[$group] = 1;
+            $this->groups[$group] = '$' . $group;
         }
 
         return $this;
@@ -246,18 +259,6 @@ class Query extends \Illuminate\Database\Query\Builder {
         return $this;
     }
 
-    /**
-     * Force the query to only return distinct results.
-     *
-     * @return Builder
-     */
-    public function distinct($column = false)
-    {
-        $this->distinct = $column;
-
-        return $this;
-    }
-
     /**
     * Compile the where array
     *
@@ -310,7 +311,7 @@ class Query extends \Illuminate\Database\Query\Builder {
 
         if ($boolean == 'or')
         {
-            return array($this->conversion[$boolean] => array($query));
+            return array('$or' => array($query));
         }
 
         return $query;
@@ -325,7 +326,7 @@ class Query extends \Illuminate\Database\Query\Builder {
 
         if ($boolean == 'or')
         {
-            return array($this->conversion[$boolean] => array($compiled));
+            return array('$or' => array($compiled));
         }
 
         return $compiled;
@@ -335,7 +336,7 @@ class Query extends \Illuminate\Database\Query\Builder {
     {
         extract($where);
 
-        return array($column => array($this->conversion['in'] => $values));
+        return array($column => array('$in' => $values));
     }
 
     public function compileWhereNull($where)