Return List<Artifact> in ArtifactDownloadManager
[aai/model-loader.git] / src / main / java / org / onap / aai / modelloader / service / ArtifactDeploymentManager.java
1 /**
2  * ============LICENSE_START=======================================================
3  * org.onap.aai
4  * ================================================================================
5  * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
6  * Copyright © 2017-2018 European Software Marketing Ltd.
7  * ================================================================================
8  * Licensed under the Apache License, Version 2.0 (the "License");
9  * you may not use this file except in compliance with the License.
10  * You may obtain a copy of the License at
11  *
12  *       http://www.apache.org/licenses/LICENSE-2.0
13  *
14  * Unless required by applicable law or agreed to in writing, software
15  * distributed under the License is distributed on an "AS IS" BASIS,
16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  * See the License for the specific language governing permissions and
18  * limitations under the License.
19  * ============LICENSE_END=========================================================
20  */
21 package org.onap.aai.modelloader.service;
22
23 import java.util.ArrayList;
24 import java.util.List;
25 import org.onap.aai.modelloader.entity.Artifact;
26 import org.onap.aai.modelloader.entity.catalog.VnfCatalogArtifactHandler;
27 import org.onap.aai.modelloader.entity.model.ModelArtifactHandler;
28 import org.onap.aai.modelloader.restclient.AaiRestClient;
29 import org.onap.sdc.api.notification.INotificationData;
30 import org.springframework.stereotype.Component;
31
32 /**
33  * This class is responsible for deploying model and catalog artifacts.
34  */
35 @Component
36 public class ArtifactDeploymentManager {
37
38     private final ModelArtifactHandler modelArtifactHandler;
39     private final VnfCatalogArtifactHandler vnfCatalogArtifactHandler;
40     private final AaiRestClient aaiClient;
41
42     public ArtifactDeploymentManager(ModelArtifactHandler modelArtifactHandler, VnfCatalogArtifactHandler vnfCatalogArtifactHandler, AaiRestClient aaiClient) {
43         this.modelArtifactHandler = modelArtifactHandler;
44         this.vnfCatalogArtifactHandler = vnfCatalogArtifactHandler;
45         this.aaiClient = aaiClient;
46     }
47
48     /**
49      * Deploys model and catalog artifacts to A&AI.
50      *
51      * @param data data about the notification that is being processed
52      * @param modelArtifacts collection of artifacts that represent yml files found in a TOSCA_CSAR file that have been
53      *        converted to XML and also those for model query specs
54      * @param catalogArtifacts collection of artifacts that represent vnf catalog files
55      * @return boolean <code>true</code> if all deployments were successful otherwise <code>false</code>
56      */
57     public boolean deploy(final INotificationData data, final List<Artifact> modelArtifacts,
58             final List<Artifact> catalogArtifacts) {
59
60         String distributionId = data.getDistributionID();
61
62         List<Artifact> completedArtifacts = new ArrayList<>();
63         boolean deploySuccess =
64                 modelArtifactHandler.pushArtifacts(modelArtifacts, distributionId, completedArtifacts, aaiClient);
65
66         if (!deploySuccess) {
67             modelArtifactHandler.rollback(completedArtifacts, distributionId, aaiClient);
68         } else {
69             List<Artifact> completedImageData = new ArrayList<>();
70             deploySuccess = vnfCatalogArtifactHandler.pushArtifacts(catalogArtifacts, distributionId,
71                     completedImageData, aaiClient);
72             if (!deploySuccess) {
73                 modelArtifactHandler.rollback(completedArtifacts, distributionId, aaiClient);
74                 vnfCatalogArtifactHandler.rollback(completedImageData, distributionId, aaiClient);
75             }
76         }
77
78         return deploySuccess;
79     }
80 }