X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=ONAP-REST%2Fsrc%2Fmain%2Fjava%2Forg%2Fonap%2Fpolicy%2Frest%2Futil%2FPolicyValidationRequestWrapper.java;h=76584e7c71258999423c0821257deaa7eba4e371;hb=c5d97e8a9a6bea71f3be329a2e44bdbe5fe50882;hp=ac97fb0eeef4e6b7ffbe7eb0af426e5ac36f3990;hpb=c579209f8a4045d7f96f2b566a82ce1b8cf8660d;p=policy%2Fengine.git

diff --git a/ONAP-REST/src/main/java/org/onap/policy/rest/util/PolicyValidationRequestWrapper.java b/ONAP-REST/src/main/java/org/onap/policy/rest/util/PolicyValidationRequestWrapper.java
index ac97fb0ee..76584e7c7 100644
--- a/ONAP-REST/src/main/java/org/onap/policy/rest/util/PolicyValidationRequestWrapper.java
+++ b/ONAP-REST/src/main/java/org/onap/policy/rest/util/PolicyValidationRequestWrapper.java
@@ -2,7 +2,7 @@
  * ============LICENSE_START=======================================================
  * ONAP Policy Engine
  * ================================================================================
- * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * Copyright (C) 2017-2018 AT&T Intellectual Property. All rights reserved.
  * ================================================================================
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -37,6 +37,7 @@ import org.onap.policy.api.AttributeType;
 import org.onap.policy.api.PolicyParameters;
 import org.onap.policy.common.logging.flexlogger.FlexLogger;
 import org.onap.policy.common.logging.flexlogger.Logger;
+import org.onap.policy.rest.adapter.ClosedLoopFaultTrapDatas;
 import org.onap.policy.rest.adapter.PolicyRestAdapter;
 import org.onap.policy.rest.adapter.RainyDayParams;
 import org.onap.policy.rest.adapter.YAMLParams;
@@ -51,16 +52,28 @@ public class PolicyValidationRequestWrapper {
 	
 	private static final Logger LOGGER	= FlexLogger.getLogger(PolicyValidationRequestWrapper.class);
 	public static final String CONFIG_NAME="configName";
+	public static final String INVALIDJSON = " improper JSON format: ";
+	public static final String ONAPNAME = "onapname";
+	public static final String SERVICETYPE_POLICY_NAME = "serviceTypePolicyName";
 
 	public PolicyRestAdapter populateRequestParameters(HttpServletRequest request) {
 		
 		PolicyRestAdapter policyData = null;
-		
+		ClosedLoopFaultTrapDatas trapDatas = null;
+		ClosedLoopFaultTrapDatas faultDatas = null;
 		try {
 			ObjectMapper mapper = new ObjectMapper();
 			mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
 			JsonNode root = mapper.readTree(request.getReader());
 			policyData = mapper.readValue(root.get("policyData").toString(), PolicyRestAdapter.class);
+			if(root.get("trapData") != null){
+				trapDatas = mapper.readValue(root.get("trapData").toString(), ClosedLoopFaultTrapDatas.class);
+				policyData.setTrapDatas(trapDatas);
+			}
+			if(root.get("faultData") != null){
+				faultDatas = mapper.readValue(root.get("faultData").toString(), ClosedLoopFaultTrapDatas.class);
+				policyData.setFaultDatas(faultDatas);
+			}
 			
 			JsonObject json;
 			json = stringToJsonObject(root.toString());
@@ -88,9 +101,16 @@ public class PolicyValidationRequestWrapper {
 		/*
 		 * set policy adapter values for Building JSON object containing policy data
 		 */
-		//Common among policy types
+		//Common Policy Fields
 		policyData.setPolicyName(parameters.getPolicyName());
 		policyData.setOnapName(parameters.getOnapName()); 
+		policyData.setPriority(parameters.getPriority()); //Micro Service
+		policyData.setConfigName(parameters.getConfigName());  //Base and Firewall
+		policyData.setRiskType(parameters.getRiskType()); //Safe parameters Attributes
+		policyData.setRiskLevel(parameters.getRiskLevel());//Safe parameters Attributes
+		policyData.setGuard(String.valueOf(parameters.getGuard()));//Safe parameters Attributes
+		policyData.setTtlDate(convertDate(parameters.getTtlDate()));//Safe parameters Attributes
+		policyData.setApiflag("API");
 		
 		//Some policies require jsonObject conversion from String for configBody (i.e. MicroService and Firewall)
 		JsonObject json = null;
@@ -99,7 +119,7 @@ public class PolicyValidationRequestWrapper {
         		json = stringToJsonObject(parameters.getConfigBody());
         	}
         } catch(JsonException| IllegalStateException e){
-            String message = XACMLErrorConstants.ERROR_DATA_ISSUE+ " improper JSON object : " + parameters.getConfigBody();
+            String message = XACMLErrorConstants.ERROR_DATA_ISSUE+ INVALIDJSON + parameters.getConfigBody();
             LOGGER.error(message, e);
             return null;
         }
@@ -215,10 +235,10 @@ public class PolicyValidationRequestWrapper {
 	            
 	            ArrayList<Object> attributeList = new ArrayList<>();
 	            if (matching != null) {
-		            for (String keyField : matching.keySet()) {
+		            for (Map.Entry<String, String> entry : matching.entrySet()) {
 						LinkedHashMap<String, String> attributeMap = new LinkedHashMap<>();
-						String key = keyField;
-						String value = matching.get(keyField);
+						String key = entry.getKey();
+						String value = entry.getValue();
 						attributeMap.put("key", key);
 						attributeMap.put("value", value);
 						attributeList.add(attributeMap);
@@ -268,7 +288,7 @@ public class PolicyValidationRequestWrapper {
 						try {
 							policyJSON = mapper.readTree(content);
 						} catch (IOException e) {
-				            String message = XACMLErrorConstants.ERROR_DATA_ISSUE+ " improper JSON object : " + parameters.getConfigBody();
+				            String message = XACMLErrorConstants.ERROR_DATA_ISSUE+ INVALIDJSON + parameters.getConfigBody();
 				            LOGGER.error(message, e);
 				            return null;					
 				        }
@@ -284,7 +304,7 @@ public class PolicyValidationRequestWrapper {
 			        }
 			        if (json.containsKey("location")){
 			            String msLocation = json.get("location").toString().replace("\"", "");
-			            policyData.setMsLocation(msLocation);
+			            policyData.setLocation(msLocation);
 			        }
 			        if (json.containsKey(CONFIG_NAME)){
 			            String configName = json.get(CONFIG_NAME).toString().replace("\"", "");
@@ -298,17 +318,36 @@ public class PolicyValidationRequestWrapper {
 			        	String version = json.get("version").toString().replace("\"", "");
 			        	policyData.setVersion(version);
 			        }
+			        if(json.containsKey("policyScope")){
+			        	String policyScope = json.get("policyScope").toString().replace("\"", "");
+			        	policyData.setPolicyScope(policyScope);
+			        }
+			        if(json.containsKey("riskType")){
+			        	String riskType = json.get("riskType").toString().replace("\"", "");
+			        	policyData.setRiskType(riskType);
+			        }
+			        if(json.containsKey("riskLevel")){
+			        	String riskLevel = json.get("riskLevel").toString().replace("\"", "");
+			        	policyData.setRiskLevel(riskLevel);
+			        }
+			        if(json.containsKey("guard")){
+			        	String guard = json.get("guard").toString().replace("\"", "");
+			        	policyData.setGuard(guard);
+			        }
+				} else {
+		            String message = XACMLErrorConstants.ERROR_DATA_ISSUE+ INVALIDJSON + parameters.getConfigBody();
+		            LOGGER.error(message);
+		            return null;				
 				}
 				
 			} else if("Fault".equals(parameters.getPolicyConfigType().toString())){
 				
 				policyData.setConfigPolicyType("ClosedLoop_Fault");
-				policyData.setApiflag("API");
 				
 				if(json != null){
 					policyData.setJsonBody(json.toString());
-			        if (json.get("onapname")!=null){
-			        	String onapName = json.get("onapname").toString().replace("\"", "");
+			        if (json.get(ONAPNAME)!=null){
+			        	String onapName = json.get(ONAPNAME).toString().replace("\"", "");
 			        	policyData.setOnapName(onapName);
 			        }
 				}
@@ -319,14 +358,14 @@ public class PolicyValidationRequestWrapper {
 				
 				if(json != null){
 					policyData.setJsonBody(json.toString());
-			        if (json.get("onapname")!=null){
-			        	String onapName = json.get("onapname").toString().replace("\"", "");
+			        if (json.get(ONAPNAME)!=null){
+			        	String onapName = json.get(ONAPNAME).toString().replace("\"", "");
 			        	policyData.setOnapName(onapName);
 			        }
-			        if (json.get("serviceTypePolicyName")!=null){
-			        	String serviceType = json.get("serviceTypePolicyName").toString().replace("\"", "");
+			        if (json.get(SERVICETYPE_POLICY_NAME)!=null){
+			        	String serviceType = json.get(SERVICETYPE_POLICY_NAME).toString().replace("\"", "");
 						LinkedHashMap<String, String> serviceTypePolicyName = new LinkedHashMap<>();
-						serviceTypePolicyName.put("serviceTypePolicyName", serviceType);
+						serviceTypePolicyName.put(SERVICETYPE_POLICY_NAME, serviceType);
 			        	policyData.setServiceTypePolicyName(serviceTypePolicyName);
 			        }
 				}
@@ -337,22 +376,14 @@ public class PolicyValidationRequestWrapper {
 				
 			}
 		}
-		
-		policyData.setPriority(parameters.getPriority()); //Micro Service
-		policyData.setConfigName(parameters.getConfigName());  //Base and Firewall
-		policyData.setRiskType(parameters.getRiskType()); //Safe parameters Attributes
-		policyData.setRiskLevel(parameters.getRiskLevel());//Safe parameters Attributes
-		policyData.setGuard(String.valueOf(parameters.getGuard()));//Safe parameters Attributes
-		policyData.setTtlDate(convertDate(parameters.getTtlDate()));//Safe parameters Attributes
 
 		return policyData;
 				
 	}
-	
+    
     private JsonObject stringToJsonObject(String value) {
-
     	try(JsonReader jsonReader = Json.createReader(new StringReader(value))){
-			return jsonReader.readObject();
+            return jsonReader.readObject();
         } catch(JsonException| IllegalStateException e){
             LOGGER.info(XACMLErrorConstants.ERROR_DATA_ISSUE+ "Improper JSON format... may or may not cause issues in validating the policy: " + value, e);
             return null;