大话设计模式-组合模式(大鸟和小菜Java版)

Posted 2021-06-29 香风智乃哈~

组合模式（Composite）：将对象组合成树形结构以表示“部分-整体”的层次结构。组合模式使得用户对单个对象和组合对象的使用具有一致性。

结构图如下：

将就着看吧，没找到怎么旋转

代码如下：

package 设计模式.组合模式;

//公司类，可以是抽象类或者接口
public abstract class Company {
    protected String name;

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

//    增加
    public abstract void add(Company c);
//    移除
    public abstract void remove(Company c);
//显示
    public abstract void display(int depth);
//    履行职责
    public abstract void lineOfDuty();
}

package 设计模式.组合模式;

import java.util.ArrayList;
import java.util.List;

//具体公司类 实现抽象类和抽象方法
public class ConcreteCompany extends Company{

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

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

    @Override
    public void add(Company c) {
        child.add(c);
    }

    @Override
    public void remove(Company c) {
        child.remove(c);
    }

    @Override
    public void display(int depth) {
        for(int i = 0;i<depth;i++){
            System.out.print("-");
        }
        System.out.println(name);
        for(Company component:child){
            component.display(depth+2);
        }
    }

    @Override
    public void lineOfDuty() {
        for (Company component : child){
            component.lineOfDuty();
        }
    }
}

package 设计模式.组合模式;

//人力资源部和财务部类  具体公司类的子类
public class HRDepartment extends Company{

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

    @Override
    public void add(Company c) {

    }

    @Override
    public void remove(Company c) {

    }

    @Override
    public void display(int depth) {
        for (int i = 0;i<depth;i++){
            System.out.print("-");
        }
        System.out.println(name);
    }

    @Override
    public void lineOfDuty() {
        System.out.println(name+" 员工招聘培训管理");
    }
}


class FinanceDepartment extends Company{

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

    @Override
    public void add(Company c) {

    }

    @Override
    public void remove(Company c) {

    }

    @Override
    public void display(int depth) {
        for (int i = 0;i<depth;i++){
            System.out.print("-");
        }
        System.out.println(name);
    }

    @Override
    public void lineOfDuty() {
        System.out.println(name+" 公司财务收支管理");
    }
}

最后是测试主类;

package 设计模式.组合模式;

//测试主类
public class Test {
    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);

        System.out.println("职责：");
        root.lineOfDuty();
    }
}

输出结果：

以上就是组合模式的Java代码展示