66 lines
1.1 KiB
PHP
66 lines
1.1 KiB
PHP
<?php
|
|
|
|
|
|
namespace Database\Base;
|
|
|
|
|
|
use Database\ActiveRecord;
|
|
use ReflectionException;
|
|
use Snowflake\Exception\NotFindClassException;
|
|
use Snowflake\Snowflake;
|
|
|
|
|
|
/**
|
|
* Class CollectionIterator
|
|
* @package Database\Base
|
|
*/
|
|
class CollectionIterator extends \ArrayIterator
|
|
{
|
|
|
|
/** @var ActiveRecord */
|
|
private $model;
|
|
|
|
|
|
/**
|
|
* CollectionIterator constructor.
|
|
* @param $model
|
|
* @param array $array
|
|
* @param int $flags
|
|
* @throws ReflectionException
|
|
* @throws NotFindClassException
|
|
*/
|
|
public function __construct($model, $array = array(), $flags = 0)
|
|
{
|
|
$this->model = $model;
|
|
if (is_string($model)) {
|
|
$this->model = Snowflake::createObject($model);
|
|
}
|
|
parent::__construct($array, $flags);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $current
|
|
* @return ActiveRecord
|
|
*/
|
|
protected function newModel($current)
|
|
{
|
|
return (clone $this->model)->setAttributes($current);
|
|
}
|
|
|
|
|
|
/**
|
|
* @return ActiveRecord|mixed
|
|
*/
|
|
public function current()
|
|
{
|
|
$current = parent::current();
|
|
if ($current instanceof ActiveRecord) {
|
|
return $current;
|
|
}
|
|
return $this->newModel($current);
|
|
}
|
|
|
|
|
|
}
|