趣解设计模式之《庞大的组织架构带来的烦恼》

〇、小故事

小王在一家创业公司，随着公司业务的发展，团队人数也从之前的十人左右拓展为了几百人的团队。那么，对于几百人的公司，就需要有一套清晰的组织架构了，这样会更便于公司员工的管理，以及跨部门的协同工作（某部门的领导是谁？包含哪些子部门？我这件事儿应该去找哪个部门？等等）这件事情显得愈发迫在眉睫。

老板找到小王跟他述说了自己的需求，小王作为公司的研发总监，毫不犹豫地就答应了老板的需求。回到工位，就开始着手对系统进行设计起来。怎么设计呢？毕竟是用来展示组织架构的系统，那么先列举出公司内的所有职位吧：

【公司的高管】董事长、CTO、COO、CFO……
【公司的中层】技术总监、人事总监、财务总监、行政总监、销售总监……
【公司的基层】HR、行政人员、产品经理、研发人员、测试人员、运维人员、运营人员、销售人员……

难道要为每一类型的职员都创建一个类？那这类爆炸了…… 而且，如果以后所有的“总监”职位都变为了“部门经理”，或者所有的研发人员都裁撤掉采用外包团队，那这些类都要变化，这个可怎么办呢？

这件事情可愁坏了小王，回到家，小王无意间看到了桌子上的优秀员工奖杯，这个是他刚到公司，公司一共10多个人的时候，公司颁给他的。这个不由得让他回忆起了过去。“哎~ 当初的日子，真的有意思，公司一共10多个人，研发团队就3个人，自己就是普通的研发人员，随着公司发展，1年后研发团队变成了20人，分为了3个研发团队，自己晋升为1个研发团队的Leader。第2年，研发团队变成了50人，分为了8个研发小组，自己也晋升为某业务线的研发主管，管理了3个小组，第3年，……”

“哎？等等！”他突然灵光乍现，公司里虽然岗位很多，但是，无论怎么说，组成元素都员工啊！基层员工是员工，高层也是员工啊，只不过基层员工不做管理，中高层员工有管理职能而已啊。换句话说，多个基层员工组成一个小组，多个小组组成一个部门，多个部门组成一条业务线，多个业务线组成子公司，多个子公司组成集团公司……

小王的思路打开了，针对员工类和组合行为，非常完美的完成了整个系统的设计和开发工作，老板也非常的满意，看来，今年公司的优秀员工奖杯，又会是小王的了。故事讲完了，那么小王的灵光乍现就是我们今天要介绍的设计模式——组合模式。

一、模式定义

组合模式（Composite Pattern）

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

我们采用树形结构，来展示组织架构图，如下所示：

二、模式类图

那么在类的设计中，我们首先创建抽象的员工类AbstractCorp，在这个类中定义职员信息，比如：姓名，职位，薪水……，然后对于普通职员，我们创建基层员工类LeafCorp，对于管理者员工，我们创建BranchCrop类，这两个类的区别就是，管理者中要包含一个变量corps，用于存储他手下团队的所有员工。具体类图如下所示：

三、代码实现

创建员工类：AbstractCorp.java

@AllArgsConstructor
public abstract class AbstractCorp {

    // 员工姓名
    private String name = "";

    // 员工职位
    private String position = "";

    // 员工薪水
    private int salary = 0;

    // 获得员工信息
    public String getCorpInfo() {
        return String.format("姓名:%s 职位:%s 薪水:%d", name, position, salary);
    }
}

创建管理层员工类（即：非叶子节点职员）：BranchCorp.java

public class BranchCorp extends AbstractCorp {
    private ArrayList corps = Lists.newArrayList();

    public BranchCorp(String name, String position, int salary) {
        super(name, position, salary);
    }

    // 添加员工
    public void addSubCorp(AbstractCorp... corp) {
        corps.addAll(Arrays.asList(corp));
    }

    // 返回手下的员工列表
    public ArrayList getSubCorp() {
        return corps;
    }
}

创建基层员工（即：叶子节点职员）类：LeafCorp.java

public class LeafCorp extends AbstractCorp {
    public LeafCorp(String name, String position, int salary) {
        super(name, position, salary);
    }
}

创建测试类：CompositeTest.java

public class CompositeTest {
    public static void main(String[] args) {
        // 创建公司所有员工
        BranchCorp a = new BranchCorp("A", "董事长", 100000);
        BranchCorp b = new BranchCorp("B", "技术总监", 70000);
        BranchCorp c = new BranchCorp("C", "人事总监", 60000);
        BranchCorp d = new BranchCorp("D", "财务总监", 50000);
        LeafCorp b1 = new LeafCorp("B1", "程序员", 30000);
        LeafCorp b2 = new LeafCorp("B2", "程序员", 20000);
        LeafCorp c1 = new LeafCorp("C1", "HR", 15000);
        LeafCorp d1 = new LeafCorp("D1", "会计", 13000);

        // 根据下属关系，构建树形结构
        a.addSubCorp(b, c, d);
        b.addSubCorp(b1, b2);
        c.addSubCorp(c1);
        d.addSubCorp(d1);

        // 获得下属信息
        System.out.println("-----a的下属-----");
        getSubCorp(a);
        System.out.println("-----b的下属-----");
        getSubCorp(b);
    }

    // 获得branchCorp的下属
    private static void getSubCorp(BranchCorp branchCorp) {
        for (AbstractCorp corp : branchCorp.getSubCorp()) {
            System.out.println("corp = " + corp.getCorpInfo());
            if (corp instanceof BranchCorp) {
                getSubCorp((BranchCorp) corp);
            }
        }
    }
}

执行后的结果：

-----a的下属-----
corp = 姓名:B 职位:技术总监 薪水:70000
corp = 姓名:B1 职位:程序员 薪水:30000
corp = 姓名:B2 职位:程序员 薪水:20000
corp = 姓名:C 职位:人事总监 薪水:60000
corp = 姓名:C1 职位:HR 薪水:15000
corp = 姓名:D 职位:财务总监 薪水:50000
corp = 姓名:D1 职位:会计 薪水:13000
-----b的下属-----
corp = 姓名:B1 职位:程序员 薪水:30000
corp = 姓名:B2 职位:程序员 薪水:20000

Process finished with exit code 0

今天的文章内容就这些了：

写作不易，笔者几个小时甚至数天完成的一篇文章，只愿换来您几秒钟的 点赞 & 分享 。

更多技术干货，欢迎大家关注公众号“爪哇缪斯” ~ \(^o^)/ ~ 「干货分享，每天更新」