refactor(jg): 优化设备数据源类型处理逻辑

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

@@ -113,6 +113,7 @@ public class BaseEnterpriseVideoServiceImpl extends BaseService<BaseEnterpriseVi
                 if (result.get("data") instanceof JSONObject) {
                     JSONObject dataObj = (JSONObject) result.get("data");
                     String cameraUrl = dataObj.getString("url");
+                    video.setType(param.getString("protocol"));
                     video.setUrl(cameraUrl);
                 }
             } catch (URISyntaxException e) {

amos-boot-system-tzs/amos-boot-module-jg/amos-boot-module-jg-api/src/main/java/com/yeejoin/amos/boot/module/jg/api/enums/EquipSourceEnum.java

@@ -10,14 +10,16 @@ import java.util.Map;
 @Getter
 @AllArgsConstructor
 public enum EquipSourceEnum {
-    jg_his("历史设备", "jg_his"),
-    jg_his_black("历史设备(无登记证)", "jg_his_black"),
-    jg("新设备", "jg");
+    jg_his("历史设备", "jg_his", "his"),
+    jg_his_black("历史设备(无登记证)", "jg_his_black", "black"),
+    jg("新设备", "jg", "new");
 
     private final String name;
 
     private final String code;
 
+    private final String type;
+
 
     public static String getMessage(String code) {
         for (EquipSourceEnum constants : values()) {
@@ -30,6 +32,7 @@ public enum EquipSourceEnum {
 
     public static Map<Integer,String> getName=new HashMap<>();
     public static Map<String,Integer> getCode=new HashMap<>();
+    public static Map<String,Integer> getType=new HashMap<>();
 
     static {
         for (ConstructionEnum e : ConstructionEnum.values()){
@@ -39,14 +42,6 @@ public enum EquipSourceEnum {
 
     }
 
-    public String getCode() {
-        return code;
-    }
-
-    public String getName() {
-        return name;
-    }
-
     public static String getDataSourceName(String dataSource) {
         if(StringUtils.isEmpty(dataSource)){
             return EquipSourceEnum.jg_his.getName();

amos-boot-system-tzs/amos-boot-module-jg/amos-boot-module-jg-biz/src/main/java/com/yeejoin/amos/boot/module/jg/biz/controller/CommonController.java

@@ -14,7 +14,6 @@ import com.yeejoin.amos.boot.module.jg.api.dto.EquipmentClassifyDto;
 import com.yeejoin.amos.boot.module.jg.api.dto.UseFlagParamDto;
 import com.yeejoin.amos.boot.module.jg.api.entity.JgRegistrationHistory;
 import com.yeejoin.amos.boot.module.jg.api.entity.JgUseRegistrationManage;
-import com.yeejoin.amos.boot.module.jg.api.enums.CompanyTypeEnum;
 import com.yeejoin.amos.boot.module.jg.api.mapper.JgUseRegistrationMapper;
 import com.yeejoin.amos.boot.module.jg.api.service.IJgCertificateReplenishService;
 import com.yeejoin.amos.boot.module.jg.api.service.IJgInstallationNoticeService;
@@ -335,8 +334,8 @@ public class CommonController extends BaseController {
     @TycloudOperation(ApiLevel = UserType.AGENCY)
     @ApiOperation(httpMethod = "GET", value = "查询监管业务名称列表", notes = "查询监管业务名称列表")
     @GetMapping(value = "/superviseBusinessCategory")
-    public ResponseModel<Object> equipTree(@RequestParam(value = "businessType") String businessType) {
-        return ResponseHelper.buildResponse(commonService.superviseBusinessCategory(businessType));
+    public ResponseModel<Object> equipTree(@RequestParam(value = "businessType") String businessType, String dataSource) {
+        return ResponseHelper.buildResponse(commonService.superviseBusinessCategory(businessType, dataSource));
     }
 
     /**

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

@@ -64,7 +64,7 @@ public interface ICommonService {
 
     List<LinkedHashMap> creatApproveTree();
 
-    List<Map<String, Object>> superviseBusinessCategory(String businessType);
+    List<Map<String, Object>> superviseBusinessCategory(String businessType, String dataSource);
 
     List<Map<String, Object>> businessScenarios(String type);
 

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/CommonServiceImpl.java

@@ -733,7 +733,7 @@ public class CommonServiceImpl implements ICommonService {
     }
 
     @Override
-    public List<Map<String, Object>> superviseBusinessCategory(String businessType) {
+    public List<Map<String, Object>> superviseBusinessCategory(String businessType, String dataSource) {
         ReginParams reginParams = getSelectedOrgInfo();
         Map<String, List<Map<String, Object>>> resourceJson = JsonUtils.getResourceJson(superviseBusinessCategory);
         if (XZSB.equals(businessType) && CompanyTypeEnum.CONSTRUCTION.getName().equals(reginParams.getCompany().getCompanyType())) {
@@ -746,6 +746,11 @@ public class CommonServiceImpl implements ICommonService {
             }
             return resourceJson.get(businessType);
         }
+        // 历史有证设备(气瓶、管道)只能通过导入进系统，不能按单位录入进系统
+        if (EquipSourceEnum.jg_his.getType().equals(dataSource)) {
+            resourceJson.get(businessType).removeIf(item -> item.get(CODE).equals(DW_LR));
+            return resourceJson.get(businessType);
+        }
         return resourceJson.get(businessType);
     }