<?php /** * @link http://www.yiiframework.com/ * @copyright Copyright (c) 2008 Yii Software LLC * @license http://www.yiiframework.com/license/ */ namespace yii\mongodb; use yii\db\ActiveRelationInterface; use yii\db\ActiveRelationTrait; /** * ActiveRelation represents a relation to Mongo Active Record class. * * @author Paul Klimov <klimov.paul@gmail.com> * @since 2.0 */ 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); } }