实现for each,通过iterable接口和iterator(实例)

java1.5提供了for each的循环方式,实现iterable接口就能让自己的类也能这么来循环,写一个例子。

[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...
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值