迭代器模式与组合模式(二) —— Head First设计模式学习

在上篇文章中，printMenu()的实现的几个问题，在这篇文章中，予以解决。

首先问题的原因是什么？很明显，不同的collection类型，煎饼屋菜单使用的是ArrayList，而餐厅的菜单使用的是数组。

代码的区别如下

• ArrayList 和 数组的元素遍历：

		for(int i = 0; i < breakfastItems.size(); i++){
			MenuItem item = (MenuItem)breakfastItems.get(i);
		}
		for(int i = 0; i < lunchItems.length; i++){
			MenuItem item = lunchItems[i];
		}

如果我们使用迭代器，上面的代码就变成了：

• 在ArrayList上使用迭代器

		//在ArrayList上使用迭代器
		Iterator iterator = breakfastItems.createIterator();
		while(iterator.hasNext()){
			MenuItem menuItem = (MenuItem)iterator.next();
		}

• 在数组上使用迭代器

		//在数组上使用迭代器
		Iterator iterator = lunchMenu.createIterator();
		while(iterator.hasNext()){
			MenuItem menuItem = (MenuItem)iterator.next();
		}

我们可以看到，在数组上和在ArrayList上使用迭代器的代码完全一致。（或许迭代器失去了collection本身的一些特性，比如ArrayList和数组通过下标来存取元素。但是迭代器本身就是为了遍历而存在）。

接下来，使用迭代器模式(Iterator Pattern)来改写代码。

迭代器接口：

package com.hufei.learnDesignPattern.improve;
public interface Iterator {
	boolean hasNext();
	Object next();
}

餐厅菜单迭代器：

public class DinerMenuIterator implements Iterator {
	MenuItem[] items;
	int position = 0;
	
	public DinerMenuIterator(MenuItem[] items){
		this.items = items;
	}
	public boolean hasNext() {
		if(position >= items.length || items[position] == null){
			return false;
		}else{
			return true;
		}
	}
	public Object next() {
		MenuItem menuItem = items[position];
		position = position + 1;
		return menuItem;
	}
}

餐厅菜单，什么都不需要修改，只是增加一个createIterator()方法。

//DinerMenu什么都不需要改，只需要增加一个createIterator()方法。
public class DinerMenu {
	static final int MAX_ITEMS = 6;
	int numberOfItems = 0;
	MenuItem[] menuItems;
	
	public DinerMenu(){
		menuItems = new MenuItem[MAX_ITEMS];
	
	}
	
	public void addItem(String name ,String description, boolean vegetarian, double price){
		MenuItem menuItem = new MenuItem(name,description,vegetarian,price);
		
		if(numberOfItems >= MAX_ITEMS){
			System.err.println("Sorry, menu is full! Can't add item to menu");
		} else{
			menuItems[numberOfItems] = menuItem;
			numberOfItems = numberOfItems +1 ;
		}
	}
	
	//不再需要MenuItems方法
/*	public MenuItem[] getMenuItems(){
		return this.menuItems;
	}*/
	public Iterator createIterator(){
		return new DinerMenuIterator(menuItems);
	}
	//还有菜单的其他方法
}

女招待类中，现在只需要一个循环即可。

public class Waitress {
	PancakeHouseMenu pancakeHouseMenu;
	DinerMenu dinerMenu;
	public Waitress(PancakeHouseMenu pancakeHouseMenu, DinerMenu dinerMenu) {
		super();
		this.pancakeHouseMenu = pancakeHouseMenu;
		this.dinerMenu = dinerMenu;
	}
	
	public void printMenu(){
		Iterator pancakeIterator = this.pancakeHouseMenu.createIterator();
		Iterator dinerIterator = dinerMenu.createIterator();
		System.out.println("MENU/n----/nBREAKFAST");
		printMenu(pancakeIterator);
		System.out.println("MENU/n----/nLUNCH");
		printMenu(dinerIterator);
		
	}
	
	public void printMenu(Iterator it){
		while(it.hasNext()){
			MenuItem menuItem = (MenuItem)it.next();
			System.out.print(menuItem.getName()+", ");
			System.out.print(menuItem.getPrice()+"-- ");
			System.out.println(menuItem.getDescription());
		}
	}
}