1 files changed, 29 insertions, 24 deletions

diff --git a/openecomp-be/lib/openecomp-sdc-model-lib/openecomp-sdc-model-impl/src/main/java/org/openecomp/sdc/model/impl/ServiceTemplateDaoCassandraImpl.java b/openecomp-be/lib/openecomp-sdc-model-lib/openecomp-sdc-model-impl/src/main/java/org/openecomp/sdc/model/impl/ServiceTemplateDaoCassandraImpl.java
index ffc9a24677..71334ce8d3 100644
--- a/openecomp-be/lib/openecomp-sdc-model-lib/openecomp-sdc-model-impl/src/main/java/org/openecomp/sdc/model/impl/ServiceTemplateDaoCassandraImpl.java
+++ b/openecomp-be/lib/openecomp-sdc-model-lib/openecomp-sdc-model-impl/src/main/java/org/openecomp/sdc/model/impl/ServiceTemplateDaoCassandraImpl.java
@@ -20,6 +20,7 @@
 
 package org.openecomp.sdc.model.impl;
 
+import com.datastax.driver.core.ResultSet;
 import com.datastax.driver.core.UDTValue;
 import com.datastax.driver.mapping.Mapper;
 import com.datastax.driver.mapping.Result;
@@ -41,33 +42,24 @@ import java.util.stream.Collectors;
 
 public class ServiceTemplateDaoCassandraImpl implements ServiceTemplateDao {
 
-
   private static final NoSqlDb noSqlDb = NoSqlDbFactory.getInstance().createInterface();
-  private static final Mapper<ServiceTemplateEntity> mapper =
-      noSqlDb.getMappingManager().mapper(ServiceTemplateEntity.class);
+  private static final Mapper<ServiceTemplateEntity> mapper = noSqlDb.getMappingManager().mapper(
+      ServiceTemplateEntity.class);
   private static final VspServiceTemplateAccessor accessor =
-      noSqlDb.getMappingManager().createAccessor(VspServiceTemplateAccessor.class);
+      noSqlDb.getMappingManager().createAccessor(
+          VspServiceTemplateAccessor.class);
   private static final UDTMapper<Version> versionMapper =
       noSqlDb.getMappingManager().udtMapper(Version.class);
 
   @Override
   public void registerVersioning(String versionableEntityType) {
-    VersioningManagerFactory.getInstance().createInterface()
-        .register(versionableEntityType, new VersionableEntityMetadata(
-            mapper.getTableMetadata().getName(),
+    VersioningManagerFactory.getInstance().createInterface().register(versionableEntityType,
+        new VersionableEntityMetadata(mapper.getTableMetadata().getName(),
             mapper.getTableMetadata().getPartitionKey().get(0).getName(),
             mapper.getTableMetadata().getPartitionKey().get(1).getName()));
   }
 
   @Override
-  public Collection<ServiceTemplate> list(String vspId, Version version) {
-
-    List<ServiceTemplateEntity> entityList = accessor.list(vspId, version).all();
-    return entityList.stream().map(entity -> entity.getServiceTemplate())
-        .collect(Collectors.toList());
-  }
-
-  @Override
   public void create(ServiceTemplate entity) {
     ServiceTemplateEntity vspServiceTemplateEntity = new ServiceTemplateEntity(entity);
     mapper.save(vspServiceTemplateEntity);
@@ -89,7 +81,6 @@ public class ServiceTemplateDaoCassandraImpl implements ServiceTemplateDao {
     mapper.delete(vspId, version);
   }
 
-
   @Override
   public Object[] getKeys(String vspId, Version version) {
     return new Object[]{vspId, versionMapper.toUDT(version)};
@@ -98,7 +89,8 @@ public class ServiceTemplateDaoCassandraImpl implements ServiceTemplateDao {
   @Override
   public ServiceTemplate getTemplateInfo(String vspId, Version version, String name) {
     ServiceTemplateEntity serviceTemplateEntity =
-        accessor.getTemplateInfo(vspId, versionMapper.toUDT(version), name).one();
+        accessor.getTemplateInfo(vspId, versionMapper.toUDT(version),
+            name).one();
     if (serviceTemplateEntity == null) {
       return null;
     }
@@ -106,6 +98,14 @@ public class ServiceTemplateDaoCassandraImpl implements ServiceTemplateDao {
   }
 
   @Override
+  public Collection<ServiceTemplate> list(String vspId, Version version) {
+
+    List<ServiceTemplateEntity> entityList = accessor.list(vspId, version).all();
+    return entityList.stream().map(entity -> entity.getServiceTemplate())
+        .collect(Collectors.toList());
+  }
+
+  @Override
   public String getBase(String vspId, Version version) {
     Result<ServiceTemplateEntity> element = accessor.getBase(vspId, versionMapper.toUDT(version));
     if (element != null) {
@@ -117,6 +117,10 @@ public class ServiceTemplateDaoCassandraImpl implements ServiceTemplateDao {
     return null;
   }
 
+  @Override
+  public void deleteAll(String vspId, Version version) {
+    accessor.deleteAll(vspId, versionMapper.toUDT(version));
+  }
 
   @Accessor
   interface VspServiceTemplateAccessor {
@@ -126,20 +130,21 @@ public class ServiceTemplateDaoCassandraImpl implements ServiceTemplateDao {
 
     @Query(
         "SELECT vsp_id, version, name, base_name ,content_data"
-                + " FROM vsp_service_template where vsp_id=? and version=?")
+            + " FROM vsp_service_template where vsp_id=? and version=?")
     Result<ServiceTemplateEntity> list(String vspId, Version version);
 
-
     @Query(
-        "SELECT vsp_id, version, name, base_name ,content_data "
-                + "FROM vsp_service_template where vsp_id=? and version=? and name=?")
+        "SELECT vsp_id, version, name, base_name ,content_data"
+            + " FROM vsp_service_template where vsp_id=? and version=? and name=?")
     Result<ServiceTemplateEntity> getTemplateInfo(String vspId, UDTValue version, String name);
 
     @Query(
-        "SELECT vsp_id, version, name, base_name  "
-                + "FROM vsp_service_template where vsp_id=? and version=?")
+        "SELECT vsp_id, version, name, base_name"
+            + "  FROM vsp_service_template where vsp_id=? and version=?")
     Result<ServiceTemplateEntity> getBase(String vspId, UDTValue version);
-  }
 
+    @Query("DELETE FROM vsp_service_template where vsp_id = ? and version = ?")
+    ResultSet deleteAll(String vspId, UDTValue version);
+  }
 
 }