Java使用访问者模式解决公司层级结构图问题详解

package com.demo.structure;import com.demo.visitor.IVisitor;/** * 抽象员工类 * * @author * */public abstract class Staff {  // 员工号  protected String no;  // 职工名字  protected String name;  // 职位  protected String position;  // 薪资  protected float salary;  // 私有属性 长度字符串  private int length;  // 构造方法  public Staff(String no, String name, String position, float salary) {    this.no = no;    this.name = name;    this.position = position;    this.salary = salary;    // 计算总字节长度    this.length += (no == null || "".equals(no.trim())) ? 0        : no.getBytes().length;    this.length += (name == null || "".equals(name.trim())) ? 0 : name        .getBytes().length;    this.length += (position == null || "".equals(position.trim())) ? 0        : position.getBytes().length;    this.length += String.valueOf(salary).getBytes().length;  }  // 获得用户基本信息  public void printUserBaseInfo() {    System.out.println("-|" + this.no + " " + this.name + " "        + this.position + " " + this.salary);  }  // 添加员工信息  public abstract void add(Staff staff);  // 删除员工  public abstract Staff remove(String no);  // 接收访问者对象  public abstract void accept(IVisitor visitor);  public String getNo() {    return no;  }  public void setNo(String no) {    this.no = no;  }  public String getName() {    return name;  }  public void setName(String name) {    this.name = name;  }  public String getPosition() {    return position;  }  public void setPosition(String position) {    this.position = position;  }  public float getSalary() {    return salary;  }  public void setSalary(float salary) {    this.salary = salary;  }}

package com.demo.structure;import java.util.ArrayList;import com.demo.visitor.IVisitor;/** * 管理人员（手下有其他员工的人） * * @author * */public class Manager extends Staff {  // 存储手下员工信息  private final ArrayList<Staff> arrayList = new ArrayList<Staff>();  // 构造方法  public Manager(String no, String name, String position, float salary) {    super(no, name, position, salary);  }  /**   * 增加一个员工   */  @Override  public void add(Staff staff) {    this.arrayList.add(staff);  }  /**   * 删除员工信息   */  @Override  public Staff remove(String no) {    Staff staff = null;    if (no != null && !"".equals(no.trim())) {      for (int i = 0; i < this.arrayList.size(); i++) {        if (this.arrayList.get(i) == null) {          continue;        }        if (no.equals(this.arrayList.get(i).getNo())) {          staff = this.arrayList.remove(i);          break;        }      }    }    return staff;  }  // 接收访问者对象  @Override  public void accept(IVisitor visitor) {    // 访问自身    visitor.visit(this);    // 遍历list列表中的各个元素对象，接收访问者对象    for (int i = 0; i < this.arrayList.size(); i++) {      if (this.arrayList.get(i) == null) {        continue;      }      // 接收访问者对象      this.arrayList.get(i).accept(visitor);    }  }}

package com.demo.structure;import com.demo.visitor.IVisitor;/** * 普通员工（真正干活的人） * * @author * */public class Employees extends Staff{  // 构造方法  public Employees(String no, String name, String position, float salary)  {    super(no, name, position, salary);  }  /**   * 添加员工信息   */  @Override  public void add(Staff staff)  {    return;  }  /**   * 删除员工信息   */  @Override  public Staff remove(String no)  {    // 直接返回null    return null;  }  // 接收访问者对象  public void accept(IVisitor visitor)  {    visitor.visit(this);  }}

package com.demo.visitor;import com.demo.structure.Employees;import com.demo.structure.Manager;/** * 访问者接口 * * @author * */public interface IVisitor {  // 访问管理者  public void visit(Manager manager);  // 访问普通员工  public void visit(Employees employees);}

package com.demo.visitor;import com.demo.structure.Employees;import com.demo.structure.Manager;/** * 打印基本信息访问者 * * @author * */public class PrintBaseInfoVisitor implements IVisitor {  /**   * 访问管理者对象   */  public void visit(Manager manager) {    System.out.print("- 管理者：");    manager.printUserBaseInfo();  }  /**   * 访问普通员工对象   */  public void visit(Employees employees) {    System.out.print("- 一般员工：");    employees.printUserBaseInfo();  }}

package com.demo.visitor;/** * 计算薪资访问者 * * @author * */public interface ISalaryVisitor extends IVisitor {  // 统计管理者薪资情况  public void printManagerTotalSalary();  // 统计一般员工薪资情况  public void printEmployeesTotalSalary();  // 统计所有员工薪资情况  public void printTotalSalary();}

package com.demo.visitor;import com.demo.structure.Employees;import com.demo.structure.Manager;/** * 计算薪资访问者具体实现 * * @author * */public class SalaryVisitor implements ISalaryVisitor {  // 管理者薪资总和  private float managerSalary;  // 普通员工薪资总和  private float employeesSalary;  public SalaryVisitor() {    managerSalary = 0;    employeesSalary = 0;  }  // 访问管理者  public void visit(Manager manager) {    managerSalary += manager.getSalary();  }  // 访问普通员工  public void visit(Employees employees) {    employeesSalary += employees.getSalary();  }  // 统计一般员工薪资情况  public void printEmployeesTotalSalary() {    System.out.println("一般员工薪资总和：" + employeesSalary);  }  // 统计管理者薪资情况  public void printManagerTotalSalary() {    System.out.println("管理者薪资总和：" + managerSalary);  }  // 统计所有员工薪资情况  public void printTotalSalary() {    System.out.println("员工薪资总和：" + (managerSalary + employeesSalary));  }}

package com.demo;import com.demo.structure.Employees;import com.demo.structure.Manager;import com.demo.structure.Staff;import com.demo.visitor.PrintBaseInfoVisitor;import com.demo.visitor.SalaryVisitor;/** * 主应用程序 * * @author * */public class Client {  /**   * @param args   */  public static void main(String[] args) {    // 公司CEO    Staff boss = new Manager("1", "大老板", "CEO", 100000);    /**     * CEO手下有若干部门经理     */    // 财务部经理    Staff financeManager = new Manager("11", "张总", "财务部经理", 60000);    // 人事部经理    Staff personnelManager = new Manager("12", "王总", "人事部经理", 60000);    // 技术部经理    Staff technicalManager = new Manager("13", "陈总", "技术部经理", 60000);    /**     * 技术部门还有助理和若干主管     */    // 技术部门助理    Staff deptAssistant = new Manager("1301", "王助理", "部门助理", 20000);    // 技术部门主管1    Staff deptManager1 = new Manager("1302", "主管1", "技术主管", 30000);    /**     * 技术主管deptManager1 下面还有软件工程师（最终干活的人）     */    Staff softwareEngineer1 = new Employees("1302001", "张三", "软件工程师", 5000);    Staff softwareEngineer2 = new Employees("1302002", "李四", "软件工程师", 5500);    Staff softwareEngineer3 = new Employees("1302003", "王五", "软件工程师", 4500);    // 为技术主管1添加员工信息    deptManager1.add(softwareEngineer1);    deptManager1.add(softwareEngineer2);    deptManager1.add(softwareEngineer3);    // 技术部门主管2    Staff deptManager2 = new Manager("1303", "主管2", "技术主管", 30000);    // 为技术部经理 添加：部门助理、技术主管1和技术主管2    technicalManager.add(deptAssistant);    technicalManager.add(deptManager1);    technicalManager.add(deptManager2);    // 市场部经理    Staff marketingManager = new Manager("14", "吴总", "市场部经理", 60000);    // 为CEO 添加：财务部经理、人事部经理、技术部经理和市场部经理    boss.add(financeManager);    boss.add(personnelManager);    boss.add(technicalManager);    boss.add(marketingManager);    // 打印CEO 信息    // boss.printUserBaseInfo();    // 打印CEO 手下员工信息    boss.accept(new PrintBaseInfoVisitor());    /**     * 统计员工薪资情况     */    // 创建统计员工薪资访问者    SalaryVisitor visitor = new SalaryVisitor();    // 让大老板接受该访问者    boss.accept(visitor);    // 管理者薪资统计情况    visitor.printManagerTotalSalary();    // 一般员工薪资统计情况    visitor.printEmployeesTotalSalary();    // 所有员工薪资统计情况    visitor.printTotalSalary();  }}