package org.jeecg.modules.tms.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.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.tms.entity.TmsToolReceiveDetail; import org.jeecg.modules.tms.entity.TmsToolReceive; import org.jeecg.modules.tms.service.ITmsToolReceiveService; import org.jeecg.modules.tms.service.ITmsToolReceiveDetailService; 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: 2025-07-28 * @Version: V1.0 */ @Api(tags="工装领用") @RestController @RequestMapping("/tms/tmsToolReceive") @Slf4j public class TmsToolReceiveController extends JeecgController { @Autowired private ITmsToolReceiveService tmsToolReceiveService; @Autowired private ITmsToolReceiveDetailService tmsToolReceiveDetailService; /*---------------------------------主表处理-begin-------------------------------------*/ /** * 分页列表查询 * @param tmsToolReceive * @param pageNo * @param pageSize * @param req * @return */ //@AutoLog(value = "工装领用-分页列表查询") @ApiOperation(value="工装领用-分页列表查询", notes="工装领用-分页列表查询") @GetMapping(value = "/list") public Result> queryPageList(TmsToolReceive tmsToolReceive, @RequestParam(name="pageNo", defaultValue="1") Integer pageNo, @RequestParam(name="pageSize", defaultValue="10") Integer pageSize, HttpServletRequest req) { QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(tmsToolReceive, req.getParameterMap()); Page page = new Page(pageNo, pageSize); IPage pageList = tmsToolReceiveService.page(page, queryWrapper); return Result.OK(pageList); } /** * 添加 * @param tmsToolReceive * @return */ @AutoLog(value = "工装领用-添加") @ApiOperation(value="工装领用-添加", notes="工装领用-添加") //@RequiresPermissions("org.jeecg.modules:tms_tool_receive:add") @PostMapping(value = "/add") public Result add(@RequestBody TmsToolReceive tmsToolReceive) { tmsToolReceiveService.save(tmsToolReceive); return Result.OK("添加成功！"); } /** * 编辑 * @param tmsToolReceive * @return */ @AutoLog(value = "工装领用-编辑") @ApiOperation(value="工装领用-编辑", notes="工装领用-编辑") //@RequiresPermissions("org.jeecg.modules:tms_tool_receive:edit") @RequestMapping(value = "/edit", method = {RequestMethod.PUT,RequestMethod.POST}) public Result edit(@RequestBody TmsToolReceive tmsToolReceive) { tmsToolReceiveService.updateById(tmsToolReceive); return Result.OK("编辑成功!"); } /** * 通过id删除 * @param id * @return */ @AutoLog(value = "工装领用-通过id删除") @ApiOperation(value="工装领用-通过id删除", notes="工装领用-通过id删除") //@RequiresPermissions("org.jeecg.modules:tms_tool_receive:delete") @DeleteMapping(value = "/delete") public Result delete(@RequestParam(name="id",required=true) String id) { tmsToolReceiveService.delMain(id); return Result.OK("删除成功!"); } /** * 批量删除 * @param ids * @return */ @AutoLog(value = "工装领用-批量删除") @ApiOperation(value="工装领用-批量删除", notes="工装领用-批量删除") //@RequiresPermissions("org.jeecg.modules:tms_tool_receive:deleteBatch") @DeleteMapping(value = "/deleteBatch") public Result deleteBatch(@RequestParam(name="ids",required=true) String ids) { this.tmsToolReceiveService.delBatchMain(Arrays.asList(ids.split(","))); return Result.OK("批量删除成功!"); } /** * 导出 * @return */ //@RequiresPermissions("org.jeecg.modules:tms_tool_receive:exportXls") @RequestMapping(value = "/exportXls") public ModelAndView exportXls(HttpServletRequest request, TmsToolReceive tmsToolReceive) { return super.exportXls(request, tmsToolReceive, TmsToolReceive.class, "工装领用"); } /** * 导入 * @return */ //@RequiresPermissions("org.jeecg.modules:tms_tool_receive:importExcel") @RequestMapping(value = "/importExcel", method = RequestMethod.POST) public Result importExcel(HttpServletRequest request, HttpServletResponse response) { return super.importExcel(request, response, TmsToolReceive.class); } /*---------------------------------主表处理-end-------------------------------------*/ /*--------------------------------子表处理-工装领用明细-begin----------------------------------------------*/ /** * 通过主表ID查询 * @return */ //@AutoLog(value = "工装领用明细-通过主表ID查询") @ApiOperation(value="工装领用明细-通过主表ID查询", notes="工装领用明细-通过主表ID查询") @GetMapping(value = "/listTmsToolReceiveDetailByMainId") public Result> listTmsToolReceiveDetailByMainId(TmsToolReceiveDetail tmsToolReceiveDetail, @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo, @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize, HttpServletRequest req) { QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(tmsToolReceiveDetail, req.getParameterMap()); Page page = new Page(pageNo, pageSize); IPage pageList = tmsToolReceiveDetailService.page(page, queryWrapper); return Result.OK(pageList); } /** * 添加 * @param tmsToolReceiveDetail * @return */ @AutoLog(value = "工装领用明细-添加") @ApiOperation(value="工装领用明细-添加", notes="工装领用明细-添加") @PostMapping(value = "/addTmsToolReceiveDetail") public Result addTmsToolReceiveDetail(@RequestBody TmsToolReceiveDetail tmsToolReceiveDetail) { tmsToolReceiveDetailService.save(tmsToolReceiveDetail); return Result.OK("添加成功！"); } /** * 编辑 * @param tmsToolReceiveDetail * @return */ @AutoLog(value = "工装领用明细-编辑") @ApiOperation(value="工装领用明细-编辑", notes="工装领用明细-编辑") @RequestMapping(value = "/editTmsToolReceiveDetail", method = {RequestMethod.PUT,RequestMethod.POST}) public Result editTmsToolReceiveDetail(@RequestBody TmsToolReceiveDetail tmsToolReceiveDetail) { tmsToolReceiveDetailService.updateById(tmsToolReceiveDetail); return Result.OK("编辑成功!"); } /** * 通过id删除 * @param id * @return */ @AutoLog(value = "工装领用明细-通过id删除") @ApiOperation(value="工装领用明细-通过id删除", notes="工装领用明细-通过id删除") @DeleteMapping(value = "/deleteTmsToolReceiveDetail") public Result deleteTmsToolReceiveDetail(@RequestParam(name="id",required=true) String id) { tmsToolReceiveDetailService.removeById(id); return Result.OK("删除成功!"); } /** * 批量删除 * @param ids * @return */ @AutoLog(value = "工装领用明细-批量删除") @ApiOperation(value="工装领用明细-批量删除", notes="工装领用明细-批量删除") @DeleteMapping(value = "/deleteBatchTmsToolReceiveDetail") public Result deleteBatchTmsToolReceiveDetail(@RequestParam(name="ids",required=true) String ids) { this.tmsToolReceiveDetailService.removeByIds(Arrays.asList(ids.split(","))); return Result.OK("批量删除成功!"); } /** * 导出 * @return */ @RequestMapping(value = "/exportTmsToolReceiveDetail") public ModelAndView exportTmsToolReceiveDetail(HttpServletRequest request, TmsToolReceiveDetail tmsToolReceiveDetail) { // Step.1 组装查询条件 QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(tmsToolReceiveDetail, request.getParameterMap()); LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal(); // Step.2 获取导出数据 List pageList = tmsToolReceiveDetailService.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()); //此处设置的filename无效,前端会重更新设置一下 mv.addObject(NormalExcelConstants.FILE_NAME, "工装领用明细"); mv.addObject(NormalExcelConstants.CLASS, TmsToolReceiveDetail.class); mv.addObject(NormalExcelConstants.PARAMS, new ExportParams("工装领用明细报表", "导出人:" + sysUser.getRealname(), "工装领用明细")); mv.addObject(NormalExcelConstants.DATA_LIST, exportList); return mv; } /** * 导入 * @return */ @RequestMapping(value = "/importTmsToolReceiveDetail/{mainId}") public Result importTmsToolReceiveDetail(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(), TmsToolReceiveDetail.class, params); for (TmsToolReceiveDetail temp : list) { temp.setOrderId(mainId); } long start = System.currentTimeMillis(); tmsToolReceiveDetailService.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----------------------------------------------*/ }