catalog-model/src/main/java/org/openecomp/sdc/be/model/operations/impl/ModelOperation.java

   1 /*
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2021 Nordix Foundation
   4  *  ================================================================================
   5  *  Licensed under the Apache License, Version 2.0 (the "License");
   6  *  you may not use this file except in compliance with the License.
   7  *  You may obtain a copy of the License at
   8  *
   9  *        http://www.apache.org/licenses/LICENSE-2.0
  10  *  Unless required by applicable law or agreed to in writing, software
  11  *  distributed under the License is distributed on an "AS IS" BASIS,
  12  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  *  See the License for the specific language governing permissions and
  14  *  limitations under the License.
  15  *
  16  *  SPDX-License-Identifier: Apache-2.0
  17  *  ============LICENSE_END=========================================================
  18  */
  19 package org.openecomp.sdc.be.model.operations.impl;
  20
  21 import fj.data.Either;
  22 import java.nio.charset.StandardCharsets;
  23 import java.util.EnumMap;
  24 import java.util.List;
  25 import java.util.Map;
  26 import java.util.Objects;
  27 import java.util.Optional;
  28 import java.util.stream.Collectors;
  29 import org.apache.commons.collections.MapUtils;
  30 import org.apache.commons.lang3.StringUtils;
  31 import org.openecomp.sdc.be.dao.api.ActionStatus;
  32 import org.openecomp.sdc.be.dao.cassandra.ToscaModelImportCassandraDao;
  33 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphGenericDao;
  34 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphOperationStatus;
  35 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
  36 import org.openecomp.sdc.be.dao.jsongraph.JanusGraphDao;
  37 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
  38 import org.openecomp.sdc.be.data.model.ToscaImportByModel;
  39 import org.openecomp.sdc.be.datatypes.enums.GraphPropertyEnum;
  40 import org.openecomp.sdc.be.model.Model;
  41 import org.openecomp.sdc.be.model.jsonjanusgraph.operations.exception.OperationException;
  42 import org.openecomp.sdc.be.model.jsonjanusgraph.operations.exception.ModelOperationExceptionSupplier;
  43 import org.openecomp.sdc.be.resources.data.ModelData;
  44 import org.openecomp.sdc.common.log.enums.EcompLoggerErrorCode;
  45 import org.openecomp.sdc.common.log.wrappers.Logger;
  46 import org.springframework.beans.factory.annotation.Autowired;
  47 import org.springframework.stereotype.Component;
  48
  49 @Component("model-operation")
  50 public class ModelOperation {
  51
  52     private static final Logger log = Logger.getLogger(ModelOperation.class);
  53
  54     private final JanusGraphGenericDao janusGraphGenericDao;
  55     private final JanusGraphDao janusGraphDao;
  56     private final ToscaModelImportCassandraDao toscaModelImportCassandraDao;
  57
  58     @Autowired
  59     public ModelOperation(final JanusGraphGenericDao janusGraphGenericDao,
  60                           final JanusGraphDao janusGraphDao,
  61                           final ToscaModelImportCassandraDao toscaModelImportCassandraDao) {
  62         this.janusGraphGenericDao = janusGraphGenericDao;
  63         this.janusGraphDao = janusGraphDao;
  64         this.toscaModelImportCassandraDao = toscaModelImportCassandraDao;
  65     }
  66
  67     public Model createModel(final Model model, final boolean inTransaction) {
  68         Model result = null;
  69         final var modelData = new ModelData(model.getName(), UniqueIdBuilder.buildModelUid(model.getName()));
  70         try {
  71             final Either<ModelData, JanusGraphOperationStatus> createNode = janusGraphGenericDao.createNode(modelData, ModelData.class);
  72             if (createNode.isRight()) {
  73                 final var janusGraphOperationStatus = createNode.right().value();
  74                 log.error(EcompLoggerErrorCode.DATA_ERROR, ModelOperation.class.getName(), "Problem while creating model, reason {}",
  75                     janusGraphOperationStatus);
  76                 if (janusGraphOperationStatus == JanusGraphOperationStatus.JANUSGRAPH_SCHEMA_VIOLATION) {
  77                     throw ModelOperationExceptionSupplier.modelAlreadyExists(model.getName()).get();
  78                 }
  79                 throw new OperationException(ActionStatus.GENERAL_ERROR,
  80                     String.format("Failed to create model %s on JanusGraph with %s error", model, janusGraphOperationStatus));
  81             }
  82             result = new Model(createNode.left().value().getName());
  83             return result;
  84         } finally {
  85             if (!inTransaction) {
  86                 if (Objects.nonNull(result)) {
  87                     janusGraphGenericDao.commit();
  88                 } else {
  89                     janusGraphGenericDao.rollback();
  90                 }
  91             }
  92         }
  93     }
  94
  95     public Optional<GraphVertex> findModelVertexByName(final String name) {
  96         if (StringUtils.isEmpty(name)) {
  97             return Optional.empty();
  98         }
  99         final Map<GraphPropertyEnum, Object> props = new EnumMap<>(GraphPropertyEnum.class);
 100         props.put(GraphPropertyEnum.NAME, name);
 101         props.put(GraphPropertyEnum.UNIQUE_ID, UniqueIdBuilder.buildModelUid(name));
 102         final Either<List<GraphVertex>, JanusGraphOperationStatus> result = janusGraphDao.getByCriteria(VertexTypeEnum.MODEL, props);
 103         if (result.isRight()) {
 104             final JanusGraphOperationStatus janusGraphOperationStatus = result.right().value();
 105             if (janusGraphOperationStatus == JanusGraphOperationStatus.NOT_FOUND) {
 106                 return Optional.empty();
 107             }
 108             log.error(EcompLoggerErrorCode.DATA_ERROR, this.getClass().getName(),
 109                 String.format("Problem while getting model %s. reason %s", name, janusGraphOperationStatus));
 110             throw new OperationException(ActionStatus.GENERAL_ERROR,
 111                 String.format("Failed to get model %s on JanusGraph with %s error", name, janusGraphOperationStatus));
 112         }
 113         return Optional.ofNullable(result.left().value().get(0));
 114     }
 115
 116     public Optional<Model> findModelByName(final String name) {
 117         if (StringUtils.isEmpty(name)) {
 118             return Optional.empty();
 119         }
 120         final Optional<GraphVertex> modelVertexOpt = findModelVertexByName(name);
 121         if (modelVertexOpt.isEmpty()) {
 122             return Optional.empty();
 123         }
 124
 125         final GraphVertex graphVertex = modelVertexOpt.get();
 126         final var model = new Model((String) graphVertex.getMetadataProperty(GraphPropertyEnum.NAME));
 127         return Optional.of(model);
 128     }
 129
 130     public void createModelImports(final String modelId, final Map<String, byte[]> zipContent) {
 131         if (MapUtils.isEmpty(zipContent)) {
 132             return;
 133         }
 134         final List<ToscaImportByModel> toscaImportByModelList = zipContent.entrySet().stream()
 135             .map(entry -> {
 136                 final String path = entry.getKey();
 137                 final byte[] bytes = entry.getValue();
 138                 final String content = new String(bytes, StandardCharsets.UTF_8);
 139                 final var toscaImportByModel = new ToscaImportByModel();
 140                 toscaImportByModel.setModelId(modelId);
 141                 toscaImportByModel.setFullPath(path);
 142                 toscaImportByModel.setContent(content);
 143                 return toscaImportByModel;
 144             }).collect(Collectors.toList());
 145         toscaModelImportCassandraDao.importAll(modelId, toscaImportByModelList);
 146     }
 147 }
 148
 149