java 标准库的 iterator 接口没有提供 peek() 方法，要实现预览下一个元素的功能，需要自定义一个包装类。具体做法是创建一个实现 iterator<t></t> 接口的类，内部持有原始迭代器并增加一个 nextelement 字段用于缓存即将返回的元素。调用 peek() 时，如果缓存为空则调用原始迭代器的 next() 并赋值给缓存，随后直接返回该缓存值；而 next() 方法则直接返回缓存值并将缓存置空。这样既保留了迭代器的状态，又实现了非消耗性的预览。
如何设计带缓存的 Iterator 包装类
核心在于利用一个成员变量来暂存从底层迭代器获取的值。在构造函数中初始化原始迭代器和缓存字段为 null。当首次调用 peek() 时，检查缓存是否为 null，如果是，则通过 hasNext() 判断是否存在元素，若存在则调用 next() 获取元素并存入缓存。后续再次调用 peek() 时，直接返回缓存中的值而不触发底层迭代器的移动。这种设计确保了 peek() 不会改变迭代器的遍历进度，除非配合 next() 使用。
import java.util.Iterator;
import java.util.List;
import java.util.ArrayList;
public class PeekableIterator<T> implements Iterator<T> {
private final Iterator<T> delegate;
private T nextElement = null;
private boolean hasPeeked = false;
public PeekableIterator(Iterator<T> delegate) {
this.delegate = delegate;
}
@Override
public boolean hasNext() {
if (hasPeeked) {
return true;
}
return delegate.hasNext();
}
@Override
public T next() {
if (!hasNext()) {
throw new java.util.NoSuchElementException();
}
if (hasPeeked) {
T result = nextElement;
nextElement = null;
hasPeeked = false;
return result;
}
return delegate.next();
}
public T peek() {
if (!hasPeeked) {
if (delegate.hasNext()) {
nextElement = delegate.next();
hasPeeked = true;
} else {
throw new java.util.NoSuchElementException();
}
}
return nextElement;
}
public static void main(String[] args) {
List<String> list = new ArrayList<>(List.of("A", "B", "C"));
PeekableIterator<String> it = new PeekableIterator<>(list.iterator());
System.out.println(it.peek()); // 输出 A，不消耗
System.out.println(it.peek()); // 输出 A，不消耗
System.out.println(it.next()); // 输出 A，消耗
System.out.println(it.next()); // 输出 B，消耗
}
}
缓存机制如何处理 hasNext 与 next 的状态同步
在实现过程中，必须确保 hasNext() 和 next() 能够正确感知缓存状态。如果仅依赖底层迭代器的 hasNext()，当用户先调用 peek() 后，底层迭代器已经前进一步，但此时 hasNext() 可能返回 false（如果这是最后一个元素），导致逻辑错误。因此，hasNext() 必须优先检查 hasPeeked 标志位。如果已缓存，说明肯定还有元素可取，直接返回 true。在 next() 中，如果存在缓存，则直接返回缓存值并清除缓存状态，否则才委托给底层迭代器。这种双向的状态同步保证了迭代器行为的一致性，避免了因预览操作导致的跳过或重复读取问题。
立即学习“Java免费学习笔记（深入）”；
// 状态同步逻辑片段
@Override
public boolean hasNext() {
if (hasPeeked) {
return true;
}
return delegate.hasNext();
}
@Override
public T next() {
if (!hasNext()) {
throw new java.util.NoSuchElementException();
}
if (hasPeeked) {
T result = nextElement;
nextElement = null;
hasPeeked = false;
return result;
}
return delegate.next();
}
当 peek 遇到空迭代器时会抛出什么异常
如果底层迭代器为空，即 delegate.hasNext() 返回 false，此时调用 peek() 会抛出 java.util.NoSuchElementException。这与标准 Iterator.next() 的行为保持一致，符合 Java 集合框架的约定。在代码实现中，peek() 方法在尝试从底层获取元素前会再次检查 hasNext() 的有效性，或者直接检查 delegate.hasNext()。如果为空，立即抛出异常，防止程序进入无效状态。这种处理方式确保了调用者可以明确知道是否还有元素可供预览，而不是返回 null 或其他默认值，从而避免潜在的 NullPointerException 或逻辑混淆。
