kiri-core/Database/Base/CollectionIterator.php

<?php

declare(strict_types=1);

namespace Database\Base;


use Database\ActiveQuery;
use Database\ActiveRecord;
use Exception;
use ReflectionException;
use Snowflake\Exception\NotFindClassException;
use Snowflake\Snowflake;


/**
 * Class CollectionIterator
 * @package Database\Base
 */
class CollectionIterator extends \ArrayIterator
{

	private ActiveRecord $model;


	/** @var ActiveQuery */
	private ActiveQuery $query;


	public function clean()
	{
		unset($this->query);
	}


	/**
	 * CollectionIterator constructor.
	 * @param $model
	 * @param $query
	 * @param array $array
	 * @param int $flags
	 * @throws Exception
	 */
	public function __construct($model, $query, $array = array(), $flags = 0)
	{
		$this->model = $model;
		$this->query = $query;
		parent::__construct($array, $flags);
	}


	/**
	 * @param $current
	 * @return ActiveRecord
	 * @throws Exception
	 */
	protected function newModel($current): ActiveRecord
	{
		$model = $this->model;
		if (is_object($model)) {
			return $model->setAttributes($current);
		}
		return objectPool($model, function () use ($model) {
			return new $model();
		})->setAttributes($current);

	}


	/**
	 * @throws Exception
	 */
	public function current(): ActiveRecord
	{
		if (!(($current = parent::current()) instanceof ActiveRecord)) {
			$current = $this->newModel($current);
		}
		return $this->query->getWith($current);
	}


}