package generic.interfaces;
public interface Generator<T> {
T next();
}
public class Coffee {
private static long counter = 0;
private final long id = counter++;
@Override
public String toString() {
// TODO Auto-generated method stub
return getClass().getSimpleName() + " " + id;
}
}
package generic.interfaces;
public class Cappuccino extends Coffee {
}
package generic.interfaces;
public class Latte extends Coffee {
}
package generic.interfaces;
public class Mocha extends Coffee {
}
package generic.interfaces;
public class Cappuccino extends Coffee {
}
package generic.interfaces;
import java.util.Iterator;
import java.util.Random;
public class CoffeeGenerator implements Generator<Coffee> , Iterable<Coffee> { //注意这里实现的是Iterable<Coffee>
private Class[] types = { Latte.class, Mocha.class, Cappuccino.class };
private static Random random = new Random(47);
public CoffeeGenerator() {
}
private int size = 0;
public CoffeeGenerator(int size) {
this.size = size;
}
@Override
public Coffee next() {
// TODO Auto-generated method stub
try {
return (Coffee) types[random.nextInt(types.length)].newInstance();
} catch (Exception e) {
// TODO Auto-generated catch block
throw new RuntimeException();
}
}
class CoffeeIterator implements Iterator<Coffee> {
int count = size;
@Override
public boolean hasNext() {
// TODO Auto-generated method stub
return count > 0;
}
@Override
public Coffee next() {
count--;
return CoffeeGenerator.this.next();
}
@Override
public void remove() {
// TODO Auto-generated method stub
throw new UnsupportedOperationException();
}
};
public Iterator<Coffee> iterator() {
return new CoffeeIterator();
}
public static void main(String[] args) {
CoffeeGenerator generator = new CoffeeGenerator();
for (int i = 0; i < 5; i++) {
System.out.println(generator.next());
}
for (Coffee coffee : new CoffeeGenerator(5)) {
System.out.println(coffee);
}
}
}
输出
Cappuccino 0
Cappuccino 1
Mocha 2
Cappuccino 3
Mocha 4
Cappuccino 5
Mocha 6
Cappuccino 7
Latte 8
Mocha 9