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 19f3a75..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.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.ToscaPolicy;\r
-import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicyFilter;\r
 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicyTypeFilter;\r
 import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;\r
 \r
@@ -40,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
@@ -67,16 +56,13 @@ public class PolicyTypeProvider {
     public ToscaServiceTemplate fetchPolicyTypes(String policyTypeId, String policyTypeVersion)\r
             throws PfModelException {\r
 \r
-        ToscaPolicyTypeFilter policyTypeFilter = ToscaPolicyTypeFilter.builder()\r
-                .name(policyTypeId).version(policyTypeVersion).build();\r
-        ToscaServiceTemplate serviceTemplate = modelsProvider.getFilteredPolicyTypes(policyTypeFilter);\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
@@ -91,15 +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
-        if (serviceTemplate.getPolicyTypes().isEmpty()) {\r
-            throw new PfModelException(Response.Status.NOT_FOUND,\r
-                    constructResourceNotFoundMessage(policyTypeId, null));\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
@@ -113,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
@@ -132,32 +117,32 @@ 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 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
-        ToscaPolicyFilter policyFilter = ToscaPolicyFilter.builder()\r
-                .type(policyTypeId).typeVersion(policyTypeVersion).build();\r
-        List<ToscaPolicy> policies = modelsProvider.getFilteredPolicyList(policyFilter);\r
-        if (!policies.isEmpty()) {\r
-            throw new PfModelException(Response.Status.CONFLICT,\r
-                    "the policy type has been parameterized by at least one policies");\r
-        }\r
+    private ToscaServiceTemplate getFilteredPolicyTypes(String policyTypeName, String policyTypeVersion)\r
+            throws PfModelException {\r
+\r
+        ToscaPolicyTypeFilter policyTypeFilter =\r
+                ToscaPolicyTypeFilter.builder().name(policyTypeName).version(policyTypeVersion).build();\r
+        return modelsProvider.getFilteredPolicyTypes(policyTypeFilter);\r
     }\r
 \r
     /**\r
@@ -172,39 +157,4 @@ public class PolicyTypeProvider {
 \r
         return "policy type with ID " + policyTypeId + ":" + policyTypeVersion + " does not exist";\r
     }\r
-\r
-    /**\r
-     * Checks if service template contains any policy type.\r
-     *\r
-     * @param serviceTemplate the service template to check against\r
-     *\r
-     * @return boolean whether service template contains any policy type\r
-     */\r
-    private boolean hasPolicyType(ToscaServiceTemplate serviceTemplate) {\r
-\r
-        if (serviceTemplate.getPolicyTypes() == null) {\r
-            return false;\r
-        } else if (serviceTemplate.getPolicyTypes().isEmpty()) {\r
-            return false;\r
-        } else if (serviceTemplate.getPolicyTypes().get(0).isEmpty()) {\r
-            return false;\r
-        } else {\r
-            return true;\r
-        }\r
-    }\r
-\r
-    /**\r
-     * Closes the connection to database.\r
-     *\r
-     * @throws PfModelException the PfModel parsing exception\r
-     */\r
-    private void close() throws PfModelException {\r
-\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
-    }\r
 }\r