3c1152ff0a31affbb9e7ffcc7f4b5297a0e5b66c
[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 implements IModelArtifact {
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     @Override
47     public void addDependentModelId(String dependentModelId) {
48         this.referencedModelIds.add(dependentModelId);
49     }
50
51     public String getModelNamespace() {
52         return modelNamespace;
53     }
54
55     @Override
56     public void setModelNamespace(String modelNamespace) {
57         this.modelNamespace = modelNamespace;
58
59         // Get the version from the namespace (in format 'http://org.openecomp.aai.inventory/v9')
60         String[] parts = modelNamespace.split("/");
61         modelNamespaceVersion = parts[parts.length - 1].trim();
62     }
63
64     public String getModelNamespaceVersion() {
65         return modelNamespaceVersion;
66     }
67
68     public abstract String getUniqueIdentifier();
69
70     public abstract boolean push(AaiRestClient aaiClient, ModelLoaderConfig config, String distId,
71             List<Artifact> completedArtifacts);
72
73     public abstract void rollbackModel(AaiRestClient aaiClient, ModelLoaderConfig config, String distId);
74
75     @Override
76     public String toString() {
77         StringBuilder sb = new StringBuilder();
78         sb.append("\nType=" + getType().toString() + "\nId=" + getUniqueIdentifier() + "\nVersion="
79                 + getModelNamespaceVersion() + "\nDependant models: ");
80         for (String dep : referencedModelIds) {
81             sb.append(dep + "  ");
82         }
83
84         return sb.toString();
85     }
86
87 }