Java实现(09)——组合模式(Composite Pattern)

组合模式将一组对象以树的形式看成一个整体。

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

/**
 * Created by wang on 16-12-17.
 */
public class Employee {
    private String name;
    private String dept;
    private int salary;
    private List<Employee> subordinates;
    public Employee(String name, String dept, int sal) {
        this.name = name;
        this.dept = dept;
        this.salary = sal;
        subordinates = new ArrayList<Employee>();
    }
    public void add(Employee e) {
        subordinates.add(e);
    }
    public void remove(Employee e) {
        subordinates.remove(e);
    }
    public List<Employee> getSubordinates() {
        return subordinates;
    }
    public String toString() {
        return ("Employee : [ Name : " + name + ", dept : " + dept + ", salary : " + salary + " ]");
    }
}

/**
 * Created by wang on 16-12-17.
 */
public class Demo {
    public static void main(String[] args) {
        Employee CEO = new Employee("John", "CEO", 30000);
        Employee headSales = new Employee("Robert", "Head Sales", 20000);
        Employee headMarketing = new Employee("Michel", "Head Marketing", 20000);
        Employee clerk1 = new Employee("Laura", "Marketing", 10000);
        Employee clerk2 = new Employee("Bob", "Marketing", 10000);
        Employee salesExecutive1 = new Employee("Richard", "Sales", 10000);
        Employee salesExecutive2 = new Employee("Rob", "Sales", 10000);
        CEO.add(headSales);
        CEO.add(headMarketing);
        headSales.add(salesExecutive1);
        headSales.add(salesExecutive2);
        headMarketing.add(clerk1);
        headMarketing.add(clerk2);
        System.out.println(CEO);
        for (Employee headEmployee : CEO.getSubordinates()) {
            System.out.println(headEmployee);
            for (Employee employee : headEmployee.getSubordinates()) {
                System.out.println(employee);
            }
        }
    }
}

输出：

Employee : [ Name : John, dept : CEO, salary : 30000 ]
Employee : [ Name : Robert, dept : Head Sales, salary : 20000 ]
Employee : [ Name : Richard, dept : Sales, salary : 10000 ]
Employee : [ Name : Rob, dept : Sales, salary : 10000 ]
Employee : [ Name : Michel, dept : Head Marketing, salary : 20000 ]
Employee : [ Name : Laura, dept : Marketing, salary : 10000 ]
Employee : [ Name : Bob, dept : Marketing, salary : 10000 ]