diff options
Diffstat (limited to 'openecomp-be/lib/openecomp-sdc-model-lib/openecomp-sdc-model-impl/src/main/java/org/openecomp/sdc/model/impl/EnrichedServiceArtifactDaoCassandraImpl.java')
-rw-r--r-- | openecomp-be/lib/openecomp-sdc-model-lib/openecomp-sdc-model-impl/src/main/java/org/openecomp/sdc/model/impl/EnrichedServiceArtifactDaoCassandraImpl.java | 137 |
1 files changed, 137 insertions, 0 deletions
diff --git a/openecomp-be/lib/openecomp-sdc-model-lib/openecomp-sdc-model-impl/src/main/java/org/openecomp/sdc/model/impl/EnrichedServiceArtifactDaoCassandraImpl.java b/openecomp-be/lib/openecomp-sdc-model-lib/openecomp-sdc-model-impl/src/main/java/org/openecomp/sdc/model/impl/EnrichedServiceArtifactDaoCassandraImpl.java new file mode 100644 index 0000000000..d9ebe849e3 --- /dev/null +++ b/openecomp-be/lib/openecomp-sdc-model-lib/openecomp-sdc-model-impl/src/main/java/org/openecomp/sdc/model/impl/EnrichedServiceArtifactDaoCassandraImpl.java @@ -0,0 +1,137 @@ +/*- + * ============LICENSE_START======================================================= + * SDC + * ================================================================================ + * Copyright (C) 2017 AT&T Intellectual Property. 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. + * ============LICENSE_END========================================================= + */ + +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; +import com.datastax.driver.mapping.UDTMapper; +import com.datastax.driver.mapping.annotations.Accessor; +import com.datastax.driver.mapping.annotations.Query; +import org.openecomp.core.model.dao.EnrichedServiceArtifactDao; +import org.openecomp.core.model.types.EnrichedServiceArtifactEntity; +import org.openecomp.core.model.types.ServiceArtifact; +import org.openecomp.core.nosqldb.api.NoSqlDb; +import org.openecomp.core.nosqldb.factory.NoSqlDbFactory; +import org.openecomp.sdc.versioning.VersioningManagerFactory; +import org.openecomp.sdc.versioning.dao.types.Version; +import org.openecomp.sdc.versioning.types.VersionableEntityMetadata; + +import java.util.Collection; +import java.util.List; +import java.util.stream.Collectors; + +public class EnrichedServiceArtifactDaoCassandraImpl implements EnrichedServiceArtifactDao { + + private static final NoSqlDb noSqlDb = NoSqlDbFactory.getInstance().createInterface(); + private static final Mapper<EnrichedServiceArtifactEntity> mapper = + noSqlDb.getMappingManager().mapper(EnrichedServiceArtifactEntity.class); + private static final VspServiceArtifactAccessor accessor = + noSqlDb.getMappingManager().createAccessor(VspServiceArtifactAccessor.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(), + mapper.getTableMetadata().getPartitionKey().get(0).getName(), + mapper.getTableMetadata().getPartitionKey().get(1).getName())); + } + + @Override + public Collection<ServiceArtifact> list(String vspId, Version version) { + List<EnrichedServiceArtifactEntity> entityList; + if (vspId != null && version != null) { + entityList = accessor.list(vspId, versionMapper.toUDT(version)).all(); + } else { + entityList = accessor.listAll().all(); + } + + return entityList.stream().map(entity -> entity.getServiceArtifact()) + .collect(Collectors.toList()); + } + + @Override + public void create(ServiceArtifact entity) { + EnrichedServiceArtifactEntity vspEnrichedServiceArtifactEntity = + new EnrichedServiceArtifactEntity(entity); + mapper.save(vspEnrichedServiceArtifactEntity); + } + + @Override + public void update(ServiceArtifact entity) { + EnrichedServiceArtifactEntity vspEnrichedServiceArtifactEntity = + new EnrichedServiceArtifactEntity(entity); + mapper.save(vspEnrichedServiceArtifactEntity); + } + + @Override + public ServiceArtifact get(String vspId, Version version) { + return mapper.get(getKeys(vspId, version)).getServiceArtifact(); + } + + @Override + public void delete(String vspId, Version version) { + accessor.delete(vspId, versionMapper.toUDT(version)); + } + + @Override + public Object[] getKeys(String vspId, Version version) { + return new Object[]{vspId, versionMapper.toUDT(version)}; + } + + @Override + public ServiceArtifact getArtifactInfo(String vspId, Version version, String name) { + EnrichedServiceArtifactEntity enrichedServiceArtifactEntity = + accessor.getArtifactInfo(vspId, versionMapper.toUDT(version), name).one(); + if (enrichedServiceArtifactEntity == null) { + return null; + } + + return enrichedServiceArtifactEntity.getServiceArtifact(); + } + + + @Accessor + interface VspServiceArtifactAccessor { + + @Query("SELECT vsp_id, version, name ,content_data FROM vsp_enriched_service_artifact") + Result<EnrichedServiceArtifactEntity> listAll(); + + @Query( + "SELECT vsp_id, version, name ,content_data FROM vsp_enriched_service_artifact " + + "where vsp_id=? and version=? ") + Result<EnrichedServiceArtifactEntity> list(String vspId, UDTValue version); + + + @Query( + "SELECT vsp_id,version,name,content_data FROM vsp_enriched_service_artifact " + + "where vsp_id=? and version=? and name=?") + Result<EnrichedServiceArtifactEntity> getArtifactInfo(String vspId, UDTValue version, + String name); + + @Query("DELETE from vsp_enriched_service_artifact where vsp_id=? and version=?") + ResultSet delete(String vspId, UDTValue version); + } +} |