孔祥富
2021-03-02 23921ae50a511f23791b9ba07b41f51ec36b1c71
febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpBaseInfoController.java
@@ -1,26 +1,56 @@
package cc.mrbird.febs.server.hr.controller;
import cc.mrbird.febs.server.hr.entity.EmpBaseInfo;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
import javax.servlet.http.HttpServletResponse;
import javax.validation.Valid;
import javax.validation.constraints.NotBlank;
import cc.mrbird.febs.common.core.entity.system.SystemUser;
import cc.mrbird.febs.server.hr.annotation.ControllerEndpoint;
import cc.mrbird.febs.server.hr.entity.EmpDimissionLog;
import cc.mrbird.febs.server.hr.entity.EmpJobChange;
import cc.mrbird.febs.server.hr.feign.IRemoteDeptService;
import cc.mrbird.febs.server.hr.feign.IRemoteUserService;
import cc.mrbird.febs.server.hr.service.IEmpBaseInfoService;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.wuwenze.poi.ExcelKit;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import cc.mrbird.febs.common.core.entity.FebsResponse;
import cc.mrbird.febs.common.core.entity.QueryRequest;
import cc.mrbird.febs.common.core.exception.FebsException;
import cc.mrbird.febs.common.core.utils.FebsUtil;
import cc.mrbird.febs.server.hr.entity.EmpBaseInfo;
import cc.mrbird.febs.server.hr.feign.IRemoteDeptService;
import cc.mrbird.febs.server.hr.feign.IRemoteUserService;
import cc.mrbird.febs.server.hr.service.IEmpAccidentCasesService;
import cc.mrbird.febs.server.hr.service.IEmpBadRecordService;
import cc.mrbird.febs.server.hr.service.IEmpBaseInfoService;
import cc.mrbird.febs.server.hr.service.IEmpContractInfoService;
import cc.mrbird.febs.server.hr.service.IEmpDimissionAttendService;
import cc.mrbird.febs.server.hr.service.IEmpInsuranceService;
import cc.mrbird.febs.server.hr.service.IEmpLaborTroubleService;
import cc.mrbird.febs.server.hr.service.IEmpLeaveInfoService;
import cc.mrbird.febs.server.hr.service.IEmpOccupationalService;
import cc.mrbird.febs.server.hr.service.IEmpPhysicalExamService;
import cc.mrbird.febs.server.hr.service.IEmpRemarkinfoService;
import cc.mrbird.febs.server.hr.service.IEmpResignService;
import cc.mrbird.febs.server.hr.service.IEmpUnemploymentService;
import cc.mrbird.febs.server.hr.service.IEmpWorkExperienceService;
import cc.mrbird.febs.server.hr.util.PoiExportExcel;
import cc.mrbird.febs.server.hr.util.PoiImportExcel;
import cn.hutool.core.date.DateUtil;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import lombok.RequiredArgsConstructor;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import javax.validation.Valid;
import javax.validation.constraints.NotBlank;
import java.util.Map;
import lombok.extern.slf4j.Slf4j;
/**
 * name:EmpBaseInfo
@@ -40,6 +70,19 @@
public class EmpBaseInfoController {
    private final IEmpBaseInfoService empBaseInfoService;
    private final IEmpWorkExperienceService empWorkExperienceService;
    private final IEmpPhysicalExamService empPhysicalExamService;
    private final IEmpContractInfoService empContractInfoService;
    private final IEmpDimissionAttendService empDimissionAttendService;
    private final IEmpLeaveInfoService empLeaveInfoService;
    private final IEmpResignService empResignService;
    private final IEmpUnemploymentService empUnemploymentService;
    private final IEmpInsuranceService empInsuranceService;
    private final IEmpAccidentCasesService empAccidentCasesService;
    private final IEmpOccupationalService empOccupationalService;
    private final IEmpLaborTroubleService empLaborTroubleService;
    private final IEmpBadRecordService empBadRecordService;
    private final IEmpRemarkinfoService empRemarkinfoService;
    private final IRemoteUserService remoteUserService;
    private final IRemoteDeptService remoteDeptService;
@@ -56,6 +99,14 @@
        Map<String, Object> dataTable = FebsUtil.getDataTable(this.empBaseInfoService.findEmpBaseInfos(request, empBaseinfo));
        return new FebsResponse().data(dataTable);
    }
    @ApiOperation(value = "智搜人员基本信息翻页列表")
    @GetMapping("zslist")
    @PreAuthorize("hasAuthority('empBaseinfo:list')")
    public FebsResponse empBaseInfoZsList(QueryRequest request, EmpBaseInfo empBaseinfo) {
        Map<String, Object> dataTable = FebsUtil.getDataTable(this.empBaseInfoService.findZsEmpBaseInfos(request, empBaseinfo));
        return new FebsResponse().data(dataTable);
    }
    @ApiOperation(value = "人员基本信息增加")
    @PostMapping
@@ -65,7 +116,6 @@
           throw new FebsException("已存在此员工编号");
       }
        try {
            //EmpBaseInfo tempInfo = this.empBaseInfoService.
            this.empBaseInfoService.createEmpBaseInfo(empBaseinfo);
        } catch (Exception e) {
            String message = "新增员工基本信息失败";
@@ -127,6 +177,69 @@
            throw new FebsException(message);
        }
    }
    @GetMapping("options")
    public FebsResponse roles() {
        QueryWrapper<EmpBaseInfo> wrapper=new QueryWrapper();
        wrapper.eq("empStatus",0);
        wrapper.eq("delFlag",0);
        wrapper.orderByDesc("createTime");
        List<EmpBaseInfo> allRoles = empBaseInfoService.list(wrapper);
        return new FebsResponse().data(allRoles);
    }
    @ApiOperation(value = "导出员工")
   @RequestMapping(value = "export",method= RequestMethod.POST)
    @ControllerEndpoint(operation = "导出用户数据", exceptionMessage = "导出Excel失败")
   public void export(QueryRequest request, EmpBaseInfo empBaseinfo, HttpServletResponse response) throws IOException {
        request.setPageSize(this.empBaseInfoService.count());
        List<EmpBaseInfo> records = this.empBaseInfoService.findZsEmpBaseInfos(request, empBaseinfo).getRecords();
        ExcelKit.$Export(EmpBaseInfo.class, response).downXlsx(records, false);
   }
    @ApiOperation(value = "导出员工")
   @RequestMapping(value = "exportWithField",method= RequestMethod.POST)
    @ControllerEndpoint(operation = "导出用户数据", exceptionMessage = "导出Excel失败")
    public void exportWithField(QueryRequest request, HttpServletResponse response, String exportField,EmpBaseInfo empBaseinfo) throws IOException {
       request.setPageSize(25535);
       request.setPageNum(1);
      List<EmpBaseInfo> exportList = this.empBaseInfoService.findZsEmpBaseInfos(request, empBaseinfo).getRecords();
      List<Map<String, Object>> listMapDicItem = new ArrayList();
      List<Map<String, Object>> allList = PoiExportExcel.getDataList(exportField, exportList, listMapDicItem);
      String fileName = "在职员工列表";
      Date currentDate = new Date();
      String dateStr = DateUtil.formatDate(currentDate);
      boolean result = PoiExportExcel.exportCommonExcel(response, fileName + dateStr, fileName, allList);
   }
    @ApiOperation(value = "导入员工")
   @PostMapping("importEmp")
   public void importEmp(MultipartFile file) throws IOException {
      InputStream in = null;
      List<List<List<Object>>> listObject = null;
      try {
         if (!file.isEmpty()) {
            in = file.getInputStream();
            listObject = PoiImportExcel.getMulitListByExcel(in, file.getOriginalFilename());
            in.close();
            empBaseInfoService.importEmpBaseInfo(listObject.get(0));
            empWorkExperienceService.importEmpWorkExperience(listObject.get(1));
            empPhysicalExamService.importEmpPhysicalExam(listObject.get(2));
            empContractInfoService.importEmpContractInfo(listObject.get(3));
            empDimissionAttendService.importEmpDimissionAttend(listObject.get(4));
            empLeaveInfoService.importEmpLeaveInfo(listObject.get(5));
            empResignService.importEmpResign(listObject.get(6));
            empUnemploymentService.importEmpUnemployment(listObject.get(7));
            empInsuranceService.importEmpInsurance(listObject.get(8));
            empAccidentCasesService.importEmpAccidentCases(listObject.get(9));
            empOccupationalService.importEmpOccupational(listObject.get(10));
            empLaborTroubleService.importEmpLaborTrouble(listObject.get(11));
            empBadRecordService.importEmpBadRecord(listObject.get(12));
            empRemarkinfoService.importEmpRemarkInfo(listObject.get(13));
         }
      } catch (Exception e) {
         log.error("导入员工", e);
      }
   }
    @ApiOperation(value = "关闭员工档案")
    @PostMapping("dimission")
@@ -150,4 +263,16 @@
            throw new FebsException(message);
        }
    }
    @GetMapping("image/{empId}")
    public void getImage(@PathVariable String empId, HttpServletResponse response) throws FebsException {
        try {
            this.empBaseInfoService.getImage(empId,response);
        } catch (Exception e) {
            String message = "获取员工图片异常";
            log.error(message, e);
            throw new FebsException(message);
        }
    }
}