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