| @@ -226,7 +226,7 @@ public class WorkflowServiceImpl implements WorkflowService { | |||||
| } | } | ||||
| } | } | ||||
| public void saveAssetWorkFlow(Workflow workflow) throws IOException { | |||||
| public void saveAssetWorkFlow(Workflow workflow) throws Exception { | |||||
| if (StringUtils.isNotEmpty(workflow.getDag())) { | if (StringUtils.isNotEmpty(workflow.getDag())) { | ||||
| Map<String, Object> dagMap = JsonUtils.jsonToMap(workflow.getDag()); | Map<String, Object> dagMap = JsonUtils.jsonToMap(workflow.getDag()); | ||||
| List<Map<String, Object>> nodes = (List<Map<String, Object>>) dagMap.get("nodes"); | List<Map<String, Object>> nodes = (List<Map<String, Object>>) dagMap.get("nodes"); | ||||
| @@ -307,69 +307,101 @@ public class WorkflowServiceImpl implements WorkflowService { | |||||
| } | } | ||||
| } | } | ||||
| public void saveImageWorkFlow(String image, Long workflowId, String workflowName) { | |||||
| String[] imageSplit = image.split(":"); | |||||
| String tagName = imageSplit[1]; | |||||
| ImageVersion imageVersion = imageVersionDao.queryByUrl(image); | |||||
| insertAssetWorkFlow(workflowId, workflowName, Long.valueOf(imageVersion.getImageId()), tagName, Constant.Asset_Type_Image); | |||||
| } | |||||
| public void saveImageWorkFlow1(String inParameters, Long workflowId, String workflowName) throws IOException { | |||||
| Map<String, Object> paramMap = JsonUtils.jsonToMap(inParameters); | |||||
| Map<String, Object> image = (Map<String, Object>) paramMap.get("--image"); | |||||
| saveImageWorkFlow((String) image.get("value"), workflowId, workflowName); | |||||
| public void saveImageWorkFlow(String image, Long workflowId, String workflowName) throws Exception { | |||||
| try { | |||||
| String[] imageSplit = image.split(":"); | |||||
| String tagName = imageSplit[1]; | |||||
| ImageVersion imageVersion = imageVersionDao.queryByUrl(image); | |||||
| insertAssetWorkFlow(workflowId, workflowName, Long.valueOf(imageVersion.getImageId()), tagName, Constant.Asset_Type_Image); | |||||
| } catch (Exception e) { | |||||
| throw new Exception("镜像参数有误:" + e); | |||||
| } | |||||
| } | } | ||||
| public void saveCodeWorkFlow(String inParameters, Long workflowId, String workflowName) throws IOException { | |||||
| Map<String, Object> paramMap = JsonUtils.jsonToMap(inParameters); | |||||
| Map<String, Object> code_config = (Map<String, Object>) paramMap.get("--code_config"); | |||||
| String value = (String) code_config.get("value"); | |||||
| Map<String, Object> valueMap = JsonUtils.jsonToMap(value); | |||||
| insertAssetWorkFlow(workflowId, workflowName, Long.valueOf((Integer) valueMap.get("id")), null, Constant.Asset_Type_Code); | |||||
| public void saveImageWorkFlow1(String inParameters, Long workflowId, String workflowName) throws Exception { | |||||
| try { | |||||
| Map<String, Object> paramMap = JsonUtils.jsonToMap(inParameters); | |||||
| Map<String, Object> image = (Map<String, Object>) paramMap.get("--image"); | |||||
| saveImageWorkFlow((String) image.get("value"), workflowId, workflowName); | |||||
| } catch (Exception e) { | |||||
| throw new Exception(e); | |||||
| } | |||||
| } | } | ||||
| public void saveDatasetWorkFlow(String inParameters, Long workflowId, String workflowName) throws IOException { | |||||
| Map<String, Object> paramMap = JsonUtils.jsonToMap(inParameters); | |||||
| Map<String, Object> dataset = (Map<String, Object>) paramMap.get("--dataset"); | |||||
| String value = (String) dataset.get("value"); | |||||
| if (!value.startsWith("{{")) { | |||||
| public void saveCodeWorkFlow(String inParameters, Long workflowId, String workflowName) throws Exception { | |||||
| try { | |||||
| Map<String, Object> paramMap = JsonUtils.jsonToMap(inParameters); | |||||
| Map<String, Object> code_config = (Map<String, Object>) paramMap.get("--code_config"); | |||||
| String value = (String) code_config.get("value"); | |||||
| Map<String, Object> valueMap = JsonUtils.jsonToMap(value); | Map<String, Object> valueMap = JsonUtils.jsonToMap(value); | ||||
| insertAssetWorkFlow(workflowId, workflowName, Long.valueOf((String) valueMap.get("id")), (String) valueMap.get("version"), Constant.Asset_Type_Dataset); | |||||
| insertAssetWorkFlow(workflowId, workflowName, Long.valueOf((Integer) valueMap.get("id")), null, Constant.Asset_Type_Code); | |||||
| } catch (IOException e) { | |||||
| throw new Exception("代码配置参数有误:" + e); | |||||
| } | } | ||||
| } | } | ||||
| public void saveDatasetWorkFlow1(String inParameters, Long workflowId, String workflowName) throws IOException { | |||||
| Map<String, Object> paramMap = JsonUtils.jsonToMap(inParameters); | |||||
| Map<String, Object> dataset = (Map<String, Object>) paramMap.get("--dataset_id"); | |||||
| String value = (String) dataset.get("value"); | |||||
| Map<String, Object> valueMap = JsonUtils.jsonToMap(value); | |||||
| insertAssetWorkFlow(workflowId, workflowName, Long.valueOf((String) valueMap.get("id")), null, Constant.Asset_Type_Dataset); | |||||
| public void saveDatasetWorkFlow(String inParameters, Long workflowId, String workflowName) throws Exception { | |||||
| try { | |||||
| Map<String, Object> paramMap = JsonUtils.jsonToMap(inParameters); | |||||
| Map<String, Object> dataset = (Map<String, Object>) paramMap.get("--dataset"); | |||||
| String value = (String) dataset.get("value"); | |||||
| if (!value.startsWith("{{")) { | |||||
| Map<String, Object> valueMap = JsonUtils.jsonToMap(value); | |||||
| insertAssetWorkFlow(workflowId, workflowName, Long.valueOf((String) valueMap.get("id")), (String) valueMap.get("version"), Constant.Asset_Type_Dataset); | |||||
| } | |||||
| } catch (Exception e) { | |||||
| throw new Exception("数据集参数有误:" + e); | |||||
| } | |||||
| } | } | ||||
| public void saveModelWorkFlow(String inParameters, Long workflowId, String workflowName) throws IOException { | |||||
| Map<String, Object> paramMap = JsonUtils.jsonToMap(inParameters); | |||||
| Map<String, Object> model = (Map<String, Object>) paramMap.get("--model_name"); | |||||
| String value = (String) model.get("value"); | |||||
| if (StringUtils.isNotEmpty(value) && !value.startsWith("{{")) { | |||||
| public void saveDatasetWorkFlow1(String inParameters, Long workflowId, String workflowName) throws Exception { | |||||
| try { | |||||
| Map<String, Object> paramMap = JsonUtils.jsonToMap(inParameters); | |||||
| Map<String, Object> dataset = (Map<String, Object>) paramMap.get("--dataset_id"); | |||||
| String value = (String) dataset.get("value"); | |||||
| Map<String, Object> valueMap = JsonUtils.jsonToMap(value); | Map<String, Object> valueMap = JsonUtils.jsonToMap(value); | ||||
| insertAssetWorkFlow(workflowId, workflowName, Long.valueOf((String) valueMap.get("id")), (String) valueMap.get("version"), Constant.Asset_Type_Model); | |||||
| insertAssetWorkFlow(workflowId, workflowName, Long.valueOf((String) valueMap.get("id")), null, Constant.Asset_Type_Dataset); | |||||
| } catch (Exception e) { | |||||
| throw new Exception("数据集参数有误:" + e); | |||||
| } | |||||
| } | |||||
| public void saveModelWorkFlow(String inParameters, Long workflowId, String workflowName) throws Exception { | |||||
| try { | |||||
| Map<String, Object> paramMap = JsonUtils.jsonToMap(inParameters); | |||||
| Map<String, Object> model = (Map<String, Object>) paramMap.get("--model_name"); | |||||
| String value = (String) model.get("value"); | |||||
| if (StringUtils.isNotEmpty(value) && !value.startsWith("{{")) { | |||||
| Map<String, Object> valueMap = JsonUtils.jsonToMap(value); | |||||
| insertAssetWorkFlow(workflowId, workflowName, Long.valueOf((String) valueMap.get("id")), (String) valueMap.get("version"), Constant.Asset_Type_Model); | |||||
| } | |||||
| } catch (Exception e) { | |||||
| throw new Exception("模型参数有误:" + e); | |||||
| } | } | ||||
| } | } | ||||
| public void saveModelWorkFlow1(String inParameters, Long workflowId, String workflowName) throws IOException { | |||||
| Map<String, Object> paramMap = JsonUtils.jsonToMap(inParameters); | |||||
| Map<String, Object> model = (Map<String, Object>) paramMap.get("--model_id"); | |||||
| String value = (String) model.get("value"); | |||||
| Map<String, Object> valueMap = JsonUtils.jsonToMap(value); | |||||
| insertAssetWorkFlow(workflowId, workflowName, Long.valueOf((String) valueMap.get("id")), null, Constant.Asset_Type_Model); | |||||
| public void saveModelWorkFlow1(String inParameters, Long workflowId, String workflowName) throws Exception { | |||||
| try { | |||||
| Map<String, Object> paramMap = JsonUtils.jsonToMap(inParameters); | |||||
| Map<String, Object> model = (Map<String, Object>) paramMap.get("--model_id"); | |||||
| String value = (String) model.get("value"); | |||||
| Map<String, Object> valueMap = JsonUtils.jsonToMap(value); | |||||
| insertAssetWorkFlow(workflowId, workflowName, Long.valueOf((String) valueMap.get("id")), null, Constant.Asset_Type_Model); | |||||
| } catch (Exception e) { | |||||
| throw new Exception("模型参数有误:" + e); | |||||
| } | |||||
| } | } | ||||
| public void saveServiceWorkFlow(String inParameters, Long workflowId, String workflowName) throws IOException { | |||||
| Map<String, Object> paramMap = JsonUtils.jsonToMap(inParameters); | |||||
| Map<String, Object> service = (Map<String, Object>) paramMap.get("--service_name"); | |||||
| String value = (String) service.get("value"); | |||||
| Map<String, Object> valueMap = JsonUtils.jsonToMap(value); | |||||
| insertAssetWorkFlow(workflowId, workflowName, Long.valueOf((Integer) valueMap.get("id")), null, Constant.Asset_Type_Service); | |||||
| public void saveServiceWorkFlow(String inParameters, Long workflowId, String workflowName) throws Exception { | |||||
| try { | |||||
| Map<String, Object> paramMap = JsonUtils.jsonToMap(inParameters); | |||||
| Map<String, Object> service = (Map<String, Object>) paramMap.get("--service_name"); | |||||
| String value = (String) service.get("value"); | |||||
| Map<String, Object> valueMap = JsonUtils.jsonToMap(value); | |||||
| insertAssetWorkFlow(workflowId, workflowName, Long.valueOf((Integer) valueMap.get("id")), null, Constant.Asset_Type_Service); | |||||
| } catch (Exception e) { | |||||
| throw new Exception("服务参数有误:" + e); | |||||
| } | |||||
| } | } | ||||
| public void insertAssetWorkFlow(Long workflowId, String workflowName, Long assetId, String version, String type) { | public void insertAssetWorkFlow(Long workflowId, String workflowName, Long assetId, String version, String type) { | ||||