Basic support for asCollection

framework/db/ActiveQuery.php

@@ -225,9 +225,25 @@ public function populate($rows)
             }
         }
 
+        if ($this->collectionClass !== null) {
+            return $this->createCollection($models);
+        }
+
         return $models;
     }
 
+    /**
+     * Instantiates collection
+     *
+     * @param array $models array of models to fill collection with.
+     * @return object collection instance.
+     * @since 2.0.10
+     */
+    protected function createCollection($models)
+    {
+        return new $this->collectionClass($models);
+    }
+
     /**
      * Removes duplicated models by checking their primary key values.
      * This method is mainly called when a join query is performed, which may cause duplicated rows being returned.

framework/db/ActiveQueryTrait.php

@@ -30,6 +30,11 @@ trait ActiveQueryTrait
      */
     public $asArray;
 
+    /**
+     * @var string collection class to be instantiated and filled with results.
+     * @since 2.0.10
+     */
+    public $collectionClass;
 
     /**
      * Sets the [[asArray]] property.
@@ -42,6 +47,18 @@ public function asArray($value = true)
         return $this;
     }
 
+    /**
+     * Sets the [[collectionClass]] property.
+     * @param string $className collection class to be instantiated and filled with results.
+     * @return $this the query object itself
+     * @since 2.0.10
+     */
+    public function asCollection($className)
+    {
+        $this->collectionClass = $className;
+        return $this;
+    }
+
     /**
      * Specifies the relations with which this query should be performed.
      *

tests/data/ar/CustomerCollection.php

@@ -0,0 +1,25 @@
+<?php
+
+
+namespace yiiunit\data\ar;
+
+
+class CustomerCollection implements \Countable, \IteratorAggregate
+{
+    private $data;
+
+    public function __construct(array $data)
+    {
+        $this->data = $data;
+    }
+
+    public function count()
+    {
+        return count($this->data);
+    }
+
+    public function getIterator()
+    {
+        return new \ArrayIterator($this->data);
+    }
+}

tests/framework/db/ActiveRecordTest.php

@@ -1293,4 +1293,14 @@ public function testUpdateAttributes()
         $this->assertTrue($newOrder->getIsNewRecord());
         $this->assertEquals($newTotal, $newOrder->total);
     }
+
+    public function testCollection()
+    {
+        $modelClass = 'yiiunit\data\ar\Customer';
+        $collectionClass = 'yiiunit\data\ar\CustomerCollection';
+        $customers = Customer::find()->asCollection($collectionClass)->all();
+        $this->assertInstanceOf($collectionClass, $customers);
+        $this->assertEquals(3, $customers->count());
+        $this->assertContainsOnlyInstancesOf($modelClass, $customers);
+    }
 }