Java语言提供了增强for操作的语法糖。
我们在遍历集合类时经常会有使用到增强for,如下:
Collection<Student> c= new ArrayList<Student>();
c.add(new Student("yqs",3303));
c.add(new Student("yqs1",3304));
c.add(new Student("yqs2",3304));
c.add(new Student("yqs",3305));
c.remove(new Student("yqs",3304));
for (Student stu:c)
{
System.out.println(stu.name+" "+stu.id);
}
那么如何让自己写的类也能像这样遍历呢?上面说了增强for是java语言为我们提供的语法糖,所以我们如果想要自己的类也能实现增强for,那么就要遵循java官方的规范,这个规范就是,实现Iterable接口,这个接口位于java.util.*
首先我们新建一个类,并实现Iterator接口的方法:
package iterable;
import java.util.Iterator;
public class MyList implements Iterable<String>
{
String[] strings=new String[]{"i","love","you"};
@Override
public Iterator<String> iterator() {
return new Iterator<String>() {
private int cursor=0;
@Override
public boolean hasNext() {
return cursor<strings.length;
}
@Override
public String next() {
return strings[cursor++];
}
};
}
}
我们需要做两件事情。
第一在我们的类中存储一些数据(没数据就没东西可遍历哈哈),这里我定义了一个字符串数组。
第二我们需要实现iterator()方法,该方法返回一个Iterator接口,该接口需要实现两个方法(hasNext()和next()),分别表示是否还有元素可遍历以及下一个元素具体的值。
这样子之后我们就可以测试一下:
public class Main
{
public static void main(String[] args) {
MyList list=new MyList();
for(String s:list)
{
System.out.println(s);
}
}
}
输出:
i
love
you
完美实现我们的需求。