Sync Integ to Master
[sdc.git] / catalog-be / src / main / java / org / openecomp / sdc / be / components / impl / CapabilityTypeImportManager.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.be.components.impl;
22
23 import fj.data.Either;
24 import org.openecomp.sdc.be.components.impl.ImportUtils.ToscaTagNamesEnum;
25 import org.openecomp.sdc.be.config.BeEcompErrorManager;
26 import org.openecomp.sdc.be.dao.api.ActionStatus;
27 import org.openecomp.sdc.be.impl.ComponentsUtils;
28 import org.openecomp.sdc.be.model.CapabilityTypeDefinition;
29 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
30 import org.openecomp.sdc.be.model.operations.impl.CapabilityTypeOperation;
31 import org.openecomp.sdc.common.config.EcompErrorName;
32 import org.openecomp.sdc.exception.ResponseFormat;
33 import org.slf4j.Logger;
34 import org.slf4j.LoggerFactory;
35 import org.springframework.stereotype.Component;
36
37 import javax.annotation.Resource;
38 import java.util.ArrayList;
39 import java.util.Iterator;
40 import java.util.List;
41 import java.util.Map;
42 import java.util.function.Consumer;
43
44 @Component("capabilityTypeImportManager")
45 public class CapabilityTypeImportManager {
46
47     private static final Logger log = LoggerFactory.getLogger(CapabilityTypeImportManager.class);
48     @Resource
49     private CapabilityTypeOperation capabilityTypeOperation;
50     @Resource
51     private ComponentsUtils componentsUtils;
52     @Resource
53     private CommonImportManager commonImportManager;
54
55     public Either<List<CapabilityTypeDefinition>, ResponseFormat> createCapabilityTypes(String capabilityYml) {
56         Either<List<CapabilityTypeDefinition>, ActionStatus> capabilityTypes = createCapabilityTypesFromYml(capabilityYml);
57         if (capabilityTypes.isRight()) {
58             ActionStatus status = capabilityTypes.right().value();
59             ResponseFormat responseFormat = componentsUtils.getResponseFormatByCapabilityType(status, null);
60             return Either.right(responseFormat);
61         }
62         return createCapabilityTypesByDao(capabilityTypes.left().value());
63
64     }
65
66     private Either<List<CapabilityTypeDefinition>, ActionStatus> createCapabilityTypesFromYml(String capabilityYml) {
67         return commonImportManager.createElementTypesFromYml(capabilityYml, (capTypeName, capTypeJsonData) -> createCapabilityType(capTypeName, capTypeJsonData));
68
69     }
70
71     private Either<List<CapabilityTypeDefinition>, ResponseFormat> createCapabilityTypesByDao(List<CapabilityTypeDefinition> capabilityTypesToCreate) {
72         List<CapabilityTypeDefinition> createdCapabilities = new ArrayList<>();
73         Either<List<CapabilityTypeDefinition>, ResponseFormat> eitherResult = Either.left(createdCapabilities);
74         Iterator<CapabilityTypeDefinition> capTypeItr = capabilityTypesToCreate.iterator();
75         boolean stopDao = false;
76         while (capTypeItr.hasNext() && !stopDao) {
77             CapabilityTypeDefinition capabilityType = capTypeItr.next();
78
79             log.info("send capabilityType {} to dao for create", capabilityType.getType());
80             Either<CapabilityTypeDefinition, StorageOperationStatus> dataModelResponse = capabilityTypeOperation.addCapabilityType(capabilityType);
81             if (dataModelResponse.isRight()) {
82                 BeEcompErrorManager.getInstance().logBeFailedAddingNodeTypeError("Create CapabilityTypes", "capability type");
83                 log.debug("failed to create capabilityType: {}", capabilityType.getType());
84                 if (dataModelResponse.right().value() != StorageOperationStatus.SCHEMA_VIOLATION) {
85                     ResponseFormat responseFormat = componentsUtils.getResponseFormatByCapabilityType(componentsUtils.convertFromStorageResponseForCapabilityType(dataModelResponse.right().value()), capabilityType);
86                     eitherResult = Either.right(responseFormat);
87                     stopDao = true;
88                 }
89
90             } else {
91                 createdCapabilities.add(capabilityType);
92             }
93             if (!capTypeItr.hasNext()) {
94                 log.info("capabilityTypes were created successfully!!!");
95             }
96
97         }
98
99         return eitherResult;
100
101     }
102
103     private CapabilityTypeDefinition createCapabilityType(String capabilityTypeName, Map<String, Object> toscaJson) {
104         CapabilityTypeDefinition capabilityType = new CapabilityTypeDefinition();
105
106         capabilityType.setType(capabilityTypeName);
107
108         // Description
109         final Consumer<String> descriptionSetter = description -> capabilityType.setDescription(description);
110         commonImportManager.setField(toscaJson, ToscaTagNamesEnum.DESCRIPTION.getElementName(), descriptionSetter);
111         // Derived From
112         final Consumer<String> derivedFromSetter = derivedFrom -> capabilityType.setDerivedFrom(derivedFrom);
113         commonImportManager.setField(toscaJson, ToscaTagNamesEnum.DERIVED_FROM.getElementName(), derivedFromSetter);
114         // Properties
115         commonImportManager.setPropertiesMap(toscaJson, (values) -> capabilityType.setProperties(values));
116
117         return capabilityType;
118     }
119
120 }