结构型模式之组合模式

 1 /**
 2  * Component
 3  * 抽象构件
 4  */
 5 public abstract class Component {
 6     //个体和整体都具有的共享
 7     public void doSomething() {
 8         //编写业务逻辑
 9     }
10 } 
11 
12 /**
13  * Composite
14  * 树枝构件
15  */
16 public class Composite extends Component {
17     //构件容器
18     private ArrayList<Component> componentArrayList = new ArrayList<Component>();
19 
20     //增加一个叶子构件或树枝构件
21     public void add(Component component) {
22         this.componentArrayList.add(component);
23     }
24 
25     //删除一个叶子构件或树枝构件
26     public void remove(Component component) {
27         this.componentArrayList.remove(component);
28     }
29 
30     //获得分支下的所有叶子构件和树枝构件
31     public ArrayList<Component> getChildren() {
32         return this.componentArrayList;
33     }
34 }  
35 
36 /**
37  * Leaf
38  * 树叶构件
39  * 树叶节点是没有子下级对象的对象，定义参加组合的原始对象行为。
40  */
41 public class Leaf extends Component {
42 
43     //  可以覆写父类方法
44     @Override
45     public void doSomething() {
46 
47     }
48 }
49 
50 /**
51  * Client
52  * 测试类负责树状结构的建立，并可以通过递归方式遍历整个树。
53  */
54 public class Client {
55     public static void main(String[] args) {
56         //创建一个根节点
57         Composite root = new Composite();
58         root.doSomething();
59         //创建一个树枝构件
60         Composite branch = new Composite();
61         //创建一个叶子节点
62         Leaf leaf = new Leaf();
63         //建立整体
64         root.add(branch);
65         branch.add(leaf);
66     }
67 
68     //通过递归遍历树
69     public static void display(Composite root) {
70         for (Component c : root.getChildren()) {
71             if (c instanceof Leaf) { //叶子节点
72                 c.doSomething();
73             } else { //树枝节点
74                 display((Composite) c);
75             }
76         }
77     }
78 }