设计模式系列:迭代器模式

package com.littlehui.design.iterator;
 * @Description TODO
 * @ClassName Iterator
 * @Author littlehui
 * @Date 2019/10/12 14:56
 * @Version 1.0
 **/
public interface  {
    boolean hasNext();
    Object next();
    void remove();
}

package com.littlehui.design.iterator;
 * @Description TODO
 * @ClassName IteratorCreator
 * @Author littlehui
 * @Date 2019/10/12 14:57
 * @Version 1.0
 **/
public interface IteratorCreator {
    public Iterator createIterator();
}

package com.littlehui.design.iterator;
 * @Description TODO
 * @ClassName MenuItem
 * @Author littlehui
 * @Date 2019/10/12 14:57
 * @Version 1.0
 **/
public class MenuItem {
    private String code;
    private String menuName;
    public MenuItem(String code, String menuName) {
        this.code = code;
        this.menuName = menuName;
    }
    public String getCode() {
        return code;
    }
    public void setCode(String code) {
        this.code = code;
    }
    public String getMenuName() {
        return menuName;
    }
    public void setMenuName(String menuName) {
        this.menuName = menuName;
    }
    public void printMenu() {
        System.out.println("编码：" + code + "名称：" + menuName);
    }
}

package com.littlehui.design.iterator;
import java.util.ArrayList;
import java.util.List;
 * @Description TODO
 * @ClassName MenuContainer
 * @Author littlehui
 * @Date 2019/10/12 14:57
 * @Version 1.0
 **/
public class MenuContainer implements IteratorCreator {
    private List<MenuItem> menuItems;
    public MenuContainer(List<MenuItem> menuItems) {
        this.menuItems = menuItems;
    }
    public MenuContainer() {
    }
    public void addMenuItem(MenuItem menuItem) {
        if (menuItems == null) {
            menuItems = new ArrayList<MenuItem>();
        }
        menuItems.add(menuItem);
    }
    
    public Iterator createIterator() {
        return new MenuContainerIterator(menuItems);
    }
}

package com.littlehui.design.iterator;
import java.util.ArrayList;
import java.util.List;
 * @Description TODO
 * @ClassName MenuContainerIterator
 * @Author littlehui
 * @Date 2019/10/12 14:58
 * @Version 1.0
 **/
public class MenuContainerIterator implements  {
    private List<MenuItem> menuItemList = null;
    private int currentIndex = -1;
    private int lastIndex = 0;
    public MenuContainerIterator(List<MenuItem> menuItemList) {
        this.menuItemList = menuItemList;
        lastIndex = menuItemList.size() - 1;
    }
    
    public boolean hasNext() {
        if (currentIndex < lastIndex) {
            return true;
        } else {
            return false;
        }
    }
    
    public Object next() {
        MenuItem currentItem = menuItemList.get(++currentIndex);
        return currentItem;
    }
    
    public void remove() {
        menuItemList.remove(currentIndex);
        --currentIndex;
        lastIndex = menuItemList.size() - 1;
    }
}

package com.littlehui.design.iterator;
import java.util.ArrayList;
import java.util.List;
 * @Description TODO
 * @ClassName Client
 * @Author littlehui
 * @Date 2019/10/12 15:16
 * @Version 1.0
 **/
public class Client {
    public static void main(String[] args) {
        List<MenuItem> menuItemList = new ArrayList();
        menuItemList.add(new MenuItem("CODE_1", "第一个菜单"));
        menuItemList.add(new MenuItem("CODE_2", "第二个菜单"));
        menuItemList.add(new MenuItem("CODE_3", "第三个菜单"));
        MenuContainer menuContainer = new MenuContainer(menuItemList);
        Iterator iterator = menuContainer.createIterator();
        while (iterator.hasNext()) {
            MenuItem item = (MenuItem)iterator.next();
            if ("CODE_2".equals(item.getCode())) {
                iterator.remove();
            } else {
                item.printMenu();
            }
        }
    }
}