1 files changed, 106 insertions, 0 deletions
diff --git a/catalog-dao/src/main/java/org/openecomp/sdc/be/dao/cassandra/ToscaModelImportCassandraDao.java b/catalog-dao/src/main/java/org/openecomp/sdc/be/dao/cassandra/ToscaModelImportCassandraDao.java
new file mode 100644
index 0000000000..a8b1ec635d
--- /dev/null
+++ b/catalog-dao/src/main/java/org/openecomp/sdc/be/dao/cassandra/ToscaModelImportCassandraDao.java
@@ -0,0 +1,106 @@
+/*
+ * ============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());
+    }
+
+    public void importAll(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())
+        );
+    }
+
+    public List<ToscaImportByModel> findAllByModel(final String modelId) {
+        return toscaImportByModelAccessor.findAllByModel(modelId).all();
+    }
+
+}