Add types to imports

Change-Id: Ifa1dea17841d8667044aff4e7a9a17b2a75b78e4
Signed-off-by: Vasyl Razinkov <vasyl.razinkov@est.tech>
Issue-ID: SDC-3684

1 files changed, 6 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
index a8b1ec635d..5d1501c0b5 100644
--- 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
@@ -99,6 +99,12 @@ public class ToscaModelImportCassandraDao extends CassandraDao {
         );
     }
 
+    public void importOnly(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();
     }