-/**\r
- * ============LICENSE_START=======================================================\r
- * Model Loader\r
- * ================================================================================\r
- * Copyright © 2017 AT&T Intellectual Property.\r
- * Copyright © 2017 Amdocs\r
- * All rights reserved.\r
- * ================================================================================\r
- * Licensed under the Apache License, Version 2.0 (the "License");\r
- * you may not use this file except in compliance with the License.\r
- * You may obtain a copy of the License at\r
- * http://www.apache.org/licenses/LICENSE-2.0\r
- * Unless required by applicable law or agreed to in writing, software\r
- * distributed under the License is distributed on an "AS IS" BASIS,\r
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
- * See the License for the specific language governing permissions and\r
- * limitations under the License.\r
- * ============LICENSE_END=========================================================\r
- *\r
- * ECOMP and OpenECOMP are trademarks\r
- * and service marks of AT&T Intellectual Property.\r
- */\r
-package org.openecomp.modelloader.entity.model;\r
-\r
-import com.sun.jersey.api.client.ClientResponse;\r
-\r
-import org.openecomp.cl.api.Logger;\r
-import org.openecomp.cl.eelf.LoggerFactory;\r
-import org.openecomp.modelloader.config.ModelLoaderConfig;\r
-import org.openecomp.modelloader.entity.Artifact;\r
-import org.openecomp.modelloader.entity.ArtifactHandler;\r
-import org.openecomp.modelloader.entity.ArtifactType;\r
-import org.openecomp.modelloader.restclient.AaiRestClient;\r
-import org.openecomp.modelloader.service.ModelLoaderMsgs;\r
-\r
-import org.w3c.dom.Node;\r
-\r
-import java.io.StringWriter;\r
-import java.util.ArrayList;\r
-import java.util.List;\r
-\r
-import javax.ws.rs.core.Response;\r
-import javax.xml.transform.OutputKeys;\r
-import javax.xml.transform.Transformer;\r
-import javax.xml.transform.TransformerException;\r
-import javax.xml.transform.TransformerFactory;\r
-import javax.xml.transform.dom.DOMSource;\r
-import javax.xml.transform.stream.StreamResult;\r
-\r
-\r
-public class ModelArtifactHandler extends ArtifactHandler {\r
-\r
- private static final String AAI_MODEL_VER = "/model-vers/model-ver";\r
- private static Logger logger = LoggerFactory.getInstance().getLogger(ArtifactHandler.class.getName());\r
-\r
-\r
- public ModelArtifactHandler(ModelLoaderConfig config) {\r
- super(config);\r
- }\r
-\r
- @Override\r
- public boolean pushArtifacts(List<Artifact> artifacts, String distributionID) {\r
- ModelSorter modelSorter = new ModelSorter();\r
- List<Artifact> sortedModelArtifacts = modelSorter.sort(artifacts);\r
-\r
- // Push the ordered list of model artifacts to A&AI. If one fails, we need to roll back\r
- // the changes.\r
- List<ModelArtifact> completedModels = new ArrayList<ModelArtifact>();\r
- AaiRestClient aaiClient = new AaiRestClient(config);\r
-\r
- for (Artifact art : sortedModelArtifacts) {\r
- ModelArtifact model = (ModelArtifact)art;\r
-\r
- boolean version = model.isV9Artifact();\r
- //Non - V9 version for models\r
- if(version == false){\r
- ClientResponse getResponse = aaiClient.getResource(getModelVerURL(model), distributionID, AaiRestClient.MimeType.XML);\r
- if ( (getResponse == null) || (getResponse.getStatus() != Response.Status.OK.getStatusCode()) ) {\r
- logger.error(ModelLoaderMsgs.DISTRIBUTION_EVENT_ERROR, " Artifact format not valid for " + \r
- model.getType().toString() + "- model-invariant-id[model-id]: " + \r
- model.getModelInvariantId() + " and model-version-id[model-name-version-id]: "+ \r
- model.getModelVerId()+ " . Rolling back distribution.");\r
- return false;\r
- }\r
- else{\r
- completedModels.add(model);\r
- logger.info(ModelLoaderMsgs.DISTRIBUTION_EVENT, model.getType().toString() + \r
- " " + model.getModelInvariantId() + " successfully ingested.");\r
- }\r
- }\r
- else\r
- {\r
- ClientResponse getResponse = aaiClient.getResource(getURL(model), distributionID, AaiRestClient.MimeType.XML);\r
- if ( (getResponse == null) || (getResponse.getStatus() != Response.Status.OK.getStatusCode()) ) {\r
- // Only attempt the PUT if the model doesn't already exist\r
- ClientResponse putResponse = aaiClient.putResource(getURL(model), model.getPayload(), distributionID, AaiRestClient.MimeType.XML);\r
- if ( (putResponse != null) && (putResponse.getStatus() == Response.Status.CREATED.getStatusCode()) ) {\r
- completedModels.add(model);\r
- logger.info(ModelLoaderMsgs.DISTRIBUTION_EVENT, model.getType().toString() + \r
- " " + model.getModelInvariantId() + " successfully ingested.");\r
- }\r
- else {\r
- logger.error(ModelLoaderMsgs.DISTRIBUTION_EVENT_ERROR, "Ingestion failed for " + \r
- model.getType().toString() + " " + model.getModelInvariantId() + ". Rolling back distribution.");\r
-\r
- for (ModelArtifact modelToDelete : completedModels) {\r
- // Best effort to delete. Nothing we can do in the event this fails.\r
- aaiClient.getAndDeleteResource(getURL(modelToDelete), distributionID);\r
- }\r
-\r
- return false;\r
- }\r
- }\r
- else {\r
- logger.info(ModelLoaderMsgs.DISTRIBUTION_EVENT, model.getType().toString() + " " + model.getModelInvariantId() + \r
- " already exists. Skipping ingestion.");\r
- getResponse = aaiClient.getResource(getModelVerURL(model), distributionID, AaiRestClient.MimeType.XML);\r
- if ( (getResponse == null) || (getResponse.getStatus() != Response.Status.OK.getStatusCode()) ) {\r
- // Only attempt the PUT if the model-ver doesn't already exist\r
- ClientResponse putResponse = null;\r
-\r
- try {\r
- putResponse = aaiClient.putResource(getModelVerURL(model), nodeToString(model.getModelVer()), distributionID, AaiRestClient.MimeType.XML);\r
- } catch (TransformerException e) {\r
- // TODO Auto-generated catch block\r
- e.printStackTrace();\r
- }\r
- if ( (putResponse != null) && (putResponse.getStatus() == Response.Status.CREATED.getStatusCode()) ) {\r
- completedModels.add(model);\r
- logger.info(ModelLoaderMsgs.DISTRIBUTION_EVENT, model.getType().toString() + " " + \r
- model.getNameVersionId() + " successfully ingested.");\r
- }\r
- else {\r
- logger.error(ModelLoaderMsgs.DISTRIBUTION_EVENT_ERROR, "Ingestion failed for " + \r
- model.getType().toString() + " " + model.getNameVersionId() + ". Rolling back distribution.");\r
-\r
- for (ModelArtifact modelToDelete : completedModels) {\r
- // Best effort to delete. Nothing we can do in the event this fails.\r
- aaiClient.getAndDeleteResource(getModelVerURL(modelToDelete), distributionID);\r
- }\r
-\r
- return false;\r
- }\r
- }\r
- else {\r
- logger.info(ModelLoaderMsgs.DISTRIBUTION_EVENT, model.getType().toString() + " " + \r
- model.getModelInvariantId() + " already exists. Skipping ingestion.");\r
- }\r
- }\r
- }\r
- }\r
-\r
- return true;\r
- }\r
-\r
-\r
- private String nodeToString(Node node) throws TransformerException {\r
- StringWriter sw = new StringWriter();\r
- Transformer t = TransformerFactory.newInstance().newTransformer();\r
- t.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "yes");\r
- t.transform(new DOMSource(node), new StreamResult(sw));\r
- System.out.println(sw.toString());\r
- return sw.toString();\r
- }\r
-\r
- private String getURL(ModelArtifact model) {\r
- String baseURL = config.getAaiBaseUrl().trim();\r
- String subURL = null;\r
- if (model.getType().equals(ArtifactType.MODEL)) {\r
- subURL = config.getAaiModelUrl().trim();\r
- }\r
- else {\r
- subURL = config.getAaiNamedQueryUrl().trim();\r
- }\r
-\r
- if ( (!baseURL.endsWith("/")) && (!subURL.startsWith("/")) ) {\r
- baseURL = baseURL + "/";\r
- }\r
-\r
- if ( baseURL.endsWith("/") && subURL.startsWith("/") ) {\r
- baseURL = baseURL.substring(0, baseURL.length()-1);\r
- }\r
-\r
- if (!subURL.endsWith("/")) {\r
- subURL = subURL + "/";\r
- }\r
-\r
- String url = baseURL + subURL + model.getModelInvariantId();\r
- return url;\r
- }\r
-\r
- private String getModelVerURL(ModelArtifact model) {\r
- String baseURL = config.getAaiBaseUrl().trim();\r
- String subURL = null;\r
- if (model.getType().equals(ArtifactType.MODEL)) {\r
- subURL = config.getAaiModelUrl().trim() + model.getModelInvariantId() + AAI_MODEL_VER;\r
- }\r
- else {\r
- subURL = config.getAaiNamedQueryUrl().trim();\r
- }\r
-\r
- if ( (!baseURL.endsWith("/")) && (!subURL.startsWith("/")) ) {\r
- baseURL = baseURL + "/";\r
- }\r
-\r
- if ( baseURL.endsWith("/") && subURL.startsWith("/") ) {\r
- baseURL = baseURL.substring(0, baseURL.length()-1);\r
- }\r
-\r
- if (!subURL.endsWith("/")) {\r
- subURL = subURL + "/";\r
- }\r
-\r
- String url = baseURL + subURL + model.getModelVerId();\r
- System.out.println(url);\r
- return url;\r
- }\r
-\r
- // This method is used for the test REST interface to load models without an ASDC\r
- public void loadModelTest(byte[] payload) {\r
- List<Artifact> modelArtifacts = new ArrayList<Artifact>();\r
- ModelArtifactParser parser = new ModelArtifactParser();\r
- modelArtifacts.addAll(parser.parse(payload, "Test-Artifact"));\r
- ModelSorter modelSorter = new ModelSorter();\r
- List<Artifact> sortedModelArtifacts = modelSorter.sort(modelArtifacts);\r
- pushArtifacts(sortedModelArtifacts, "Test-Distribution");\r
- }\r
-}\r
+/**
+ * ============LICENSE_START=======================================================
+ * Model Loader
+ * ================================================================================
+ * Copyright © 2017 AT&T Intellectual Property.
+ * Copyright © 2017 Amdocs
+ * All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * http://www.apache.org/licenses/LICENSE-2.0
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ *
+ * ECOMP and OpenECOMP are trademarks
+ * and service marks of AT&T Intellectual Property.
+ */
+package org.openecomp.modelloader.entity.model;
+
+import org.openecomp.cl.api.Logger;
+import org.openecomp.cl.eelf.LoggerFactory;
+import org.openecomp.modelloader.config.ModelLoaderConfig;
+import org.openecomp.modelloader.entity.Artifact;
+import org.openecomp.modelloader.entity.ArtifactHandler;
+import org.openecomp.modelloader.restclient.AaiRestClient;
+import org.openecomp.modelloader.service.ModelLoaderMsgs;
+
+import java.util.ArrayList;
+import java.util.List;
+
+
+public class ModelArtifactHandler extends ArtifactHandler {
+
+ private static Logger logger = LoggerFactory.getInstance().getLogger(ModelArtifactHandler.class.getName());
+
+ public ModelArtifactHandler(ModelLoaderConfig config) {
+ super(config);
+ }
+
+ @Override
+ public boolean pushArtifacts(List<Artifact> artifacts, String distributionID) {
+ ModelSorter modelSorter = new ModelSorter();
+ List<Artifact> sortedModelArtifacts;
+ try {
+ sortedModelArtifacts = modelSorter.sort(artifacts);
+ }
+ catch (RuntimeException ex) {
+ logger.error(ModelLoaderMsgs.DISTRIBUTION_EVENT_ERROR, "Unable to resolve models: " + ex.getMessage());
+ return false;
+ }
+
+ // Push the ordered list of model artifacts to A&AI. If one fails, we need to roll back
+ // the changes.
+ List<AbstractModelArtifact> completedModels = new ArrayList<AbstractModelArtifact>();
+ AaiRestClient aaiClient = new AaiRestClient(config);
+
+ for (Artifact art : sortedModelArtifacts) {
+ AbstractModelArtifact model = (AbstractModelArtifact)art;
+ if (model.push(aaiClient, config, distributionID, completedModels) != true) {
+ for (AbstractModelArtifact modelToDelete : completedModels) {
+ modelToDelete.rollbackModel(aaiClient, config, distributionID);
+ }
+
+ return false;
+ }
+ }
+
+ return true;
+ }
+
+ // This method is used for the test REST interface to load models without an ASDC
+ public void loadModelTest(byte[] payload) {
+ List<Artifact> modelArtifacts = new ArrayList<Artifact>();
+ ModelArtifactParser parser = new ModelArtifactParser();
+ modelArtifacts.addAll(parser.parse(payload, "Test-Artifact"));
+ ModelSorter modelSorter = new ModelSorter();
+ List<Artifact> sortedModelArtifacts = modelSorter.sort(modelArtifacts);
+ pushArtifacts(sortedModelArtifacts, "Test-Distribution");
+ }
+}
Code Review / aai / model-loader.git / blobdiff