/**
 * 個(gè)體和整體的抽象
 */
public abstract class Component {
    // 個(gè)體和整體都有的共享
    public void doSomething(){
        // 通用業(yè)務(wù)邏輯
        System.out.println("通用業(yè)務(wù)邏輯");
    }

/**
 * 樹(shù)枝節(jié)點(diǎn)
 */
public class Composite extends Component{
    // 構(gòu)件容器
    private ArrayList<Component> componentArrayList = new ArrayList<>();

    // 增加一個(gè)葉子節(jié)點(diǎn)或者樹(shù)枝節(jié)點(diǎn)
    public void add(Component component){
        this.componentArrayList.add(component);
    }

    // 刪除一個(gè)葉子節(jié)點(diǎn)或者樹(shù)枝節(jié)點(diǎn)
    public void remove(Component component){
        this.componentArrayList.remove(component);
    }

    // 獲取分支下所有葉子節(jié)點(diǎn)和樹(shù)枝節(jié)點(diǎn)
    public List<Component> getChildren(){
        return this.componentArrayList;
    }
}

/**
 * 葉子節(jié)點(diǎn)
 */
public class Leaf extends Component {

    // 覆寫(xiě)父類(lèi)方法
    @Override
    public void doSomething() {
        // 葉子節(jié)點(diǎn)邏輯
        System.out.println("葉子節(jié)點(diǎn)邏輯");
    }
}

public class ClientTest {

    public static void main(String[] args) {
        // 創(chuàng)建一個(gè)根節(jié)點(diǎn)
        Composite root = new Composite();
        root.doSomething();
        // 創(chuàng)建一個(gè)樹(shù)枝構(gòu)件
        Composite branch = new Composite();
        // 創(chuàng)建一個(gè)葉子節(jié)點(diǎn)
        Leaf leaf = new Leaf();

        // 串聯(lián)起來(lái)
        root.add(branch);
        branch.add(leaf);

        display(root);
    }

    // 通過(guò)遞歸遍歷數(shù)
    public static void display(Composite root){
        for(Component c : root.getChildren()){
            if(c instanceof Leaf){ // 葉子節(jié)點(diǎn)
                c.doSomething();
            }else{
                display((Composite) c);
            }
        }
    }
}

/**
 * 部門(mén)類(lèi)和員工類(lèi)的抽象類(lèi)
 */
public abstract class HumanResource {
    protected long id;
    protected double salary;

    public HumanResource(long id){
        this.id = id;
    }

    public long getId(){
        return id;
    }

    public abstract double calculateSalary();

}

public class Department extends HumanResource{
    private List<HumanResource> subNodes = new ArrayList<>();

    public Department(long id){
        super(id);
    }

    @Override
    public double calculateSalary() {
        double totalSalary = 0d;
        for (HumanResource hr : subNodes){
            totalSalary += hr.calculateSalary();
        }
        this.salary = totalSalary;
        return totalSalary;
    }

    public void addSubNode(HumanResource humanResource){
        subNodes.add(humanResource);
    }
}

public class Employee extends HumanResource{
    public Employee(long id,double salary){
        super(id);
        this.salary = salary;
    }

    @Override
    public double calculateSalary() {
        return salary;
    }
}

public class PersonClientTest {
    private static final long ORGANIZATION_ROOT_ID = 1;

    public static void main(String[] args) {
        // 創(chuàng)建總部門(mén)
        Department root = new Department(ORGANIZATION_ROOT_ID);

        // 創(chuàng)建子部門(mén)
        Department branch = new Department(2L);

        // 創(chuàng)建員工
        Employee employee1 = new Employee(21L,2000);
        Employee employee2 = new Employee(22L,4000);

        root.addSubNode(branch);
        branch.addSubNode(employee1);
        branch.addSubNode(employee2);

        double v = root.calculateSalary();
        System.out.println(v);
    }

    private void buildOrganization(Department department){
        // 根據(jù) 部門(mén)id 查詢(xún)數(shù)據(jù)庫(kù) 所有下屬部門(mén) id
        // List<Long> subDepartmentIds = departmentRepo.getSubDepartmentIds(department.getId());
        List<Long> subDepartmentIds = new ArrayList<>();

        for (Long subDepartmentId : subDepartmentIds){
            Department subDepartment = new Department(subDepartmentId);
            department.addSubNode(subDepartment);
            buildOrganization(subDepartment);
        }

        // 根據(jù)部門(mén)id 查詢(xún)數(shù)據(jù)庫(kù) 其關(guān)聯(lián)員工所有 id
        // List<Long> employeeIds = employeeRepo.getDepartmentEmployeeIds(department.getId());
        List<Long> employeeIds = new ArrayList<>();
        for (Long employeeId : employeeIds){
            // 根據(jù) employeeId 查詢(xún)數(shù)據(jù)庫(kù)得到 salary
            // 假設(shè)為 1000
            double salary = 1000d;
            department.addSubNode(new Employee(employeeId,salary));
        }

    }
}