fix(jg):西安数据导入接口修改

amos-boot-data/amos-boot-data-openapi/src/main/java/com/yeejoin/amos/api/openapi/dto/XiAnEquipInfoExcelDto.java

@@ -987,4 +987,8 @@ public class XiAnEquipInfoExcelDto extends BaseDto {
     @ExcelProperty(value = "序号")
     private String xaSerial;
 
+    @ApiModelProperty(value = "资料是否齐全（0齐全， 1不齐全）-西安数据导入")
+    @ExcelIgnore
+    private Integer isCompleteXa;
+
 }
\ No newline at end of file

amos-boot-data/amos-boot-data-openapi/src/main/java/com/yeejoin/amos/api/openapi/service/XiAnDataDockServiceImpl.java

@@ -27,6 +27,7 @@ import java.util.*;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
+import java.util.stream.Stream;
 
 @Slf4j
 @Service
@@ -205,7 +206,7 @@ public class XiAnDataDockServiceImpl {
 
             // 通用字段检查 每一个sheet页都有的基本信息，设计信息，制造信息和使用信息
             this.commonFieldCheck(data, rowError);
-
+            data.setIsCompleteXa(this.checkIsComplete(data) ? 0 : 1);
             // 起重机械----技术参数 检查
             if (isQZJX) {
                 this.QZJXTechnicalParamsCheck(data, rowError);
@@ -259,6 +260,20 @@ public class XiAnDataDockServiceImpl {
     }
 
     /**
+     * 检查资料是否齐全，16个导入必填项
+     *
+     * @param data 导入项
+     */
+    private boolean checkIsComplete(XiAnEquipInfoExcelDto data) {
+        return Stream.of(data.getEquList(), data.getEquCategory(), data.getUseInnerCode(),
+                data.getProductName(), data.getEquType(), data.getUseUnitCode(),
+                data.getUseUnit(), data.getDesignUnitCreditCode(), data.getDesignUnitName(),
+                data.getDesignUseDate(), data.getDesignDate(), data.getProduceUnitCreditCode(),
+                data.getProduceUnitName(), data.getProduceLicenseNum(), data.getFactoryNum(),
+                data.getProduceDate()).noneMatch(StringUtils::isBlank);
+    }
+
+    /**
      * 通用字段检查 每一个sheet页都有的基本信息，设计信息，制造信息和使用信息
      *
      * @param data     源数据

amos-boot-system-tzs/amos-boot-module-common/amos-boot-module-common-api/src/main/java/com/yeejoin/amos/boot/module/common/api/dto/ESEquipmentCategoryDto.java

@@ -173,10 +173,15 @@ public class ESEquipmentCategoryDto {
     @Field(type = FieldType.Text)
     private String INFORMATION_SITUATION;
 
-
     /**
      * 工程装置id（工业管道使用）
      */
     @Field(type = FieldType.Keyword, name = "PROJECT_CONTRAPTION_ID")
     private String projectContraptionId;
+
+    /**
+     * 资料是否齐全（0齐全， 1不齐全）-西安数据导入
+     */
+    @Field(type = FieldType.Keyword, name = "IS_COMPLETE_XA")
+    private String IS_COMPLETE_XA;
 }

amos-boot-system-tzs/amos-boot-module-jg/amos-boot-module-jg-biz/src/main/java/com/yeejoin/amos/boot/module/jg/biz/service/impl/DataDockServiceImpl.java

@@ -60,6 +60,7 @@ import java.util.*;
 import java.util.concurrent.CompletableFuture;
 import java.util.regex.Pattern;
 import java.util.stream.Collectors;
+import java.util.stream.Stream;
 import static com.alibaba.fastjson.JSON.toJSONString;
 
 @Slf4j
@@ -120,7 +121,7 @@ public class DataDockServiceImpl {
         String product = RequestContext.getProduct();
         String token = RequestContext.getToken();
         CompletableFuture.allOf(
-                equLists.stream().map(equ -> CompletableFuture.runAsync(() -> {
+                equLists.parallelStream().map(equ -> CompletableFuture.runAsync(() -> {
                     RequestContext.setAppKey(appKey);
                     RequestContext.setProduct(product);
                     RequestContext.setToken(token);
@@ -135,6 +136,7 @@ public class DataDockServiceImpl {
             // transactionTemplate.execute(status -> {
             String record = UUID.randomUUID().toString();
             String equList = String.valueOf(equ.get("equList"));
+            String isCompleteXa = String.valueOf(equ.get("isCompleteXa"));
             // 压力管道保存 工程装置表信息  必须在saveUseInfo之前进行，需要提前生成工程装置id
             saveProjectContraption(equ, equList);
             // 保存到设备表
@@ -147,7 +149,7 @@ public class DataDockServiceImpl {
             // 保存技术参数
             saveTechParams(equ, record, equList);
             // 保存到ES
-            saveEquInfoToEs(record);
+            saveEquInfoToEs(record, isCompleteXa);
             // return null;
             // });
         } catch (Exception e) {
@@ -164,7 +166,7 @@ public class DataDockServiceImpl {
         idxBizJgSupervisionInfoService.lambdaUpdate().set(IdxBizJgSupervisionInfo::getOrgBranchCode, "50*74*160*12478").set(IdxBizJgSupervisionInfo::getOrgBranchName, "交口河镇市场监管所").in(IdxBizJgSupervisionInfo::getRecord, records).update();
         records.forEach(record -> {
             esEquipmentCategory.deleteById(record);
-            this.saveEquInfoToEs(record);
+            this.saveEquInfoToEs(record,"1");
         });
         return records.size();
     }
@@ -350,15 +352,21 @@ public class DataDockServiceImpl {
         supervisionInfo.setRecDate(new Date());
 
         // 获取符合条件的公司 Map
-        Map<String, String> countyMap;
+        Map<String, Map<String, String>> countyMap;
         try {
             countyMap = Optional.ofNullable(Privilege.companyClient.queryAgencyList("county").getResult())
                     .orElse(Collections.emptyList())
                     .stream()
                     .filter(v -> "1442687786065854466".equals(String.valueOf(v.getParentId())))
-                    .collect(Collectors.toMap(CompanyModel::getCompanyCode, CompanyModel::getCompanyName, (a, b) -> a));
+                    .collect(Collectors.toMap(
+                            CompanyModel::getCompanyCode,
+                            v -> Stream.of(new AbstractMap.SimpleEntry<>("companyName", v.getCompanyName()),
+                                            new AbstractMap.SimpleEntry<>("orgCode", v.getOrgCode()))
+                                    .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue)),
+                            (a, b) -> a
+                    ));
         } catch (Exception e) {
-            throw new RuntimeException(e);
+            countyMap = Collections.emptyMap();
         }
         String companyCode = "1".equals(equ.get("equCodeType"))
                 ? String.valueOf(equ.get("equCode"))
@@ -368,11 +376,14 @@ public class DataDockServiceImpl {
         if (companyCode.length() > 9) {
             companyCode = companyCode.substring(4, 10);
         }
-        // 直接获取公司名称
-        String companyName = countyMap.get(companyCode);
+        Map<String, String> companyInfo = countyMap.get(companyCode);
+        if (companyInfo != null) {
+            String companyName = companyInfo.get("companyName");
+            String orgCode = companyInfo.get("orgCode");
             if (companyName != null && !companyName.trim().isEmpty()) {
-            supervisionInfo.setOrgBranchCode(companyCode);
+                supervisionInfo.setOrgBranchCode(orgCode);
                 supervisionInfo.setOrgBranchName(companyName);
+            }
         } else {
             // 通过 useUnitCode 查询属地监管部门
             String useUnitCode = String.valueOf(equ.get("useUnitCode")).trim();
@@ -586,12 +597,13 @@ public class DataDockServiceImpl {
      *
      * @param record 设备唯一编码
      */
-    private void saveEquInfoToEs(String record) {
+    private void saveEquInfoToEs(String record, String isCompleteXa) {
         Map<String, Object> map = categoryOtherInfoMapper.selectDataById(record);
         ESEquipmentCategoryDto equipmentCategoryDto = JSON.parseObject(toJSONString(map), ESEquipmentCategoryDto.class);
         if (!ObjectUtils.isEmpty(equipmentCategoryDto)) {
             long time = Timestamp.valueOf(map.get("REC_DATE").toString().substring(0, 19)).getTime();
             equipmentCategoryDto.setREC_DATE(time);
+            equipmentCategoryDto.setIS_COMPLETE_XA(isCompleteXa);
             esEquipmentCategory.save(equipmentCategoryDto);
         }
     }

amos-boot-system-tzs/amos-boot-module-ymt/amos-boot-module-ymt-api/src/main/java/com/yeejoin/amos/boot/module/ymt/api/entity/IdxBizJgRegisterInfo.java

@@ -233,4 +233,10 @@ public class IdxBizJgRegisterInfo extends TzsBaseEntity {
      */
     @TableField("\"XA_SERIAL\"")
     private String xaSerial;
+
+    /**
+     * 资料是否齐全（0齐全， 1不齐全）-西安数据导入
+     */
+    @TableField("\"IS_COMPLETE_XA\"")
+    private String isCompleteXa;
 }