Consolidate PolicyRestAdapter setup
[policy/engine.git] / ONAP-PAP-REST / src / main / java / org / onap / policy / pap / xacml / rest / controller / ActionPolicyDictionaryController.java
index 02b0707..f576bff 100644 (file)
@@ -2,7 +2,7 @@
  * ============LICENSE_START=======================================================
  * ONAP-PAP-REST
  * ================================================================================
- * Copyright (C) 2017-2018 AT&T Intellectual Property. All rights reserved.
+ * Copyright (C) 2017-2019 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.
 
 package org.onap.policy.pap.xacml.rest.controller;
 
+import com.fasterxml.jackson.databind.DeserializationFeature;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+
 import java.io.IOException;
-import java.io.PrintWriter;
 import java.util.Date;
-import java.util.HashMap;
-import java.util.LinkedHashMap;
 import java.util.List;
-import java.util.Map;
 
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 
-import org.json.JSONObject;
-import org.onap.policy.common.logging.flexlogger.FlexLogger;
-import org.onap.policy.common.logging.flexlogger.Logger;
-import org.onap.policy.pap.xacml.rest.util.JsonMessage;
+import org.onap.policy.pap.xacml.rest.util.DictionaryUtils;
 import org.onap.policy.rest.dao.CommonClassDao;
 import org.onap.policy.rest.jpa.ActionPolicyDict;
 import org.onap.policy.rest.jpa.UserInfo;
-import org.onap.policy.utils.PolicyUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.MediaType;
 import org.springframework.stereotype.Controller;
 import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestMethod;
 import org.springframework.web.servlet.ModelAndView;
 
-import com.fasterxml.jackson.databind.DeserializationFeature;
-import com.fasterxml.jackson.databind.JsonNode;
-import com.fasterxml.jackson.databind.ObjectMapper;
-
 @Controller
 public class ActionPolicyDictionaryController {
 
-       private static final Logger LOGGER  = FlexLogger.getLogger(ActionPolicyDictionaryController.class);
-
-       private static CommonClassDao commonClassDao;
-       private static String utf8 = "UTF-8";
-       private static String attributeName = "attributeName";
-       @Autowired
-       public ActionPolicyDictionaryController(CommonClassDao commonClassDao){
-               ActionPolicyDictionaryController.commonClassDao = commonClassDao;
-       }
-       
-       public void setCommonClassDao(CommonClassDao commonClassDao){
-               ActionPolicyDictionaryController.commonClassDao = commonClassDao;
-       }
-       /*
-        * This is an empty constructor
-        */     
-       public ActionPolicyDictionaryController(){
-               
-       }
-
-       public UserInfo getUserInfo(String loginId){
-               return (UserInfo) commonClassDao.getEntityItem(UserInfo.class, "userLoginId", loginId);
-       }
-
-       @RequestMapping(value={"/get_ActionPolicyDictDataByName"}, method={org.springframework.web.bind.annotation.RequestMethod.GET} , produces=MediaType.APPLICATION_JSON_VALUE)
-       public void getActionEntitybyName(HttpServletResponse response){
-               try{
-                       Map<String, Object> model = new HashMap<>();
-                       ObjectMapper mapper = new ObjectMapper();
-                       model.put("actionPolicyDictionaryDatas", mapper.writeValueAsString(commonClassDao.getDataByColumn(ActionPolicyDict.class, attributeName)));
-                       JsonMessage msg = new JsonMessage(mapper.writeValueAsString(model));
-                       JSONObject j = new JSONObject(msg);
-                       response.getWriter().write(j.toString());
-               }
-               catch (Exception e){
-                       LOGGER.error(e.getMessage(),e);
-               }
-       }
-
-       @RequestMapping(value={"/get_ActionPolicyDictData"}, method={org.springframework.web.bind.annotation.RequestMethod.GET} , produces=MediaType.APPLICATION_JSON_VALUE)
-       public void getActionPolicyDictionaryEntityData(HttpServletResponse response){
-               try{
-                       Map<String, Object> model = new HashMap<>();
-                       ObjectMapper mapper = new ObjectMapper();
-                       model.put("actionPolicyDictionaryDatas", mapper.writeValueAsString(commonClassDao.getData(ActionPolicyDict.class)));
-                       JsonMessage msg = new JsonMessage(mapper.writeValueAsString(model));
-                       JSONObject j = new JSONObject(msg);
-                       response.addHeader("successMapKey", "success"); 
-                       response.addHeader("operation", "getDictionary");
-                       response.getWriter().write(j.toString());
-               }
-               catch (Exception e){
-                       LOGGER.error(e.getMessage(),e);
-                       response.setStatus(HttpServletResponse.SC_BAD_REQUEST);                             
-                       response.addHeader("error", "dictionaryDBQuery");
-               }
-       }
-
-       @RequestMapping(value={"/action_dictionary/save_ActionDict"}, method={org.springframework.web.bind.annotation.RequestMethod.POST})
-       public ModelAndView saveActionPolicyDictionary(HttpServletRequest request, HttpServletResponse response) throws IOException   {
-               try {
-                       boolean duplicateflag = false;
-                       boolean isFakeUpdate = false;
-                       boolean fromAPI = false;
-
-                       if (request.getParameter("apiflag")!=null && ("api").equalsIgnoreCase(request.getParameter("apiflag"))) {
-                               fromAPI = true;
-                       }
-                       ObjectMapper mapper = new ObjectMapper();
-                       mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
-                       JsonNode root = mapper.readTree(request.getReader());
-                       ActionPolicyDict actionPolicyDict = null;
-                       ActionAdapter adapter = null;
-                       String userId = null;
-
-                       if(fromAPI) {
-                               actionPolicyDict = mapper.readValue(root.get("dictionaryFields").toString(), ActionPolicyDict.class);
-                               adapter = mapper.readValue(root.get("dictionaryFields").toString(), ActionAdapter.class);
-                               userId = "API";
-
-                               //check if update operation or create, get id for data to be updated and update attributeData
-                               if (("update").equals(request.getParameter("operation"))) {
-                                       List<Object> duplicateData =  commonClassDao.checkDuplicateEntry(actionPolicyDict.getAttributeName(), attributeName, ActionPolicyDict.class);
-                                       ActionPolicyDict data = (ActionPolicyDict) duplicateData.get(0);
-                                       int id = data.getId();
-                                       if(id==0){
-                                               isFakeUpdate=true;
-                                               actionPolicyDict.setId(1);
-                                       } else {
-                                               actionPolicyDict.setId(id);
-                                       }
-                                       actionPolicyDict.setUserCreatedBy(this.getUserInfo(userId));
-                               }
-                       } else {
-                               actionPolicyDict = mapper.readValue(root.get("actionPolicyDictionaryData").toString(), ActionPolicyDict.class);
-                               adapter = mapper.readValue(root.get("actionPolicyDictionaryData").toString(), ActionAdapter.class);
-                               userId = root.get("userid").textValue();
-                       }
-                       StringBuilder header = new StringBuilder();
-                       int counter = 0;
-                       if(!adapter.getHeaders().isEmpty()){
-                               for(Object attribute : adapter.getHeaders()){
-                                       if(attribute instanceof LinkedHashMap<?, ?>){
-                                               String key = ((LinkedHashMap<?, ?>) attribute).get("option").toString();
-                                               String value = ((LinkedHashMap<?, ?>) attribute).get("number").toString();
-                                               if(counter>0){
-                                                       header.append(":");
-                                               }
-                                               header.append(key).append("=").append(value);
-                                               counter ++;
-                                       }
-                               }
-                       }
-                       actionPolicyDict.setHeader(header.toString());
-                       if(actionPolicyDict.getId() == 0){
-                               List<Object> duplicateData =  commonClassDao.checkDuplicateEntry(actionPolicyDict.getAttributeName(), attributeName, ActionPolicyDict.class);
-                               if(!duplicateData.isEmpty()){
-                                       duplicateflag = true;
-                               }else{
-                                       actionPolicyDict.setUserCreatedBy(this.getUserInfo(userId));
-                                       actionPolicyDict.setUserModifiedBy(this.getUserInfo(userId));
-                                       commonClassDao.save(actionPolicyDict);
-                               }
-                       }else{
-                               if(!isFakeUpdate) {
-                                       actionPolicyDict.setUserModifiedBy(this.getUserInfo(userId));
-                                       actionPolicyDict.setModifiedDate(new Date());
-                                       commonClassDao.update(actionPolicyDict); 
-                               }
-                       } 
-
-                       String responseString = null;
-                       if(duplicateflag) {
-                               responseString = "Duplicate";
-                       } else {
-                               responseString = mapper.writeValueAsString(commonClassDao.getData(ActionPolicyDict.class));
-                       }
-
-                       if (fromAPI) {
-                               if (responseString!=null && !("Duplicate").equals(responseString)) {
-                                       if(isFakeUpdate) {
-                                               responseString = "Exists";
-                                       } else {
-                                               responseString = "Success";
-                                       }   
-                               }
-
-                               ModelAndView result = new ModelAndView();
-                               result.setViewName(responseString);
-                               return result;
-                       } else {
-                               response.setCharacterEncoding(utf8);
-                               response.setContentType("application / json");
-                               request.setCharacterEncoding(utf8); 
-
-                               PrintWriter out = response.getWriter();
-                               JSONObject j = new JSONObject("{actionPolicyDictionaryDatas: " + responseString + "}");
-                               out.write(j.toString());
-
-                               return null;
-                       }
-               }
-               catch (Exception e){
-                       LOGGER.error(e);
-                       response.setCharacterEncoding(utf8);
-                       request.setCharacterEncoding(utf8);
-                       PrintWriter out = response.getWriter();
-                       out.write(PolicyUtils.CATCH_EXCEPTION);
-               }
-               return null;
-       }
-
-       @RequestMapping(value={"/action_dictionary/remove_actionPolicyDict"}, method={org.springframework.web.bind.annotation.RequestMethod.POST})
-       public void removeActionPolicyDictionary(HttpServletRequest request, HttpServletResponse response) throws IOException {
-               try{
-                       ObjectMapper mapper = new ObjectMapper();
-                       mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
-                       JsonNode root = mapper.readTree(request.getReader());
-                       ActionPolicyDict actionPolicyDict = mapper.readValue(root.get("data").toString(), ActionPolicyDict.class);
-                       commonClassDao.delete(actionPolicyDict);
-                       response.setCharacterEncoding(utf8);
-                       response.setContentType("application / json");
-                       request.setCharacterEncoding(utf8);
-
-                       PrintWriter out = response.getWriter();
-
-                       String responseString = mapper.writeValueAsString(ActionPolicyDictionaryController.commonClassDao.getData(ActionPolicyDict.class));
-                       JSONObject j = new JSONObject("{actionPolicyDictionaryDatas: " + responseString + "}");
-                       out.write(j.toString());
-               }
-               catch (Exception e){
-                       LOGGER.error(e);
-                       response.setCharacterEncoding(utf8);
-                       request.setCharacterEncoding(utf8);
-                       PrintWriter out = response.getWriter();
-                       out.write(e.getMessage());
-               }
-       }
+    private static CommonClassDao commonClassDao;
+    private static String operation = "operation";
+    private static String attributeName = "attributeName";
+    private static String actionDatas = "actionPolicyDictionaryDatas";
+
+    @Autowired
+    public ActionPolicyDictionaryController(CommonClassDao commonClassDao) {
+        ActionPolicyDictionaryController.commonClassDao = commonClassDao;
+    }
+
+    public void setCommonClassDao(CommonClassDao commonClassDao) {
+        ActionPolicyDictionaryController.commonClassDao = commonClassDao;
+    }
+
+    public ActionPolicyDictionaryController() {
+        super();
+    }
+
+    private DictionaryUtils getDictionaryUtilsInstance() {
+        return DictionaryUtils.getDictionaryUtils();
+    }
+
+    @RequestMapping(
+            value = {"/get_ActionPolicyDictDataByName"},
+            method = {RequestMethod.GET},
+            produces = MediaType.APPLICATION_JSON_VALUE)
+    public void getActionEntitybyName(HttpServletResponse response) {
+        DictionaryUtils utils = getDictionaryUtilsInstance();
+        utils.getDataByEntity(response, actionDatas, attributeName, ActionPolicyDict.class);
+    }
+
+    @RequestMapping(
+            value = {"/get_ActionPolicyDictData"},
+            method = {RequestMethod.GET},
+            produces = MediaType.APPLICATION_JSON_VALUE)
+    public void getActionPolicyDictionaryEntityData(HttpServletResponse response) {
+        DictionaryUtils utils = getDictionaryUtilsInstance();
+        utils.getData(response, actionDatas, ActionPolicyDict.class);
+    }
+
+    @RequestMapping(value = {"/action_dictionary/save_ActionDict"}, method = {RequestMethod.POST})
+    public ModelAndView saveActionPolicyDictionary(HttpServletRequest request, HttpServletResponse response)
+            throws IOException {
+        DictionaryUtils utils = getDictionaryUtilsInstance();
+        try {
+            boolean fromAPI = utils.isRequestFromAPI(request);
+            ObjectMapper mapper = new ObjectMapper();
+            mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
+            JsonNode root = mapper.readTree(request.getReader());
+            ActionPolicyDict actionPolicyDict = null;
+            ActionAdapter adapter = null;
+            String userId = null;
+            if (fromAPI) {
+                actionPolicyDict = mapper.readValue(root.get("dictionaryFields").toString(), ActionPolicyDict.class);
+                adapter = mapper.readValue(root.get("dictionaryFields").toString(), ActionAdapter.class);
+                userId = "API";
+            } else {
+                actionPolicyDict =
+                        mapper.readValue(root.get("actionPolicyDictionaryData").toString(), ActionPolicyDict.class);
+                adapter = mapper.readValue(root.get("actionPolicyDictionaryData").toString(), ActionAdapter.class);
+                userId = root.get("userid").textValue();
+            }
+            UserInfo userInfo = utils.getUserInfo(userId);
+
+            List<Object> duplicateData = commonClassDao.checkDuplicateEntry(actionPolicyDict.getAttributeName(),
+                    attributeName, ActionPolicyDict.class);
+            boolean duplicateflag = false;
+            if (!duplicateData.isEmpty()) {
+                ActionPolicyDict data = (ActionPolicyDict) duplicateData.get(0);
+                if (request.getParameter(operation) != null && "update".equals(request.getParameter(operation))) {
+                    actionPolicyDict.setId(data.getId());
+                } else if ((request.getParameter(operation) != null
+                        && !"update".equals(request.getParameter(operation)))
+                        || (request.getParameter(operation) == null && (data.getId() != actionPolicyDict.getId()))) {
+                    duplicateflag = true;
+                }
+            }
+            actionPolicyDict.setHeader(utils.appendKeyValue(adapter.getHeaders(), ":", "="));
+
+            String responseString = null;
+            if (!duplicateflag) {
+                actionPolicyDict.setUserModifiedBy(userInfo);
+                if (actionPolicyDict.getId() == 0) {
+                    actionPolicyDict.setUserCreatedBy(userInfo);
+                    commonClassDao.save(actionPolicyDict);
+                } else {
+                    actionPolicyDict.setModifiedDate(new Date());
+                    commonClassDao.update(actionPolicyDict);
+                }
+                responseString = mapper.writeValueAsString(commonClassDao.getData(ActionPolicyDict.class));
+            } else {
+                responseString = "Duplicate";
+            }
+            if (fromAPI) {
+                return utils.getResultForApi(responseString);
+            } else {
+                utils.setResponseData(response, actionDatas, responseString);
+            }
+        } catch (Exception e) {
+            utils.setErrorResponseData(response, e);
+        }
+        return null;
+    }
+
+    @RequestMapping(value = {"/action_dictionary/remove_actionPolicyDict"}, method = {RequestMethod.POST})
+    public void removeActionPolicyDictionary(HttpServletRequest request, HttpServletResponse response)
+            throws IOException {
+        DictionaryUtils utils = getDictionaryUtilsInstance();
+        utils.removeData(request, response, actionDatas, ActionPolicyDict.class);
+    }
 }
 
-class ActionAdapter{
-       private List<Object> headers;
 
-       public List<Object> getHeaders() {
-               return headers;
-       }
+class ActionAdapter {
+    private List<Object> headers;
+
+    public List<Object> getHeaders() {
+        return headers;
+    }
 
-       public void setHeaders(List<Object> headers) {
-               this.headers = headers;
-       }
+    public void setHeaders(List<Object> headers) {
+        this.headers = headers;
+    }
 }