gerrit.onap Code Review - sdc.git/blob

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