[SDC-29] Amdocs OnBoard 1707 initial commit.
[sdc.git] / openecomp-be / lib / openecomp-sdc-versioning-lib / openecomp-sdc-versioning-core / src / main / java / org / openecomp / sdc / versioning / dao / impl / VersionHistoryCassandraDaoImpl.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.core.UDTValue;
24 import com.datastax.driver.mapping.Mapper;
25 import com.datastax.driver.mapping.Result;
26 import com.datastax.driver.mapping.UDTMapper;
27 import com.datastax.driver.mapping.annotations.Accessor;
28 import com.datastax.driver.mapping.annotations.Query;
29 import org.openecomp.core.dao.impl.CassandraBaseDao;
30 import org.openecomp.core.nosqldb.api.NoSqlDb;
31 import org.openecomp.core.nosqldb.factory.NoSqlDbFactory;
32 import org.openecomp.sdc.versioning.dao.VersionHistoryDao;
33 import org.openecomp.sdc.versioning.dao.types.VersionHistoryEntity;
34 import org.openecomp.sdc.versioning.dao.types.VersionableEntityId;
35
36 import java.util.Collection;
37
38 public class VersionHistoryCassandraDaoImpl extends CassandraBaseDao<VersionHistoryEntity>
39     implements
40     VersionHistoryDao {
41
42   private static NoSqlDb noSqlDb = NoSqlDbFactory.getInstance().createInterface();
43   private static Mapper<VersionHistoryEntity> mapper =
44       noSqlDb.getMappingManager().mapper(VersionHistoryEntity.class);
45   private static VersionHistoryAccessor accessor =
46       noSqlDb.getMappingManager().createAccessor(VersionHistoryAccessor.class);
47   private static UDTMapper<VersionableEntityId> versionedEntityIdMapper =
48       noSqlDb.getMappingManager().udtMapper(VersionableEntityId.class);
49
50   @Override
51   protected Mapper<VersionHistoryEntity> getMapper() {
52     return mapper;
53   }
54
55   @Override
56   protected Object[] getKeys(VersionHistoryEntity entity) {
57     return new Object[]{versionedEntityIdMapper.toUDT(entity.getEntityId())};
58   }
59
60   @Override
61   public Collection<VersionHistoryEntity> list(VersionHistoryEntity entity) {
62     return accessor.getAll(versionedEntityIdMapper.toUDT(entity.getEntityId())).all();
63   }
64
65   @Accessor
66   interface VersionHistoryAccessor {
67     @Query("select * from version_history where entity_id=?")
68     Result<VersionHistoryEntity> getAll(UDTValue entityId);
69   }
70 }