package org.jeecg.modules.demo.hzz.yhyc.controller; import org.jeecg.common.system.query.QueryGenerator; 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 org.jeecg.common.system.base.controller.JeecgController; import org.jeecg.common.api.vo.Result; import org.jeecg.modules.demo.hzz.yhyc.geo.service.IRmWtqdgeoService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import org.springframework.web.servlet.ModelAndView; import java.util.Arrays; import org.jeecg.common.util.oConvertUtils; import org.jeecg.modules.demo.hzz.yhyc.entity.RmMbqd; import org.jeecg.modules.demo.hzz.yhyc.entity.RmWtqd; import org.jeecg.modules.demo.hzz.yhyc.entity.RmMbfjb; import org.jeecg.modules.demo.hzz.yhyc.entity.RmRwqd; import org.jeecg.modules.demo.hzz.yhyc.entity.RmZrycsqd; import org.jeecg.modules.demo.hzz.yhyc.entity.RmJbxx; import org.jeecg.modules.demo.hzz.yhyc.service.IRmJbxxService; import org.jeecg.modules.demo.hzz.yhyc.service.IRmMbqdService; import org.jeecg.modules.demo.hzz.yhyc.service.IRmWtqdService; import org.jeecg.modules.demo.hzz.yhyc.service.IRmMbfjbService; import org.jeecg.modules.demo.hzz.yhyc.service.IRmRwqdService; import org.jeecg.modules.demo.hzz.yhyc.service.IRmZrycsqdService; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import org.jeecg.common.aspect.annotation.AutoLog; import org.apache.shiro.SecurityUtils; import org.jeecg.common.system.vo.LoginUser; 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.springframework.web.multipart.MultipartFile; import org.springframework.web.multipart.MultipartHttpServletRequest; import java.io.IOException; import java.util.List; import java.util.Map; import java.util.stream.Collectors; /** * @Description: 基本信息表 * @Author: jeecg-boot * @Date: 2021-11-26 * @Version: V1.0 */ @Api(tags="基本信息表") @RestController @RequestMapping("/hzz.yhyc/rmJbxx") @Slf4j public class RmJbxxController extends JeecgController { @Autowired private IRmJbxxService rmJbxxService; @Autowired private IRmMbqdService rmMbqdService; @Autowired private IRmWtqdService rmWtqdService; @Autowired private IRmWtqdgeoService rmWtqdgeoService; @Autowired private IRmMbfjbService rmMbfjbService; @Autowired private IRmRwqdService rmRwqdService; @Autowired private IRmZrycsqdService rmZrycsqdService; /*---------------------------------主表处理-begin-------------------------------------*/ /** * 分页列表查询 * @param rmJbxx * @param pageNo * @param pageSize * @param req * @return */ @AutoLog(value = "基本信息表-分页列表查询") @ApiOperation(value="基本信息表-分页列表查询", notes="基本信息表-分页列表查询") @GetMapping(value = "/list") public Result queryPageList(RmJbxx rmJbxx, @RequestParam(name="pageNo", defaultValue="1") Integer pageNo, @RequestParam(name="pageSize", defaultValue="10") Integer pageSize, HttpServletRequest req) { QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(rmJbxx, req.getParameterMap()); Page page = new Page(pageNo, pageSize); IPage pageList = rmJbxxService.page(page, queryWrapper); return Result.OK(pageList); } /** * 添加 * @param rmJbxx * @return */ @AutoLog(value = "基本信息表-添加") @ApiOperation(value="基本信息表-添加", notes="基本信息表-添加") @PostMapping(value = "/add") public Result add(@RequestBody RmJbxx rmJbxx) { rmJbxxService.save(rmJbxx); return Result.OK("添加成功！"); } /** * 编辑 * @param rmJbxx * @return */ @AutoLog(value = "基本信息表-编辑") @ApiOperation(value="基本信息表-编辑", notes="基本信息表-编辑") @PostMapping(value = "/edit") public Result edit(@RequestBody RmJbxx rmJbxx) { rmJbxxService.updateById(rmJbxx); return Result.OK("编辑成功!"); } /** * 通过id删除 * @param id * @return */ @AutoLog(value = "基本信息表-通过id删除") @ApiOperation(value="基本信息表-通过id删除", notes="基本信息表-通过id删除") @PostMapping(value = "/delete") public Result delete(@RequestParam(name="id",required=true) String id) { rmJbxxService.delMain(id); return Result.OK("删除成功!"); } /** * 批量删除 * @param ids * @return */ @AutoLog(value = "基本信息表-批量删除") @ApiOperation(value="基本信息表-批量删除", notes="基本信息表-批量删除") @PostMapping(value = "/deleteBatch") public Result deleteBatch(@RequestParam(name="ids",required=true) String ids) { this.rmJbxxService.delBatchMain(Arrays.asList(ids.split(","))); return Result.OK("批量删除成功!"); } /** * 导出 * @return */ @RequestMapping(value = "/exportXls") public ModelAndView exportXls(HttpServletRequest request, RmJbxx rmJbxx) { return super.exportXls(request, rmJbxx, RmJbxx.class, "基本信息表"); } /** * 导入 * @return */ @RequestMapping(value = "/importExcel", method = RequestMethod.POST) public Result importExcel(HttpServletRequest request, HttpServletResponse response) { return super.importExcel(request, response, RmJbxx.class); } /*---------------------------------主表处理-end-------------------------------------*/ /*--------------------------------子表处理-目标清单-begin----------------------------------------------*/ /** * 通过主表ID查询 * @return */ @AutoLog(value = "目标清单-通过主表ID查询") @ApiOperation(value="目标清单-通过主表ID查询", notes="目标清单-通过主表ID查询") @GetMapping(value = "/listRmMbqdByMainId") public Result listRmMbqdByMainId(RmMbqd rmMbqd, @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo, @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize, HttpServletRequest req) { QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(rmMbqd, req.getParameterMap()); Page page = new Page(pageNo, pageSize); IPage pageList = rmMbqdService.page(page, queryWrapper); return Result.OK(pageList); } /** * 添加 * @param rmMbqd * @return */ @AutoLog(value = "目标清单-添加") @ApiOperation(value="目标清单-添加", notes="目标清单-添加") @PostMapping(value = "/addRmMbqd") public Result addRmMbqd(@RequestBody RmMbqd rmMbqd) { rmMbqdService.save(rmMbqd); return Result.OK("添加成功！"); } /** * 编辑 * @param rmMbqd * @return */ @AutoLog(value = "目标清单-编辑") @ApiOperation(value="目标清单-编辑", notes="目标清单-编辑") @PostMapping(value = "/editRmMbqd") public Result editRmMbqd(@RequestBody RmMbqd rmMbqd) { rmMbqdService.updateById(rmMbqd); return Result.OK("编辑成功!"); } /** * 通过id删除 * @param id * @return */ @AutoLog(value = "目标清单-通过id删除") @ApiOperation(value="目标清单-通过id删除", notes="目标清单-通过id删除") @PostMapping(value = "/deleteRmMbqd") public Result deleteRmMbqd(@RequestParam(name="id",required=true) String id) { rmMbqdService.removeById(id); return Result.OK("删除成功!"); } /** * 批量删除 * @param ids * @return */ @AutoLog(value = "目标清单-批量删除") @ApiOperation(value="目标清单-批量删除", notes="目标清单-批量删除") @PostMapping(value = "/deleteBatchRmMbqd") public Result deleteBatchRmMbqd(@RequestParam(name="ids",required=true) String ids) { this.rmMbqdService.removeByIds(Arrays.asList(ids.split(","))); return Result.OK("批量删除成功!"); } /** * 导出 * @return */ @RequestMapping(value = "/exportRmMbqd") public ModelAndView exportRmMbqd(HttpServletRequest request, RmMbqd rmMbqd) { // Step.1 组装查询条件 QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(rmMbqd, request.getParameterMap()); LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal(); // Step.2 获取导出数据 List pageList = rmMbqdService.list(queryWrapper); List exportList = null; // 过滤选中数据 String selections = request.getParameter("selections"); if (oConvertUtils.isNotEmpty(selections)) { List selectionList = Arrays.asList(selections.split(",")); exportList = pageList.stream().filter(item -> selectionList.contains(item.getId())).collect(Collectors.toList()); } else { exportList = pageList; } // Step.3 AutoPoi 导出Excel ModelAndView mv = new ModelAndView(new JeecgEntityExcelView()); mv.addObject(NormalExcelConstants.FILE_NAME, "目标清单"); //此处设置的filename无效 ,前端会重更新设置一下 mv.addObject(NormalExcelConstants.CLASS, RmMbqd.class); mv.addObject(NormalExcelConstants.PARAMS, new ExportParams("目标清单报表", "导出人:" + sysUser.getRealname(), "目标清单")); mv.addObject(NormalExcelConstants.DATA_LIST, exportList); return mv; } /** * 导入 * @return */ @RequestMapping(value = "/importRmMbqd/{mainId}") public Result importRmMbqd(HttpServletRequest request, HttpServletResponse response, @PathVariable("mainId") String mainId) { 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(), RmMbqd.class, params); for (RmMbqd temp : list) { temp.setMainId(mainId); } long start = System.currentTimeMillis(); rmMbqdService.saveBatch(list); log.info("消耗时间" + (System.currentTimeMillis() - start) + "毫秒"); 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.error("文件导入失败！"); } /*--------------------------------子表处理-目标清单-end----------------------------------------------*/ /*--------------------------------子表处理-问题清单-begin----------------------------------------------*/ /** * 通过主表ID查询 * @return */ @AutoLog(value = "问题清单-通过主表ID查询") @ApiOperation(value="问题清单-通过主表ID查询", notes="问题清单-通过主表ID查询") @GetMapping(value = "/listRmWtqdByMainId") public Result listRmWtqdByMainId(RmWtqd rmWtqd, @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo, @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize, HttpServletRequest req) { QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(rmWtqd, req.getParameterMap()); Page page = new Page(pageNo, pageSize); IPage pageList = rmWtqdService.page(page, queryWrapper); return Result.OK(pageList); } /** * 添加 * @param rmWtqd * @return */ @AutoLog(value = "问题清单-添加") @ApiOperation(value="问题清单-添加", notes="问题清单-添加") @PostMapping(value = "/addRmWtqd") public Result addRmWtqd(@RequestBody RmWtqd rmWtqd) { rmWtqdService.save(rmWtqd); if(rmWtqd.getJd()>180&&rmWtqd.getWd()>90){ rmWtqdgeoService.addWtqd("POINT("+rmWtqd.getJd()+" "+rmWtqd.getWd()+")",rmWtqd.getId()); }else{ rmWtqdgeoService.addWtqd2("POINT("+rmWtqd.getJd()+" "+rmWtqd.getWd()+")",rmWtqd.getId()); } rmWtqdService.saveOrUpdate(rmWtqd); return Result.OK("添加成功！"); } /** * 编辑 * @param rmWtqd * @return */ @AutoLog(value = "问题清单-编辑") @ApiOperation(value="问题清单-编辑", notes="问题清单-编辑") @PostMapping(value = "/editRmWtqd") public Result editRmWtqd(@RequestBody RmWtqd rmWtqd) { rmWtqdService.updateById(rmWtqd); return Result.OK("编辑成功!"); } /** * 通过id删除 * @param id * @return */ @AutoLog(value = "问题清单-通过id删除") @ApiOperation(value="问题清单-通过id删除", notes="问题清单-通过id删除") @PostMapping(value = "/deleteRmWtqd") public Result deleteRmWtqd(@RequestParam(name="id",required=true) String id) { rmWtqdService.removeById(id); rmWtqdgeoService.delWtqd(id); return Result.OK("删除成功!"); } /** * 批量删除 * @param ids * @return */ @AutoLog(value = "问题清单-批量删除") @ApiOperation(value="问题清单-批量删除", notes="问题清单-批量删除") @PostMapping(value = "/deleteBatchRmWtqd") public Result deleteBatchRmWtqd(@RequestParam(name="ids",required=true) String ids) { this.rmWtqdService.removeByIds(Arrays.asList(ids.split(","))); return Result.OK("批量删除成功!"); } /** * 导出 * @return */ @RequestMapping(value = "/exportRmWtqd") public ModelAndView exportRmWtqd(HttpServletRequest request, RmWtqd rmWtqd) { // Step.1 组装查询条件 QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(rmWtqd, request.getParameterMap()); LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal(); // Step.2 获取导出数据 List pageList = rmWtqdService.list(queryWrapper); List exportList = null; // 过滤选中数据 String selections = request.getParameter("selections"); if (oConvertUtils.isNotEmpty(selections)) { List selectionList = Arrays.asList(selections.split(",")); exportList = pageList.stream().filter(item -> selectionList.contains(item.getId())).collect(Collectors.toList()); } else { exportList = pageList; } // Step.3 AutoPoi 导出Excel ModelAndView mv = new ModelAndView(new JeecgEntityExcelView()); mv.addObject(NormalExcelConstants.FILE_NAME, "问题清单"); //此处设置的filename无效 ,前端会重更新设置一下 mv.addObject(NormalExcelConstants.CLASS, RmWtqd.class); mv.addObject(NormalExcelConstants.PARAMS, new ExportParams("问题清单报表", "导出人:" + sysUser.getRealname(), "问题清单")); mv.addObject(NormalExcelConstants.DATA_LIST, exportList); return mv; } /** * 导入 * @return */ @RequestMapping(value = "/importRmWtqd/{mainId}") public Result importRmWtqd(HttpServletRequest request, HttpServletResponse response, @PathVariable("mainId") String mainId) { 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(), RmWtqd.class, params); for (RmWtqd temp : list) { temp.setMainId(mainId); } long start = System.currentTimeMillis(); rmWtqdService.saveBatch(list); log.info("消耗时间" + (System.currentTimeMillis() - start) + "毫秒"); 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.error("文件导入失败！"); } /*--------------------------------子表处理-问题清单-end----------------------------------------------*/ /*--------------------------------子表处理-目标分解表-begin----------------------------------------------*/ /** * 通过主表ID查询 * @return */ @AutoLog(value = "目标分解表-通过主表ID查询") @ApiOperation(value="目标分解表-通过主表ID查询", notes="目标分解表-通过主表ID查询") @GetMapping(value = "/listRmMbfjbByMainId") public Result listRmMbfjbByMainId(RmMbfjb rmMbfjb, @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo, @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize, HttpServletRequest req) { QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(rmMbfjb, req.getParameterMap()); Page page = new Page(pageNo, pageSize); IPage pageList = rmMbfjbService.page(page, queryWrapper); return Result.OK(pageList); } /** * 添加 * @param rmMbfjb * @return */ @AutoLog(value = "目标分解表-添加") @ApiOperation(value="目标分解表-添加", notes="目标分解表-添加") @PostMapping(value = "/addRmMbfjb") public Result addRmMbfjb(@RequestBody RmMbfjb rmMbfjb) { rmMbfjbService.save(rmMbfjb); return Result.OK("添加成功！"); } /** * 编辑 * @param rmMbfjb * @return */ @AutoLog(value = "目标分解表-编辑") @ApiOperation(value="目标分解表-编辑", notes="目标分解表-编辑") @PostMapping(value = "/editRmMbfjb") public Result editRmMbfjb(@RequestBody RmMbfjb rmMbfjb) { rmMbfjbService.updateById(rmMbfjb); return Result.OK("编辑成功!"); } /** * 通过id删除 * @param id * @return */ @AutoLog(value = "目标分解表-通过id删除") @ApiOperation(value="目标分解表-通过id删除", notes="目标分解表-通过id删除") @PostMapping(value = "/deleteRmMbfjb") public Result deleteRmMbfjb(@RequestParam(name="id",required=true) String id) { rmMbfjbService.removeById(id); return Result.OK("删除成功!"); } /** * 批量删除 * @param ids * @return */ @AutoLog(value = "目标分解表-批量删除") @ApiOperation(value="目标分解表-批量删除", notes="目标分解表-批量删除") @PostMapping(value = "/deleteBatchRmMbfjb") public Result deleteBatchRmMbfjb(@RequestParam(name="ids",required=true) String ids) { this.rmMbfjbService.removeByIds(Arrays.asList(ids.split(","))); return Result.OK("批量删除成功!"); } /** * 导出 * @return */ @RequestMapping(value = "/exportRmMbfjb") public ModelAndView exportRmMbfjb(HttpServletRequest request, RmMbfjb rmMbfjb) { // Step.1 组装查询条件 QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(rmMbfjb, request.getParameterMap()); LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal(); // Step.2 获取导出数据 List pageList = rmMbfjbService.list(queryWrapper); List exportList = null; // 过滤选中数据 String selections = request.getParameter("selections"); if (oConvertUtils.isNotEmpty(selections)) { List selectionList = Arrays.asList(selections.split(",")); exportList = pageList.stream().filter(item -> selectionList.contains(item.getId())).collect(Collectors.toList()); } else { exportList = pageList; } // Step.3 AutoPoi 导出Excel ModelAndView mv = new ModelAndView(new JeecgEntityExcelView()); mv.addObject(NormalExcelConstants.FILE_NAME, "目标分解表"); //此处设置的filename无效 ,前端会重更新设置一下 mv.addObject(NormalExcelConstants.CLASS, RmMbfjb.class); mv.addObject(NormalExcelConstants.PARAMS, new ExportParams("目标分解表报表", "导出人:" + sysUser.getRealname(), "目标分解表")); mv.addObject(NormalExcelConstants.DATA_LIST, exportList); return mv; } /** * 导入 * @return */ @RequestMapping(value = "/importRmMbfjb/{mainId}") public Result importRmMbfjb(HttpServletRequest request, HttpServletResponse response, @PathVariable("mainId") String mainId) { 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(), RmMbfjb.class, params); for (RmMbfjb temp : list) { temp.setMainId(mainId); } long start = System.currentTimeMillis(); rmMbfjbService.saveBatch(list); log.info("消耗时间" + (System.currentTimeMillis() - start) + "毫秒"); 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.error("文件导入失败！"); } /*--------------------------------子表处理-目标分解表-end----------------------------------------------*/ /*--------------------------------子表处理-任务清单-begin----------------------------------------------*/ /** * 通过主表ID查询 * @return */ @AutoLog(value = "任务清单-通过主表ID查询") @ApiOperation(value="任务清单-通过主表ID查询", notes="任务清单-通过主表ID查询") @GetMapping(value = "/listRmRwqdByMainId") public Result listRmRwqdByMainId(RmRwqd rmRwqd, @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo, @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize, HttpServletRequest req) { QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(rmRwqd, req.getParameterMap()); Page page = new Page(pageNo, pageSize); IPage pageList = rmRwqdService.page(page, queryWrapper); return Result.OK(pageList); } /** * 添加 * @param rmRwqd * @return */ @AutoLog(value = "任务清单-添加") @ApiOperation(value="任务清单-添加", notes="任务清单-添加") @PostMapping(value = "/addRmRwqd") public Result addRmRwqd(@RequestBody RmRwqd rmRwqd) { rmRwqdService.save(rmRwqd); return Result.OK("添加成功！"); } /** * 编辑 * @param rmRwqd * @return */ @AutoLog(value = "任务清单-编辑") @ApiOperation(value="任务清单-编辑", notes="任务清单-编辑") @PostMapping(value = "/editRmRwqd") public Result editRmRwqd(@RequestBody RmRwqd rmRwqd) { rmRwqdService.updateById(rmRwqd); return Result.OK("编辑成功!"); } /** * 通过id删除 * @param id * @return */ @AutoLog(value = "任务清单-通过id删除") @ApiOperation(value="任务清单-通过id删除", notes="任务清单-通过id删除") @PostMapping(value = "/deleteRmRwqd") public Result deleteRmRwqd(@RequestParam(name="id",required=true) String id) { rmRwqdService.removeById(id); return Result.OK("删除成功!"); } /** * 批量删除 * @param ids * @return */ @AutoLog(value = "任务清单-批量删除") @ApiOperation(value="任务清单-批量删除", notes="任务清单-批量删除") @PostMapping(value = "/deleteBatchRmRwqd") public Result deleteBatchRmRwqd(@RequestParam(name="ids",required=true) String ids) { this.rmRwqdService.removeByIds(Arrays.asList(ids.split(","))); return Result.OK("批量删除成功!"); } /** * 导出 * @return */ @RequestMapping(value = "/exportRmRwqd") public ModelAndView exportRmRwqd(HttpServletRequest request, RmRwqd rmRwqd) { // Step.1 组装查询条件 QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(rmRwqd, request.getParameterMap()); LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal(); // Step.2 获取导出数据 List pageList = rmRwqdService.list(queryWrapper); List exportList = null; // 过滤选中数据 String selections = request.getParameter("selections"); if (oConvertUtils.isNotEmpty(selections)) { List selectionList = Arrays.asList(selections.split(",")); exportList = pageList.stream().filter(item -> selectionList.contains(item.getId())).collect(Collectors.toList()); } else { exportList = pageList; } // Step.3 AutoPoi 导出Excel ModelAndView mv = new ModelAndView(new JeecgEntityExcelView()); mv.addObject(NormalExcelConstants.FILE_NAME, "任务清单"); //此处设置的filename无效 ,前端会重更新设置一下 mv.addObject(NormalExcelConstants.CLASS, RmRwqd.class); mv.addObject(NormalExcelConstants.PARAMS, new ExportParams("任务清单报表", "导出人:" + sysUser.getRealname(), "任务清单")); mv.addObject(NormalExcelConstants.DATA_LIST, exportList); return mv; } /** * 导入 * @return */ @RequestMapping(value = "/importRmRwqd/{mainId}") public Result importRmRwqd(HttpServletRequest request, HttpServletResponse response, @PathVariable("mainId") String mainId) { 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(), RmRwqd.class, params); for (RmRwqd temp : list) { temp.setMainId(mainId); } long start = System.currentTimeMillis(); rmRwqdService.saveBatch(list); log.info("消耗时间" + (System.currentTimeMillis() - start) + "毫秒"); 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.error("文件导入失败！"); } /*--------------------------------子表处理-任务清单-end----------------------------------------------*/ /*--------------------------------子表处理-责任与措施清单-begin----------------------------------------------*/ /** * 通过主表ID查询 * @return */ @AutoLog(value = "责任与措施清单-通过主表ID查询") @ApiOperation(value="责任与措施清单-通过主表ID查询", notes="责任与措施清单-通过主表ID查询") @GetMapping(value = "/listRmZrycsqdByMainId") public Result listRmZrycsqdByMainId(RmZrycsqd rmZrycsqd, @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo, @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize, HttpServletRequest req) { QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(rmZrycsqd, req.getParameterMap()); Page page = new Page(pageNo, pageSize); IPage pageList = rmZrycsqdService.page(page, queryWrapper); return Result.OK(pageList); } /** * 添加 * @param rmZrycsqd * @return */ @AutoLog(value = "责任与措施清单-添加") @ApiOperation(value="责任与措施清单-添加", notes="责任与措施清单-添加") @PostMapping(value = "/addRmZrycsqd") public Result addRmZrycsqd(@RequestBody RmZrycsqd rmZrycsqd) { rmZrycsqdService.save(rmZrycsqd); return Result.OK("添加成功！"); } /** * 编辑 * @param rmZrycsqd * @return */ @AutoLog(value = "责任与措施清单-编辑") @ApiOperation(value="责任与措施清单-编辑", notes="责任与措施清单-编辑") @PostMapping(value = "/editRmZrycsqd") public Result editRmZrycsqd(@RequestBody RmZrycsqd rmZrycsqd) { rmZrycsqdService.updateById(rmZrycsqd); return Result.OK("编辑成功!"); } /** * 通过id删除 * @param id * @return */ @AutoLog(value = "责任与措施清单-通过id删除") @ApiOperation(value="责任与措施清单-通过id删除", notes="责任与措施清单-通过id删除") @PostMapping(value = "/deleteRmZrycsqd") public Result deleteRmZrycsqd(@RequestParam(name="id",required=true) String id) { rmZrycsqdService.removeById(id); return Result.OK("删除成功!"); } /** * 批量删除 * @param ids * @return */ @AutoLog(value = "责任与措施清单-批量删除") @ApiOperation(value="责任与措施清单-批量删除", notes="责任与措施清单-批量删除") @PostMapping(value = "/deleteBatchRmZrycsqd") public Result deleteBatchRmZrycsqd(@RequestParam(name="ids",required=true) String ids) { this.rmZrycsqdService.removeByIds(Arrays.asList(ids.split(","))); return Result.OK("批量删除成功!"); } /** * 导出 * @return */ @RequestMapping(value = "/exportRmZrycsqd") public ModelAndView exportRmZrycsqd(HttpServletRequest request, RmZrycsqd rmZrycsqd) { // Step.1 组装查询条件 QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(rmZrycsqd, request.getParameterMap()); LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal(); // Step.2 获取导出数据 List pageList = rmZrycsqdService.list(queryWrapper); List exportList = null; // 过滤选中数据 String selections = request.getParameter("selections"); if (oConvertUtils.isNotEmpty(selections)) { List selectionList = Arrays.asList(selections.split(",")); exportList = pageList.stream().filter(item -> selectionList.contains(item.getId())).collect(Collectors.toList()); } else { exportList = pageList; } // Step.3 AutoPoi 导出Excel ModelAndView mv = new ModelAndView(new JeecgEntityExcelView()); mv.addObject(NormalExcelConstants.FILE_NAME, "责任与措施清单"); //此处设置的filename无效 ,前端会重更新设置一下 mv.addObject(NormalExcelConstants.CLASS, RmZrycsqd.class); mv.addObject(NormalExcelConstants.PARAMS, new ExportParams("责任与措施清单报表", "导出人:" + sysUser.getRealname(), "责任与措施清单")); mv.addObject(NormalExcelConstants.DATA_LIST, exportList); return mv; } /** * 导入 * @return */ @RequestMapping(value = "/importRmZrycsqd/{mainId}") public Result importRmZrycsqd(HttpServletRequest request, HttpServletResponse response, @PathVariable("mainId") String mainId) { 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(), RmZrycsqd.class, params); for (RmZrycsqd temp : list) { temp.setMainId(mainId); } long start = System.currentTimeMillis(); rmZrycsqdService.saveBatch(list); log.info("消耗时间" + (System.currentTimeMillis() - start) + "毫秒"); 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.error("文件导入失败！"); } /*--------------------------------子表处理-责任与措施清单-end----------------------------------------------*/ }