gerrit.onap Code Review - sdc.git/blob

   1 /*
   2  * Copyright © 2016-2018 European Support Limited
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package org.openecomp.sdc.versioning.dao.impl;
  18
  19 import com.datastax.driver.mapping.Mapper;
  20 import com.datastax.driver.mapping.Result;
  21 import com.datastax.driver.mapping.annotations.Accessor;
  22 import com.datastax.driver.mapping.annotations.Query;
  23 import java.util.Collection;
  24 import org.openecomp.core.dao.impl.CassandraBaseDao;
  25 import org.openecomp.core.nosqldb.api.NoSqlDb;
  26 import org.openecomp.sdc.versioning.dao.VersionInfoDao;
  27 import org.openecomp.sdc.versioning.dao.types.VersionInfoEntity;
  28
  29 public class VersionInfoDaoImpl extends CassandraBaseDao<VersionInfoEntity>
  30     implements VersionInfoDao {
  31
  32
  33   private final NoSqlDb noSqlDb;
  34   private final Mapper<VersionInfoEntity> mapper;
  35   private final VersionInfoAccessor accessor;
  36
  37
  38   public VersionInfoDaoImpl(NoSqlDb noSqlDb) {
  39     this.noSqlDb = noSqlDb;
  40     this.mapper = this.noSqlDb.getMappingManager().mapper(VersionInfoEntity.class);
  41     this.accessor = this.noSqlDb.getMappingManager().createAccessor(VersionInfoAccessor.class);
  42   }
  43
  44   @Override
  45   protected Mapper<VersionInfoEntity> getMapper() {
  46     return mapper;
  47   }
  48
  49   @Override
  50   protected Object[] getKeys(VersionInfoEntity entity) {
  51     return new Object[]{entity.getEntityType(), entity.getEntityId()};
  52   }
  53
  54   @Override
  55   public Collection<VersionInfoEntity> list(VersionInfoEntity entity) {
  56     return accessor.getAll(entity.getEntityType()).all();
  57   }
  58
  59   @Accessor
  60   interface VersionInfoAccessor {
  61     @Query("select * from version_info where entity_type=?")
  62     Result<VersionInfoEntity> getAll(String entityType);
  63   }
  64 }