Add changes for safe delete in policy-models
[policy/api.git] / main / src / main / java / org / onap / policy / api / main / rest / provider / PolicyTypeProvider.java
index 8fab2bd..6c8e73e 100644 (file)
@@ -2,7 +2,8 @@
  * ============LICENSE_START=======================================================\r
  * ONAP Policy API\r
  * ================================================================================\r
- * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.\r
+ * Copyright (C) 2019-2020 AT&T Intellectual Property. All rights reserved.\r
+ * Modifications Copyright (C) 2020 Nordix Foundation.\r
  * ================================================================================\r
  * Licensed under the Apache License, Version 2.0 (the "License");\r
  * you may not use this file except in compliance with the License.\r
 \r
 package org.onap.policy.api.main.rest.provider;\r
 \r
-import java.util.List;\r
 import javax.ws.rs.core.Response;\r
-import org.onap.policy.api.main.parameters.ApiParameterGroup;\r
-import org.onap.policy.common.parameters.ParameterService;\r
+\r
 import org.onap.policy.models.base.PfModelException;\r
-import org.onap.policy.models.pdp.concepts.PdpGroup;\r
-import org.onap.policy.models.pdp.concepts.PdpGroupFilter;\r
-import org.onap.policy.models.provider.PolicyModelsProvider;\r
-import org.onap.policy.models.provider.PolicyModelsProviderFactory;\r
-import org.onap.policy.models.provider.PolicyModelsProviderParameters;\r
 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicyTypeFilter;\r
-import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicyTypeIdentifier;\r
 import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;\r
 \r
 /**\r
@@ -41,18 +34,13 @@ import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;
  *\r
  * @author Chenfei Gao (cgao@research.att.com)\r
  */\r
-public class PolicyTypeProvider {\r
-\r
-    private PolicyModelsProvider modelsProvider;\r
+public class PolicyTypeProvider extends CommonModelProvider {\r
 \r
     /**\r
      * Default constructor.\r
      */\r
     public PolicyTypeProvider() throws PfModelException {\r
-\r
-        ApiParameterGroup parameterGroup = ParameterService.get("ApiGroup");\r
-        PolicyModelsProviderParameters providerParameters = parameterGroup.getDatabaseProviderParameters();\r
-        modelsProvider = new PolicyModelsProviderFactory().createPolicyModelsProvider(providerParameters);\r
+        super();\r
     }\r
 \r
     /**\r
@@ -68,16 +56,13 @@ public class PolicyTypeProvider {
     public ToscaServiceTemplate fetchPolicyTypes(String policyTypeId, String policyTypeVersion)\r
             throws PfModelException {\r
 \r
-        ToscaServiceTemplate serviceTemplate;\r
-        if (policyTypeId == null || policyTypeVersion == null) {\r
-            ToscaPolicyTypeFilter policyTypeFilter = ToscaPolicyTypeFilter.builder()\r
-                    .name(policyTypeId).version(policyTypeVersion).build();\r
-            serviceTemplate = modelsProvider.getFilteredPolicyTypes(policyTypeFilter);\r
-        } else {\r
-            serviceTemplate = modelsProvider.getPolicyTypes(policyTypeId, policyTypeVersion);\r
+        ToscaServiceTemplate serviceTemplate = getFilteredPolicyTypes(policyTypeId, policyTypeVersion);\r
+\r
+        if (policyTypeId != null && !hasPolicyType(serviceTemplate)) {\r
+            throw new PfModelException(Response.Status.NOT_FOUND,\r
+                    constructResourceNotFoundMessage(policyTypeId, policyTypeVersion));\r
         }\r
 \r
-        close();\r
         return serviceTemplate;\r
     }\r
 \r
@@ -92,11 +77,12 @@ public class PolicyTypeProvider {
      */\r
     public ToscaServiceTemplate fetchLatestPolicyTypes(String policyTypeId) throws PfModelException {\r
 \r
-        ToscaPolicyTypeFilter policyTypeFilter = ToscaPolicyTypeFilter.builder()\r
-                .name(policyTypeId).version(ToscaPolicyTypeFilter.LATEST_VERSION).build();\r
-        ToscaServiceTemplate serviceTemplate = modelsProvider.getFilteredPolicyTypes(policyTypeFilter);\r
+        ToscaServiceTemplate serviceTemplate =\r
+                getFilteredPolicyTypes(policyTypeId, ToscaPolicyTypeFilter.LATEST_VERSION);\r
+        if (!hasPolicyType(serviceTemplate)) {\r
+            throw new PfModelException(Response.Status.NOT_FOUND, constructResourceNotFoundMessage(policyTypeId, null));\r
+        }\r
 \r
-        close();\r
         return serviceTemplate;\r
     }\r
 \r
@@ -110,10 +96,12 @@ public class PolicyTypeProvider {
      */\r
     public ToscaServiceTemplate createPolicyType(ToscaServiceTemplate body) throws PfModelException {\r
 \r
-        ToscaServiceTemplate serviceTemplate = modelsProvider.createPolicyTypes(body);\r
+        if (!hasPolicyType(body)) {\r
+            throw new PfModelException(Response.Status.BAD_REQUEST,\r
+                    "no policy types specified in the service template");\r
+        }\r
 \r
-        close();\r
-        return serviceTemplate;\r
+        return modelsProvider.createPolicyTypes(body);\r
     }\r
 \r
     /**\r
@@ -129,48 +117,44 @@ public class PolicyTypeProvider {
     public ToscaServiceTemplate deletePolicyType(String policyTypeId, String policyTypeVersion)\r
             throws PfModelException {\r
 \r
-        validateDeleteEligibility(policyTypeId, policyTypeVersion);\r
-\r
         ToscaServiceTemplate serviceTemplate = modelsProvider.deletePolicyType(policyTypeId, policyTypeVersion);\r
 \r
-        close();\r
+        if (!hasPolicyType(serviceTemplate)) {\r
+            throw new PfModelException(Response.Status.NOT_FOUND,\r
+                    constructResourceNotFoundMessage(policyTypeId, policyTypeVersion));\r
+        }\r
+\r
         return serviceTemplate;\r
     }\r
 \r
     /**\r
-     * Validates whether specified policy type can be deleted based on the rule that\r
-     * policy type parameterized by at least one policies cannot be deleted.\r
+     * Retrieves the specified version of the policy type.\r
      *\r
-     * @param policyTypeId the ID of policy type\r
-     * @param policyTypeVersion the version of policy type\r
-     * @param policyId the ID of policy\r
-     * @param policyVersion the version of policy\r
+     * @param policyTypeName the name of the policy type\r
+     * @param policyTypeVersion the version of the policy type\r
+     *\r
+     * @return the TOSCA service template containing the specified version of the policy type\r
      *\r
      * @throws PfModelException the PfModel parsing exception\r
      */\r
-    private void validateDeleteEligibility(String policyTypeId, String policyTypeVersion) throws PfModelException {\r
-\r
-        PdpGroupFilter pdpGroupFilter = PdpGroupFilter.builder()\r
-                .policyType(new ToscaPolicyTypeIdentifier(policyTypeId, policyTypeVersion)).build();\r
+    private ToscaServiceTemplate getFilteredPolicyTypes(String policyTypeName, String policyTypeVersion)\r
+            throws PfModelException {\r
 \r
-        List<PdpGroup> pdpGroups = modelsProvider.getFilteredPdpGroups(pdpGroupFilter);\r
-        if (!pdpGroups.isEmpty()) {\r
-            throw new PfModelException(Response.Status.CONFLICT,\r
-                    "the policy type is parameterized by at least one policies that have been deployed in pdp group");\r
-        }\r
+        ToscaPolicyTypeFilter policyTypeFilter =\r
+                ToscaPolicyTypeFilter.builder().name(policyTypeName).version(policyTypeVersion).build();\r
+        return modelsProvider.getFilteredPolicyTypes(policyTypeFilter);\r
     }\r
 \r
     /**\r
-     * Closes the connection to database.\r
+     * Constructs returned message for not found resource.\r
      *\r
-     * @throws PfModelException the PfModel parsing exception\r
+     * @param policyTypeId the ID of policy type\r
+     * @param policyTypeVersion the version of policy type\r
+     *\r
+     * @return constructed message\r
      */\r
-    private void close() throws PfModelException {\r
-        try {\r
-            modelsProvider.close();\r
-        } catch (Exception e) {\r
-            throw new PfModelException(\r
-                    Response.Status.INTERNAL_SERVER_ERROR, "error closing connection to database", e);\r
-        }\r
+    private String constructResourceNotFoundMessage(String policyTypeId, String policyTypeVersion) {\r
+\r
+        return "policy type with ID " + policyTypeId + ":" + policyTypeVersion + " does not exist";\r
     }\r
 }\r