diff --git a/extensions/yii/mongodb/ActiveRelation.php b/extensions/yii/mongodb/ActiveRelation.php index d4541cc..785b701 100644 --- a/extensions/yii/mongodb/ActiveRelation.php +++ b/extensions/yii/mongodb/ActiveRelation.php @@ -19,4 +19,35 @@ use yii\db\ActiveRelationTrait; class ActiveRelation extends ActiveQuery implements ActiveRelationInterface { use ActiveRelationTrait; + + /** + * @inheritdoc + */ + protected function buildCursor($db = null) + { + if ($this->primaryModel !== null) { + // lazy loading + if ($this->via instanceof self) { + // via pivot collection + $viaModels = $this->via->findPivotRows([$this->primaryModel]); + $this->filterByModels($viaModels); + } elseif (is_array($this->via)) { + // via relation + /** @var ActiveRelation $viaQuery */ + list($viaName, $viaQuery) = $this->via; + if ($viaQuery->multiple) { + $viaModels = $viaQuery->all(); + $this->primaryModel->populateRelation($viaName, $viaModels); + } else { + $model = $viaQuery->one(); + $this->primaryModel->populateRelation($viaName, $model); + $viaModels = $model === null ? [] : [$model]; + } + $this->filterByModels($viaModels); + } else { + $this->filterByModels([$this->primaryModel]); + } + } + return parent::buildCursor($db); + } } \ No newline at end of file diff --git a/tests/unit/extensions/mongodb/ActiveRelationTest.php b/tests/unit/extensions/mongodb/ActiveRelationTest.php index 8c77d23..8736d52 100644 --- a/tests/unit/extensions/mongodb/ActiveRelationTest.php +++ b/tests/unit/extensions/mongodb/ActiveRelationTest.php @@ -52,7 +52,7 @@ class ActiveRelationTest extends MongoDbTestCase ]; $customerOrders[] = [ 'customer_id' => $customer['_id'], - 'number' => $customer['status'] + 1, + 'number' => $customer['status'] + 100, ]; } $customerOrderCollection->batchInsert($customerOrders); @@ -65,9 +65,10 @@ class ActiveRelationTest extends MongoDbTestCase /** @var CustomerOrder $order */ $order = CustomerOrder::find(['number' => 2]); $this->assertFalse($order->isRelationPopulated('customer')); - $index = $order->customer; + $customer = $order->customer; $this->assertTrue($order->isRelationPopulated('customer')); - $this->assertTrue($index instanceof Customer); + $this->assertTrue($customer instanceof Customer); + $this->assertEquals((string)$customer->_id, (string)$order->customer_id); $this->assertEquals(1, count($order->populatedRelations)); } @@ -78,6 +79,8 @@ class ActiveRelationTest extends MongoDbTestCase $this->assertTrue($orders[0]->isRelationPopulated('customer')); $this->assertTrue($orders[1]->isRelationPopulated('customer')); $this->assertTrue($orders[0]->customer instanceof Customer); + $this->assertEquals((string)$orders[0]->customer->_id, (string)$orders[0]->customer_id); $this->assertTrue($orders[1]->customer instanceof Customer); + $this->assertEquals((string)$orders[1]->customer->_id, (string)$orders[1]->customer_id); } } \ No newline at end of file