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