4a07ca3aad6a52ccc186de39cf346ec0b49d3ec2
[clamp.git] / src / main / java / org / onap / clamp / clds / client / req / tca / TcaRequestFormatter.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * ONAP CLAMP
4  * ================================================================================
5  * Copyright (C) 2017-2018 AT&T Intellectual Property. All rights
6  *                             reserved.
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  * 
22  */
23
24 package org.onap.clamp.clds.client.req.tca;
25
26 import com.att.eelf.configuration.EELFLogger;
27 import com.att.eelf.configuration.EELFManager;
28 import com.fasterxml.jackson.databind.JsonNode;
29 import com.fasterxml.jackson.databind.node.ArrayNode;
30 import com.fasterxml.jackson.databind.node.ObjectNode;
31
32 import java.io.IOException;
33 import java.util.Map;
34
35 import org.onap.clamp.clds.config.ClampProperties;
36 import org.onap.clamp.clds.exception.TcaRequestFormatterException;
37 import org.onap.clamp.clds.model.properties.ModelProperties;
38 import org.onap.clamp.clds.model.properties.Tca;
39 import org.onap.clamp.clds.model.properties.TcaItem;
40 import org.onap.clamp.clds.model.properties.TcaThreshold;
41 import org.yaml.snakeyaml.Yaml;
42
43 /**
44  * Construct the requests for TCA policy and SDC.
45  */
46 public class TcaRequestFormatter {
47
48     protected static final EELFLogger logger = EELFManager.getInstance().getLogger(TcaRequestFormatter.class);
49     protected static final EELFLogger metricsLogger = EELFManager.getInstance().getMetricsLogger();
50
51     /**
52      * Hide the default constructor.
53      */
54     private TcaRequestFormatter() {
55     }
56
57     /**
58      * Format Tca Policy JSON request.
59      *
60      * @param refProp
61      *            The refProp generally created by Spring, it's an access on the
62      *            clds-references.properties file
63      * @param modelProperties
64      *            The Model Prop created from BPMN JSON and BPMN properties JSON
65      * @return The Json string containing that should be sent to policy
66      */
67     public static String createPolicyJson(ClampProperties refProp, ModelProperties modelProperties) {
68         try {
69             String service = modelProperties.getGlobal().getService();
70             Tca tca = modelProperties.getType(Tca.class);
71             modelProperties.setCurrentModelElementId(tca.getId());
72             ObjectNode rootNode = (ObjectNode) refProp.getJsonTemplate("tca.policy.template", service);
73             String policyName = modelProperties.getCurrentPolicyScopeAndPolicyName();
74             rootNode.put("policyName", policyName);
75             ((ObjectNode) rootNode.get("content")).replace("tca_policy",
76                     createPolicyContent(refProp, modelProperties, service, policyName, tca));
77             String tcaPolicyReq = rootNode.toString();
78             logger.info("tcaPolicyReq=" + tcaPolicyReq);
79             return tcaPolicyReq;
80         } catch (IOException e) {
81             throw new TcaRequestFormatterException("Exception caught when attempting to create the policy JSON", e);
82         }
83     }
84
85     /**
86      * Format Tca Policy Content JSON
87      *
88      * @param refProp
89      *            The refProp generally created by Spring, it's an access on the
90      *            clds-references.properties file
91      * @param modelProperties
92      *            The Model Prop created from BPMN JSON and BPMN properties JSON
93      * @param service
94      *            The service ID, if not specified getGlobal.getService will be
95      *            used
96      * @param policyName
97      *            The policyName, if not specified the
98      *            modelProperties.getCurrentPolicyScopeAndPolicyName will be
99      *            used
100      * @param tca
101      *            The Tca object, if not specified the
102      *            modelProperties.setCurrentModelElementId will be used
103      * @return The Json node containing what should be sent to policy
104      */
105     public static JsonNode createPolicyContent(ClampProperties refProp, ModelProperties modelProperties, String service,
106             String policyName, Tca tca) {
107         try {
108             String serviceToUse = service;
109             String policyNameToUse = policyName;
110             Tca tcaToUse = tca;
111             if (serviceToUse == null) {
112                 serviceToUse = modelProperties.getGlobal().getService();
113             }
114             if (tcaToUse == null) {
115                 tcaToUse = modelProperties.getType(Tca.class);
116                 modelProperties.setCurrentModelElementId(tcaToUse.getId());
117             }
118             if (policyNameToUse == null) {
119                 policyNameToUse = modelProperties.getCurrentPolicyScopeAndPolicyName();
120             }
121             ObjectNode rootNode = (ObjectNode) refProp.getJsonTemplate("tca.template", serviceToUse);
122             ((ObjectNode) rootNode.get("metricsPerEventName").get(0)).put("eventName",
123                     tcaToUse.getTcaItem().getEventName());
124             ((ObjectNode) rootNode.get("metricsPerEventName").get(0)).put("policyName", policyNameToUse);
125             ((ObjectNode) rootNode.get("metricsPerEventName").get(0)).put("controlLoopSchemaType",
126                     tcaToUse.getTcaItem().getControlLoopSchemaType());
127             ObjectNode thresholdsParent = ((ObjectNode) rootNode.get("metricsPerEventName").get(0));
128             addThresholds(refProp, serviceToUse, thresholdsParent, tcaToUse.getTcaItem(), modelProperties);
129             logger.info("tcaPolicyContent=" + rootNode.toString());
130             return rootNode;
131         } catch (IOException e) {
132             throw new TcaRequestFormatterException("Exception caught when attempting to create the policy content JSON",
133                     e);
134         }
135     }
136
137     /**
138      * Add threshold values to the existing policy JSON.
139      *
140      * @param refProp
141      *            The refProp generally created by Spring, it's an access on the
142      *            clds-references.properties file
143      * @param service
144      *            The Service value extracted from Global section of the Bpmn
145      *            Properties JSON
146      * @param appendToNode
147      *            The JSON structure from where the thresholds section must be
148      *            added
149      * @param tcaItem
150      *            The TCA item contained in the Tca object
151      * @param modelProperties
152      *            The Model Properties created from BPMN JSON and BPMN
153      *            properties JSON
154      */
155     private static void addThresholds(ClampProperties refProp, String service, ObjectNode appendToNode, TcaItem tcaItem,
156             ModelProperties modelProperties) {
157         ArrayNode tcaNodes = appendToNode.withArray("thresholds");
158         ObjectNode tcaNode;
159         try {
160             tcaNode = (ObjectNode) refProp.getJsonTemplate("tca.thresholds.template", service);
161             for (TcaThreshold tcaThreshold : tcaItem.getTcaThresholds()) {
162                 tcaNode.put("closedLoopControlName", modelProperties.getControlNameAndPolicyUniqueId());
163                 tcaNode.put("fieldPath", tcaThreshold.getFieldPath());
164                 tcaNode.put("thresholdValue", tcaThreshold.getThreshold());
165                 tcaNode.put("direction", tcaThreshold.getOperator());
166                 tcaNode.put("closedLoopEventStatus", tcaThreshold.getClosedLoopEventStatus());
167                 tcaNodes.add(tcaNode);
168             }
169         } catch (IOException e) {
170             throw new TcaRequestFormatterException("Exception caught when attempting to create the thresholds JSON", e);
171         }
172     }
173
174     /**
175      * This method updates the blueprint that is received in the UI with the TCA
176      * Json.
177      * 
178      * @param refProp
179      *            * The refProp generally created by Spring, it's an access on
180      *            the clds-references.properties file
181      * @param modelProperties
182      *            The Model Prop created from BPMN JSON and BPMN properties JSON
183      * @param yamlValue
184      *            The yaml string received from the UI
185      * @return The updated YAML as a string
186      */
187     public static String updatedBlueprintWithConfiguration(ClampProperties refProp, ModelProperties modelProperties,
188             String yamlValue) {
189         String jsonPolicy = ((ObjectNode) createPolicyContent(refProp, modelProperties, null, null, null)).toString();
190         logger.info("Yaml that will be updated:" + yamlValue);
191         Yaml yaml = new Yaml();
192         Map<String, Object> loadedYaml = (Map<String, Object>) yaml.load(yamlValue);
193         Map<String, Object> nodeTemplates = (Map<String, Object>) loadedYaml.get("node_templates");
194         Map<String, Object> tcaObject = (Map<String, Object>) nodeTemplates.get("tca_tca");
195         Map<String, Object> propsObject = (Map<String, Object>) tcaObject.get("properties");
196         Map<String, Object> appPreferences = (Map<String, Object>) propsObject.get("app_preferences");
197         appPreferences.put("tca_policy", jsonPolicy);
198         String blueprint = yaml.dump(loadedYaml);
199         logger.info("Yaml updated:" + blueprint);
200         return blueprint;
201     }
202 }