Upgrade SDC from Titan to Janus Graph
[sdc.git] / asdctool / src / main / java / org / openecomp / sdc / asdctool / impl / validator / executers / TopologyTemplateValidatorExecuter.java
1 package org.openecomp.sdc.asdctool.impl.validator.executers;
2
3 import fj.data.Either;
4 import org.openecomp.sdc.asdctool.impl.validator.tasks.TopologyTemplateValidationTask;
5 import org.openecomp.sdc.asdctool.impl.validator.utils.ReportManager;
6 import org.openecomp.sdc.asdctool.impl.validator.utils.VertexResult;
7 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
8 import org.openecomp.sdc.be.dao.jsongraph.JanusGraphDao;
9 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
10 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphOperationStatus;
11 import org.openecomp.sdc.be.datatypes.enums.ComponentTypeEnum;
12 import org.openecomp.sdc.be.datatypes.enums.GraphPropertyEnum;
13 import org.openecomp.sdc.be.datatypes.enums.ResourceTypeEnum;
14 import org.openecomp.sdc.common.log.wrappers.Logger;
15 import org.springframework.beans.factory.annotation.Autowired;
16
17 import java.util.*;
18
19 /**
20  * Created by chaya on 7/3/2017.
21  */
22 public class TopologyTemplateValidatorExecuter {
23
24     private static Logger log = Logger.getLogger(VfValidatorExecuter.class.getName());
25
26     @Autowired
27     protected JanusGraphDao janusGraphDao;
28
29     protected String name;
30
31     public void setName(String name) {
32         this.name = name;
33     }
34
35     public String getName() {
36         return name;
37     }
38
39     protected List<GraphVertex> getVerticesToValidate(ComponentTypeEnum type) {
40         Map<GraphPropertyEnum, Object> props = new EnumMap<>(GraphPropertyEnum.class);
41         props.put(GraphPropertyEnum.COMPONENT_TYPE, type.name());
42         if(type.equals(ComponentTypeEnum.RESOURCE)) {
43             props.put(GraphPropertyEnum.RESOURCE_TYPE, ResourceTypeEnum.VF);
44         }
45
46         Either<List<GraphVertex>, JanusGraphOperationStatus> results = janusGraphDao
47             .getByCriteria(VertexTypeEnum.TOPOLOGY_TEMPLATE, props);
48         if (results.isRight()) {
49             log.error("getVerticesToValidate failed "+ results.right().value());
50             return new ArrayList<>();
51         }
52         log.info("getVerticesToValidate: "+results.left().value().size()+" vertices to scan");
53         return results.left().value();
54     }
55
56     protected boolean validate(List<? extends TopologyTemplateValidationTask> tasks, List<GraphVertex> vertices) {
57         ReportManager.reportStartValidatorRun(getName(), vertices.size());
58         Set<String> failedTasks = new HashSet<>();
59         Set<String> successTasks = new HashSet<>();
60         boolean successAllVertices = true;
61         int vertexNum = 0;
62         int verticesSize = vertices.size();
63
64         for (GraphVertex vertex: vertices) {
65             vertexNum++;
66             boolean successAllTasks = true;
67             for (TopologyTemplateValidationTask task: tasks) {
68                 ReportManager.reportStartTaskRun(vertex, task.getTaskName());
69                 VertexResult result = task.validate(vertex);
70                 if (!result.getStatus()) {
71                     failedTasks.add(task.getTaskName());
72                     successAllVertices = false;
73                     successAllTasks = false;
74                 } else if (successAllTasks && vertexNum == verticesSize) {
75                     successTasks.add(task.getTaskName());
76                 }
77                 ReportManager.printValidationTaskStatus(vertex, task.getTaskName(), result.getStatus());
78                 ReportManager.reportTaskEnd(vertex.getUniqueId(), task.getTaskName(), result);
79             }
80             String componentScanStatus = successAllTasks? "success" : "failed";
81             log.info("Topology Template "+vertex.getUniqueId()+" Validation finished with "+componentScanStatus);
82         }
83         ReportManager.reportValidatorTypeSummary(getName(), failedTasks, successTasks);
84         return successAllVertices;
85     }
86 }