板材、配件导入重复代码移除、代码规范修正

This commit is contained in:
gaoqr
2025-03-28 15:32:11 +08:00
parent d72d6961aa
commit cf110a3c37
2 changed files with 124 additions and 83 deletions
@@ -13,7 +13,9 @@ import com.cf.imes.module.executor.controller.admin.manage.parts.vo.PartsCadPage
import com.cf.imes.module.executor.controller.admin.manage.parts.vo.PartsImportExcelVO;
import com.cf.imes.module.executor.controller.admin.manage.parts.vo.PartsPageReqVO;
import com.cf.imes.module.executor.controller.admin.manage.parts.vo.PartsSaveReqVO;
import com.cf.imes.module.executor.controller.admin.manage.plate.vo.plate.PlateImportExcelVO;
import com.cf.imes.module.executor.dal.dataobject.orderParts.PartsDO;
import com.cf.imes.module.executor.dal.dataobject.plate.PlateGoodDO;
import com.cf.imes.module.executor.dal.mysql.orderParts.PartsBatchMapper;
import com.cf.imes.module.executor.dal.mysql.orderParts.PartsMapper;
import com.cf.imes.module.executor.enums.manage.parts.CategoryEnum;
@@ -274,7 +276,6 @@ public class PartsServiceImpl implements PartsService {
String peek = DynamicDataSourceContextHolder.peek();
for (List<PartsImportExcelVO> batch : batches) {
threadPoolTaskExecutor.submit(() -> {
threadPoolTaskExecutor.submit(() -> {
try {
// 子线程同步主线程的多数据源
@@ -285,11 +286,46 @@ public class PartsServiceImpl implements PartsService {
new LambdaQueryWrapper<PartsDO>().in(PartsDO::getGoodsId, goodsIds).eq(PartsDO::getOrganId, organId))
.stream().collect(Collectors.toMap(PartsDO::getGoodsId, p -> p));
// 整理到insertLis、updateList
doBatch(batch, partsMap, organId, isUpdateSupport, insertList, updateList);
} catch (Exception e) {
boolean b = hasError.get();
if (!b) {
hasError.set(true);
}
} finally {
latch.countDown(); // 任务完成,计数器减1
}
});
}
try {
// 等待所有任务查询完成
latch.await();
} catch (InterruptedException e) {
Thread.currentThread().interrupt();
throw new ServiceException(PARTS_IMPORT_INTERRUPT_ERROR);
}
return hasError.get();
}
/**
* 遍历批次列表,存在更新,不存在新增
*
* @param batch
* @param partsMap 已经存在goodsId的part map
* @param organId
* @param isUpdateSupport
* @param insertList
* @param updateList
*/
private void doBatch(List<PartsImportExcelVO> batch, Map<String, PartsDO> partsMap, Long organId,
boolean isUpdateSupport, List<PartsDO> insertList, List<PartsDO> updateList) {
for (PartsImportExcelVO importPart : batch) {
PartsDO existPlate = partsMap.get(importPart.getGoodsId());
if (existPlate == null) {
PartsDO partsDO = BeanUtils.toBean(importPart, PartsDO.class).setOrganId(organId);
partsDO.setIsComposite(ObjectUtil.isNull(partsDO.getIsComposite()) ? false : partsDO.getIsComposite())
partsDO.setIsComposite(getDefaultFalse(partsDO.getIsComposite()))
.setCategory(getEmpty(partsDO.getCategory()))
.setSubparts(getEmpty(partsDO.getType()))
.setColor(getEmpty(partsDO.getColor()))
@@ -314,26 +350,6 @@ public class PartsServiceImpl implements PartsService {
updateList.add(partsDO);
}
}
} catch (Exception e) {
boolean b = hasError.get();
if (!b) {
hasError.set(true);
}
} finally {
latch.countDown(); // 任务完成,计数器减1
}
});
});
}
try {
// 等待所有任务查询完成
latch.await();
} catch (InterruptedException e) {
Thread.currentThread().interrupt();
throw new ServiceException(PARTS_IMPORT_INTERRUPT_ERROR);
}
return hasError.get();
}
/**
@@ -394,6 +410,16 @@ public class PartsServiceImpl implements PartsService {
return ObjectUtil.defaultIfNull(value, 0.0);
}
/**
* value为空返回false
*
* @param value
* @return
*/
private Boolean getDefaultFalse(Boolean value) {
return ObjectUtil.defaultIfNull(value, false);
}
@Override
public List<PartsDO> getPartAll(String category) {
return partsMapper.getAllByOrganId(OrganContextHolder.getOrganId(), category);
@@ -208,7 +208,6 @@ public class PlateManageServiceImpl implements PlateManageService {
String peek = DynamicDataSourceContextHolder.peek();
for (List<PlateImportExcelVO> batch : batches) {
threadPoolTaskExecutor.submit(() -> {
threadPoolTaskExecutor.submit(() -> {
try {
// 子线程同步主线程的多数据源
@@ -219,25 +218,9 @@ public class PlateManageServiceImpl implements PlateManageService {
new LambdaQueryWrapper<PlateGoodDO>().in(PlateGoodDO::getGoodsId, goodsIds).eq(PlateGoodDO::getOrganId, organId))
.stream().collect(Collectors.toMap(PlateGoodDO::getGoodsId, p -> p));
for (PlateImportExcelVO importPart : batch) {
PlateGoodDO existPlate = plateGoodMap.get(importPart.getGoodsId());
if (existPlate == null) {
PlateGoodDO plateGoodDO = BeanUtils.toBean(importPart, PlateGoodDO.class).setOrganId(organId);
plateGoodDO.setColorBlack(EMPTY_STRING)
.setBrand(EMPTY_STRING)
.setSpec(getEmpty(plateGoodDO.getSpec()))
.setRemark(getEmpty(plateGoodDO.getRemark()))
.setTexture(ObjectUtil.isNull(plateGoodDO.getTexture()) ? false : plateGoodDO.getTexture())
.setOrganId(organId)
.setDeleted(false);
insertList.add(plateGoodDO);
} else if (!isUpdateSupport) {
PlateGoodDO plateGoodDO = BeanUtils.toBean(importPart, PlateGoodDO.class)
.setOrganId(organId)
.setId(existPlate.getId());
updateList.add(plateGoodDO);
}
}
// 整理到insertLis、updateList
doBatch(batch, plateGoodMap, organId, isUpdateSupport, insertList, updateList);
} catch (Exception e) {
boolean b = hasError.get();
if (!b) {
@@ -247,7 +230,6 @@ public class PlateManageServiceImpl implements PlateManageService {
latch.countDown(); // 任务完成,计数器减1
}
});
});
}
try {
@@ -260,6 +242,39 @@ public class PlateManageServiceImpl implements PlateManageService {
return hasError.get();
}
/**
* 遍历批次列表,存在更新,不存在新增
*
* @param batch
* @param plateGoodMap 已经存在goodsId的plate map
* @param organId
* @param isUpdateSupport
* @param insertList
* @param updateList
*/
private void doBatch(List<PlateImportExcelVO> batch, Map<String, PlateGoodDO> plateGoodMap, Long organId,
boolean isUpdateSupport, List<PlateGoodDO> insertList, List<PlateGoodDO> updateList) {
for (PlateImportExcelVO importPart : batch) {
PlateGoodDO existPlate = plateGoodMap.get(importPart.getGoodsId());
if (existPlate == null) {
PlateGoodDO plateGoodDO = BeanUtils.toBean(importPart, PlateGoodDO.class).setOrganId(organId);
plateGoodDO.setColorBlack(EMPTY_STRING)
.setBrand(EMPTY_STRING)
.setSpec(getEmpty(plateGoodDO.getSpec()))
.setRemark(getEmpty(plateGoodDO.getRemark()))
.setTexture(getDefaultFalse(plateGoodDO.getTexture()))
.setOrganId(organId)
.setDeleted(false);
insertList.add(plateGoodDO);
} else if (!isUpdateSupport) {
PlateGoodDO plateGoodDO = BeanUtils.toBean(importPart, PlateGoodDO.class)
.setOrganId(organId)
.setId(existPlate.getId());
updateList.add(plateGoodDO);
}
}
}
/**
* 每batchSize条数据处理一次DO
*
@@ -309,13 +324,13 @@ public class PlateManageServiceImpl implements PlateManageService {
}
/**
* value为空返回0.0
* value为空返回false
*
* @param value
* @return
*/
private double getZeroDouble(Double value) {
return ObjectUtil.defaultIfNull(value, 0.0);
private Boolean getDefaultFalse(Boolean value) {
return ObjectUtil.defaultIfNull(value, false);
}
@Override