From a1b85ef72062ca80db35546e4216dd564f3e0f57 Mon Sep 17 00:00:00 2001 From: WangHan <wwh_work@126,com> Date: 星期四, 03 四月 2025 15:58:19 +0800 Subject: [PATCH] 问题与漏洞修改 --- consum-base/src/main/java/com/consum/base/controller/LWhFormOutputController.java | 122 ++++++++++++++++++++++------------------ 1 files changed, 66 insertions(+), 56 deletions(-) diff --git a/consum-base/src/main/java/com/consum/base/controller/LWhFormOutputController.java b/consum-base/src/main/java/com/consum/base/controller/LWhFormOutputController.java index 1de90ad..e5dc74e 100644 --- a/consum-base/src/main/java/com/consum/base/controller/LWhFormOutputController.java +++ b/consum-base/src/main/java/com/consum/base/controller/LWhFormOutputController.java @@ -5,11 +5,12 @@ import java.util.HashMap; import java.util.List; import java.util.Map; +import java.util.Map.Entry; import java.util.Optional; +import java.util.stream.Collectors; -import javax.annotation.Resource; -import javax.servlet.http.HttpServletResponse; - +import jakarta.annotation.Resource; +import jakarta.servlet.http.HttpServletResponse; import org.apache.commons.compress.utils.Lists; import org.apache.poi.hssf.usermodel.HSSFWorkbook; import org.apache.poi.ss.usermodel.Workbook; @@ -23,16 +24,16 @@ import com.consum.base.BaseController; import com.consum.base.core.utils.CommonUtil; -import com.consum.base.core.utils.MapUtils; -import com.consum.base.core.utils.MapperUtil; +import com.consum.base.core.utils.CurrencyUtil; import com.consum.base.core.utils.PageUtil; import com.consum.base.pojo.LWhFormOutputParam; +import com.consum.base.pojo.dto.GoodModelInfoDTO; import com.consum.base.pojo.excel.OutputExcelTemplate; import com.consum.base.pojo.query.LWhFormOutputQry; -import com.consum.base.pojo.response.FormOutputGoodsVO; import com.consum.base.pojo.response.FormOutputVO; import com.consum.base.pojo.response.GoodsModelVO; import com.consum.base.pojo.response.GoodsTemplateCountVO; +import com.consum.base.pojo.response.GoodsTemplateInfoVO; import com.consum.base.pojo.response.LWhFormOutputExtendVO; import com.consum.base.service.LWarehouseFlowService; import com.consum.base.service.LWhFormOutputService; @@ -48,17 +49,14 @@ import cn.afterturn.easypoi.excel.ExcelExportUtil; import cn.afterturn.easypoi.excel.entity.TemplateExportParams; import cn.hutool.core.util.ReflectUtil; -import io.swagger.annotations.Api; -import io.swagger.annotations.ApiImplicitParam; -import io.swagger.annotations.ApiImplicitParams; -import io.swagger.annotations.ApiOperation; + /** * @Description 鍑哄簱鍗� * @Author 鍗㈠簡闃� * @Date 2023/10/27 */ -@Api(value = "鍑哄簱鍗�", tags = "鍑哄簱鍗�") +// @Api(value = "鍑哄簱鍗�", tags = "鍑哄簱鍗�") @RestController @RequestMapping("/pc/l/wh/form/output") public class LWhFormOutputController extends BaseController { @@ -68,16 +66,16 @@ @Autowired private LWhProcureModelService lWhProcureModelService; @Resource - private LWarehouseFlowService lWarehouseFlowService; + private LWarehouseFlowService lWarehouseFlowServiceImpl; /** * @Description 鏂板鍑哄簱鍗� * @Author 鍗㈠簡闃� * @Date 2023/10/27 */ - @ApiOperation(value = "鏂板鍑哄簱鍗�", notes = "鏂板鍑哄簱鍗�") - @ApiImplicitParams({@ApiImplicitParam(name = "param", value = "鍑哄簱鍗曞疄浣�", dataType = "lWhFormOutputParam", - dataTypeClass = LWhFormOutputParam.class, paramType = "body")}) + // @ApiOperation(value = "鏂板鍑哄簱鍗�", notes = "鏂板鍑哄簱鍗�") +// @ApiImplicitParams({@ApiImplicitParam(name = "param", value = "鍑哄簱鍗曞疄浣�", dataType = "lWhFormOutputParam", +// dataTypeClass = LWhFormOutputParam.class, paramType = "body")}) @PostMapping("/add") public ResponseValue add() { LWhFormOutputParam param = CommonUtil.getObjFromReqBody(LWhFormOutputParam.class); @@ -95,9 +93,9 @@ /** * @Description 鍒楄〃鏌ヨ */ - @ApiOperation(value = "鍒楄〃鏌ヨ", notes = "鍒楄〃鏌ヨ") - @ApiImplicitParam(name = "param", value = "鍑哄簱鍗曞疄浣�", required = true, dataType = "LWhFormOutputParam", - paramType = "query") + // @ApiOperation(value = "鍒楄〃鏌ヨ", notes = "鍒楄〃鏌ヨ") +// @ApiImplicitParam(name = "param", value = "鍑哄簱鍗曞疄浣�", required = true, dataType = "LWhFormOutputParam", +// paramType = "query") @GetMapping("/list") public ResponseValue queryFormOutputList() { LWhFormOutputQry param = CommonUtil.getObjFromReq(LWhFormOutputQry.class); @@ -130,8 +128,6 @@ } try { Field fieldDatas = GenericPager.class.getDeclaredField("datas"); - // fieldDatas.setAccessible(true); - // fieldDatas.set(genericPager, resultList); ReflectUtil.setFieldValue(genericPager, fieldDatas, resultList); } catch (Exception e) { logger.error("鏌ヨ鍑哄簱鍗曞垪琛ㄥけ璐�", e); @@ -142,10 +138,10 @@ /** * 鏍规嵁id鏌ヨ璇︽儏 */ - @ApiOperation(value = "鏍规嵁id鏌ヨ璇︽儏", notes = "鏍规嵁id鏌ヨ璇︽儏") - @ApiImplicitParam(name = "id", value = "鍑哄簱鍗昳d", required = true, dataType = "Long", paramType = "query") + // @ApiOperation(value = "鏍规嵁id鏌ヨ璇︽儏", notes = "鏍规嵁id鏌ヨ璇︽儏") + // @ApiImplicitParam(name = "id", value = "鍑哄簱鍗昳d", required = true, dataType = "Long", paramType = "query") @GetMapping("/detail") - public ResponseValue getById(Long id) throws IllegalAccessException { + public ResponseValue getById(Long id) throws Exception { if (id == null) { return ResponseValue.error("鍑哄簱鍗昳d涓虹┖"); } @@ -154,53 +150,66 @@ BeanUtils.copyProperties(lWhFormOutput, lWhFormOutputExtendVO); lWhFormOutputExtendVO.setDoc(lWhFormOutput.getOutputDoc()); - String sql = "SELECT bgt.id, CATEGORY_ID, CATEGORY_NAME, GOODS_NAME " - + "FROM l_wh_procure_model pm LEFT JOIN base_goods_models bgm ON bgm.id = pm.BASE_GOODS_MODELS_ID " - + "LEFT JOIN base_goods_template bgt ON bgt.id = bgm.GOODS_TEMPLATES_ID " - + "WHERE pm.BUSINESS_ID =:id GROUP BY bgt.id "; - Map<String, Object> paramMap = new HashMap<>(); - paramMap.put("id", lWhFormOutputExtendVO.getId()); - List<Map<String, Object>> procureModelList = lWhFormOutputService.select(sql, paramMap, new MapperUtil()); - List<FormOutputGoodsVO> fromOutputGoods = Lists.newArrayList(); - for (Map<String, Object> map : procureModelList) { - FormOutputGoodsVO procureTemplateInfoVO = MapUtils.convertMapToObj(map, FormOutputGoodsVO.class); - - // 鏌ヨ鍨嬪彿鏁伴噺 - List<GoodsModelVO> goodsModelVOList = Lists.newArrayList(); - String sql2 = "SELECT pm.id,pm.BASE_GOODS_MODELS_NAME,bgm.UNIT,COUNTS,total_amount " - + "FROM l_wh_procure_model pm LEFT JOIN base_goods_models bgm ON bgm.id = pm.BASE_GOODS_MODELS_ID " - + "LEFT JOIN base_goods_template bgt ON bgt.id = bgm.GOODS_TEMPLATES_ID " + "WHERE pm.BUSINESS_ID =:id"; - List<Map<String, Object>> modelList = lWhFormOutputService.select(sql2, paramMap, new MapperUtil()); - modelList.forEach(item -> { - GoodsModelVO goodsModelVO = MapUtils.convertMapToObj(item, GoodsModelVO.class); - goodsModelVOList.add(goodsModelVO); - }); - procureTemplateInfoVO.setModels(goodsModelVOList); - fromOutputGoods.add(procureTemplateInfoVO); + List<GoodModelInfoDTO> goodsModelList = + lWhProcureModelService.getGoodsModelListByBusinessId(lWhFormOutputExtendVO.getId(), null); + if (CollectionUtils.isEmpty(goodsModelList)) { + return ResponseValue.success(lWhFormOutputExtendVO); } - lWhFormOutputExtendVO.setFromOutputGoods(fromOutputGoods); + Map<Long, List<GoodModelInfoDTO>> collect = + goodsModelList.stream().collect(Collectors.groupingBy(GoodModelInfoDTO::getBaseGoodsTemplateId)); + List<GoodsTemplateInfoVO> goodsTemplateInfoList = Lists.newArrayList(); + for (Map.Entry<Long, List<GoodModelInfoDTO>> entry : collect.entrySet()) { + GoodsTemplateInfoVO goodsTemplateInfoVO = getFormOutputGoodsVO(entry); + goodsTemplateInfoList.add(goodsTemplateInfoVO); + } + lWhFormOutputExtendVO.setFromOutputGoods(goodsTemplateInfoList); return ResponseValue.success("鏌ヨ鎴愬姛!", lWhFormOutputExtendVO); } - @ApiOperation(value = "鏌ヨ鍑哄簱鍗曡鎯呮槑缁�", notes = "鏌ヨ鍑哄簱鍗曡鎯呮槑缁�") - @ApiImplicitParam(name = "formOutputQry", value = "鍑哄簱鍗曡鎯呮煡璇㈡潯浠�", required = true, dataType = "LWhFormOutputQry", - paramType = "query") + private GoodsTemplateInfoVO getFormOutputGoodsVO(Entry<Long, List<GoodModelInfoDTO>> entry) { + Long baseGoodsTemplateId = entry.getKey(); + List<GoodModelInfoDTO> goodModelInfoDTOS = entry.getValue(); + GoodModelInfoDTO goodModelInfoDTO = goodModelInfoDTOS.stream().findFirst().orElse(null); + GoodsTemplateInfoVO goodsTemplateInfoVO = new GoodsTemplateInfoVO(); + if (goodModelInfoDTO != null) { + goodsTemplateInfoVO.setId(baseGoodsTemplateId); + goodsTemplateInfoVO.setCategoryName(goodModelInfoDTO.getCategoryName()); + goodsTemplateInfoVO.setGoodsName(goodModelInfoDTO.getBaseGoodsModelsName()); + } + List<GoodsModelVO> goodsModelList = Lists.newArrayList(); + for (GoodModelInfoDTO goodModelInfo : goodModelInfoDTOS) { + GoodsModelVO goodsModelVO = new GoodsModelVO(); + goodsModelVO.setId(goodModelInfo.getId()); + goodsModelVO.setBaseGoodsModelsName(goodModelInfo.getBaseGoodsModelsName()); + goodsModelVO.setUnit(goodModelInfo.getUnit()); + goodsModelVO.setCounts(goodModelInfo.getCounts()); + goodsModelVO.setTotalAmount(CurrencyUtil.convertFenToYuan(goodModelInfo.getTotalAmount())); + goodsModelList.add(goodsModelVO); + } + goodsTemplateInfoVO.setModels(goodsModelList); + return goodsTemplateInfoVO; + } + + // @ApiOperation(value = "鏌ヨ鍑哄簱鍗曡鎯呮槑缁�", notes = "鏌ヨ鍑哄簱鍗曡鎯呮槑缁�") +// @ApiImplicitParam(name = "formOutputQry", value = "鍑哄簱鍗曡鎯呮煡璇㈡潯浠�", required = true, dataType = "LWhFormOutputQry", +// paramType = "query") @GetMapping("/detail/list") public ResponseValue queryFormOutputDetailList() { LWhFormOutputQry formOutputQry = CommonUtil.getObjFromReq(LWhFormOutputQry.class); - S_user_core currentUser = this.getCurrentUser(); + FinSysTenantUser currentUser = this.getSysInfo(); if (currentUser == null) { return ResponseValue.error("鐧诲綍鐢ㄦ埛淇℃伅涓嶅瓨鍦�"); } - PageUtil genericPager = lWarehouseFlowService.queryAllBusinessFlow(formOutputQry); + formOutputQry.setAgencyId(Long.valueOf(currentUser.getTenantId())); + PageUtil genericPager = lWarehouseFlowServiceImpl.queryAllBusinessFlow(formOutputQry); return ResponseValue.success(genericPager); } - @ApiOperation(value = "鍑哄簱鍗曞鍑�", notes = "鍑哄簱鍗曞鍑�") - @ApiImplicitParam(name = "id", value = "鍑哄簱鍗昳d", required = true, dataType = "Long", paramType = "query") + // @ApiOperation(value = "鍑哄簱鍗曞鍑�", notes = "鍑哄簱鍗曞鍑�") + // @ApiImplicitParam(name = "id", value = "鍑哄簱鍗昳d", required = true, dataType = "Long", paramType = "query") @GetMapping("/list/export") - public void export(Long id, HttpServletResponse response) throws Exception { + public ResponseValue export(Long id, HttpServletResponse response) throws Exception { FinSysTenantUser sysInfo = this.getSysInfo(); if (sysInfo == null) { throw new RuntimeException("鐧诲綍鐢ㄦ埛淇℃伅涓嶅瓨鍦�"); @@ -236,7 +245,8 @@ map.put("totalAmount", totalAmount); Workbook workbook = ExcelExportUtil.exportExcel(params, OutputExcelTemplate.class, exportList, map); - downLoadExcel("鍑哄簱鍗�", response, workbook); + String filePath = downLoadExcel("鍑哄簱鍗�", workbook); + return ResponseValue.success("瀵煎嚭鎴愬姛", filePath); } -- Gitblit v1.9.1