如何通过PHP Iterator从单个元素返回多个元素？

当然可以！PHP的Iterator接口完全支持这种场景——让单个元素在遍历过程中返回多个值。针对你给出的数组$a = ['1','2','3','double 6','4']，我们可以自定义一个Iterator，在遍历到特定元素（比如包含double的字符串）时，将其拆分成多个元素逐个返回，最终得到1 2 3 6 6 4的输出。

实现思路 #

核心是在自定义Iterator中维护两个关键状态：

• 原数组的遍历指针
• 一个临时缓冲区，用来存储拆分后的待返回元素

当遍历到需要拆分的元素时，我们先把它拆成多个子元素存入缓冲区，后续的遍历优先从缓冲区取元素；缓冲区空了之后，再继续处理原数组的下一个元素。

具体代码实现 #

class CustomElementIterator implements Iterator {
    private $originalArray;
    private $currentKey;
    private $buffer = [];

    public function __construct(array $array) {
        $this->originalArray = $array;
    }

    public function rewind(): void {
        // 重置原数组指针和缓冲区
        $this->currentKey = 0;
        $this->buffer = [];
    }

    public function current(): mixed {
        // 如果缓冲区有元素，直接返回第一个
        if (!empty($this->buffer)) {
            return $this->buffer[0];
        }

        // 处理当前原数组元素
        $currentElement = $this->originalArray[$this->currentKey] ?? null;
        if ($currentElement === null) {
            return null;
        }

        // 判断是否需要拆分（这里匹配"double X"格式的元素）
        if (str_starts_with($currentElement, 'double ')) {
            $value = trim(str_replace('double ', '', $currentElement));
            // 拆分成两个相同的元素存入缓冲区
            $this->buffer = [$value, $value];
        } else {
            // 不需要拆分的元素直接存入缓冲区
            $this->buffer = [$currentElement];
        }

        return $this->buffer[0];
    }

    public function key(): mixed {
        // 这里可以根据需求调整key的返回逻辑，比如拆分的元素共用原元素的key
        return $this->currentKey;
    }

    public function next(): void {
        // 如果缓冲区有元素，移除第一个（因为已经返回过了）
        if (!empty($this->buffer)) {
            array_shift($this->buffer);
            // 缓冲区空了之后，再移动原数组指针
            if (empty($this->buffer)) {
                $this->currentKey++;
            }
            return;
        }

        // 缓冲区为空时直接移动原数组指针
        $this->currentKey++;
    }

    public function valid(): bool {
        // 只要原数组还有未处理的元素，或者缓冲区还有元素，就说明遍历有效
        return isset($this->originalArray[$this->currentKey]) || !empty($this->buffer);
    }
}

测试代码 #

$a = ['1','2','3','double 6','4'];
$iterator = new CustomElementIterator($a);

foreach ($iterator as $value) {
    echo $value . ' ';
}
// 输出：1 2 3 6 6 4 

扩展说明 #

• 你可以修改current()方法中的判断逻辑，比如支持更多拆分规则（比如"triple X"拆成三个元素）
• 如果需要给拆分后的元素设置不同的key，可以在key()方法中做额外处理，比如维护一个独立的key计数器

内容的提问来源于stack exchange，提问作者joachim