Java 經典實例：自定義迭代器

編寫自己的Iterator，實現Iterator接口，這裡多說一句，實現Iterable後，可以用“foreach”循環遍歷你的對象。

import java.util.Iterator;
import java.util.NoSuchElementException;

/**
* 演示Iterator和Iterable接口，並說明怎樣編寫一個用於對象數組的簡單迭代器。
*/
public class AarrayIterator<T> implements Iterable<T>, Iterator<T> {
private final static String[] names = {"rose", "petunia", "tulip"};

public static void main(String[] args) {
AarrayIterator<String> arrayIterator = new AarrayIterator<>(names);

// Java 5,6的方式
for (String s : arrayIterator) {
System.out.println(s);
}

// Java 8的形式
arrayIterator.forEach(System.out::println);
}

/**
* 要遍歷的數據
**/
protected T[] data;

protected int index = 0;

/**
* 構造一個AarryIterator對象。
*
* @param data 被迭代的對象數組
*/
public AarrayIterator(final T[] data) {
setData(data);
}

/**
* 設置（重置）數組為給定的數組，重置迭代器。
* 參數d代表被迭代的數組對象。
*
* @param d 被迭代的數組對象
*/

public void setData(final T[] d) {
this.data = d;
index = 0;
}

/**
* 如果不是末尾，返回true，例如，if next()語句將成功執行。
* 否則返回false,執行if next()語句會拋出異常。
*
* @return
*/
public boolean hasNext() {
return index < data.length;
}

/**
* 返回該數據的下一個元素
*
* @return
*/
public T next() {
if (hasNext()) {
return data[index++];
}
throw new NoSuchElementException("only " + data.length + " elements");
}

public void remove() {
throw new UnsupportedOperationException("This demo Iterator does not implement the remove method");
}

/**
* Iterator的方法
*
* @return
*/
public Iterator<T> iterator() {
index = 0;
return this;
}
}

執行結果：

rose
petunia
tulip
rose
petunia
tulip