push addional code
[sdc.git] / openecomp-be / lib / openecomp-core-lib / openecomp-nosqldb-lib / openecomp-nosqldb-core / src / main / java / org / openecomp / core / dao / impl / UniqueValueCassandraDaoImpl.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.core.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.UniqueValueDao;
28 import org.openecomp.core.dao.types.UniqueValueEntity;
29 import org.openecomp.core.nosqldb.api.NoSqlDb;
30 import org.openecomp.core.nosqldb.factory.NoSqlDbFactory;
31
32 import java.util.Collection;
33
34 public class UniqueValueCassandraDaoImpl extends CassandraBaseDao<UniqueValueEntity>
35     implements UniqueValueDao {
36
37   private static NoSqlDb noSqlDb = NoSqlDbFactory.getInstance().createInterface();
38   private static Mapper<UniqueValueEntity> mapper =
39       noSqlDb.getMappingManager().mapper(UniqueValueEntity.class);
40   private static UniqueValueAccessor accessor =
41       noSqlDb.getMappingManager().createAccessor(UniqueValueAccessor.class);
42
43
44   @Override
45   protected Mapper<UniqueValueEntity> getMapper() {
46     return mapper;
47   }
48
49   @Override
50   protected Object[] getKeys(UniqueValueEntity entity) {
51     return new Object[]{entity.getType(), entity.getValue()};
52   }
53
54   @Override
55   public Collection<UniqueValueEntity> list(UniqueValueEntity entity) {
56     return accessor.listAll().all();
57   }
58
59   @Accessor
60   interface UniqueValueAccessor {
61
62     @Query("select * from unique_value")
63     Result<UniqueValueEntity> listAll();
64   }
65 }