push addional code
[sdc.git] / openecomp-be / lib / openecomp-sdc-vendor-license-lib / openecomp-sdc-vendor-license-core / src / main / java / org / openecomp / sdc / vendorlicense / dao / impl / VendorLicenseModelCassandraDaoImpl.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.vendorlicense.dao.impl;
22
23 import com.datastax.driver.mapping.Mapper;
24 import com.datastax.driver.mapping.Result;
25 import com.datastax.driver.mapping.UDTMapper;
26 import com.datastax.driver.mapping.annotations.Accessor;
27 import com.datastax.driver.mapping.annotations.Query;
28 import org.openecomp.core.dao.impl.CassandraBaseDao;
29 import org.openecomp.core.nosqldb.api.NoSqlDb;
30 import org.openecomp.core.nosqldb.factory.NoSqlDbFactory;
31 import org.openecomp.sdc.vendorlicense.dao.VendorLicenseModelDao;
32 import org.openecomp.sdc.vendorlicense.dao.types.VendorLicenseModelEntity;
33 import org.openecomp.sdc.versioning.VersioningManagerFactory;
34 import org.openecomp.sdc.versioning.dao.types.Version;
35 import org.openecomp.sdc.versioning.types.VersionableEntityMetadata;
36
37 import java.util.Collection;
38
39 public class VendorLicenseModelCassandraDaoImpl extends CassandraBaseDao<VendorLicenseModelEntity>
40     implements VendorLicenseModelDao {
41
42   private static NoSqlDb noSqlDb = NoSqlDbFactory.getInstance().createInterface();
43   private static Mapper<VendorLicenseModelEntity> mapper =
44       noSqlDb.getMappingManager().mapper(VendorLicenseModelEntity.class);
45   private static VendorLicenseModelAccessor accessor =
46       noSqlDb.getMappingManager().createAccessor(VendorLicenseModelAccessor.class);
47   private static UDTMapper<Version> versionMapper =
48       noSqlDb.getMappingManager().udtMapper(Version.class);
49
50   @Override
51   public void registerVersioning(String versionableEntityType) {
52     VersioningManagerFactory.getInstance().createInterface()
53         .register(versionableEntityType, new VersionableEntityMetadata(
54             mapper.getTableMetadata().getName(),
55             mapper.getTableMetadata().getPartitionKey().get(0).getName(),
56             mapper.getTableMetadata().getPartitionKey().get(1).getName()));
57   }
58
59   @Override
60   protected Mapper<VendorLicenseModelEntity> getMapper() {
61     return mapper;
62   }
63
64   @Override
65   protected Object[] getKeys(VendorLicenseModelEntity entity) {
66     return new Object[]{entity.getId(), versionMapper.toUDT(entity.getVersion())};
67   }
68
69   @Override
70   public Collection<VendorLicenseModelEntity> list(VendorLicenseModelEntity vendorLicenseModel) {
71     return accessor.getAll().all();
72   }
73
74   //    @Override
75   //    public void updateLastModificationTime(VendorLicenseModelEntity vendorLicenseModel){
76   //        accessor.updateLastModificationTime(vendorLicenseModel.getLastModificationTime(),
77   // vendorLicenseModel.getId(), versionMapper.toUDT(vendorLicenseModel.getVersion()));
78   //    }
79
80   @Accessor
81   interface VendorLicenseModelAccessor {
82
83     @Query("SELECT * FROM vendor_license_model")
84     Result<VendorLicenseModelEntity> getAll();
85
86     //        @Query("UPDATE vendor_license_model set last_modification_time
87     // = ? where vlm_id = ? and version = ?")
88     //        ResultSet updateLastModificationTime(Date lastModificationTime,
89     // String vlmId, UDTValue version);
90   }
91 }