结构型模式之组合模式

// 定义抽象构件接口
public interface Component {
    public void operation();
}

// 定义树枝构件
public class Composite implements Component {
    // 构件容器
    private ArrayList<Component> componentList = new ArrayList<Component>();
    // 添加构件
    public void add(Component component) {
        this.componentList.add(component);
    }
    // 删除构件
    public void remove(Component component) {
        this.componentList.remove(component);
    }
    // 获取子构件
    public ArrayList<Component> getChild() {
        return this.componentList;
    }
    @Override
    public void operation() {
        System.out.println("业务逻辑代码");
    }
}

// 定义叶子结构
public class Leaf implements Component {
    @Override
    public void operation() {
        System.out.println("业务逻辑代码Leaf");
    }
}

public class Client {
    public static void main(String[] args) {
        // 创建根节点
        Composite root = new Composite();
        // 创建树枝节点
        Composite branch = new Composite();
        // 创建叶子节点
        Leaf leaf = new Leaf();
        root.add(branch);
        branch.add(leaf);
        display(root);
    }
    // 遍历树递归
    public static void display(Composite root) {
        for (Component c : root.getChild()) {
            if (c instanceof Leaf) { // 如果节点类型是叶子节点
                c.operation();
            } else { // 树枝节点
                c.operation();
                display((Composite) c);
            }
        }
    }
}