From fb036b3ca33d483afbf83eede9fd806a2f77deaa Mon Sep 17 00:00:00 2001
From: futian.liu <liufutianyoo@163.com>
Date: 星期三, 06 十二月 2023 13:06:41 +0800
Subject: [PATCH] 代码优化、漏洞修复
---
consum-base/src/main/java/com/consum/base/controller/LWhFormProcureController.java | 277 +++++++++++++++++++++----------------------------------
1 files changed, 106 insertions(+), 171 deletions(-)
diff --git a/consum-base/src/main/java/com/consum/base/controller/LWhFormProcureController.java b/consum-base/src/main/java/com/consum/base/controller/LWhFormProcureController.java
index 79df3ac..a6e626a 100644
--- a/consum-base/src/main/java/com/consum/base/controller/LWhFormProcureController.java
+++ b/consum-base/src/main/java/com/consum/base/controller/LWhFormProcureController.java
@@ -1,37 +1,60 @@
package com.consum.base.controller;
-import cn.hutool.core.util.ReflectUtil;
+import java.lang.reflect.Field;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Optional;
+
+import javax.annotation.Resource;
+import javax.servlet.http.HttpServletResponse;
+
+import org.apache.commons.compress.utils.Lists;
+import org.apache.poi.ss.usermodel.Workbook;
+import org.springframework.beans.BeanUtils;
+import org.springframework.web.bind.annotation.DeleteMapping;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
import com.consum.base.BaseController;
-import com.consum.base.core.CodeGeneratorEnum;
-import com.consum.base.core.CodeGeneratorService;
-import com.consum.base.core.WhBusinessEnum;
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.PageUtil;
import com.consum.base.pojo.LWhFormProcureGoodsInfoParam;
import com.consum.base.pojo.LWhFormProcureParam;
-import com.consum.base.pojo.LWhProcureModelParam;
+import com.consum.base.pojo.excel.ProcureExcelTemplate;
import com.consum.base.pojo.query.FormProcureQry;
-import com.consum.base.pojo.response.*;
-import com.consum.base.service.*;
-import com.consum.model.po.*;
+import com.consum.base.pojo.response.FormProcureVO;
+import com.consum.base.pojo.response.GoodsModelVO;
+import com.consum.base.pojo.response.GoodsTemplateCountVO;
+import com.consum.base.pojo.response.LWhFormProcureExtendVO;
+import com.consum.base.pojo.response.LWhFormProcureGoodsVO;
+import com.consum.base.service.BaseGoodsTemplateService;
+import com.consum.base.service.impl.LWhFormProcureCoreService;
+import com.consum.base.service.LWhFormProcureGoodsService;
+import com.consum.base.service.LWhFormProcureService;
+import com.consum.base.service.LWhProcureModelService;
+import com.consum.model.po.BaseGoodsTemplate;
+import com.consum.model.po.FinSysTenantUser;
+import com.consum.model.po.LWhFormProcure;
+import com.consum.model.po.LWhFormProcureGoods;
import com.iplatform.model.po.S_user_core;
import com.walker.db.page.GenericPager;
import com.walker.infrastructure.utils.CollectionUtils;
-import com.walker.infrastructure.utils.NumberGenerator;
+import com.walker.infrastructure.utils.DateUtils;
import com.walker.web.ResponseValue;
+
+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;
-import org.apache.commons.compress.utils.Lists;
-import org.springframework.beans.BeanUtils;
-import org.springframework.web.bind.annotation.*;
-
-import javax.annotation.Resource;
-import java.lang.reflect.Field;
-import java.util.*;
/**
* @ClassName LWhFormProcureController
@@ -47,23 +70,13 @@
@Resource
private LWhFormProcureService lWhFormProcureService;
@Resource
- private BaseWarehouseServiceImpl baseWarehouseService;
- @Resource
- private FinSysTenantServiceImpl finSysTenantService;
- @Resource
- private CodeGeneratorService codeGeneratorService;
- @Resource
- private LWhFormProcureGoodsService lWhFormProcureGoodsService;
+ private LWhFormProcureGoodsService lWhFormProcureGoodsServiceImpl;
@Resource
private LWhFormProcureCoreService lWhFormProcureCoreService;
@Resource
- private LWhGoodsService lWhGoodsService;
- @Resource
- private BaseGoodsTemplateServiceImpl baseGoodsTemplateService;
+ private BaseGoodsTemplateService baseGoodsTemplateService;
@Resource
private LWhProcureModelService lWhProcureModelService;
- @Resource
- private LOrgSupplierServiceImpl lOrgSupplierService;
/**
* @Description 鏂板
@@ -78,134 +91,26 @@
}
private ResponseValue add(LWhFormProcureParam param) {
- Long warehouseId = param.getWarehouseId();
if (param.getWarehouseId() == null) {
return ResponseValue.error("浠撳簱ID涓嶈兘涓虹┖");
}
- BaseWarehouse baseWarehouse = baseWarehouseService.getById(warehouseId);
- FinSysTenantUser sysTenantUser = this.getSysInfo();
- String tenantId = sysTenantUser.getTenantId();
- String tenantName = sysTenantUser.getTenantName();
List<LWhFormProcureGoodsInfoParam> procureGoods = param.getProcureGoods();
if (CollectionUtils.isEmpty(procureGoods)) {
return ResponseValue.error("閲囪喘鍗曚笉鑳戒负绌�");
}
-
+ FinSysTenantUser sysTenantUser = this.getSysInfo();
S_user_core currentUser = this.getCurrentUser();
- // 閲囪喘鍗昳d
- Long whFormProcureId = param.getId();
- if (whFormProcureId == null) {
- whFormProcureId = NumberGenerator.getLongSequenceNumber();
- }
- long procureTime = Long.valueOf(param.getProcureTime());
-
- Set<String> supplierList = new HashSet<>();
-
- ArrayList<LWhFormProcureGoods> procureGoodList = new ArrayList<>();
- ArrayList<LWhProcureModel> procureModelList = new ArrayList<>();
-
- for (int i = 0; i < procureGoods.size(); i++) {
- LWhFormProcureGoodsInfoParam procureGoodT = procureGoods.get(i);
- Long baseCategoryId = procureGoodT.getBaseCategoryId();
- Long baseGoodsTemplateId = procureGoodT.getBaseGoodsTemplateId();
- String supplier = procureGoodT.getSupplier();
- // 鍒ゆ柇褰撳墠渚涘簲鍟嗘槸鍚﹀凡瀛樺湪
- if (!supplierList.contains(supplier) && !lOrgSupplierService.supplierIsContain(tenantId, supplier)) {
- supplierList.add(supplier);
- }
-
- List<LWhProcureModelParam> models = procureGoodT.getModels();
- if (CollectionUtils.isEmpty(models)) {
- return ResponseValue.error("閲囪喘鍗曚笉鑳戒负绌�");
- }
- LWhFormProcureGoods procureGood = new LWhFormProcureGoods();
- long fromProcureGoodsId = NumberGenerator.getLongSequenceNumber();
- procureGood.setId(fromProcureGoodsId);
- procureGood.setWhFormProcureId(whFormProcureId);
- procureGood.setBaseCategoryId(baseCategoryId);
- procureGood.setBaseGoodsTemplateId(baseGoodsTemplateId);
- // TODO 鐗╁搧鍘嗗彶鐗堟湰
- procureGood.setGoodsTemplateName(procureGoodT.getGoodsTemplateName());
- procureGood.setSupplier(supplier);
- procureGood.setSort(i);
- for (LWhProcureModelParam modelT : models) {
- Integer counts = modelT.getCounts();
- if (counts == null || counts <= 0) {
- return ResponseValue.error("鍨嬪彿鐨勯噰璐暟閲忓彧鑳戒负姝ｆ暣鏁�");
- }
- Long price = modelT.getPrice();
- if (price == null || price <= 0) {
- return ResponseValue.error("鍨嬪彿鐨勯噰璐环鏍煎彧鑳戒负姝ｆ暟");
- }
- Long baseGoodsModelsId = modelT.getBaseGoodsModelsId();
- if (baseGoodsModelsId == null) {
- return ResponseValue.error("閲囪喘鐨勪环鍨嬪彿鏈壘鍒�");
- }
- //鏌ヨ鍨嬪彿淇℃伅
- Map<String, Object> tempGoodsInfo = baseGoodsTemplateService.queryGoodsInfoByModelId(baseGoodsModelsId);
- if (tempGoodsInfo == null) {
- return ResponseValue.error("閲囪喘鐨勪环鍨嬪彿鏈壘鍒�");
- }
- String modelName = (String) tempGoodsInfo.get("modelname");
- // 鏌ヨ鍨嬪彿搴撳瓨
- int goodsModelNum = lWhGoodsService.queryGoodsModelNum(0, warehouseId, baseGoodsModelsId, 1, null);
- LWhProcureModel model = new LWhProcureModel();
- model.setId(NumberGenerator.getLongSequenceNumber());
- model.setFromProcureGoodsId(fromProcureGoodsId);
- model.setBusinessType(WhBusinessEnum.CAIGOU.getValue() + 0);
- model.setBusinessId(whFormProcureId);
- model.setBaseGoodsModelsId(baseGoodsModelsId);
- model.setBaseGoodsModelsName(modelName);
- model.setPrice(price);
- model.setCounts(counts);
- model.setWorehouseCount(goodsModelNum);
- model.setSupplier(supplier);
- model.setTotalAmount(price * counts);
- procureModelList.add(model);
- }
- procureGoodList.add(procureGood);
- }
- lWhFormProcureGoodsService.insertBatch(procureGoodList);
- lWhProcureModelService.insertBatch(procureModelList);
-
- // 鎻掑叆 閲囪喘鍗曠墿鍝併�怢_WH_FORM_PROCURE_GOODS銆�
- LWhFormProcure lWhFormProcure = new LWhFormProcure();
- lWhFormProcure.setId(whFormProcureId);
- lWhFormProcure.setBusinessFormCode(codeGeneratorService.createBusinessFormCode(CodeGeneratorEnum.Procure_Warehouse));
- lWhFormProcure.setWarehouseId(warehouseId);
- lWhFormProcure.setWarehouseName(baseWarehouse.getWarehouseName());
- lWhFormProcure.setBuyerId(currentUser.getId());
- lWhFormProcure.setBuyerName(currentUser.getNick_name());
- lWhFormProcure.setProcureTime(procureTime);
- lWhFormProcure.setProcureDoc(param.getProcureDoc());
- lWhFormProcure.setBuyType(param.getBuyType());
- lWhFormProcure.setStates(1);
- lWhFormProcure.setAgencyId(baseWarehouse.getAgencyId());
- lWhFormProcure.setAgencyName(baseWarehouse.getAgencyName());
- int result = lWhFormProcureService.insert(lWhFormProcure);
-
- supplierList.forEach(supplier -> {
- LOrgSupplier lOrgSupplier = new LOrgSupplier();
- lOrgSupplier.setId(NumberGenerator.getLongSequenceNumber());
- lOrgSupplier.setAgencyId(tenantId);
- lOrgSupplier.setAgencyName(tenantName);
- lOrgSupplier.setSupplier(supplier);
- lOrgSupplierService.insert(lOrgSupplier);
- });
- if (result > 0) {
- return ResponseValue.success(1);
- }
- return ResponseValue.error("鏂板澶辫触锛�");
+ lWhFormProcureService.add(param, sysTenantUser, currentUser);
+ return ResponseValue.success();
}
/**
* @Description 鍒楄〃鏌ヨ
*/
@ApiOperation(value = "閲囪喘鍗曞垪琛ㄦ煡璇�", notes = "閲囪喘鍗曞垪琛ㄦ煡璇�")
- @ApiImplicitParams({
- @ApiImplicitParam(name = "param", value = "閲囪喘鏌ヨ鏉′欢", required = true, dataType = "FormProcureQryDto", paramType = "query")
- })
+ @ApiImplicitParams({@ApiImplicitParam(name = "param", value = "閲囪喘鏌ヨ鏉′欢", required = true,
+ dataType = "FormProcureQryDto", paramType = "query")})
@GetMapping("/list")
public ResponseValue queryFormProcureList() {
FormProcureQry param = CommonUtil.getObjFromReq(FormProcureQry.class);
@@ -227,31 +132,21 @@
ArrayList<FormProcureVO> result = new ArrayList<>();
if (!CollectionUtils.isEmpty(data)) {
data.forEach(item -> {
-
FormProcureVO fromProcureVO = new FormProcureVO();
BeanUtils.copyProperties(item, fromProcureVO);
- List<FromProcureTemplateInfoVO> procureTemplateInfoList = Lists.newArrayList();
// 鏌ヨ鍨嬪彿鏁伴噺
- String sql = "select fpg.id,BASE_CATEGORY_ID ,BASE_GOODS_TEMPLATE_ID ,GOODS_TEMPLATE_NAME ,sum(counts) count "
- + "from L_WH_FORM_PROCURE_GOODS fpg left join L_WH_PROCURE_MODEL pm on fpg.id = pm.FROM_PROCURE_GOODS_ID "
- + "where fpg.WH_FORM_PROCURE_ID =:id group by pm.FROM_PROCURE_GOODS_ID";
- Map<String, Object> paramMap = new HashMap<>();
- paramMap.put("id", item.getId());
- List<Map<String, Object>> procureModelList = lWhProcureModelService.select(sql, paramMap, new MapperUtil());
- for (Map<String, Object> map : procureModelList) {
- FromProcureTemplateInfoVO procureTemplateInfoVO = MapUtils.convertMapToObj(map, FromProcureTemplateInfoVO.class);
- procureTemplateInfoList.add(procureTemplateInfoVO);
- }
- fromProcureVO.setFromProcureTemplateInfoList(procureTemplateInfoList);
+ List<GoodsTemplateCountVO> procureCount =
+ lWhProcureModelService.getProcureCountByBusinessId(item.getId());
+ fromProcureVO.setFromProcureTemplateInfoList(procureCount);
result.add(fromProcureVO);
});
}
try {
Field fieldDatas = GenericPager.class.getDeclaredField("datas");
-// fieldDatas.setAccessible(true);
-// fieldDatas.set(genericPager, result);
+ // fieldDatas.setAccessible(true);
+ // fieldDatas.set(genericPager, result);
ReflectUtil.setFieldValue(genericPager, fieldDatas, result);
} catch (Exception e) {
e.printStackTrace();
@@ -295,7 +190,7 @@
if (num == 0) {
return ResponseValue.error("鍒犻櫎澶辫触锛�");
}
- lWhFormProcureGoodsService.update("delete from L_WH_FORM_PROCURE_GOODS where WH_FORM_PROCURE_ID=" + id);
+ lWhFormProcureGoodsServiceImpl.update("delete from L_WH_FORM_PROCURE_GOODS where WH_FORM_PROCURE_ID=" + id);
lWhProcureModelService.update("delete from L_WH_PROCURE_MODEL where BUSINESS_TYPE =1 and BUSINESS_ID=" + id);
return ResponseValue.success(1);
@@ -331,15 +226,15 @@
// 鐗╁搧
LWhFormProcureGoods lWhFormProcureGoods = new LWhFormProcureGoods();
lWhFormProcureGoods.setWhFormProcureId(id);
- List<LWhFormProcureGoods> formProcureGoods = lWhFormProcureGoodsService.select(lWhFormProcureGoods);
+ List<LWhFormProcureGoods> formProcureGoods = lWhFormProcureGoodsServiceImpl.select(lWhFormProcureGoods);
-// Field fieldModels = null;
-// try {
-// fieldModels = LWhFormProcureGoodsVO.class.getDeclaredField("models");
-// } catch (NoSuchFieldException e) {
-// e.printStackTrace();
-// }
-// fieldModels.setAccessible(true);
+ // Field fieldModels = null;
+ // try {
+ // fieldModels = LWhFormProcureGoodsVO.class.getDeclaredField("models");
+ // } catch (NoSuchFieldException e) {
+ // e.printStackTrace();
+ // }
+ // fieldModels.setAccessible(true);
ArrayList<LWhFormProcureGoodsVO> procureGoodsVOList = new ArrayList<>();
for (LWhFormProcureGoods formProcureGood : formProcureGoods) {
LWhFormProcureGoodsVO lWhFormProcureGoodsVO = new LWhFormProcureGoodsVO();
@@ -353,16 +248,17 @@
// 鏌ヨ鍨嬪彿鏁伴噺
Map<String, Object> paramMap = new HashMap<>();
paramMap.put("formProcureGoodId", formProcureGood.getId());
- String sql = "SELECT\n" + "pm.BASE_GOODS_MODELS_ID,pm.BASE_GOODS_MODELS_NAME,pm.PRICE,pm.COUNTS,pm.WOREHOUSE_COUNT,bgm.unit " + "FROM\n"
- + "\tl_wh_procure_model pm\n"
- + "\tLEFT JOIN base_goods_models bgm on pm.BASE_GOODS_MODELS_ID = bgm.id where pm.FROM_PROCURE_GOODS_ID =:formProcureGoodId";
+ String sql = "SELECT\n"
+ + "pm.BASE_GOODS_MODELS_ID,pm.BASE_GOODS_MODELS_NAME,pm.PRICE,pm.COUNTS,pm.WOREHOUSE_COUNT,bgm.unit "
+ + "FROM\n" + "\tl_wh_procure_model pm\n"
+ + "\tLEFT JOIN base_goods_models bgm on pm.BASE_GOODS_MODELS_ID = bgm.id where pm.FROM_PROCURE_GOODS_ID =:formProcureGoodId";
List<Map<String, Object>> procureModelList = lWhProcureModelService.select(sql, paramMap, new MapperUtil());
List<GoodsModelVO> goodsModelVOList = Lists.newArrayList();
procureModelList.forEach(item -> {
GoodsModelVO goodsModelVO = MapUtils.convertMapToObj(item, GoodsModelVO.class);
goodsModelVOList.add(goodsModelVO);
});
-// fieldModels.set(lWhFormProcureGoodsVO, goodsModelVOList);
+ // fieldModels.set(lWhFormProcureGoodsVO, goodsModelVOList);
ReflectUtil.setFieldValue(lWhFormProcureGoodsVO, "models", goodsModelVOList);
procureGoodsVOList.add(lWhFormProcureGoodsVO);
@@ -371,9 +267,9 @@
return ResponseValue.success("鏌ヨ鎴愬姛!", lWhFormProcureExtendVO);
}
-
@ApiOperation(value = "閲囪喘鍗曟槑缁嗘煡璇�", notes = "閲囪喘鍗曟槑缁嗘煡璇�")
- @ApiImplicitParams({@ApiImplicitParam(name = "param", value = "閲囪喘鍗曟槑缁嗘煡璇�", required = true, dataType = "FormProcureQryDto", paramType = "query")})
+ @ApiImplicitParams({@ApiImplicitParam(name = "param", value = "閲囪喘鍗曟槑缁嗘煡璇�", required = true,
+ dataType = "FormProcureQryDto", paramType = "query")})
@GetMapping("detail/list")
public ResponseValue queryFormProcureDetailList() {
FormProcureQry formProcureQry = CommonUtil.getObjFromReq(FormProcureQry.class);
@@ -389,4 +285,43 @@
GenericPager genericPager = lWhFormProcureService.queryFormProcureDetailList(formProcureQry);
return ResponseValue.success(genericPager);
}
+
+ @ApiOperation(value = "閲囪喘鍗曞鍑�", notes = "閲囪喘鍗曞鍑�")
+ @ApiImplicitParams({
+ @ApiImplicitParam(name = "id", value = "閲囪喘鍗昳d", required = true, dataType = "Long", paramType = "query")})
+ @GetMapping("/list/export")
+ public void export(Long id, HttpServletResponse response) throws Exception {
+ TemplateExportParams params = new TemplateExportParams("import/閲囪喘鍏ュ簱鍗�.xls");
+ params.setHeadingStartRow(2);
+ FinSysTenantUser sysInfo = this.getSysInfo();
+ if (sysInfo == null) {
+ throw new RuntimeException("鐧诲綍鐢ㄦ埛淇℃伅涓嶅瓨鍦�");
+ }
+
+ List<ProcureExcelTemplate> exportList = lWhFormProcureService.getExportList(id);
+ if (CollectionUtils.isEmpty(exportList)) {
+ throw new RuntimeException("鏁版嵁涓虹┖");
+ }
+
+ int countNum =
+ exportList.stream().filter(item -> item.getNum() != null).mapToInt(ProcureExcelTemplate::getNum).sum();
+ double totalAmount = exportList.stream().filter(export -> export.getTotalAmount() != null)
+ .mapToDouble(ProcureExcelTemplate::getAmount).sum();
+ Optional<ProcureExcelTemplate> first = exportList.stream().findFirst();
+ ProcureExcelTemplate templateExcelExport = first.get();
+ String businessFormCode = templateExcelExport.getBusinessFormCode();
+ Long createTime = templateExcelExport.getCreateTime();
+ String operatorName = templateExcelExport.getOperatorName();
+
+ Map<String, Object> map = new HashMap<>();
+ map.put("code", businessFormCode);
+ map.put("date", DateUtils.toShowDate(createTime));
+ map.put("name", operatorName);
+ map.put("countNum", countNum);
+ map.put("totalAmount", totalAmount);
+
+ Workbook workbook = ExcelExportUtil.exportExcel(params, ProcureExcelTemplate.class, exportList, map);
+ downLoadExcel("閲囪喘鍏ュ簱鍗�", response, workbook);
+
+ }
}
--
Gitblit v1.9.1