首页 > 代码库 > Java 经典实例:自定义迭代器

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 =http://www.mamicode.com/ 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

 

Java 经典实例:自定义迭代器