Java 练习(开发团队调度软件三)

Data.java

package com.klvchen.team.service;


public class Data {
    public static final int EMPLOYEE = 10;
    public static final int PROGRAMMER = 11;
    public static final int DESIGNER = 12;
    public static final int ARCHITECT = 13;

    public static final int PC = 21;
    public static final int NOTEBOOK = 22;
    public static final int PRINTER = 23;

    //Employee  :  10, id, name, age, salary
    //Programmer:  11, id, name, age, salary
    //Designer  :  12, id, name, age, salary, bonus
    //Architect :  13, id, name, age, salary, bonus, stock
    public static final String[][] EMPLOYEES = {
        {"10", "1", "马云", "22", "3000"},
        {"13", "2", "马化腾", "32", "18000", "15000", "2000"},
        {"11", "3", "李彦宏", "23", "7000"},
        {"11", "4", "刘强东", "24", "7300"},
        {"12", "5", "雷军", "28", "10000", "5000"},
        {"11", "6", "任志强", "22", "6800"},
        {"12", "7", "柳传志", "29", "10800","5200"},
        {"13", "8", "杨元庆", "30", "19800", "15000", "2500"},
        {"12", "9", "史玉柱", "26", "9800", "5500"},
        {"11", "10", "丁磊", "21", "6600"},
        {"11", "11", "张朝阳", "25", "7100"},
        {"12", "12", "杨致远", "27", "9600", "4800"}
    };
    
    //如下的EQIPMENTS数组与上面的EMPLOYEES数组元素一一对应
    //PC      :21, model, display
    //NoteBook:22, model, price
    //Printer :23, type, name
    public static final String[][] EQUIPMENTS = {
        {},
        {"22", "联想T4", "6000"},
        {"21", "戴尔", "NEC17寸"},
        {"21", "戴尔", "三星 17寸"},
        {"23", "激光", "佳能 2900"},
        {"21", "华硕", "三星 17寸"},
        {"21", "华硕", "三星 17寸"},
        {"23", "针式", "爱普生20K"},
        {"22", "惠普m6", "5800"},
        {"21", "戴尔", "NEC 17寸"},
        {"21", "华硕","三星 17寸"},
        {"22", "惠普m6", "5800"}
    };
}

NameListService.java

package com.klvchen.team.service;

import com.klvchen.team.domain.Architect;
import com.klvchen.team.domain.Designer;
import com.klvchen.team.domain.Employee;
import com.klvchen.team.domain.Equipment;
import com.klvchen.team.domain.NoteBook;
import com.klvchen.team.domain.PC;
import com.klvchen.team.domain.Printer;
import com.klvchen.team.domain.Programmer;

import static com.klvchen.team.service.Data.*;

// 负责将 Data 中的数据封装到 Employee[] 数组中，同时提供相关操作 Employee[] 的方法。

public class NameListService {
	
	private Employee[] employees;

	public NameListService() {
		employees = new Employee[EMPLOYEES.length];
		
		for(int i = 0; i < employees.length; i++) {
			//获取员工的类型
			int type = Integer.parseInt(EMPLOYEES[i][0]);
			
			//获取Employee的4个基本信息
			int id = Integer.parseInt(EMPLOYEES[i][1]);
			String name = EMPLOYEES[i][2];
			int age = Integer.parseInt(EMPLOYEES[i][3]);
			double salary = Double.parseDouble(EMPLOYEES[i][4]);
			
			Equipment equipment;
			double bonus;
			int stock;
			
			switch(type) {
			case EMPLOYEE:
				employees[i] = new Employee(id, name, age, salary);
				break;
			case PROGRAMMER:
				equipment = createEquipment(i);
				employees[i] = new Programmer(id, name, age, salary, equipment);
				break;
			case DESIGNER:
				equipment = createEquipment(i);
				bonus = Double.parseDouble(EMPLOYEES[i][5]);
				employees[i] = new Designer(id, name, age, salary, equipment, bonus);	
				break;
			case ARCHITECT:
				equipment = createEquipment(i);
				bonus = Double.parseDouble(EMPLOYEES[i][5]);
				stock = Integer.parseInt(EMPLOYEES[i][6]);
				employees[i] = new Architect(id, name, age, salary, equipment, bonus, stock);	
				break;
			}
		}
		
	}
	
	//获取指定 Index 上的员工的设备
	private Equipment createEquipment(int index) {
		
		int key = Integer.parseInt(EQUIPMENTS[index][0]);
		
		String modelOrName = EQUIPMENTS[index][1];
		switch(key) {
		case PC:
			String display = EQUIPMENTS[index][2];
			return new PC(modelOrName, display);
		case NOTEBOOK:
			double price = Double.parseDouble(EQUIPMENTS[index][2]);
			return new NoteBook(modelOrName, price);
		case PRINTER:
			String type = EQUIPMENTS[index][2];
			return new Printer(modelOrName, type);
		}
		return null;
	}
	
	//获取所有的员工
	public Employee[] getAllEmployees() {
		return employees;
	}

	//获取指定ID的员工对象
	public Employee getEmployee(int id) throws TeamException {
		for(int i = 0; i< employees.length; i++) {
			if(employees[i].getId() == id) {
				return employees[i];
			}       
		}
		
		throw new TeamException("找不到指定的员工");
	}
	
}

Status.java

package com.klvchen.team.service;
/*
 * 表示员工的状态
 */

public class Status {
	private final String NAME;
	
	private Status(String name) {
		this.NAME = name;
	}
	
	public static final Status FREE = new Status("FREE");
	public static final Status BUSY = new Status("BUSY");
	public static final Status VOCATION = new Status("VOCATION");
	
	public String getName() {
		return NAME;
	}
	
	@Override
		public String toString() {
			return NAME;
		}

}

TeamException.java

package com.klvchen.team.service;

public class TeamException extends Exception {
	static final long serialVersionUID = -333876149948L;
	
	public TeamException() {
		super();
	}
	
	public TeamException(String msg) {
		super(msg);
	}

}

TeamService.java

package com.klvchen.team.service;

import com.klvchen.team.domain.Architect;
import com.klvchen.team.domain.Designer;
import com.klvchen.team.domain.Employee;
import com.klvchen.team.domain.Programmer;

// 关于开发团队程序的管理，添加，删除等

public class TeamService {

	private static int counter = 1;   //给 memberID赋值使用
	private final int MAX_MEMBER = 5; //限制开发团队的人数
	private Programmer[] team = new Programmer[MAX_MEMBER]; //保存开发团队成员
	private int total; //记录开发团队中的实际的人数
	
	public TeamService() {
		super();
	}
	
	//获取开发团队中的所有成员
	public Programmer[] getTeam() {
		Programmer[] team = new Programmer[total];
		for(int i = 0; i< team.length; i++) {
			team[i] = this.team[i];
		}
		return team;
	}
	
	//将指定的员工添加到开发团队中
	public void addMember(Employee e) throws TeamException {
		//成员已满，无法添加
		if(total >= MAX_MEMBER) {
			throw new TeamException("成员已满，无法添加");
		}
		//该成员不是开发人员，无法添加
		if(!(e instanceof Programmer)) {
			throw new TeamException("该成员不是开发人员，无法添加");
		}
		//该员工已在本开发团队中
		if(isExist(e)) {
			throw new TeamException("该员工已在本开发团队中");
		}
		
		//该员工已是某团队成员
		
		//该员正在休假，无法添加
		Programmer p = (Programmer)e; //一定不会出现 classCastException
		if("BUSY".equalsIgnoreCase(p.getStatus().getName())) {
			throw new TeamException("该员工已是某团队成员");
		}else if("VOCATION".equalsIgnoreCase(p.getStatus().getName())) {
			throw new TeamException("该员正在休假，无法添加");
		}
		
		//获取team已有成员中架构师,设计师，程序员的人数
		int numOfArch = 0, numOfDes = 0, numOfPro = 0;
		for(int i = 0; i< total; i++) {
			if(team[i] instanceof Architect) {
				numOfArch++;
			}else if (team[i] instanceof Designer) {
				numOfDes++;
			}else if (team[i] instanceof Programmer) {
				numOfPro++;
			}
		}
		
		if(p instanceof Architect) {
			if(numOfArch >= 1) {
				throw new TeamException("团队中至多只能有一名架构师");
			} 
		}else if(p instanceof Designer) {
			if(numOfDes >= 2) {
				throw new TeamException("团队中至多只能有两名设计师");
			}
		}else if(p instanceof Programmer) {
			if(numOfPro >= 3) {
				throw new TeamException("团队中至多只能有三名程序员");
			}
		}
		
		//将p(或e)添加到现有的 team 中
		team[total++] = p;
		//p的属性赋值
		p.setStatus(Status.BUSY);
		p.setMemberId(counter++);

		
	}
	
	//判断指定的员工是否已经存在于现有的开发团队中
	private boolean isExist(Employee e) {
		for(int i = 0; i < total; i++) {
			return team[i].getId() == e.getId();
		}
		return false;
	}

	//从团队中删除成员
	public void removeMember(int memberId) throws TeamException {
		int i = 0;
		for(; i < total; i++) {
			if(team[i].getMemberId() == memberId) {
				team[i].setStatus(Status.FREE);
				break;
			}
		}
		
		//无法找到指定 memberId的情况
		if(i == total) {
			throw new TeamException("找不到指定 memberId 的员工，删除失败");
		}
		
		//后一个元素覆盖前一个元素，实现删除操作
		for(int j = i + 1; j < total; j++) {
			team[j - 1] = team[j];
		}
		
		//写法一：
//		team[total-1] = null;
//		total--;
		//写法二：
		team[--total] = null;
		
		
	}
	
}