summaryrefslogtreecommitdiffstats
path: root/openecomp-be/lib/openecomp-sdc-versioning-lib/openecomp-sdc-versioning-core/src/main/java/org/openecomp/sdc/versioning/dao/impl/VersionInfoDaoImpl.java
blob: e9bb17efd281cc99c7e3ed25ef0285cf3b8ea183 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
/*
 * Copyright © 2016-2018 European Support Limited
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.openecomp.sdc.versioning.dao.impl;

import com.datastax.driver.extras.codecs.enums.EnumNameCodec;
import com.datastax.driver.mapping.Mapper;
import com.datastax.driver.mapping.Result;
import com.datastax.driver.mapping.annotations.Accessor;
import com.datastax.driver.mapping.annotations.Query;
import java.util.Collection;
import org.openecomp.core.dao.impl.CassandraBaseDao;
import org.openecomp.core.nosqldb.api.NoSqlDb;
import org.openecomp.sdc.versioning.dao.VersionInfoDao;
import org.openecomp.sdc.versioning.dao.types.VersionInfoEntity;
import org.openecomp.sdc.versioning.dao.types.VersionStatus;

public class VersionInfoDaoImpl extends CassandraBaseDao<VersionInfoEntity> implements VersionInfoDao {

    private final NoSqlDb noSqlDb;
    private final Mapper<VersionInfoEntity> mapper;
    private final VersionInfoAccessor accessor;

    public VersionInfoDaoImpl(NoSqlDb noSqlDb) {
        this.noSqlDb = noSqlDb;
        this.mapper = this.noSqlDb.getMappingManager().mapper(VersionInfoEntity.class);
        this.accessor = this.noSqlDb.getMappingManager().createAccessor(VersionInfoAccessor.class);
        this.noSqlDb.getMappingManager().getSession().getCluster().getConfiguration().getCodecRegistry()
            .register(new EnumNameCodec<>(VersionStatus.class));
    }

    @Override
    protected Mapper<VersionInfoEntity> getMapper() {
        return mapper;
    }

    @Override
    protected Object[] getKeys(VersionInfoEntity entity) {
        return new Object[]{entity.getEntityType(), entity.getEntityId()};
    }

    @Override
    public Collection<VersionInfoEntity> list(VersionInfoEntity entity) {
        return accessor.getAll(entity.getEntityType()).all();
    }

    @Accessor
    interface VersionInfoAccessor {

        @Query("select * from version_info where entity_type=?")
        Result<VersionInfoEntity> getAll(String entityType);
    }
}