读设计模式之禅--组合模式

• 定义：将对象组合成树形结构以表示“部分-整体”的层次结构，使得用户对单个对象和组合对象的使用具有一致性。
• 简言之：就是将数据组装成树结构

package design.Combination;

import java.util.ArrayList;

/**
 * @Author: hyh
 * @Date: 2021/8/21 17:36
 * 举一个族谱  抽象一个家庭成员类
 **/
public abstract class FamilyMember {

    // 成员姓名
    private String name;
    // 上代长辈
    private FamilyMember parent;

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

    // 获取信息
    public void getInfo() {
        System.out.println(this.name);
    }

    public void setParent(FamilyMember parent) {
        this.parent = parent;
        parent.addChildren(this);
    }

    public void addParent(FamilyMember parent) {
        this.parent = parent;
    }

    public FamilyMember getParent() {
        return parent;
    }

    public abstract ArrayList<FamilyMember> getChildren();

    public abstract void setChildren(FamilyMember member);

    public abstract void addChildren(FamilyMember member);
}

// 定义一个父辈 包含有子辈
class Parents extends FamilyMember {

    // 子辈集合
    private ArrayList<FamilyMember> children = new ArrayList();

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

    @Override
    public void setChildren(FamilyMember member) {
        // 将当前对象 Parents 传给 父类
        member.addParent(this);
        this.children.add(member);
    }

    @Override
    public void addChildren(FamilyMember member) {
        this.children.add(member);
    }

    @Override
    public ArrayList<FamilyMember> getChildren() {
        return children;
    }

}

// 没有下一代的 子辈
class Children extends FamilyMember {

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

    @Override
    @Deprecated
    public ArrayList<FamilyMember> getChildren() {
        return null;
    }

    @Override
    @Deprecated
    public void setChildren(FamilyMember member) {
    }

    @Override
    @Deprecated
    public void addChildren(FamilyMember member) {

    }
}

class Test {
    public static void main(String[] args) {

        FamilyMember father = new Parents("爸爸");

        Parents grandF = new Parents("爷爷");

        Children child = new Children("孩子");

        father.setParent(grandF);
        //grandF.addChildren(father);
        father.setChildren(child);
        // 输出孩子信息
        printInfo(grandF);
        // 输出父类信息
        father.getParent().getInfo();
    }

    public static void printInfo(FamilyMember member) {
        for (FamilyMember children : member.getChildren()) {
            if (children instanceof Children) {
                children.getInfo();
            } else {
                children.getInfo();
                printInfo(children);
            }
        }
    }
}
输出：
爸爸
孩子
爷爷