首页 > 文章列表 > 如何实现java中的迭代器功能

如何实现java中的迭代器功能

java
226 2023-04-24

如何实现java中的迭代器功能

两行代码

 List<String> list=new ArrayList<>(2);

        list.add("java");

        list.add("C#");

        Iterator<String> iterator=list.iterator();

        while (iterator.hasNext()){

            System.out.println(iterator.next());

        }

上面的代码是java中很常见的一个迭代的功能。

于是自己也想要写一个泛型类,然后支持这种迭代的功能。

于是乎写了一个类似ArrayList的动态数组功能。

package a;



import javax.swing.text.html.HTMLDocument;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Iterator;



public class Gys<T>{

    private final  static int default_capacity =10;

    private int endIndex =0;

    private Object[] elemts;



    public Gys() {

        this.elemts = new Object[default_capacity];

    }



    public T[] add(T t){

        if(elemts.length-1< endIndex){

            int newCapcti= default_capacity *2;

            elemts= Arrays.copyOf(elemts,newCapcti);

        }

        elemts[endIndex++]=t;

        return (T[])elemts;

    }



    public int size(){

        return endIndex;

    }



    public T get(int i){

        if(i< endIndex){

            return (T) elemts[i];

        }

        throw new RuntimeException("索引超出界限");

    }





    public static void main(String[] args) {

        Gys<Integer> gys=new Gys<>();

        gys.add(5);

        gys.add(45);

        System.out.println(gys.get(0));

        System.out.println(gys.get(1));        

    }

}

上面的代码怎么都没办法实现Iterator的功能,在idea下怎么都点不出来Iterator的提示;

于是只能去翻阅jdk原码。在ArrayList中看到如下的代码。

在ArrayList中申明一个内部类Itr,并且继承Iterator<E>这个接口,然后实现hasNext()和next()方法。

在定义一个方法专门获取迭代器实例。

public Iterator<E> iterator() { return new Itr(); }

这才明白如何实现迭代器功能。

所以对上面的泛型代码进行改造。

package a;



import java.util.Arrays;

import java.util.Iterator;



public class Gys<T>{

    private final  static int default_capacity =10;

    private int endIndex =0;

    private Object[] elemts;



    public Gys() {

        this.elemts = new Object[default_capacity];

    }



    public T[] add(T t){

        if(elemts.length-1< endIndex){

            int newCapcti= default_capacity *2;

            elemts= Arrays.copyOf(elemts,newCapcti);

        }

        elemts[endIndex++]=t;

        return (T[])elemts;

    }



    public int size(){

        return endIndex;

    }



    class Itr implements Iterator<T>{



        private int point;

        private int len;



        public Itr() {

            this.point=0;

            this.len=endIndex;

        }



        @Override

        public boolean hasNext() {

            return point<endIndex?true:false;

        }



        @Override

        public T next() {

            return (T) elemts[point++];

        }

    }



    public Iterator<T> iterator(){

        return new Itr();

    } 



   public T get(int i){

        if(i< endIndex){

            return (T) elemts[i];

        }

        throw new RuntimeException("索引超出界限");

    }





    public static void main(String[] args) {

        Gys<Integer> gys=new Gys<>();

        gys.add(5);

        gys.add(45);

        /*System.out.println(gys.get(0));

        System.out.println(gys.get(1));*/

        Iterator iterator= gys.iterator();

        while (iterator.hasNext()){

            System.out.println(iterator.next());

        }

    }

}

怎么样、这个时候就可以对自己的泛型类Gys实现迭代器的功能了。

同时另一个疑问也来了,和Iterator长得异常相似的接口Iterable是干什么的?他和Iterator又是什么关系?

翻开源码看看。

 

源码中可以看出Iterable接口提供了一个获取迭代器的接口方法。那么又有哪些类实现了接口呢?

使用idea的ctrl+h快捷键调出查看类的全部继承关系。

  我们看到熟悉的Collection接口。

看到Colllection接口中并没有实现这个接口,依然是一个接口方法。继续向下追踪

 看到我们熟悉的ArrayList这个类型实现了iterator方法。

从这个角度来看ArrayList中的iterator()方法不是空穴来风的,他是通过继承collection和Iterable这些接口而来的。

虽然我们上面的泛型类实现了迭代的功能,但是学习了新知识后总要练练手,于是这个时候画蛇添足的对上面的代码继续改造。

package a;



import java.util.Arrays;

import java.util.Iterator;



public class Gys<T> implements Iterable<T>{

    private final  static int default_capacity =10;

    private int endIndex =0;

    private Object[] elemts;



    public Gys() {

        this.elemts = new Object[default_capacity];

    }



    public T[] add(T t){

        if(elemts.length-1< endIndex){

            int newCapcti= default_capacity *2;

            elemts= Arrays.copyOf(elemts,newCapcti);

        }

        elemts[endIndex++]=t;

        return (T[])elemts;

    }



    public int size(){

        return endIndex;

    }



    class Itr implements Iterator<T>{



        private int point;

        private int len;



        public Itr() {

            this.point=0;

            this.len=endIndex;

        }



        @Override

        public boolean hasNext() {

            return point<endIndex?true:false;

        }



        @Override

        public T next() {

            return (T) elemts[point++];

        }

    }    

   @Override

    public Iterator<T> iterator(){

        return new Itr();

    }





    public T get(int i){

        if(i< endIndex){

            return (T) elemts[i];

        }

        throw new RuntimeException("索引超出界限");

    }





    public static void main(String[] args) {

        Gys<Integer> gys=new Gys<>();

        gys.add(5);

        gys.add(45);

        /*System.out.println(gys.get(0));

        System.out.println(gys.get(1));*/

        Iterator iterator= gys.iterator();

        while (iterator.hasNext()){

            System.out.println(iterator.next());

        }

    }

}