openecomp-be/lib/openecomp-sdc-vendor-software-product-lib/openecomp-sdc-vendor-software-product-core/src/main/java/org/openecomp/sdc/vendorsoftwareproduct/dao/impl/DeploymentFlavorDaoCassandraImpl.java

   1 package org.openecomp.sdc.vendorsoftwareproduct.dao.impl;
   2
   3 import com.datastax.driver.core.ResultSet;
   4 import com.datastax.driver.core.UDTValue;
   5 import com.datastax.driver.mapping.Mapper;
   6 import com.datastax.driver.mapping.Result;
   7 import com.datastax.driver.mapping.UDTMapper;
   8 import com.datastax.driver.mapping.annotations.Accessor;
   9 import com.datastax.driver.mapping.annotations.Query;
  10 import org.openecomp.core.dao.impl.CassandraBaseDao;
  11 import org.openecomp.core.nosqldb.api.NoSqlDb;
  12 import org.openecomp.core.nosqldb.factory.NoSqlDbFactory;
  13 import org.openecomp.sdc.vendorsoftwareproduct.VendorSoftwareProductConstants;
  14 import org.openecomp.sdc.vendorsoftwareproduct.dao.DeploymentFlavorDao;
  15 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.DeploymentFlavorEntity;
  16 import org.openecomp.sdc.versioning.VersioningManagerFactory;
  17 import org.openecomp.sdc.versioning.dao.types.Version;
  18 import org.openecomp.sdc.versioning.types.UniqueValueMetadata;
  19 import org.openecomp.sdc.versioning.types.VersionableEntityMetadata;
  20
  21 import java.util.Arrays;
  22 import java.util.Collection;
  23 import java.util.Collections;
  24
  25
  26 public class DeploymentFlavorDaoCassandraImpl extends CassandraBaseDao<DeploymentFlavorEntity> implements DeploymentFlavorDao {
  27     private static final NoSqlDb noSqlDb = NoSqlDbFactory.getInstance().createInterface();
  28     private static final Mapper<DeploymentFlavorEntity> mapper = noSqlDb.getMappingManager().mapper(DeploymentFlavorEntity.class);
  29     private static final DeploymentFlavorAccessor accessor =
  30             noSqlDb.getMappingManager().createAccessor(DeploymentFlavorAccessor.class);
  31     private static final UDTMapper<Version> versionMapper =
  32             noSqlDb.getMappingManager().udtMapper(Version.class);
  33     @Override
  34     protected Mapper<DeploymentFlavorEntity> getMapper() {
  35         return mapper;
  36     }
  37
  38     @Override
  39     protected Object[] getKeys(DeploymentFlavorEntity entity) {
  40         return new Object[]{entity.getVspId(), versionMapper.toUDT(entity.getVersion()),
  41                 entity.getId()};
  42     }
  43
  44     @Override
  45     public Collection<DeploymentFlavorEntity> list(DeploymentFlavorEntity entity) {
  46         return accessor.list(entity.getVspId(), versionMapper.toUDT(entity.getVersion())).all();
  47     }
  48
  49     @Override
  50     public void update(DeploymentFlavorEntity entity) {
  51         accessor.updateCompositionData(entity.getVspId(), versionMapper.toUDT(entity.getVersion()),
  52             entity.getId(), entity.getCompositionData());
  53     }
  54
  55     @Override
  56     public void registerVersioning(String versionableEntityType) {
  57         VersionableEntityMetadata metadata = new VersionableEntityMetadata(
  58                 mapper.getTableMetadata().getName(),
  59                 mapper.getTableMetadata().getPartitionKey().get(0).getName(),
  60                 mapper.getTableMetadata().getPartitionKey().get(1).getName());
  61
  62
  63         metadata.setUniqueValuesMetadata(Collections.singletonList(new UniqueValueMetadata(
  64                 VendorSoftwareProductConstants.UniqueValues.DEPLOYMENT_FLAVOR_NAME,
  65                 Arrays.asList(mapper.getTableMetadata().getPartitionKey().get(0).getName(),
  66                         mapper.getTableMetadata().getPartitionKey().get(1).getName(), "name"))));
  67
  68         VersioningManagerFactory.getInstance().createInterface()
  69                 .register(versionableEntityType, metadata);
  70     }
  71
  72     @Override
  73     public void deleteAll(String vspId, Version version) {
  74         accessor.deleteAll(vspId, version);
  75     }
  76
  77     @Accessor
  78     interface DeploymentFlavorAccessor {
  79         @Query(
  80                 "select vsp_id, version, deployment_flavor_id, composition_data from vsp_deployment_flavor where vsp_id=?"
  81                         + " and version=?")
  82         Result<DeploymentFlavorEntity> list(String vspId, UDTValue version);
  83
  84         @Query(
  85                 "insert into vsp_deployment_flavor (vsp_id, version, deployment_flavor_id, composition_data) values (?,?,?,?)")
  86         ResultSet updateCompositionData(String vspId, UDTValue version, String id,
  87                                         String compositionData);
  88
  89         @Query("delete from vsp_deployment_flavor where vsp_id=? and version=?")
  90         ResultSet deleteAll(String vspId, Version version);
  91
  92     }
  93 }