From 8fc8b4788ed8be27ff6aae1cdd3fe9a584972ae3 Mon Sep 17 00:00:00 2001 From: cy <1664593601@qq.com> Date: 星期三, 22 十一月 2023 18:41:05 +0800 Subject: [PATCH] feat: 物品统计;库存预警模板下载、上传、下载异常文件 --- consum-base/src/main/java/com/consum/base/controller/LWhWarningConfigController.java | 186 ++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 186 insertions(+), 0 deletions(-) diff --git a/consum-base/src/main/java/com/consum/base/controller/LWhWarningConfigController.java b/consum-base/src/main/java/com/consum/base/controller/LWhWarningConfigController.java index 4143c23..783f3de 100644 --- a/consum-base/src/main/java/com/consum/base/controller/LWhWarningConfigController.java +++ b/consum-base/src/main/java/com/consum/base/controller/LWhWarningConfigController.java @@ -1,8 +1,13 @@ package com.consum.base.controller; +import cn.afterturn.easypoi.excel.ExcelImportUtil; +import cn.afterturn.easypoi.excel.entity.ImportParams; +import cn.afterturn.easypoi.excel.entity.result.ExcelImportResult; import cn.hutool.core.convert.Convert; import com.consum.base.BaseController; +import com.consum.base.config.WhWarningImportVerifyHandler; import com.consum.base.core.utils.CommonUtil; +import com.consum.base.pojo.WarnConfImEntity; import com.consum.base.pojo.WhWarningConfigParam; import com.consum.base.pojo.query.WhWarningConfigQry; import com.consum.base.service.BaseWarehouseServiceImpl; @@ -12,17 +17,32 @@ import com.consum.model.po.WhWarningConfig; import com.fasterxml.jackson.databind.ObjectMapper; import com.walker.db.page.GenericPager; +import com.walker.file.FileInfo; import com.walker.infrastructure.utils.NumberGenerator; +import com.walker.web.ResponseCode; import com.walker.web.ResponseValue; import io.swagger.annotations.Api; import org.apache.commons.lang3.StringUtils; +import org.springframework.core.io.ClassPathResource; +import org.springframework.core.io.InputStreamResource; +import org.springframework.http.HttpHeaders; +import org.springframework.http.HttpStatus; +import org.springframework.http.MediaType; +import org.springframework.http.ResponseEntity; import org.springframework.util.CollectionUtils; import org.springframework.web.bind.annotation.*; +import org.springframework.web.multipart.MultipartFile; import javax.annotation.Resource; +import java.io.ByteArrayInputStream; +import java.io.IOException; +import java.io.InputStream; +import java.net.URLEncoder; +import java.nio.charset.StandardCharsets; import java.util.ArrayList; import java.util.List; import java.util.Map; +import java.util.stream.Collectors; /** * @ClassName LWhWarningConfigController @@ -175,4 +195,170 @@ } return ResponseValue.success(lWhWarningConfigService.select(param)); } + + /** + * @return 鑾峰彇椤圭洰瀵煎叆鐨勬ā鏉� + * @throws IOException + */ + @PostMapping("getWarnConfImportTemplate") + public ResponseEntity<InputStreamResource> getProjectImportTemplate() throws IOException { + // 浠庡綋鍓嶉」鐩祫婧愮洰褰曡幏鍙栨枃浠� + org.springframework.core.io.Resource resource = new ClassPathResource("import/warningConfImp.xls"); + // 鑾峰彇鏂囦欢杈撳叆娴� + InputStream inputStream = resource.getInputStream(); + // 璁剧疆HTTP鍝嶅簲澶� + HttpHeaders headers = new HttpHeaders(); + headers.setContentType(MediaType.APPLICATION_OCTET_STREAM); + String encodedFilename = URLEncoder.encode("搴撳瓨棰勮閰嶇疆瀵煎叆妯℃澘.xls", "UTF-8"); + headers.setContentDispositionFormData("attachment", encodedFilename); + // 鍒涘缓InputStreamResource瀵硅薄锛屽皢鏂囦欢杈撳叆娴佸寘瑁呰捣鏉� + InputStreamResource resourceToDownload = new InputStreamResource(inputStream); + // 杩斿洖甯︽湁鏂囦欢杈撳叆娴佺殑ResponseEntity瀵硅薄 + return ResponseEntity + .status(HttpStatus.OK) + .headers(headers) + .body(resourceToDownload); + } + + + @PostMapping("import") + public ResponseValue warnConfImport(MultipartFile multipartFile) throws Exception { + String originalFilename = multipartFile.getOriginalFilename(); + if (!"xls".equals(originalFilename.substring(originalFilename.lastIndexOf(".") + 1))) { + return ResponseValue.error(ResponseCode.ERROR.getCode(), "鏂囦欢鏍煎紡鏈夎!"); + } + FinSysTenantUser sysInfo = this.getSysInfo(); + if (sysInfo == null) { + return ResponseValue.error(ResponseCode.ERROR.getCode(), "褰撳墠鐧诲綍鐢ㄦ埛涓虹┖"); + } + + ImportParams params = new ImportParams(); + params.setHeadRows(1); + params.setNeedVerify(true); + WhWarningImportVerifyHandler whWarningImportVerifyHandler = new WhWarningImportVerifyHandler(Long.valueOf(sysInfo.getTenantId())); + params.setVerifyHandler(whWarningImportVerifyHandler); + + ExcelImportResult<WarnConfImEntity> result = ExcelImportUtil.importExcelMore(multipartFile.getInputStream(), WarnConfImEntity.class, params); + + boolean verfiyFail = result.isVerfiyFail(); + if (verfiyFail) { + StringBuilder errmsg = new StringBuilder(); + List<WarnConfImEntity> failList = result.getFailList(); + for (int i = 0; i < failList.size() && i < 20; i++) { + WarnConfImEntity entity = failList.get(i); + errmsg.append("绗�") + .append(entity.getRowNum()) + .append("琛岀殑閿欒鏄細") + .append(entity.getErrorMsg()) + .append("\r\n"); + } + InputStream inputStream = new ByteArrayInputStream(errmsg.toString().getBytes(StandardCharsets.UTF_8)); + FileInfo fileInfo = uploadFileToLocal(inputStream, NumberGenerator.generatorHexUUID() + ".txt", "-1", 111, 0, "-1"); + String url = fileInfo.getUrl(); + return ResponseValue.error(ResponseCode.PARAM_NOT_VALID.getCode(), url); + } + + + List<WarnConfImEntity> excelList = result.getList(); + ArrayList<WhWarningConfig> configArrayList = new ArrayList<>(excelList.size()); + // 鎸塨aseWarehouseId灞炴�у垎缁� + Map<Long, List<WarnConfImEntity>> groupedData = excelList.stream().collect(Collectors.groupingBy(WarnConfImEntity::getBaseWarehouseId)); + for (Map.Entry<Long, List<WarnConfImEntity>> entry : groupedData.entrySet()) { + Long baseWarehouseId = entry.getKey(); + List<WarnConfImEntity> houseModelList = entry.getValue(); + // 鎻愬彇id骞跺瓨鍌ㄥ埌鏂板垪琛� + List<Long> modelIdList = houseModelList.stream() + .map(WarnConfImEntity::getBaseGoodsModelsId) + .collect(Collectors.toList()); + + List<Map<String, Object>> queryModelIsExistList = lWhWarningConfigService.queryModelIsExist(0, baseWarehouseId, modelIdList); + if (CollectionUtils.isEmpty(queryModelIsExistList)) { + // 鍏ㄩ儴涓烘柊澧� + for (WarnConfImEntity warnConfImEntity : houseModelList) { + // 鏂板 + WhWarningConfig whWarningConfig = new WhWarningConfig(); + whWarningConfig.setId(NumberGenerator.getLongSequenceNumber()); + whWarningConfig.setWarehouseType(0); + whWarningConfig.setBaseWarehouseId(warnConfImEntity.getBaseWarehouseId()); + whWarningConfig.setBaseGoodsTemplateId(warnConfImEntity.getGoodsTemplateId()); + whWarningConfig.setBaseGoodsModelsId(warnConfImEntity.getBaseGoodsModelsId()); + whWarningConfig.setUpperLimit(warnConfImEntity.getUpperLimit()); + whWarningConfig.setLowerLimit(warnConfImEntity.getLowerLimit()); + if (whWarningConfig.getBaseGoodsModelsId() == null) { + whWarningConfig.setGoodsType(1); + } else { + whWarningConfig.setGoodsType(2); + } + configArrayList.add(whWarningConfig); + } + } else { + // 鎻愬彇id骞跺瓨鍌ㄥ埌鏂板垪琛� + List<Long> dbExistModelIdList = queryModelIsExistList.stream() + .map(map -> Convert.toLong(map.get("baseGoodsModelsId"))) + .collect(Collectors.toList()); + for (WarnConfImEntity warnConfImEntity : houseModelList) { + Long baseGoodsModelsId = warnConfImEntity.getBaseGoodsModelsId(); + WhWarningConfig whWarningConfig = new WhWarningConfig(); + whWarningConfig.setWarehouseType(0); + whWarningConfig.setBaseWarehouseId(warnConfImEntity.getBaseWarehouseId()); + whWarningConfig.setBaseGoodsModelsId(warnConfImEntity.getBaseGoodsModelsId()); + whWarningConfig.setUpperLimit(warnConfImEntity.getUpperLimit()); + whWarningConfig.setLowerLimit(warnConfImEntity.getLowerLimit()); + + if (dbExistModelIdList.contains(baseGoodsModelsId)) { + // 淇敼 + limitEditByModeInfo(whWarningConfig); + } else { + // 鏂板 + whWarningConfig.setId(NumberGenerator.getLongSequenceNumber()); + whWarningConfig.setBaseGoodsTemplateId(warnConfImEntity.getGoodsTemplateId()); + if (whWarningConfig.getBaseGoodsModelsId() == null) { + whWarningConfig.setGoodsType(1); + } else { + whWarningConfig.setGoodsType(2); + } + configArrayList.add(whWarningConfig); + } + } + } + } + if (!CollectionUtils.isEmpty(configArrayList)) { + int flag = lWhWarningConfigService.insertBatch(configArrayList); + } + return ResponseValue.success("瀵煎叆鎴愬姛!", 1); + } + + + private String KU_CUN_UPD_BY_MODE_INFO = "update WH_WARNING_CONFIG set "; + + public int limitEditByModeInfo(WhWarningConfig config) { + StringBuilder sql = new StringBuilder(KU_CUN_UPD_BY_MODE_INFO); + Integer lowerLimit = config.getLowerLimit(); + if (lowerLimit == null) { + sql.append("lower_limit=null"); + } else { + sql.append("lower_limit=" + lowerLimit); + } + Integer upperLimit = config.getUpperLimit(); + if (upperLimit == null) { + sql.append(",upper_Limit=null"); + } else { + sql.append(",upper_Limit=" + upperLimit); + } + sql.append(" where 1=1"); + Integer warehouseType = config.getWarehouseType(); + if (warehouseType != null) { + sql.append(" and warehouse_Type=" + warehouseType); + } + Long baseWarehouseId = config.getBaseWarehouseId(); + if (baseWarehouseId != null) { + sql.append(" and BASE_WAREHOUSE_ID=" + baseWarehouseId); + } + Long baseGoodsModelsId = config.getBaseGoodsModelsId(); + if (baseGoodsModelsId != null) { + sql.append(" and BASE_GOODS_MODELS_ID=" + baseGoodsModelsId); + } + lWhWarningConfigService.update(sql.toString()); + return 1; + } } -- Gitblit v1.9.1