添加重复校验

amos-boot-system-tzs/amos-boot-module-ymt/amos-boot-module-ymt-biz/src/main/java/com/yeejoin/amos/boot/module/ymt/biz/controller/EquipmentCategoryController.java

@@ -12,14 +12,20 @@ import com.yeejoin.amos.boot.biz.common.utils.TreeParser;
 import com.yeejoin.amos.boot.module.ymt.api.dto.EquipExportDto;
 import com.yeejoin.amos.boot.module.ymt.api.dto.EquipmentCategoryDto;
 import com.yeejoin.amos.boot.module.ymt.api.entity.EquipmentCategory;
+import com.yeejoin.amos.boot.module.ymt.api.entity.IdxBizJgOtherInfo;
+import com.yeejoin.amos.boot.module.ymt.api.entity.TzsJgOtherInfo;
 import com.yeejoin.amos.boot.module.ymt.api.enums.ImageSizeEnums;
 import com.yeejoin.amos.boot.module.ymt.api.mapper.EquipmentCategoryMapper;
+import com.yeejoin.amos.boot.module.ymt.api.mapper.IdxBizJgOtherInfoMapper;
 import com.yeejoin.amos.boot.module.ymt.api.service.IEquipmentCategoryService;
+import com.yeejoin.amos.boot.module.ymt.api.service.ITzsJgOtherInfoService;
 import com.yeejoin.amos.boot.module.ymt.api.vo.EquipExportVo;
 import com.yeejoin.amos.boot.module.ymt.biz.service.impl.EquipmentCategoryServiceImpl;
 import com.yeejoin.amos.boot.module.ymt.biz.utils.ImageUtils;
+import com.yeejoin.amos.component.robot.BadRequest;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
+import org.checkerframework.checker.units.qual.A;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.http.HttpStatus;
@@ -63,6 +69,9 @@ public class EquipmentCategoryController extends BaseController {
 	String REGULATORY_CODE_PREFIX ;
 
 
+	@Autowired
+	IdxBizJgOtherInfoMapper idxBizJgOtherInfoMapper;
+
 	/**
 	 * 新增装备分类
 	 *
@@ -384,6 +393,19 @@ public class EquipmentCategoryController extends BaseController {
 		// 获取resources目录下的文件输入流
 		InputStream bgImgFile = getClass().getClassLoader().getResourceAsStream("temp/" + imageSizeEnums.getBgPath());
 		File QrCodeFile;
+
+		LambdaQueryWrapper<IdxBizJgOtherInfo> lambda = new QueryWrapper<IdxBizJgOtherInfo>().lambda();
+		lambda.eq(IdxBizJgOtherInfo::getClaimStatus, "已认领");
+		if (ImageSizeEnums.DT.getCode().equals(type)) {
+			lambda.eq(IdxBizJgOtherInfo::getCode96333, elevatorCode);
+		} else {
+			lambda.eq(IdxBizJgOtherInfo::getSupervisoryCode, code);
+		}
+		Integer integer = idxBizJgOtherInfoMapper.selectCount(lambda);
+		if (integer > 1) {
+			throw new BadRequest("存在重复数据，请联系管理员！");
+		}
+
 		if (ImageSizeEnums.DT.getCode().equals(type)) {
 			QrCodeFile = File.createTempFile(elevatorCode, ".png");
 			ImageUtils.creatQRCode(QrCodeFile, bgImgFile, imageSizeEnums.getWith(), imageSizeEnums.getHeight(), REGULATORY_CODE_PREFIX + code, "", elevatorCode, imageSizeEnums.getSize(), imageSizeEnums.getImagesX(), imageSizeEnums.getImagesY(), imageSizeEnums.getText1X(), imageSizeEnums.getText1Y(), imageSizeEnums.getText2X(), imageSizeEnums.getText2Y());
@@ -424,6 +446,17 @@ public class EquipmentCategoryController extends BaseController {
 			ImageUtils.downloadResourceZip("设备码信息.zip", zipFile.getAbsolutePath(), response);
 			zipFile.delete();
 		}
+		LambdaQueryWrapper<IdxBizJgOtherInfo> lambda = new QueryWrapper<IdxBizJgOtherInfo>().lambda();
+		lambda.eq(IdxBizJgOtherInfo::getClaimStatus, "已认领");
+		if (ImageSizeEnums.DT.getCode().equals(type)) {
+			lambda.in(IdxBizJgOtherInfo::getCode96333, equipExportData.stream().map(item -> item.get("CODE96333")).collect(Collectors.toList()));
+		} else {
+			lambda.in(IdxBizJgOtherInfo::getSupervisoryCode, equipExportData.stream().map(item -> item.get("SUPERVISORY_CODE")).collect(Collectors.toList()));
+		}
+		Integer integer = idxBizJgOtherInfoMapper.selectCount(lambda);
+		if (!map.get("size").equals(String.valueOf(integer))) {
+			throw new BadRequest("存在重复数据，请联系管理员！");
+		}
 		if (ImageSizeEnums.DT.getCode().equals(type)) {
 			for (JSONObject equipExportDatum : equipExportData.stream().filter(item -> !ObjectUtils.isEmpty(item.get("CODE96333"))).collect(Collectors.toList())) {
 				InputStream bgImgFile = getClass().getClassLoader().getResourceAsStream("temp/" + imageSizeEnums.getBgPath());