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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
/*
* ============LICENSE_START=======================================================
* Copyright (C) 2021 Nordix Foundation
* ================================================================================
* 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.
*
* SPDX-License-Identifier: Apache-2.0
* ============LICENSE_END=========================================================
*/
package org.openecomp.sdc.be.dao.cassandra;
import static java.util.function.Predicate.not;
import com.datastax.driver.core.Session;
import com.datastax.driver.mapping.Mapper;
import com.datastax.driver.mapping.MappingManager;
import fj.data.Either;
import java.util.List;
import java.util.stream.Collectors;
import javax.annotation.PostConstruct;
import org.apache.commons.lang3.tuple.ImmutablePair;
import org.openecomp.sdc.be.dao.api.exception.CassandraDaoInitException;
import org.openecomp.sdc.be.dao.api.exception.CassandraDaoInitExceptionProvider;
import org.openecomp.sdc.be.data.model.ToscaImportByModel;
import org.openecomp.sdc.be.resources.data.auditing.AuditingTypesConstants;
import org.openecomp.sdc.common.log.enums.EcompLoggerErrorCode;
import org.openecomp.sdc.common.log.wrappers.Logger;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
@Component("tosca-model-import-cassandra-dao")
public class ToscaModelImportCassandraDao extends CassandraDao {
private static final Logger LOGGER = Logger.getLogger(ToscaModelImportCassandraDao.class.getName());
private ToscaImportByModelAccessor toscaImportByModelAccessor;
private Mapper<ToscaImportByModel> toscaImportByModelMapper;
@Autowired
public ToscaModelImportCassandraDao(final CassandraClient cassandraClient) {
super(cassandraClient);
}
/**
* For test purposes.
*
* @param toscaImportByModelAccessor the sdcartifact.tosca_import_by_model accessor
*/
ToscaModelImportCassandraDao(final ToscaImportByModelAccessor toscaImportByModelAccessor,
final Mapper<ToscaImportByModel> toscaImportByModelMapper) {
super(null);
this.toscaImportByModelAccessor = toscaImportByModelAccessor;
this.toscaImportByModelMapper = toscaImportByModelMapper;
}
@PostConstruct
public void init() {
final var keyspace = AuditingTypesConstants.ARTIFACT_KEYSPACE;
if (!client.isConnected()) {
LOGGER.error(EcompLoggerErrorCode.SCHEMA_ERROR, ToscaModelImportCassandraDao.class.getName(), "Cassandra client isn't connected");
return;
}
final Either<ImmutablePair<Session, MappingManager>, CassandraOperationStatus> connectionResult = client.connect(keyspace);
if (connectionResult.isRight()) {
final CassandraDaoInitException exception =
CassandraDaoInitExceptionProvider.keySpaceConnectError(keyspace, connectionResult.right().value()).get();
LOGGER.error(EcompLoggerErrorCode.SCHEMA_ERROR, ToscaModelImportCassandraDao.class.getName(), exception.getMessage());
throw exception;
}
session = connectionResult.left().value().getLeft();
manager = connectionResult.left().value().getRight();
toscaImportByModelMapper = manager.mapper(ToscaImportByModel.class);
toscaImportByModelAccessor = manager.createAccessor(ToscaImportByModelAccessor.class);
LOGGER.info("{} successfully initialized", ToscaModelImportCassandraDao.class.getName());
}
/**
* Completely replaces the previous model imports by the imports on the given list that are from the same model.
* New imports will be added, existing will be replaced and the remaining will be deleted.
*
* @param modelId the model id
* @param toscaImportByModelList the new list of imports
*/
public void replaceImports(final String modelId, final List<ToscaImportByModel> toscaImportByModelList) {
final List<ToscaImportByModel> importOfModelList = toscaImportByModelList.stream()
.filter(toscaImportByModel -> modelId.equals(toscaImportByModel.getModelId()))
.collect(Collectors.toList());
final List<ToscaImportByModel> actualImportOfModelList = toscaImportByModelAccessor.findAllByModel(modelId).all();
final List<ToscaImportByModel> removedImportList = actualImportOfModelList.stream()
.filter(not(importOfModelList::contains))
.collect(Collectors.toList());
importOfModelList.forEach(toscaImportByModelMapper::save);
removedImportList.forEach(toscaImportByModel ->
toscaImportByModelMapper.delete(toscaImportByModel.getModelId(), toscaImportByModel.getFullPath())
);
}
/**
* Saves all imports provided on the list that are from the given modelId.
*
* @param modelId the model id
* @param toscaImportByModelList the list of imports to save
*/
public void saveAll(final String modelId, final List<ToscaImportByModel> toscaImportByModelList) {
toscaImportByModelList.stream()
.filter(toscaImportByModel -> modelId.equals(toscaImportByModel.getModelId()))
.forEach(toscaImportByModelMapper::save);
}
public List<ToscaImportByModel> findAllByModel(final String modelId) {
return toscaImportByModelAccessor.findAllByModel(modelId).all();
}
public void deleteAllByModel(final String modelId) {
final List<ToscaImportByModel> allByModel = findAllByModel(modelId);
allByModel.forEach(toscaImportByModelMapper::delete);
}
}
|
