push addional code
[sdc.git] / openecomp-be / lib / openecomp-sdc-enrichment-lib / openecomp-sdc-enrichment-impl / src / main / java / org / openecomp / sdc / enrichment / impl / tosca / ToscaEnricher.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * SDC
4  * ================================================================================
5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  * 
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  * 
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.openecomp.sdc.enrichment.impl.tosca;
22
23 import org.openecomp.sdc.datatypes.error.ErrorMessage;
24 import org.openecomp.sdc.enrichment.inter.Enricher;
25 import org.openecomp.sdc.tosca.datatypes.ToscaServiceModel;
26 import org.openecomp.sdc.tosca.datatypes.model.NodeType;
27 import org.openecomp.sdc.tosca.services.ToscaUtil;
28
29 import java.util.HashMap;
30 import java.util.List;
31 import java.util.Map;
32
33 public class ToscaEnricher extends Enricher {
34
35   Map<String, List<NodeType>> componentTypNodeTypeMap;
36
37   @Override
38   public Map<String, List<ErrorMessage>> enrich() {
39     Map<String, List<ErrorMessage>> errors = new HashMap<>();
40     Map<String, List<ErrorMessage>> enrichResponse;
41     enrichResponse = enrichCilometer();
42     errors.putAll(enrichResponse);
43     enrichResponse = enrichSnmp();
44     errors.putAll(enrichResponse);
45
46
47     return errors;
48   }
49
50   private Map<String, List<ErrorMessage>> enrichCilometer() {
51     Map<String, List<ErrorMessage>> enrichResponse;
52
53     ToscaServiceModel toscaModel = (ToscaServiceModel) model;
54
55     componentTypNodeTypeMap =
56         ToscaUtil.normalizeComponentNameNodeType(toscaModel, input.getEntityInfo().keySet());
57
58     enrichResponse = CeilometerEnricher.enrich(toscaModel, componentTypNodeTypeMap, this.input);
59
60     return enrichResponse;
61   }
62
63
64   private Map<String, List<ErrorMessage>> enrichSnmp() {
65     Map<String, List<ErrorMessage>> enrichResponse;
66
67     enrichResponse = SnmpEnricher.enrich(componentTypNodeTypeMap, this.input);
68
69     return enrichResponse;
70   }
71
72
73 }