asdctool/src/main/java/org/openecomp/sdc/asdctool/impl/validator/executers/ArtifactValidatorExecuter.java

   1 package org.openecomp.sdc.asdctool.impl.validator.executers;
   2
   3 import fj.data.Either;
   4 import java.io.BufferedWriter;
   5 import java.io.FileOutputStream;
   6 import java.io.IOException;
   7 import java.io.OutputStreamWriter;
   8 import java.io.Writer;
   9 import java.util.ArrayList;
  10 import java.util.Collection;
  11 import java.util.HashMap;
  12 import java.util.HashSet;
  13 import java.util.List;
  14 import java.util.Map;
  15 import java.util.Optional;
  16 import java.util.Set;
  17 import java.util.stream.Collectors;
  18
  19 import org.openecomp.sdc.asdctool.impl.validator.config.ValidationConfigManager;
  20 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
  21 import org.openecomp.sdc.be.dao.jsongraph.TitanDao;
  22 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
  23 import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
  24 import org.openecomp.sdc.be.datatypes.enums.GraphPropertyEnum;
  25 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
  26 import org.openecomp.sdc.be.model.ArtifactDefinition;
  27 import org.openecomp.sdc.be.model.Component;
  28 import org.openecomp.sdc.be.model.ComponentParametersView;
  29 import org.openecomp.sdc.be.model.jsontitan.operations.TopologyTemplateOperation;
  30 import org.openecomp.sdc.be.model.jsontitan.operations.ToscaOperationFacade;
  31 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  32 import org.springframework.beans.factory.annotation.Autowired;
  33 import org.slf4j.Logger;
  34 import org.slf4j.LoggerFactory;
  35
  36 public class ArtifactValidatorExecuter {
  37
  38     @Autowired
  39     protected TitanDao titanDao;
  40
  41     @Autowired
  42     private ToscaOperationFacade toscaOperationFacade;
  43     private static Logger log = LoggerFactory.getLogger(ArtifactValidatorExecuter.class.getName());
  44
  45     protected String name;
  46
  47     public void setName(String name) {
  48         this.name = name;
  49     }
  50
  51     public String getName() {
  52         return name;
  53     }
  54
  55     public Map<String, List<Component>> getVerticesToValidate(VertexTypeEnum type,
  56                                                               Map<GraphPropertyEnum, Object> hasProps) {
  57
  58         Map<String, List<Component>> result = new HashMap<>();
  59         Either<List<GraphVertex>, TitanOperationStatus> resultsEither = titanDao.getByCriteria(type, hasProps);
  60         if (resultsEither.isRight()) {
  61             System.out.println("getVerticesToValidate failed " + resultsEither.right().value());
  62             return result;
  63         }
  64         System.out.println("getVerticesToValidate: " + resultsEither.left().value().size() + " vertices to scan");
  65         List<GraphVertex> componentsList = resultsEither.left().value();
  66         componentsList.forEach(vertex -> {
  67             String ivariantUuid = (String)vertex.getMetadataProperty(GraphPropertyEnum.INVARIANT_UUID);
  68             if (!result.containsKey(ivariantUuid)) {
  69                 result.put(ivariantUuid, new ArrayList<>());
  70             }
  71             List<Component> compList = result.get(ivariantUuid);
  72
  73             ComponentParametersView filter = new ComponentParametersView(true);
  74             filter.setIgnoreArtifacts(false);
  75
  76             Either<Component, StorageOperationStatus> toscaElement
  77                 = toscaOperationFacade.getToscaElement(vertex.getUniqueId(), filter);
  78             if (toscaElement.isRight()) {
  79                 System.out.println("getVerticesToValidate: failed to find element"
  80                     + vertex.getUniqueId() + " staus is" + toscaElement.right().value());
  81             } else {
  82                 compList.add(toscaElement.left().value());
  83             }
  84         });
  85         return result;
  86     }
  87
  88     public boolean validate(Map<String, List<Component>> vertices) {
  89         boolean result = true;
  90         long time = System.currentTimeMillis();
  91         String fileName = ValidationConfigManager.getOutputFilePath() + this.getName() + "_" + time + ".csv";
  92
  93         try (Writer writer = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(fileName), "utf-8"))) {
  94             writer.write("name, UUID, invariantUUID, state, version\n");
  95             Collection<List<Component>> collection = vertices.values();
  96             for (List<Component> compList: collection) {
  97                 Set<String> artifactEsId = new HashSet<>();
  98                 for (Component component: compList) {
  99                     Map<String, ArtifactDefinition> toscaArtifacts = component.getToscaArtifacts();
 100                     Optional<ArtifactDefinition> op = toscaArtifacts.values()
 101                         .stream().filter(a -> artifactEsId.contains(a.getEsId())).findAny();
 102                     if (op.isPresent()) {
 103                         result = false;
 104                         writeModuleResultToFile(writer, compList);
 105                         writer.flush();
 106                         break;
 107                     } else {
 108                         artifactEsId.addAll(toscaArtifacts.values()
 109                             .stream().map(ArtifactDefinition::getEsId).collect(Collectors.toList()));
 110                     }
 111                 }
 112             }
 113         } catch (Exception e) {
 114             log.info("Failed to fetch vf resources ", e);
 115             return false;
 116         } finally {
 117             titanDao.commit();
 118         }
 119         return result;
 120     }
 121
 122     private void writeModuleResultToFile(Writer writer, List<Component> components) {
 123         try {
 124             // "service name, service id, state, version
 125             for (Component component: components ) {
 126                 StringBuilder sb = new StringBuilder(component.getName());
 127                 sb.append(",").append(component.getUniqueId())
 128                     .append(",").append(component.getInvariantUUID())
 129                     .append(",").append(component.getLifecycleState())
 130                     .append(",").append(component.getVersion())
 131                     .append("\n");
 132
 133                 writer.write(sb.toString());
 134             }
 135         } catch (IOException e) {
 136             log.debug("Cannot write module result to file", e);
 137         }
 138     }
 139 }