java 标准库的 iterator 接口没有提供 peek() 方法，要实现预览下一个元素的功能，需要自定义一个包装类。具体做法是创建一个实现 iterator 接口的类，内部持有原始迭代器并增加一个 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，消耗
}
}</string></string></t></t></t></t>
缓存机制如何处理 hasNext 与 next 的状态同步
在实现过程中，必须确保 hasNext() 和 next() 能够正确感知缓存状态。如果仅依赖底层迭代器的 hasNext()，当用户先调用 peek() 后，底层迭代器已经前进一步，但此时 hasNext() 可能返回 false（如果这是最后一个元素），导致逻辑错误。因此，hasNext() 必须优先检查 hasPeeked 标志位。如果已缓存，说明肯定还有元素可取，直接返回 true。在 next() 中，如果存在缓存，则直接返回缓存值并清除缓存状态，否则才委托给底层迭代器。这种双向的状态同步保证了迭代器行为的一致性，避免了因预览操作导致的跳过或重复读取问题。
// 状态同步逻辑片段
@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 或逻辑混淆。
Java免费学习笔记：立即使用
解锁 Java 大师之旅：从入门到精通的终极指南
