X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Forg%2Fonap%2Fclamp%2Floop%2Ftemplate%2FPolicyModelsService.java;h=0a09dd8d79d4a2de0c6083051dd3f6e99c074cb5;hb=3a83e2a2ff88ef49535973df8dc77dc8015170da;hp=8e22852a96a773e732a66f93b778c7247a3e0bef;hpb=29e3cd1afbd3202a0c16ef6a4057662942ddefa2;p=clamp.git

diff --git a/src/main/java/org/onap/clamp/loop/template/PolicyModelsService.java b/src/main/java/org/onap/clamp/loop/template/PolicyModelsService.java
index 8e22852a..0a09dd8d 100644
--- a/src/main/java/org/onap/clamp/loop/template/PolicyModelsService.java
+++ b/src/main/java/org/onap/clamp/loop/template/PolicyModelsService.java
@@ -2,7 +2,7 @@
  * ============LICENSE_START=======================================================
  * ONAP CLAMP
  * ================================================================================
- * Copyright (C) 2019 AT&T Intellectual Property. All rights
+ * Copyright (C) 2020 AT&T Intellectual Property. All rights
  *                             reserved.
  * ================================================================================
  * Licensed under the Apache License, Version 2.0 (the "License");
@@ -23,24 +23,62 @@
 
 package org.onap.clamp.loop.template;
 
+import com.google.gson.JsonObject;
 import java.util.List;
-
+import org.onap.clamp.clds.tosca.ToscaSchemaConstants;
+import org.onap.clamp.clds.tosca.ToscaYamlToJsonConvertor;
+import org.onap.clamp.util.SemanticVersioning;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 @Service
 public class PolicyModelsService {
     private final PolicyModelsRepository policyModelsRepository;
+    private ToscaYamlToJsonConvertor toscaYamlToJsonConvertor;
 
     @Autowired
-    public PolicyModelsService(PolicyModelsRepository policyModelrepo) {
+    public PolicyModelsService(PolicyModelsRepository policyModelrepo,
+        ToscaYamlToJsonConvertor convertor) {
         policyModelsRepository = policyModelrepo;
+        toscaYamlToJsonConvertor = convertor;
     }
 
     public PolicyModel saveOrUpdatePolicyModel(PolicyModel policyModel) {
         return policyModelsRepository.save(policyModel);
     }
 
+    /**
+     * Creates or updates the Tosca Policy Model.
+     *
+     * @param policyModelType
+     *        The policyModeltype in Tosca yaml
+     * @param policyModel
+     *        The Policymodel object
+     * @return The Policy Model
+     */
+    public PolicyModel saveOrUpdateByPolicyModelType(String policyModelType,
+        String policyModelTosca) {
+        JsonObject jsonObject = toscaYamlToJsonConvertor.validateAndConvertToJson(policyModelTosca);
+
+        String policyModelTypeName = toscaYamlToJsonConvertor.getValueFromMetadata(jsonObject,
+            ToscaSchemaConstants.METADATA_POLICY_MODEL_TYPE);
+        String acronym = toscaYamlToJsonConvertor.getValueFromMetadata(jsonObject,
+            ToscaSchemaConstants.METADATA_ACRONYM);
+        PolicyModel model = getPolicyModelByType(
+            policyModelTypeName != null ? policyModelTypeName : policyModelType);
+
+        if (model == null) {
+            model = new PolicyModel(policyModelTypeName, policyModelTosca,
+                SemanticVersioning.incrementMajorVersion(null), acronym);
+        } else {
+            model.setVersion(SemanticVersioning
+                .incrementMajorVersion(model.getVersion() != null ? model.getVersion() : null));
+            model.setPolicyModelType(policyModelTypeName);
+            model.setPolicyAcronym(acronym);
+        }
+        return saveOrUpdatePolicyModel(model);
+    }
+
     public List<String> getAllPolicyModelTypes() {
         return policyModelsRepository.getAllPolicyModelType();
     }
@@ -56,4 +94,20 @@ public class PolicyModelsService {
     public Iterable<PolicyModel> getAllPolicyModelsByType(String type) {
         return policyModelsRepository.findByPolicyModelType(type);
     }
+
+    public PolicyModel getPolicyModelByType(String type) {
+        List<PolicyModel> list = policyModelsRepository.findByPolicyModelType(type);
+        return list.stream().sorted().findFirst().orElse(null);
+    }
+
+    /**
+     * Retrieves the Tosca model Yaml string.
+     *
+     * @param type The PolicyModelType
+     * @return The Tosca model Yaml string
+     */
+    public String getPolicyModelTosca(String type) {
+        PolicyModel policyModel = getPolicyModelByType(type);
+        return policyModel != null ? policyModel.getPolicyModelTosca() : null;
+    }
 }