《大话设计模式》java实现之组合模式

组合模式主要是体现部分与整体层次的结构，统一使用结构中的所有对象（书中原话）。

个人理解：组合模式多用于树状展现的内容中。

本示例省略书中的履行职责方法，这个对该模式没什么影响

UML图



代码结构

public class Client {
public static void main(String[] args) {
ConcreteCompany root = new ConcreteCompany("北京总公司");
root.add(new HRDepartment("总公司人力资源部"));
root.add(new FinanceDepartment("总公司财务部"));

ConcreteCompany comp = new ConcreteCompany("上海华东分公司");
comp.add(new HRDepartment("东分公司人力资源部"));
comp.add(new FinanceDepartment("东分公司财务部"));
root.add(comp);

ConcreteCompany comp1 = new ConcreteCompany("南京办事处");
comp1.add(new HRDepartment("南京办事处人力资源部"));
comp1.add(new FinanceDepartment("南京办事处财务部"));
comp.add(comp1);

ConcreteCompany comp2 = new ConcreteCompany("杭州办事处");
comp2.add(new HRDepartment("杭州办事处人力资源部"));
comp2.add(new FinanceDepartment("杭州办事处财务部"));
comp.add(comp2);

System.out.println("结构图");
root.display(1);
}
}

public abstract class Company {
private String name;     //书中是procteced，为了和以往编程习惯一致，这里使用private

public String getName() {
return name;
}

public void setName(String name) {
this.name = name;
}

public Company() {                  //保留无参构造方法
super();
}

public Company(String name) {
super();
this.name = name;
}

public abstract void add(Company company);
public abstract void remove(Company company);
public abstract void display(int depth);
}

public class ConcreteCompany extends Company {

private List<Company> children = new ArrayList<>();

public ConcreteCompany(String name) {
super(name);
}

@Override
public void add(Company company) {
children.add(company);
}

@Override
public void remove(Company company) {
children.remove(company);
}

@Override
public void display(int depth) {
for(int i = 0; i < depth; i++) {   //由于没有找到书中的类似方法，这里作了一个小循环
System.out.print("-");
}
System.out.println(getName());

for(Company component : children) {
component.display(depth + 2);
}
}

}

public class FinanceDepartment extends Company {

public FinanceDepartment(String name) {
super(name);
}

@Override
public void add(Company company) {

}

@Override
public void remove(Company company) {

}

@Override
public void display(int depth) {
for(int i = 0; i < depth; i++) {   //由于没有找到书中的类似方法，这里作了一个小循环
System.out.print("-");
}
System.out.println(getName());
}

}

public class HRDepartment extends Company {

public HRDepartment(String name) {
super(name);
}

@Override
public void add(Company company) {

}

@Override
public void remove(Company company) {

}

@Override
public void display(int depth) {
for(int i = 0; i < depth; i++) {   //由于没有找到书中的类似方法，这里作了一个小循环
System.out.print("-");
}
System.out.println(getName());
}

}

结果



多说一句，从上面可以看到HRDepartment和FinanceDepartment的display方法都是重复的代码，这个可以用模版方法模式进行优化，在父类定义一个方法template，然后子类调用，ConcreCompany调用父类后再添加自己的逻辑即可。