java1.5提供了for each的循环方式,实现iterable接口就能让自己的类也能这么来循环,写一个例子。
[size=large][color=red]MainTest[/color][/size]
[size=large][color=red]Dept类:实现了iterable接口[/color][/size]
[size=large][color=red]Staff类[/color][/size]
[size=large][color=red]MainTest[/color][/size]
package org.hx.itr;
import java.util.Iterator;
import java.util.UUID;
public class MainTest {
public static void main(String[] args) {
Dept dept = randomDept(3);
for (Staff staff : dept) {
System.out.println(staff);
}
System.out.println();
for (Iterator<Staff> iterator = dept.iterator(); iterator.hasNext();) {
Staff staff = iterator.next();
System.out.println(staff);
}
}
/**
* 随机产生一些数据用作测试
* @param staffCount
* @return
*/
private static Dept randomDept(int staffCount){
Dept dept = new Dept();
for (int i = 0; i < staffCount; i++) {
Staff staff = new Staff();
staff.setName(UUID.randomUUID().toString().substring(0,5));
staff.setGender((int)Math.round(Math.random()*2));
staff.setAge((int)Math.round(Math.random()*100)+1);
dept.addStaff(staff);
}
return dept;
}
}
[size=large][color=red]Dept类:实现了iterable接口[/color][/size]
package org.hx.itr;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
public class Dept implements Iterable<Staff> {
List<Staff> staffList = new ArrayList<Staff>();
public void addStaff(Staff staff) {
this.staffList.add(staff);
}
public List<Staff> getStaffList() {
return staffList;
}
public Iterator<Staff> iterator() {
Itr itr = new Itr();
itr.setDept(this);
return itr;
}
}
class Itr implements Iterator<Staff> {
private int cursor = -1;
private Dept dept;
public void setDept(Dept dept) {
this.dept = dept;
}
public boolean hasNext() {
return dept.getStaffList() != null && dept.getStaffList().size() > cursor + 1;
}
public Staff next() {
return dept.getStaffList().get(++cursor);
}
public void remove() {
dept.getStaffList().remove(cursor);
}
}
[size=large][color=red]Staff类[/color][/size]
package org.hx.itr;
public class Staff {
private String name;
private int gender;
private int age;
@Override
public String toString() {
return "name = " + name + ", gender = " + gender + ", age = " + age;
}
// ...getter and setters...
}