（尚硅谷练习案例）开发团队人员调度软件

记：TSUtility.java 是工具类，是事先提供好的，Data.java是开发团队人员数据信息，也是事先提供好的，联系项目的整体结构如下，记录自己实现的代码，敲代码敲得头晕，不过敲完还是挺有成就感的：

package com.atguigu.java.domain;

/**
 * @author fanjunjie
 * @create 2022-05-08 11:28
 */
public class Architect extends Designer{

    private int stock;

    public Architect(){

    }

    public Architect(int id, String name, int age, double salary, Equipment equipment, double bonus, int stock) {
        super(id, name, age, salary, equipment, bonus);
        this.stock = stock;
    }

    public int getStock() {
        return stock;
    }

    public void setStock(int stock) {
        this.stock = stock;
    }

    @Override
    public String toString() {
        return getDetail()+"\t"+getStatus()+"\t"+getBonus()+"\t"+stock+"\t"+getEquipment().getDescription();
    }

    public String gettDetail(){
        return getTeamDetail()+"\t架构师\t"+getBonus()+"\t"+stock;
    }
}

package com.atguigu.java.domain;

/**
 * @author fanjunjie
 * @create 2022-05-08 11:28
 */
public class Designer extends Programmer {

    private double bonus;

    public Designer(){

    }

    public Designer(int id, String name, int age, double salary, Equipment equipment, double bonus) {
        super(id, name, age, salary, equipment);
        this.bonus = bonus;
    }

    public double getBonus() {
        return bonus;
    }

    public void setBonus(double bonus) {
        this.bonus = bonus;
    }


    @Override
    public String toString() {
        return getDetail()+"\t"+ getStatus()+"\t"+bonus+"\t\t"+getEquipment().getDescription();
    }

    public String gettDetail(){
        return getTeamDetail()+"\t设计师\t"+bonus;
    }
}

package com.atguigu.java.domain;

/**
 * @author fanjunjie
 * @create 2022-05-08 11:27
 */
public class Employee {
    private int id;
    private String name;
    private int age;
    private double salary;

    public Employee(){

    }

    public Employee(int id, String name, int age, double salary) {
        this.id = id;
        this.name = name;
        this.age = age;
        this.salary = salary;
    }

    public int getId() {
        return id;
    }

    public void setId(int id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

    public double getSalary() {
        return salary;
    }

    public void setSalary(double salary) {
        this.salary = salary;
    }

    public String getDetail(){
        return id+"\t"+name+"\t"+age+"\t"+salary;
    }

    @Override
    public String toString() {

        return getDetail();
    }
}

package com.atguigu.java.domain;

/**
 * @author fanjunjie
 * @create 2022-05-08 11:28
 */
public interface Equipment {

    String getDescription();
}

package com.atguigu.java.domain;

/**
 * @author fanjunjie
 * @create 2022-05-08 11:29
 */
public class NoteBook implements Equipment {

    private String model;
    private double price;

    public NoteBook(){

    }
    public NoteBook(String model, double price) {
        this.model = model;
        this.price = price;
    }

    public String getModel() {
        return model;
    }

    public void setModel(String model) {
        this.model = model;
    }

    public double getPrice() {
        return price;
    }

    public void setPrice(double price) {
        this.price = price;
    }

    @Override
    public String getDescription() {
        return model+"("+price+")";
    }

}

package com.atguigu.java.domain;

/**
 * @author fanjunjie
 * @create 2022-05-08 11:30
 */
public class PC implements Equipment {

    private String model;
    private String display;

    public PC(){

    }

    public PC(String model, String display) {
        this.model = model;
        this.display = display;
    }

    public String getModel() {
        return model;
    }

    public void setModel(String model) {
        this.model = model;
    }

    public String getDisplay() {
        return display;
    }

    public void setDisplay(String display) {
        this.display = display;
    }

    @Override
    public String getDescription() {
        return model+"("+display+")";
    }
}

package com.atguigu.java.domain;

/**
 * @author fanjunjie
 * @create 2022-05-08 11:30
 */
public class Printer implements Equipment {

    private String name;
    private String type;

    public Printer(){

    }
    public Printer(String name, String type) {
        this.name = name;
        this.type = type;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public String getType() {
        return type;
    }

    public void setType(String type) {
        this.type = type;
    }

    @Override
    public String getDescription() {
        return name+"("+type+")";
    }
}

package com.atguigu.java.domain;

import com.atguigu.java.service.Status;

/**
 * @author fanjunjie
 * @create 2022-05-08 11:27
 */
public class Programmer extends Employee{
    private int memberId;
    private Status status=Status.FREE;
    private Equipment equipment;

    public Programmer(){

    }

    public Programmer(int id, String name, int age, double salary, Equipment equipment) {
        super(id, name, age, salary);
        this.equipment = equipment;
    }

    public int getMemberId() {
        return memberId;
    }

    public void setMemberId(int memberId) {
        this.memberId = memberId;
    }

    public Status getStatus() {
        return status;
    }

    public void setStatus(Status status) {
        this.status = status;
    }

    public Equipment getEquipment() {
        return equipment;
    }

    public void setEquipment(Equipment equipment) {
        this.equipment = equipment;
    }

    public String getTeamDetail(){
        return memberId+"/"+getId()+"\t"+getName()+"\t"+getAge()+"\t"+getSalary();
    }

    @Override
    public String toString() {
        return getDetail()+"\t"+status+"\t\t\t"+equipment.getDescription();
    }

    public String gettDetail(){
        return getTeamDetail()+"\t程序员";
    }
}

package com.atguigu.java.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"}
    };
    
    //如下的EQUIPMENTS数组与上面的EMPLOYEES数组元素一一对应
    //PC      :21, model, display
    //NoteBook:22, model, price
    //Printer :23, name, type 
    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"}
    };
}

package com.atguigu.java.service;

import com.atguigu.java.domain.*;
import com.atguigu.java.domain.PC;

import static com.atguigu.java.service.Data.*;

/**
 * @author fanjunjie
 * @create 2022-05-08 11:56
 */
public class NameListService {
//    @Test
    public static void main(String[] args) {
//        NameListService nameListService=new NameListService();
//        nameListService.getAllEmployee();

    }

    private Employee employees[];

    //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 NameListService(){
        employees=new Employee[EMPLOYEES.length];
        for(int i=0;i<employees.length;i++){
            int type=Integer.parseInt(EMPLOYEES[i][0]);
            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;
            switch (type){
                case EMPLOYEE:
                    employees[i]=new Employee(id,name,age,salary);
                    break;
                case PROGRAMMER:
                    equipment=creatEquipment(i);
                    employees[i]=new Programmer(id,name,age,salary,equipment);
                    break;
                case DESIGNER:
                    equipment=creatEquipment(i);
                    bonus=Double.parseDouble(EMPLOYEES[i][5]);
                    employees[i]=new Designer(id,name,age,salary,equipment,bonus);
                    break;
                case ARCHITECT:
                    equipment=creatEquipment(i);
                    int stock=Integer.parseInt(EMPLOYEES[i][6]);
                    bonus=Double.parseDouble(EMPLOYEES[i][5]);
                    employees[i]=new Architect(id,name,age,salary,equipment,bonus,stock);
                    break;
            }


        }
    }

    public Equipment creatEquipment(int index){
        int type=Integer.parseInt(EQUIPMENTS[index][0]);
        String model=EQUIPMENTS[index][1];
        switch (type){
            case PC:
                String display=EQUIPMENTS[index][2];
                PC pc=new PC(model,display);
                return pc;
            case NOTEBOOK:
                double price=Double.parseDouble(EQUIPMENTS[index][2]);
                return new NoteBook(model,price);
            case PRINTER:
                String name=EQUIPMENTS[index][1];
                String typep=EQUIPMENTS[index][2];
                return new Printer(name,typep);
        }
        return null;
    }

    public Employee[] getAllEmployee() {

        return employees;
    }

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

package com.atguigu.java.service;

/**
 * @author fanjunjie
 * @create 2022-05-08 11:42
 */
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 VOCATION=new Status("VOCATION");
    public static final Status BUSY=new Status("BUSY");

    public String getNAME() {
        return NAME;
    }

    @Override
    public String toString() {
        return NAME;
    }
}

package com.atguigu.java.service;

/**
 * @author fanjunjie
 * @create 2022-05-08 12:07
 */
public class TeamException extends Exception {
    static final long serialVersionUID = -33875169931248L;

    public TeamException(){
        super();
    }

    public TeamException(String message){
        super(message);
    }

}

package com.atguigu.java.service;

import com.atguigu.java.domain.Architect;
import com.atguigu.java.domain.Designer;
import com.atguigu.java.domain.Employee;
import com.atguigu.java.domain.Programmer;

/**
 * @author fanjunjie
 * @create 2022-05-08 11:57
 */
public class TeamService {
    public static void main(String[] args) {

    }

    private static int counter=1;
    private static final int MAX_MEMBER=5;
    private int total=0;
    private Programmer team[]=new Programmer[MAX_MEMBER];

    public TeamService(){

    }

    public Programmer[] getTeam() {
        Programmer team1[]=new Programmer[total];
        for(int i=0;i<total;i++){
            team1[i]=team[i];
        }
        return team1;
    }

    public void addMember(Employee e) throws TeamException{

// 成员已满，无法添加
        if(total>=MAX_MEMBER)
            throw new TeamException("成员已满，无法添加！");
//        该成员不是开发人员，无法添加
        if(!(e instanceof Programmer))
            throw new TeamException("该成员不是开发人员，无法添加");
        Programmer p=(Programmer)e;
// 该员工已在本开发团队中
        for(int i=0;i<total;i++) {
            if(team[i].getId()==p.getId()){
                throw new TeamException("该员工已在本开发团队中");
            }
        }
//        该员工已是某团队成员
//        该员正在休假，无法添加
        if(p.getStatus().getNAME().equals("BUSY")){
            throw new TeamException("该员工已是其他团队成员");
        }else if(p.getStatus().getNAME().equals("VOCATION")){
            throw new TeamException("该员工正在休假，无法添加");
        }
//        团队中至多只能有一名架构师
//        团队中至多只能有两名设计师
//        团队中至多只能有三名程序员
        int numOfArch=0,numOfDesi=0,numOfPro=0;
        for(int i=0;i<total;i++){
            if(team[i] instanceof Architect) numOfArch++;
            else if(team[i] instanceof Designer) numOfDesi++;
            else if(team[i] instanceof Programmer) numOfPro++;
        }
        if(p instanceof Architect){
            if(numOfArch>=1)
                throw new TeamException("团队中至多只能有一名架构师");
        }else if(p instanceof Designer){
            if(numOfDesi>=2)
                throw new TeamException("团队中至多只能有两名设计师");
        }else if(p instanceof Programmer){
            if(numOfPro>=3)
                throw new TeamException("团队中至多只能有三名程序员");
        }

        p.setStatus(Status.BUSY);
        p.setMemberId(counter++);
        team[total++]=p;

    }

    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;
            }
        }
        if(i==total)
            throw new TeamException("找不到指定memberId的员工，删除失败");
        for(int j=i;j<total-1;j++){
            team[j]=team[j+1];
        }

        team[total--]=null;

    }

}

package com.atguigu.java.view;

import com.atguigu.java.domain.Employee;
import com.atguigu.java.domain.Programmer;
import com.atguigu.java.service.NameListService;
import com.atguigu.java.service.TeamException;
import com.atguigu.java.service.TeamService;

/**
 * @author fanjunjie
 * @create 2022-05-08 17:38
 */
public class TeamView {
    public static void main(String[] args) {
        TeamView teamView=new TeamView();
        teamView.enterMainMenu();
    }

    NameListService listSvc=new NameListService();
    TeamService teamSvc=new TeamService();

    public void enterMainMenu(){

        boolean flag=true;
        char key=0;
        while(flag){
            if(key!='1'){
                listAllEmployees();
            }

            System.out.println("1-团队列表 2-添加团队成员 3-删除团队成员 4-退出 请选择(1-4):");
            key=TSUtility.readMenuSelection();
            switch (key){
                case '1':
                    getTeam();
                    break;
                case '2':
                    addMember();
                    break;
                case '3':
                    deleteMember();
                    break;
                case '4':
                    System.out.println("是否确认退出(Y/N):");
                    char isExit=TSUtility.readConfirmSelection();
                    if(isExit=='Y'){
                        flag=false;
                        break;
                    }
            }
        }

    }

    public void listAllEmployees(){
        System.out.println("------------------------开发团队调度软件--------------------------\n");
        System.out.println("ID\t姓名\t年龄\t工资\t职位\t状态\t奖金\t股票\t领用设备");
        Employee employees[]=listSvc.getAllEmployee();
        for(int i=0;i<employees.length;i++){
            System.out.println(employees[i]+"\n");
        }
        System.out.println("----------------------------------------------------------------");
    }

    public void getTeam(){
        System.out.println("------------------------团队成员列表--------------------------\n");
        System.out.println("TID/ID\t姓名\t年龄\t工资\t职位\t奖金\t股票");
        Programmer programmer[]=teamSvc.getTeam();
        for(int i=0;i<programmer.length;i++){
            System.out.println(programmer[i].gettDetail());
        }
        System.out.println("------------------------------------------------------------");
    }

    public void addMember(){
        System.out.println("------------------------添加成员--------------------------");
        System.out.println("请输入要添加的员工ID：");
        int id=TSUtility.readInt();

        try{
            Employee a=listSvc.getEmployee(id);
            teamSvc.addMember(a);
        }catch (TeamException e){
            System.out.println(e.getMessage());
        }
        TSUtility.readReturn();

    }

    public void deleteMember(){
        System.out.println("------------------------删除成员--------------------------");
        System.out.println("请输入要添加的员工ID：");
        int id=TSUtility.readInt();
        System.out.println("确认是否删除(Y/N):");
        char isDele=TSUtility.readConfirmSelection();
        if(isDele=='N'){
            return;
        }

        try {
            teamSvc.removeMember(id);
        }catch (TeamException e){
            System.out.println(e.getMessage());
        }
        System.out.println("删除成功");

        TSUtility.readReturn();
    }

}

package com.atguigu.java.view;

import java.util.*;
/**
 * 
 * @Description 项目中提供了TSUtility.java类，可用来方便地实现键盘访问。
 * @author shkstart  Email:shkstart@126.com
 * @version 
 * @date 2019年2月12日上午12:02:58
 *
 */
public class TSUtility {
    private static Scanner scanner = new Scanner(System.in);
    /**
     * 
     * @Description 该方法读取键盘，如果用户键入’1’-’4’中的任意字符，则方法返回。返回值为用户键入字符。
     * @author shkstart
     * @date 2019年2月12日上午12:03:30
     * @return
     */
	public static char readMenuSelection() {
        char c;
        for (; ; ) {
            String str = readKeyBoard(1, false);
            c = str.charAt(0);
            if (c != '1' && c != '2' &&
                c != '3' && c != '4') {
                System.out.print("选择错误，请重新输入：");
            } else break;
        }
        return c;
    }
	/**
	 * 
	 * @Description 该方法提示并等待，直到用户按回车键后返回。
	 * @author shkstart
	 * @date 2019年2月12日上午12:03:50
	 */
    public static void readReturn() {
        System.out.print("按回车键继续...");
        readKeyBoard(100, true);
    }
    /**
     * 
     * @Description 该方法从键盘读取一个长度不超过2位的整数，并将其作为方法的返回值。
     * @author shkstart
     * @date 2019年2月12日上午12:04:04
     * @return
     */
    public static int readInt() {
        int n;
        for (; ; ) {
            String str = readKeyBoard(2, false);
            try {
                n = Integer.parseInt(str);
                break;
            } catch (NumberFormatException e) {
                System.out.print("数字输入错误，请重新输入：");
            }
        }
        return n;
    }
    /**
     * 
     * @Description 从键盘读取‘Y’或’N’，并将其作为方法的返回值。
     * @author shkstart
     * @date 2019年2月12日上午12:04:45
     * @return
     */
    public static char readConfirmSelection() {
        char c;
        for (; ; ) {
            String str = readKeyBoard(1, false).toUpperCase();
            c = str.charAt(0);
            if (c == 'Y' || c == 'N') {
                break;
            } else {
                System.out.print("选择错误，请重新输入：");
            }
        }
        return c;
    }

    private static String readKeyBoard(int limit, boolean blankReturn) {
        String line = "";

        while (scanner.hasNextLine()) {
            line = scanner.nextLine();
            if (line.length() == 0) {
                if (blankReturn) return line;
                else continue;
            }

            if (line.length() < 1 || line.length() > limit) {
                System.out.print("输入长度（不大于" + limit + "）错误，请重新输入：");
                continue;
            }
            break;
        }

        return line;
    }
}