new method for policy client
[clamp.git] / src / main / java / org / onap / clamp / clds / client / PolicyClient.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * ONAP CLAMP
4  * ================================================================================
5  * Copyright (C) 2017 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  * ECOMP is a trademark and service mark of AT&T Intellectual Property.
22  */
23
24 package org.onap.clamp.clds.client;
25
26 import com.att.eelf.configuration.EELFLogger;
27 import com.att.eelf.configuration.EELFManager;
28
29 import java.io.IOException;
30 import java.util.ArrayList;
31 import java.util.Collection;
32 import java.util.Collections;
33 import java.util.Date;
34 import java.util.List;
35 import java.util.Map;
36 import java.util.UUID;
37
38 import javax.ws.rs.BadRequestException;
39
40 import org.onap.clamp.clds.exception.policy.PolicyClientException;
41 import org.onap.clamp.clds.model.prop.ModelProperties;
42 import org.onap.clamp.clds.model.refprop.RefProp;
43 import org.onap.clamp.clds.util.LoggingUtils;
44 import org.onap.policy.api.AttributeType;
45 import org.onap.policy.api.ConfigRequestParameters;
46 import org.onap.policy.api.DeletePolicyCondition;
47 import org.onap.policy.api.DeletePolicyParameters;
48 import org.onap.policy.api.PolicyChangeResponse;
49 import org.onap.policy.api.PolicyConfig;
50 import org.onap.policy.api.PolicyConfigException;
51 import org.onap.policy.api.PolicyConfigType;
52 import org.onap.policy.api.PolicyEngine;
53 import org.onap.policy.api.PolicyEngineException;
54 import org.onap.policy.api.PolicyParameters;
55 import org.onap.policy.api.PolicyType;
56 import org.onap.policy.api.PushPolicyParameters;
57 import org.springframework.beans.factory.annotation.Autowired;
58 import org.springframework.beans.factory.annotation.Value;
59 import org.springframework.context.ApplicationContext;
60
61 /**
62  * Policy utility methods - specifically, send the policy.
63  */
64 public class PolicyClient {
65
66     protected static final String     POLICY_PREFIX_BASE         = "Config_";
67     protected static final String     POLICY_PREFIX_BRMS_PARAM   = "Config_BRMS_Param_";
68     protected static final String     POLICY_PREFIX_MICROSERVICE = "Config_MS_";
69
70     protected static final String     LOG_POLICY_PREFIX          = "Response is ";
71
72     protected static final EELFLogger logger                     = EELFManager.getInstance()
73             .getLogger(PolicyClient.class);
74     protected static final EELFLogger metricsLogger              = EELFManager.getInstance().getMetricsLogger();
75
76     @Value("${org.onap.clamp.config.files.cldsPolicyConfig:'classpath:/clds/clds-policy-config.properties'}")
77     protected String                  cldsPolicyConfigFile;
78
79     @Autowired
80     protected ApplicationContext      appContext;
81
82     @Autowired
83     protected RefProp                 refProp;
84
85     /**
86      * Perform BRMS policy type.
87      *
88      * @param attributes
89      *            A map of attributes
90      * @param prop
91      *            The ModelProperties
92      * @param policyRequestUuid
93      *            PolicyRequest UUID
94      * @return The response message of policy
95      * 
96      */
97     public String sendBrmsPolicy(Map<AttributeType, Map<String, String>> attributes, ModelProperties prop,
98             String policyRequestUuid) {
99
100         PolicyParameters policyParameters = new PolicyParameters();
101
102         // Set Policy Type(Mandatory)
103         policyParameters.setPolicyConfigType(PolicyConfigType.BRMS_PARAM);
104
105         // Set Policy Name(Mandatory)
106         policyParameters.setPolicyName(prop.getPolicyScopeAndNameWithUniqueId());
107
108         // documentation says this is options, but when tested, got the
109         // following failure: java.lang.Exception: Policy send failed: PE300 -
110         // Data Issue: No policyDescription given.
111         policyParameters.setPolicyDescription(refProp.getStringValue("op.policyDescription"));
112
113         policyParameters.setAttributes(attributes);
114
115         // Set a random UUID(Mandatory)
116         policyParameters.setRequestID(UUID.fromString(policyRequestUuid));
117         String policyNamePrefix = refProp.getStringValue("policy.op.policyNamePrefix");
118         String rtnMsg = send(policyParameters, prop, policyNamePrefix);
119
120         String policyType = refProp.getStringValue("policy.op.type");
121         push(policyType, prop);
122
123         return rtnMsg;
124     }
125
126     /**
127      * Perform send of microservice policy in JSON.
128      *
129      * @param policyJson
130      *            The policy JSON
131      * @param prop
132      *            The ModelProperties
133      * @param policyRequestUuid
134      *            The policy Request UUID
135      * @return The response message of policy
136      */
137     public String sendMicroServiceInJson(String policyJson, ModelProperties prop, String policyRequestUuid) {
138
139         PolicyParameters policyParameters = new PolicyParameters();
140
141         // Set Policy Type
142         policyParameters.setPolicyConfigType(PolicyConfigType.MicroService);
143         policyParameters.setEcompName(refProp.getStringValue("policy.onap.name"));
144         policyParameters.setPolicyName(prop.getCurrentPolicyScopeAndPolicyName());
145
146         policyParameters.setConfigBody(policyJson);
147         policyParameters.setConfigBodyType(PolicyType.JSON);
148
149         policyParameters.setRequestID(UUID.fromString(policyRequestUuid));
150         String policyNamePrefix = refProp.getStringValue("policy.ms.policyNamePrefix");
151
152         // Adding this line to clear the policy id from policy name while
153         // pushing to policy engine
154         prop.setPolicyUniqueId("");
155
156         String rtnMsg = send(policyParameters, prop, policyNamePrefix);
157         String policyType = refProp.getStringValue("policy.ms.type");
158         push(policyType, prop);
159
160         return rtnMsg;
161     }
162
163     /**
164      * Perform send of base policy in OTHER type.
165      * 
166      * @param configBody
167      *            The config policy string body
168      * @param prop
169      *            The ModelProperties
170      * @param policyRequestUuid
171      *            The policy request UUID
172      * @return The answer from policy call
173      */
174     public String sendBasePolicyInOther(String configBody, ModelProperties prop, String policyRequestUuid) {
175
176         PolicyParameters policyParameters = new PolicyParameters();
177
178         // Set Policy Type
179         policyParameters.setPolicyConfigType(PolicyConfigType.Base);
180         policyParameters.setEcompName(refProp.getStringValue("policy.onap.name"));
181         policyParameters.setPolicyName(prop.getCurrentPolicyScopeAndPolicyName());
182
183         policyParameters.setConfigBody(configBody);
184         policyParameters.setConfigBodyType(PolicyType.OTHER);
185         policyParameters.setConfigName("HolmesPolicy");
186
187         policyParameters.setRequestID(UUID.fromString(policyRequestUuid));
188
189         // Adding this line to clear the policy id from policy name while
190         // pushing to policy engine
191         prop.setPolicyUniqueId("");
192
193         String rtnMsg = send(policyParameters, prop, POLICY_PREFIX_BASE);
194         push(PolicyConfigType.Base.toString(), prop);
195
196         return rtnMsg;
197     }
198
199     /**
200      * Perform send of Microservice policy in OTHER type.
201      * 
202      * @param configBody
203      *            The config policy string body
204      * @param prop
205      *            The ModelProperties
206      * @param policyRequestUuid
207      *            The policy request UUID
208      * @return The answer from policy call
209      */
210     public String sendMicroServiceInOther(String configBody, ModelProperties prop, String policyRequestUuid) {
211
212         PolicyParameters policyParameters = new PolicyParameters();
213
214         // Set Policy Type
215         policyParameters.setPolicyConfigType(PolicyConfigType.MicroService);
216         policyParameters.setEcompName(refProp.getStringValue("policy.onap.name"));
217         policyParameters.setPolicyName(prop.getCurrentPolicyScopeAndPolicyName());
218
219         policyParameters.setConfigBody(configBody);
220         policyParameters.setConfigBodyType(PolicyType.OTHER);
221
222         policyParameters.setRequestID(UUID.fromString(policyRequestUuid));
223         String policyNamePrefix = refProp.getStringValue("policy.ms.policyNamePrefix");
224
225         // Adding this line to clear the policy id from policy name while
226         // pushing to policy engine
227         prop.setPolicyUniqueId("");
228
229         String rtnMsg = send(policyParameters, prop, policyNamePrefix);
230         String policyType = refProp.getStringValue("policy.ms.type");
231         push(policyType, prop);
232
233         return rtnMsg;
234     }
235
236     /**
237      * Perform send of policy.
238      *
239      * @param policyParameters
240      *            The PolicyParameters
241      * @param prop
242      *            The ModelProperties
243      * @return The response message of Policy
244      */
245     protected String send(PolicyParameters policyParameters, ModelProperties prop, String policyNamePrefix) {
246         // Verify whether it is triggered by Validation Test button from UI
247         if (prop.isTest()) {
248             return "send not executed for test action";
249         }
250
251         // API method to create or update Policy.
252         PolicyChangeResponse response = null;
253         String responseMessage = "";
254         Date startTime = new Date();
255         try {
256             List<Integer> versions = getVersions(policyNamePrefix, prop);
257             if (versions.isEmpty()) {
258                 LoggingUtils.setTargetContext("Policy", "createPolicy");
259                 logger.info("Attempting to create policy for action=" + prop.getActionCd());
260                 response = getPolicyEngine().createPolicy(policyParameters);
261                 responseMessage = response.getResponseMessage();
262             } else {
263                 LoggingUtils.setTargetContext("Policy", "updatePolicy");
264                 logger.info("Attempting to update policy for action=" + prop.getActionCd());
265                 response = getPolicyEngine().updatePolicy(policyParameters);
266                 responseMessage = response.getResponseMessage();
267             }
268         } catch (Exception e) {
269             logger.error("Exception occurred during policy communication", e);
270             throw new PolicyClientException("Exception while communicating with Policy", e);
271         }
272         logger.info(LOG_POLICY_PREFIX + responseMessage);
273
274         LoggingUtils.setTimeContext(startTime, new Date());
275
276         if (response.getResponseCode() == 200) {
277             logger.info("Policy send successful");
278             metricsLogger.info("Policy send success");
279         } else {
280             logger.warn("Policy send failed: " + responseMessage);
281             metricsLogger.info("Policy send failure");
282             throw new BadRequestException("Policy send failed: " + responseMessage);
283         }
284
285         return responseMessage;
286     }
287
288     /**
289      * Format and send push of policy.
290      *
291      * @param policyType
292      *            The policy Type
293      * @param prop
294      *            The ModelProperties
295      * @return The response message of policy
296      */
297     protected String push(String policyType, ModelProperties prop) {
298         // Verify whether it is triggered by Validation Test button from UI
299         if (prop.isTest()) {
300             return "push not executed for test action";
301         }
302
303         PushPolicyParameters pushPolicyParameters = new PushPolicyParameters();
304
305         // Parameter arguments
306         if (prop.getPolicyUniqueId() != null && !prop.getPolicyUniqueId().isEmpty()) {
307             pushPolicyParameters.setPolicyName(prop.getPolicyScopeAndNameWithUniqueId());
308         } else {
309             pushPolicyParameters.setPolicyName(prop.getCurrentPolicyScopeAndPolicyName());
310         }
311         logger.info("Policy Name in Push policy method - " + pushPolicyParameters.getPolicyName());
312
313         pushPolicyParameters.setPolicyType(policyType);
314         pushPolicyParameters.setPdpGroup(refProp.getStringValue("policy.pdp.group"));
315         pushPolicyParameters.setRequestID(null);
316
317         // API method to create or update Policy.
318         PolicyChangeResponse response = null;
319         String responseMessage = "";
320         try {
321             logger.info("Attempting to push policy...");
322             response = getPolicyEngine().pushPolicy(pushPolicyParameters);
323             responseMessage = response.getResponseMessage();
324         } catch (Exception e) {
325             logger.error("Exception occurred during policy communication", e);
326         }
327         logger.info(LOG_POLICY_PREFIX + responseMessage);
328
329         if (response != null && (response.getResponseCode() == 200 || response.getResponseCode() == 204)) {
330             logger.info("Policy push successful");
331         } else {
332             logger.warn("Policy push failed: " + responseMessage);
333             throw new BadRequestException("Policy push failed: " + responseMessage);
334         }
335
336         return responseMessage;
337     }
338
339     /**
340      * Use Get Config Policy API to retrieve the versions for a policy. Return
341      * versions in sorted order. Return empty list if none found.
342      *
343      * @param policyNamePrefix
344      *            The Policy Name Prefix
345      * @param prop
346      *            The ModelProperties
347      * @return The response message from policy
348      * @throws PolicyConfigException
349      *             In case of issues with policy engine
350      */
351     protected List<Integer> getVersions(String policyNamePrefix, ModelProperties prop) throws PolicyConfigException {
352
353         ArrayList<Integer> versions = new ArrayList<>();
354         ConfigRequestParameters configRequestParameters = new ConfigRequestParameters();
355         String policyName = "";
356
357         if (prop.getPolicyUniqueId() != null && !prop.getPolicyUniqueId().isEmpty()) {
358             policyName = prop.getCurrentPolicyScopeAndFullPolicyName(policyNamePrefix) + "_" + prop.getPolicyUniqueId();
359         } else {
360             policyName = prop.getCurrentPolicyScopeAndFullPolicyName(policyNamePrefix);
361         }
362
363         logger.info("policyName=" + policyName);
364         configRequestParameters.setPolicyName(policyName);
365
366         Collection<PolicyConfig> response = getPolicyEngine().getConfig(configRequestParameters);
367         for (PolicyConfig policyConfig : response) {
368             Integer version = Integer.valueOf(policyConfig.getPolicyVersion());
369             versions.add(version);
370         }
371         Collections.sort(versions);
372         logger.info("Policy versions.size()=" + versions.size());
373
374         return versions;
375
376     }
377
378     /**
379      * This method create a new policy engine.
380      * 
381      * @return A new policy engine
382      */
383     private PolicyEngine getPolicyEngine() {
384         PolicyEngine policyEngine;
385         try {
386             policyEngine = new PolicyEngine(appContext.getResource(cldsPolicyConfigFile).getFile().getAbsolutePath());
387         } catch (IOException e1) {
388             throw new PolicyClientException("Exception when opening policy config file", e1);
389         } catch (PolicyEngineException e) {
390             throw new PolicyClientException("Exception when creating a new policy engine", e);
391         }
392         return policyEngine;
393     }
394
395     /**
396      * Format and send delete Micro Service requests to Policy.
397      *
398      * @param prop
399      *            The ModelProperties
400      * @return The response message from Policy
401      */
402     public String deleteMicrosService(ModelProperties prop) {
403         String policyType = refProp.getStringValue("policy.ms.type");
404         return deletePolicy(prop, policyType);
405     }
406
407     /**
408      * This method delete the Base policy.
409      *
410      * @param prop
411      *            The model Properties
412      * @return A string with the answer from policy
413      */
414     public String deleteBasePolicy(ModelProperties prop) {
415         return deletePolicy(prop, PolicyConfigType.Base.toString());
416     }
417
418     /**
419      * Format and send delete BRMS requests to Policy.
420      *
421      * @param prop
422      *            The ModelProperties
423      * @return The response message from policy
424      */
425     public String deleteBrms(ModelProperties prop) {
426         String policyType = refProp.getStringValue("policy.op.type");
427         return deletePolicy(prop, policyType);
428     }
429
430     /**
431      * Format and send delete PAP and PDP requests to Policy.
432      *
433      * @param prop
434      *            The ModelProperties
435      *
436      * @return The response message from policy
437      */
438     protected String deletePolicy(ModelProperties prop, String policyType) {
439         DeletePolicyParameters deletePolicyParameters = new DeletePolicyParameters();
440
441         if (prop.getPolicyUniqueId() != null && !prop.getPolicyUniqueId().isEmpty()) {
442             deletePolicyParameters.setPolicyName(prop.getPolicyScopeAndNameWithUniqueId());
443         } else {
444             deletePolicyParameters.setPolicyName(prop.getCurrentPolicyScopeAndPolicyName());
445         }
446         logger.info("Policy Name in delete policy method - " + deletePolicyParameters.getPolicyName());
447         deletePolicyParameters.setPolicyComponent("PDP");
448         deletePolicyParameters.setDeleteCondition(DeletePolicyCondition.ALL);
449         deletePolicyParameters.setPdpGroup(refProp.getStringValue("policy.pdp.group"));
450         deletePolicyParameters.setPolicyType(policyType);
451         // send delete request
452         StringBuilder responseMessage = new StringBuilder(sendDeletePolicy(deletePolicyParameters, prop));
453
454         logger.info("Deleting policy from PAP...");
455         deletePolicyParameters.setPolicyComponent("PAP");
456         deletePolicyParameters.setDeleteCondition(DeletePolicyCondition.ALL);
457
458         // send delete request
459         responseMessage.append(sendDeletePolicy(deletePolicyParameters, prop));
460
461         return responseMessage.toString();
462     }
463
464     /**
465      * Send delete request to Policy.
466      *
467      * @param deletePolicyParameters
468      *            The DeletePolicyParameters
469      * @param prop
470      *            The ModelProperties
471      * @return The response message from policy
472      */
473     protected String sendDeletePolicy(DeletePolicyParameters deletePolicyParameters, ModelProperties prop) {
474         // Verify whether it is triggered by Validation Test button from UI
475         if (prop.isTest()) {
476             return "delete not executed for test action";
477         }
478
479         // API method to create or update Policy.
480         PolicyChangeResponse response = null;
481         String responseMessage = "";
482         try {
483             logger.info("Attempting to delete policy...");
484             response = getPolicyEngine().deletePolicy(deletePolicyParameters);
485             responseMessage = response.getResponseMessage();
486         } catch (Exception e) {
487             logger.error("Exception occurred during policy communnication", e);
488         }
489         logger.info(LOG_POLICY_PREFIX + responseMessage);
490
491         if (response != null && response.getResponseCode() == 200) {
492             logger.info("Policy delete successful");
493         } else {
494             logger.warn("Policy delete failed: " + responseMessage);
495             throw new BadRequestException("Policy delete failed: " + responseMessage);
496         }
497
498         return responseMessage;
499     }
500 }