889db552fccc6e77940110f9f811e10248eea95b
[aai/model-loader.git] / src / main / java / org / onap / aai / modelloader / entity / model / AbstractModelArtifact.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 Amdocs
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.entity.model;
22
23 import java.util.HashSet;
24 import java.util.List;
25 import java.util.Set;
26
27 import org.onap.aai.modelloader.config.ModelLoaderConfig;
28 import org.onap.aai.modelloader.entity.Artifact;
29 import org.onap.aai.modelloader.entity.ArtifactType;
30 import org.onap.aai.modelloader.restclient.AaiRestClient;
31
32 public abstract class AbstractModelArtifact extends Artifact {
33                  
34   private String modelNamespace;
35   private String modelNamespaceVersion;
36         private Set<String> referencedModelIds = new HashSet<>(); 
37
38         public AbstractModelArtifact(ArtifactType type) {
39           super(type);
40         }
41         
42         public Set<String> getDependentModelIds() {
43                 return referencedModelIds;
44         }
45         
46         public void addDependentModelId(String dependentModelId) {
47                 this.referencedModelIds.add(dependentModelId);
48         }
49         
50   public String getModelNamespace() {
51     return modelNamespace;
52   }
53   
54   public void setModelNamespace(String modelNamespace) {
55     this.modelNamespace = modelNamespace;
56     
57     // Get the version from the namespace (in format 'http://org.openecomp.aai.inventory/v9')
58     String[] parts = modelNamespace.split("/");
59     modelNamespaceVersion = parts[parts.length-1].trim();
60   }
61   
62   public String getModelNamespaceVersion() {
63     return modelNamespaceVersion;
64   }
65   
66   public abstract String getUniqueIdentifier();
67   
68   public abstract boolean push(AaiRestClient aaiClient, ModelLoaderConfig config, String distId, List<AbstractModelArtifact> addedModels);
69   
70   public abstract void rollbackModel(AaiRestClient aaiClient, ModelLoaderConfig config, String distId);
71     @Override
72         public String toString() {
73                 StringBuilder sb = new StringBuilder();
74                 sb.append("\nType=" + getType().toString() +"\nId=" + getUniqueIdentifier() +"\nVersion=" + getModelNamespaceVersion() + "\nDependant models: ");
75                 for (String dep : referencedModelIds) {
76                         sb.append(dep + "  ");
77                 }
78                 
79                 return sb.toString();
80         }
81         
82         
83 }