Support edit query

Issue-ID: CPS-511
Signed-off-by: Niranjana <niranjana.y60@wipro.com>
Change-Id: Icc22fb6143c01dcc388339ffa57818f41ed50cfd

3 files changed, 71 insertions, 15 deletions

diff --git a/cps-tbdmt-service/src/main/java/org/onap/cps/tbdmt/client/CpsRestClient.java b/cps-tbdmt-service/src/main/java/org/onap/cps/tbdmt/client/CpsRestClient.java
index 0d356d8..e1301fb 100644
--- a/cps-tbdmt-service/src/main/java/org/onap/cps/tbdmt/client/CpsRestClient.java
+++ b/cps-tbdmt-service/src/main/java/org/onap/cps/tbdmt/client/CpsRestClient.java
@@ -21,6 +21,7 @@
 package org.onap.cps.tbdmt.client;
 
 import java.util.Arrays;
+import java.util.Map;
 import org.onap.cps.tbdmt.exception.CpsClientException;
 import org.onap.cps.tbdmt.model.AppConfiguration;
 import org.onap.cps.tbdmt.model.CpsConfiguration;
@@ -30,6 +31,7 @@ import org.springframework.http.HttpHeaders;
 import org.springframework.http.HttpMethod;
 import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
+import org.springframework.http.client.HttpComponentsClientHttpRequestFactory;
 import org.springframework.stereotype.Component;
 import org.springframework.util.LinkedMultiValueMap;
 import org.springframework.util.MultiValueMap;
@@ -43,6 +45,8 @@ public class CpsRestClient {
 
     private static final String QUERY_API_PATH = "/anchors/{anchor}/nodes/query";
 
+    private static final String POST_API_PATH = "/anchors/{anchor}/nodes";
+
     @Autowired
     private RestTemplate restTemplate;
 
@@ -91,6 +95,49 @@ public class CpsRestClient {
         }
     }
 
+    /**
+     * Post data to CPS using xpath.
+     *
+     * @param anchor anchor
+     * @param xpath xpath query
+     * @param requestType http request type
+     * @param payload request body
+     * @return result Response string from CPS
+     */
+    public String addData(final String anchor, final String xpath, final String requestType,
+            final Map<String, Object> payload) throws CpsClientException {
+
+        final MultiValueMap<String, String> queryParams = new LinkedMultiValueMap<>();
+        queryParams.add("xpath", xpath);
+        final CpsConfiguration cpsConfiguration = "cpsCore".equals(appConfiguration.getCpsClient())
+            ? appConfiguration.getCpsCoreConfiguration() : appConfiguration.getNcmpConfiguration();
+
+        final HttpHeaders headers = new HttpHeaders();
+        headers.setAccept(Arrays.asList(MediaType.APPLICATION_JSON));
+        headers.setContentType(MediaType.APPLICATION_JSON);
+        headers.setBasicAuth(cpsConfiguration.getUsername(), cpsConfiguration.getPassword());
+        final HttpEntity<String> entity = new HttpEntity<>(new com.google.gson.Gson().toJson(payload), headers);
+
+        String uri = buildCpsUrl(cpsConfiguration.getUrl(), POST_API_PATH, anchor, queryParams);
+        try {
+            if (requestType.equalsIgnoreCase("post")) {
+                uri = buildCpsUrl(cpsConfiguration.getUrl(), POST_API_PATH, anchor, new LinkedMultiValueMap<>());
+                return restTemplate.postForEntity(uri, entity, String.class).getBody();
+            } else if (requestType.equalsIgnoreCase("patch")) {
+                final HttpComponentsClientHttpRequestFactory requestFactory
+                        = new HttpComponentsClientHttpRequestFactory();
+                requestFactory.setConnectTimeout(10000);
+                requestFactory.setReadTimeout(10000);
+                restTemplate.setRequestFactory(requestFactory);
+                return restTemplate.patchForObject(uri, entity, String.class);
+            } else {
+                return restTemplate.exchange(uri, HttpMethod.PUT, entity, String.class).getBody();
+            }
+        } catch (final Exception e) {
+            throw new CpsClientException(e.getLocalizedMessage());
+        }
+    }
+
     private String buildCpsUrl(final String baseUrl, final String path, final String anchor,
         final MultiValueMap<String, String> queryParams) {
 
diff --git a/cps-tbdmt-service/src/main/java/org/onap/cps/tbdmt/model/ExecutionRequest.java b/cps-tbdmt-service/src/main/java/org/onap/cps/tbdmt/model/ExecutionRequest.java
index 322c6d4..8b6db17 100644
--- a/cps-tbdmt-service/src/main/java/org/onap/cps/tbdmt/model/ExecutionRequest.java
+++ b/cps-tbdmt-service/src/main/java/org/onap/cps/tbdmt/model/ExecutionRequest.java
@@ -33,4 +33,6 @@ import lombok.Setter;
 public class ExecutionRequest {
 
     private Map<String, String> inputParameters;
+
+    private Map<String, Object> payload;
 }
diff --git a/cps-tbdmt-service/src/main/java/org/onap/cps/tbdmt/service/ExecutionBusinessLogic.java b/cps-tbdmt-service/src/main/java/org/onap/cps/tbdmt/service/ExecutionBusinessLogic.java
index 05aed38..6032b05 100644
--- a/cps-tbdmt-service/src/main/java/org/onap/cps/tbdmt/service/ExecutionBusinessLogic.java
+++ b/cps-tbdmt-service/src/main/java/org/onap/cps/tbdmt/service/ExecutionBusinessLogic.java
@@ -72,17 +72,18 @@ public class ExecutionBusinessLogic {
         final Optional<Template> templateOptional = templateRepository.findById(new TemplateKey(templateId));
         if (templateOptional.isPresent()) {
             if (!StringUtils.isBlank(templateOptional.get().getMultipleQueryTemplateId())) {
-                return executeMultipleQuery(templateOptional.get(), executionRequest.getInputParameters());
+                return executeMultipleQuery(templateOptional.get(), executionRequest.getInputParameters(),
+                        executionRequest.getPayload());
             } else {
-                return execute(templateOptional.get(), executionRequest.getInputParameters());
+                return execute(templateOptional.get(), executionRequest.getInputParameters(),
+                        executionRequest.getPayload());
             }
         }
         throw new TemplateNotFoundException("Template does not exist");
     }
 
-    private String executeMultipleQuery(final Template template, final Map<String, String> inputParameters)
-                    throws OutputTransformationException {
-
+    private String executeMultipleQuery(final Template template, final Map<String, String> inputParameters,
+            final Map<String, Object> payload) throws OutputTransformationException {
         final List<Object> processedQueryOutput = new ArrayList<Object>();
         final String multipleQuerytemplateId = template.getMultipleQueryTemplateId();
         final Optional<Template> multipleQueryTemplate =
@@ -97,7 +98,7 @@ public class ExecutionBusinessLogic {
             final List<String> transformParamList = new ArrayList<String>(
                     Arrays.asList(multipleQueryTemplate.get().getTransformParam().split("\\s*,\\s*")));
             final String inputKey = transformParamList.get(transformParamList.size() - 1);
-            final String queryParamString = execute(multipleQueryTemplate.get(), inputParameters);
+            final String queryParamString = execute(multipleQueryTemplate.get(), inputParameters, payload);
             final List<String> queryParamList = new ArrayList<String>();
             final JsonParser jsonParser = new JsonParser();
             final Gson gson = new Gson();
@@ -114,7 +115,7 @@ public class ExecutionBusinessLogic {
                 queryParamList.forEach(queryParam -> {
                     final Map<String, String> inputParameter = new HashMap<String, String>();
                     inputParameter.put(inputKey, queryParam);
-                    final Object result = execute(template, inputParameter);
+                    final Object result = execute(template, inputParameter, payload);
                     processedQueryOutput.add(result);
                 });
             } catch (final Exception e) {
@@ -124,21 +125,27 @@ public class ExecutionBusinessLogic {
         }
     }
 
-    private String execute(final Template template, final Map<String, String> inputParameters) {
+    private String execute(final Template template, final Map<String, String> inputParameters,
+            final Map<String, Object> payload) {
+
         final String anchor = appConfiguration.getSchemaToAnchor().get(template.getModel());
         if (anchor == null) {
             throw new ExecuteException("Anchor not found for the schema");
         }
         final String xpath = generateXpath(template.getXpathTemplate(), inputParameters);
-
         try {
-            final String result = cpsRestClient.fetchNode(anchor, xpath, template.getRequestType(),
-                        template.getIncludeDescendants());
-            if (StringUtils.isBlank(template.getTransformParam())) {
-                return result;
+            if (template.getRequestType().equalsIgnoreCase("put") || template.getRequestType().equalsIgnoreCase("patch")
+                    || template.getRequestType().equalsIgnoreCase("post")) {
+                return cpsRestClient.addData(anchor, xpath, template.getRequestType(), payload);
             } else {
-                final List<JsonElement> json = transform(template, result);
-                return new Gson().toJson(json);
+                final String result = cpsRestClient.fetchNode(anchor, xpath, template.getRequestType(),
+                        template.getIncludeDescendants());
+                if (StringUtils.isBlank(template.getTransformParam())) {
+                    return result;
+                } else {
+                    final List<JsonElement> json = transform(template, result);
+                    return new Gson().toJson(json);
+                }
             }
         } catch (final CpsClientException e) {
             throw new ExecuteException(e.getLocalizedMessage());