模拟数据库操作记录

发布于 2021-04-05  624 次阅读


package com.yang.dao;

import com.yang.mapper.DepartmentMapper;
import com.yang.pojo.Department;
import com.yang.pojo.Employee;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Repository;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

//员工Dao
//标识为Bean
@Repository
public class EmployeeDao {

    //模拟数据库中的数据
    private static Map<Integer, Employee> employees = null;
    //员工有所属部门,注入部门
    @Autowired
    DepartmentMapper departmentMapper;

    static {
        employees = new HashMap<Integer, Employee>();//创建一个部门表

        employees.put(1001, new Employee(1001, "AA", "A1606598203@qq.com", 0, new Department(101, "教学部")));
        employees.put(1002, new Employee(1002, "BB", "B1606598203@qq.com", 1, new Department(102, "市场部")));
        employees.put(1003, new Employee(1003, "CC", "C1606598203@qq.com", 0, new Department(103, "教研部")));
        employees.put(1004, new Employee(1004, "DD", "D1606598203@qq.com", 1, new Department(104, "运营部")));
        employees.put(1005, new Employee(1005, "EE", "E1606598203@qq.com", 0, new Department(105, "后勤部")));
    }

    //增加一个员工,主键自增!
    private static Integer initID = 1006;

    //增加一个员工
    public void save(Employee employee) {
        if(employee.getId()==null){
            employee.setId(initID++);
        }

        employee.setDepartment(departmentMapper.getDepartmentById(employee.getDepartment().getId()));

        employees.put(employee.getId(),employee);
    }

    //查询全部员工信息
    public Collection<Employee> getAll(){

        return employees.values();
    }

    //通过ID查询员工
    public Employee getEmployeeById(Integer id){

        return employees.get(id);
    }

    //通过ID删除员工
    public void delete(Integer id){

        employees.remove(id);
    }

}

她喜欢所以就做咯