1 files changed, 152 insertions, 0 deletions
diff --git a/main/src/main/java/org/onap/policy/pap/main/service/PolicyStatusService.java b/main/src/main/java/org/onap/policy/pap/main/service/PolicyStatusService.java
new file mode 100644
index 00000000..09870314
--- /dev/null
+++ b/main/src/main/java/org/onap/policy/pap/main/service/PolicyStatusService.java
@@ -0,0 +1,152 @@
+/*-
+ * ============LICENSE_START=======================================================
+ *  Copyright (C) 2022 Bell Canada. 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.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.pap.main.service;
+
+import java.util.Collection;
+import java.util.Collections;
+import java.util.List;
+import java.util.stream.Collectors;
+import javax.ws.rs.core.Response;
+import lombok.NonNull;
+import lombok.RequiredArgsConstructor;
+import org.onap.policy.common.parameters.BeanValidationResult;
+import org.onap.policy.models.base.PfModelRuntimeException;
+import org.onap.policy.models.pdp.concepts.PdpPolicyStatus;
+import org.onap.policy.models.pdp.persistence.concepts.JpaPdpPolicyStatus;
+import org.onap.policy.models.tosca.authorative.concepts.ToscaConceptIdentifierOptVersion;
+import org.onap.policy.pap.main.repository.PolicyStatusRepository;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+@Service
+@Transactional
+@RequiredArgsConstructor
+public class PolicyStatusService {
+
+    private final PolicyStatusRepository policyStatusRepository;
+
+    /**
+     * Gets all status for policies in a group.
+     *
+     * @param pdpGroup the group's name
+     * @return the policy status list found
+     */
+    public List<PdpPolicyStatus> getGroupPolicyStatus(@NonNull String pdpGroup) {
+        return asPolicyStatusList(policyStatusRepository.findByPdpGroup(pdpGroup));
+    }
+
+    /**
+     * Gets all status for policies.
+     *
+     * @return the policy status list found
+     */
+    public List<PdpPolicyStatus> getAllPolicyStatus() {
+        return asPolicyStatusList(policyStatusRepository.findAll());
+    }
+
+    /**
+     * Gets all status for a policy.
+     *
+     * @param policy the policy
+     * @return the policy status list found
+     */
+    public List<PdpPolicyStatus> getAllPolicyStatus(@NonNull ToscaConceptIdentifierOptVersion policy) {
+
+        if (policy.getVersion() != null) {
+            return asPolicyStatusList(policyStatusRepository
+                .findByKeyParentKeyNameAndKeyParentKeyVersion(policy.getName(), policy.getVersion()));
+
+        } else {
+            return asPolicyStatusList(policyStatusRepository.findByKeyParentKeyName(policy.getName()));
+        }
+    }
+
+    /**
+     * Gets all status for a policy in a group.
+     *
+     * @param pdpGroup the group's name
+     * @param policy the policy
+     * @return the policy status list found
+     */
+    public List<PdpPolicyStatus> getAllPolicyStatus(@NonNull String pdpGroup,
+        @NonNull ToscaConceptIdentifierOptVersion policy) {
+        if (policy.getVersion() != null) {
+            return asPolicyStatusList(policyStatusRepository.findByPdpGroupAndKeyParentKeyNameAndKeyParentKeyVersion(
+                pdpGroup, policy.getName(), policy.getVersion()));
+        } else {
+            return asPolicyStatusList(
+                policyStatusRepository.findByPdpGroupAndKeyParentKeyName(pdpGroup, policy.getName()));
+        }
+    }
+
+    /**
+     * Creates, updates, and deletes collections of policy status.
+     *
+     * @param createObjs the objects to create
+     * @param updateObjs the objects to update
+     * @param deleteObjs the objects to delete
+     */
+    public void cudPolicyStatus(Collection<PdpPolicyStatus> createObjs, Collection<PdpPolicyStatus> updateObjs,
+        Collection<PdpPolicyStatus> deleteObjs) {
+        try {
+            policyStatusRepository.deleteAll(fromAuthorativeStatus(deleteObjs, "deletePdpPolicyStatusList"));
+            policyStatusRepository.saveAll(fromAuthorativeStatus(createObjs, "createPdpPolicyStatusList"));
+            policyStatusRepository.saveAll(fromAuthorativeStatus(updateObjs, "updatePdpPolicyStatusList"));
+        } catch (Exception exc) {
+            throw new PfModelRuntimeException(Response.Status.BAD_REQUEST,
+                "Policy status operation failed." + exc.getMessage(), exc);
+        }
+    }
+
+    /**
+     * Converts a collection of authorative policy status to a collection of JPA policy
+     * status. Validates the resulting list.
+     *
+     * @param objs authorative policy status to convert
+     * @param fieldName name of the field containing the collection
+     * @return a list of JPA policy status
+     */
+    private List<JpaPdpPolicyStatus> fromAuthorativeStatus(Collection<PdpPolicyStatus> objs, String fieldName) {
+        if (objs == null) {
+            return Collections.emptyList();
+        }
+
+        List<JpaPdpPolicyStatus> jpas = objs.stream().map(JpaPdpPolicyStatus::new).collect(Collectors.toList());
+
+        // validate the objects
+        var result = new BeanValidationResult(fieldName, jpas);
+
+        var count = 0;
+        for (JpaPdpPolicyStatus jpa : jpas) {
+            result.addResult(jpa.validate(String.valueOf(count++)));
+        }
+
+        if (!result.isValid()) {
+            throw new PfModelRuntimeException(Response.Status.BAD_REQUEST, result.getResult());
+        }
+
+        return jpas;
+    }
+
+    private List<PdpPolicyStatus> asPolicyStatusList(List<JpaPdpPolicyStatus> jpaPdpPolicyStatusList) {
+        return jpaPdpPolicyStatusList.stream().map(JpaPdpPolicyStatus::toAuthorative).collect(Collectors.toList());
+    }
+}