| | |
| | | |
| | | import cn.hutool.core.util.IdUtil; |
| | | import com.aps.common.core.utils.DateUtils; |
| | | import com.aps.common.core.utils.poi.ExcelUtil; |
| | | import com.aps.common.security.utils.DictUtils; |
| | | import com.aps.common.security.utils.SecurityUtils; |
| | | import com.aps.core.domain.ApsGasPipelinePrediction; |
| | | import com.aps.core.domain.ApsMaterialProductGroupManagement; |
| | | import com.aps.core.mapper.ApsGasPipelinePredictionMapper; |
| | | import com.aps.core.service.IApsGasPipelinePredictionService; |
| | | import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; |
| | | import io.micrometer.common.util.StringUtils; |
| | | import lombok.SneakyThrows; |
| | | import org.apache.poi.ss.usermodel.Row; |
| | | import org.apache.poi.ss.usermodel.Sheet; |
| | | import org.apache.poi.ss.usermodel.Workbook; |
| | | import org.apache.poi.ss.usermodel.WorkbookFactory; |
| | | import org.springframework.beans.factory.annotation.Autowired; |
| | | import org.springframework.stereotype.Service; |
| | | import org.springframework.web.multipart.MultipartFile; |
| | | |
| | | import java.math.BigDecimal; |
| | | import java.sql.Timestamp; |
| | | import java.text.SimpleDateFormat; |
| | | import java.util.*; |
| | | import java.util.stream.Collectors; |
| | | |
| | |
| | | @SneakyThrows |
| | | @Override |
| | | public int batchInsertApsGasPipelinePrediction(MultipartFile file) { |
| | | Workbook workbook = WorkbookFactory.create(file.getInputStream()); |
| | | |
| | | Sheet sheet = workbook.getSheetAt(0); |
| | | int rows = sheet.getLastRowNum(); |
| | | if (rows > 0) { |
| | | List<ApsGasPipelinePrediction> list = new ArrayList<>(); |
| | | ExcelUtil<ApsGasPipelinePrediction> util = new ExcelUtil<>(ApsGasPipelinePrediction.class); |
| | | List<ApsGasPipelinePrediction> list = util.importExcel(file.getInputStream()); |
| | | DictUtils.CacheValue cacheValue = DictUtils.getCacheValue("aps_factory"); |
| | | Set<String> keys = new HashSet<>(); |
| | | list.forEach(item -> { |
| | | item.setId(IdUtil.getSnowflakeNextId()); |
| | | item.setFactory(cacheValue.get(item.getFactory())); |
| | | item.setCreateBy(SecurityUtils.getUsername()); |
| | | item.setCreateTime(DateUtils.getNowDate()); |
| | | }); |
| | | |
| | | /*数据列从1开始*/ |
| | | for (int i = 1; i <= rows; i++) { |
| | | Row row = sheet.getRow(i); |
| | | if (row.getCell(0) == null) { |
| | | continue; |
| | | } |
| | | for (int i = list.size() - 1; i >= 0; --i) { |
| | | list.get(i).setKey(null); |
| | | if (!keys.contains(list.get(i).getKey())) { |
| | | keys.add(list.get(i).getKey()); |
| | | } else { |
| | | list.remove(i); |
| | | } |
| | | } |
| | | |
| | | String materialCode = row.getCell(0).getStringCellValue(); |
| | | if (StringUtils.isEmpty(materialCode)) { |
| | | continue; |
| | | } |
| | | String factory = row.getCell(1).getStringCellValue(); |
| | | double quantity = row.getCell(2).getNumericCellValue(); |
| | | Date date = row.getCell(3).getDateCellValue(); |
| | | if (StringUtils.isNotBlank(factory) && |
| | | date != null) { |
| | | ApsGasPipelinePrediction item = new ApsGasPipelinePrediction(); |
| | | item.setId(IdUtil.getSnowflakeNextId()); |
| | | item.setFactory(factory); |
| | | item.setMaterialCode(materialCode); |
| | | item.setCreateBy(SecurityUtils.getUsername()); |
| | | item.setCreateTime(DateUtils.getNowDate()); |
| | | item.setPredictQuantity(new BigDecimal(quantity)); |
| | | item.setPredictDate(new Timestamp(date.getTime())); |
| | | list.add(item); |
| | | if (!list.isEmpty()) { |
| | | List<Map<String, Object>> facCodeKey = apsGasPipelinePredictionMapper.selectByFacOrMaterial(keys); |
| | | if (!facCodeKey.isEmpty()) { |
| | | Iterator<ApsGasPipelinePrediction> it = list.iterator(); |
| | | while (it.hasNext()) { |
| | | ApsGasPipelinePrediction item = it.next(); |
| | | int count = 0; |
| | | for (Map<String, Object> map : facCodeKey) { |
| | | if (map.get("key").equals(item.getKey())) { |
| | | item.setId((Long) map.get("id")); |
| | | item.setCreateBy(null); |
| | | item.setCreateTime(null); |
| | | item.setUpdateBy(SecurityUtils.getUsername()); |
| | | item.setUpdateTime(new Timestamp(System.currentTimeMillis())); |
| | | apsGasPipelinePredictionMapper.updateById(item); |
| | | count++; |
| | | } |
| | | } |
| | | if (count > 0) { |
| | | it.remove(); |
| | | } |
| | | } |
| | | } |
| | | if (!list.isEmpty()) { |
| | | |
| | | List<Map<String, Object>> facCodeKey = apsGasPipelinePredictionMapper.selectByFacOrMaterial( |
| | | list.stream().map(ApsGasPipelinePrediction::getFactory).collect(Collectors.toSet()), |
| | | list.stream().map(ApsGasPipelinePrediction::getMaterialCode).collect(Collectors.toSet()) |
| | | ); |
| | | if (!facCodeKey.isEmpty()) { |
| | | SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM"); |
| | | Iterator<ApsGasPipelinePrediction> it = list.iterator(); |
| | | while (it.hasNext()) { |
| | | ApsGasPipelinePrediction item = it.next(); |
| | | String key = item.getFactory() + "##" + item.getMaterialCode() + "##" + sdf.format(item.getPredictDate()); |
| | | facCodeKey.stream() |
| | | .filter(map -> map.get("key").equals(key)) |
| | | .forEach(row -> { |
| | | item.setId((Long) row.get("id")); |
| | | item.setCreateBy(null); |
| | | item.setCreateTime(null); |
| | | item.setUpdateBy(SecurityUtils.getUsername()); |
| | | item.setUpdateTime(new Timestamp(System.currentTimeMillis())); |
| | | apsGasPipelinePredictionMapper.updateById(item); |
| | | it.remove(); |
| | | }); |
| | | } |
| | | } |
| | | if (!list.isEmpty()) { |
| | | apsGasPipelinePredictionMapper.insert(list); |
| | | } |
| | | apsGasPipelinePredictionMapper.insert(list); |
| | | } |
| | | return list.size(); |
| | | } |
| | | return 0; |
| | | return list.size(); |
| | | } |
| | | } |