package org.jeecg.modules.eam.controller; import java.io.UnsupportedEncodingException; import java.io.IOException; import java.net.URLDecoder; import java.time.LocalDate; import java.util.*; import java.util.stream.Collectors; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import org.jeecg.common.util.DateUtils; import org.jeecg.modules.eam.entity.OperationCertificateApplyDetail; import org.jeecg.modules.eam.service.IdentityService; import org.jeecg.modules.system.service.SysIdentityService; import org.jeecgframework.poi.excel.ExcelImportUtil; import org.jeecgframework.poi.excel.def.NormalExcelConstants; import org.jeecgframework.poi.excel.entity.ExportParams; import org.jeecgframework.poi.excel.entity.ImportParams; import org.jeecgframework.poi.excel.view.JeecgEntityExcelView; import org.jeecg.common.system.vo.LoginUser; import org.apache.shiro.SecurityUtils; import org.jeecg.common.api.vo.Result; import org.jeecg.common.system.query.QueryGenerator; import org.jeecg.common.util.oConvertUtils; import org.jeecg.modules.eam.entity.OperationCertificateDetail; import org.jeecg.modules.eam.entity.OperationCertificate; import org.jeecg.modules.eam.vo.OperationCertificatePage; import org.jeecg.modules.eam.service.IOperationCertificateService; import org.jeecg.modules.eam.service.IOperationCertificateDetailService; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import org.springframework.web.servlet.ModelAndView; import org.springframework.web.multipart.MultipartFile; import org.springframework.web.multipart.MultipartHttpServletRequest; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import lombok.extern.slf4j.Slf4j; import com.alibaba.fastjson.JSON; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import org.jeecg.common.aspect.annotation.AutoLog; /** * @Description: 操作证书管理 * @Author: jeecg-boot * @Date: 2023-10-17 * @Version: V1.0 */ @Api(tags="操作证书管理") @RestController @RequestMapping("/eam/operationCertificate") @Slf4j public class OperationCertificateController { @Autowired private IOperationCertificateService operationCertificateService; @Autowired private IOperationCertificateDetailService operationCertificateDetailService; @Autowired private IdentityService identityService; /** * 分页列表查询 * * @param operationCertificate * @param pageNo * @param pageSize * @param req * @return */ //@AutoLog(value = "操作证书管理-分页列表查询") @ApiOperation(value="操作证书管理-分页列表查询", notes="操作证书管理-分页列表查询") @GetMapping(value = "/list") public Result> queryPageList(OperationCertificate operationCertificate, @RequestParam(name="pageNo", defaultValue="1") Integer pageNo, @RequestParam(name="pageSize", defaultValue="10") Integer pageSize, HttpServletRequest req) { QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(operationCertificate, req.getParameterMap()); Page page = new Page(pageNo, pageSize); IPage pageList = operationCertificateService.page(page, queryWrapper); return Result.OK(pageList); } /** * 添加 * * @param operationCertificatePage * @return */ @AutoLog(value = "操作证书管理-添加") @ApiOperation(value="操作证书管理-添加", notes="操作证书管理-添加") //@RequiresPermissions("org.jeecg.modules.mdc:mom_eam_operation_certificate:add") @PostMapping(value = "/add") public Result add(@RequestBody OperationCertificatePage operationCertificatePage) { OperationCertificate operationCertificate = new OperationCertificate(); BeanUtils.copyProperties(operationCertificatePage, operationCertificate); operationCertificateService.saveMain(operationCertificate, operationCertificatePage.getOperationCertificateDetailList()); return Result.OK("添加成功!"); } @PostMapping(value = "/addOperationCertificate") public Result addOperationCertificate(@RequestBody List operationCertificateList) { for(OperationCertificateApplyDetail operationCertificateApplyDetail : operationCertificateList){ if("pass".equals(operationCertificateApplyDetail.getExaminationConclusion())){ Date currentDate = new Date(); Calendar c = Calendar.getInstance(); c.setTime(currentDate); c.add(Calendar.YEAR, 1); currentDate = c.getTime(); OperationCertificate operationCertificate = new OperationCertificate(); operationCertificate.setNum(identityService.getNumByTypeAndLength("OperationCertificate", 4)); operationCertificate.setUserId(operationCertificateApplyDetail.getUserId()); operationCertificate.setEquipmentIds(operationCertificateApplyDetail.getEquipmentIds()); operationCertificate.setCurrentCycleScore(12); operationCertificate.setIssueDate(new Date()); operationCertificate.setStartTime(new Date()); operationCertificate.setEndTime(currentDate); operationCertificate.setStatus("1"); operationCertificateService.save(operationCertificate); } } return Result.OK("添加成功!"); } /** * 编辑 * * @param operationCertificatePage * @return */ @AutoLog(value = "操作证书管理-编辑") @ApiOperation(value="操作证书管理-编辑", notes="操作证书管理-编辑") //@RequiresPermissions("org.jeecg.modules.mdc:mom_eam_operation_certificate:edit") @RequestMapping(value = "/edit", method = {RequestMethod.PUT,RequestMethod.POST}) public Result edit(@RequestBody OperationCertificatePage operationCertificatePage) { OperationCertificate operationCertificate = new OperationCertificate(); BeanUtils.copyProperties(operationCertificatePage, operationCertificate); OperationCertificate operationCertificateEntity = operationCertificateService.getById(operationCertificate.getId()); if(operationCertificateEntity==null) { return Result.error("未找到对应数据"); } operationCertificateService.updateMain(operationCertificate, operationCertificatePage.getOperationCertificateDetailList()); return Result.OK("编辑成功!"); } /** * 通过id删除 * * @param id * @return */ @AutoLog(value = "操作证书管理-通过id删除") @ApiOperation(value="操作证书管理-通过id删除", notes="操作证书管理-通过id删除") //@RequiresPermissions("org.jeecg.modules.mdc:mom_eam_operation_certificate:delete") @DeleteMapping(value = "/delete") public Result delete(@RequestParam(name="id",required=true) String id) { operationCertificateService.delMain(id); return Result.OK("删除成功!"); } /** * 批量删除 * * @param ids * @return */ @AutoLog(value = "操作证书管理-批量删除") @ApiOperation(value="操作证书管理-批量删除", notes="操作证书管理-批量删除") //@RequiresPermissions("org.jeecg.modules.mdc:mom_eam_operation_certificate:deleteBatch") @DeleteMapping(value = "/deleteBatch") public Result deleteBatch(@RequestParam(name="ids",required=true) String ids) { this.operationCertificateService.delBatchMain(Arrays.asList(ids.split(","))); return Result.OK("批量删除成功!"); } /** * 通过id查询 * * @param id * @return */ //@AutoLog(value = "操作证书管理-通过id查询") @ApiOperation(value="操作证书管理-通过id查询", notes="操作证书管理-通过id查询") @GetMapping(value = "/queryById") public Result queryById(@RequestParam(name="id",required=true) String id) { OperationCertificate operationCertificate = operationCertificateService.getById(id); if(operationCertificate==null) { return Result.error("未找到对应数据"); } return Result.OK(operationCertificate); } /** * 通过id查询 * * @param id * @return */ //@AutoLog(value = "操作证管理明细通过主表ID查询") @ApiOperation(value="操作证管理明细主表ID查询", notes="操作证管理明细-通主表ID查询") @GetMapping(value = "/queryOperationCertificateDetailByMainId") public Result> queryOperationCertificateDetailListByMainId(@RequestParam(name="id",required=true) String id) { List operationCertificateDetailList = operationCertificateDetailService.selectByMainId(id); return Result.OK(operationCertificateDetailList); } /** * 导出excel * * @param request * @param operationCertificate */ //@RequiresPermissions("org.jeecg.modules.mdc:mom_eam_operation_certificate:exportXls") @RequestMapping(value = "/exportXls") public ModelAndView exportXls(HttpServletRequest request, OperationCertificate operationCertificate) { // Step.1 组装查询条件查询数据 QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(operationCertificate, request.getParameterMap()); LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal(); //配置选中数据查询条件 String selections = request.getParameter("selections"); if(oConvertUtils.isNotEmpty(selections)) { List selectionList = Arrays.asList(selections.split(",")); queryWrapper.in("id",selectionList); } //Step.2 获取导出数据 List operationCertificateList = operationCertificateService.list(queryWrapper); // Step.3 组装pageList List pageList = new ArrayList(); for (OperationCertificate main : operationCertificateList) { OperationCertificatePage vo = new OperationCertificatePage(); BeanUtils.copyProperties(main, vo); List operationCertificateDetailList = operationCertificateDetailService.selectByMainId(main.getId()); vo.setOperationCertificateDetailList(operationCertificateDetailList); pageList.add(vo); } // Step.4 AutoPoi 导出Excel ModelAndView mv = new ModelAndView(new JeecgEntityExcelView()); mv.addObject(NormalExcelConstants.FILE_NAME, "操作证书管理列表"); mv.addObject(NormalExcelConstants.CLASS, OperationCertificatePage.class); mv.addObject(NormalExcelConstants.PARAMS, new ExportParams("操作证书管理数据", "导出人:"+sysUser.getRealname(), "操作证书管理")); mv.addObject(NormalExcelConstants.DATA_LIST, pageList); return mv; } /** * 通过excel导入数据 * * @param request * @param response * @return */ //@RequiresPermissions("org.jeecg.modules.mdc:mom_eam_operation_certificate:importExcel") @RequestMapping(value = "/importExcel", method = RequestMethod.POST) public Result importExcel(HttpServletRequest request, HttpServletResponse response) { MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request; Map fileMap = multipartRequest.getFileMap(); for (Map.Entry entity : fileMap.entrySet()) { // 获取上传文件对象 MultipartFile file = entity.getValue(); ImportParams params = new ImportParams(); params.setTitleRows(2); params.setHeadRows(1); params.setNeedSave(true); try { List list = ExcelImportUtil.importExcel(file.getInputStream(), OperationCertificatePage.class, params); for (OperationCertificatePage page : list) { OperationCertificate po = new OperationCertificate(); BeanUtils.copyProperties(page, po); operationCertificateService.saveMain(po, page.getOperationCertificateDetailList()); } return Result.OK("文件导入成功!数据行数:" + list.size()); } catch (Exception e) { log.error(e.getMessage(),e); return Result.error("文件导入失败:"+e.getMessage()); } finally { try { file.getInputStream().close(); } catch (IOException e) { e.printStackTrace(); } } } return Result.OK("文件导入失败!"); } /** * 通过主表ID查询 * @return */ //@AutoLog(value = "操作证管理明细-通过主表ID查询") @ApiOperation(value="操作证管理明细-通过主表ID查询", notes="操作证管理明细-通过主表ID查询") @GetMapping(value = "/listOperationCertificateDetailByMainId") public Result> listOperationCertificateDetailByMainId(OperationCertificateDetail operationCertificateDetail, @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo, @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize, HttpServletRequest req) { QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(operationCertificateDetail, req.getParameterMap()); Page page = new Page(pageNo, pageSize); IPage pageList = operationCertificateDetailService.page(page, queryWrapper); return Result.OK(pageList); } }