feature: 白名单IP添加时增加对“执行方式”的判断，如仅开文件分发则在脚本执行中不可选中 TencentBlueKing#115

hLinx · Sep 16, 2021 · f2d473b · f2d473b
1 parent dc89450
commit f2d473b
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/...e-job-manage/src/main/java/com/tencent/bk/job/manage/service/impl/WhiteIPServiceImpl.java b/...e-job-manage/src/main/java/com/tencent/bk/job/manage/service/impl/WhiteIPServiceImpl.java
@@ -198,10 +198,7 @@ public List<CloudIPDTO> listWhiteIP(Long appId, ActionScopeEnum actionScope) {
         );
     }
 
-    @Override
-    public Long saveWhiteIP(String username, WhiteIPRecordCreateUpdateReq createUpdateReq) {
-        LOG.infoWithRequestId("Input(" + username + "," + createUpdateReq.toString() + ")");
-        //1.参数校验、预处理
+    private List<String> checkReqAndGetIpList(WhiteIPRecordCreateUpdateReq createUpdateReq) {
         String appIdStr = createUpdateReq.getAppIdStr();
         if (null == appIdStr || appIdStr.isEmpty()) {
             throw new InvalidParamException("appIdStr", "appIdStr cannot be null or empty");
@@ -235,6 +232,16 @@ public Long saveWhiteIP(String username, WhiteIPRecordCreateUpdateReq createUpda
                     throw new InvalidParamException("ipStr", "not a valid ip format");
             });
         }
+        return ipList;
+    }
+
+    @Override
+    public Long saveWhiteIP(String username, WhiteIPRecordCreateUpdateReq createUpdateReq) {
+        LOG.infoWithRequestId("Input(" + username + "," + createUpdateReq.toString() + ")");
+        //1.参数校验、预处理
+        List<String> ipList = checkReqAndGetIpList(createUpdateReq);
+        List<Long> appIdList = Arrays.stream(createUpdateReq.getAppIdStr().split(","))
+            .map(Long::parseLong).collect(Collectors.toList());
         List<Long> actionScopeIdList = createUpdateReq.getActionScopeIdList();
         val ipDtoList = ipList.stream().map(ip -> new WhiteIPIPDTO(
             null,