use of org.drools.impact.analysis.graph.Node in project drools by kiegroup.
the class GraphImageGenerator method convertGraph.
private guru.nidi.graphviz.model.Graph convertGraph(Graph g) {
guru.nidi.graphviz.model.Graph graph = graph(graphName).directed().graphAttr().with(Rank.dir(rankDir).sep(sep));
List<Node> nodeList = g.getNodeMap().values().stream().collect(Collectors.toList());
for (Node n : nodeList) {
guru.nidi.graphviz.model.Node node = node(n.getRuleName());
if (n.getStatus() == Node.Status.CHANGED) {
node = node.with(Color.RED, Style.FILLED);
} else if (n.getStatus() == Node.Status.IMPACTED) {
node = node.with(Color.YELLOW, Style.FILLED);
} else if (n.getStatus() == Node.Status.TARGET) {
node = node.with(Color.ORANGE, Style.FILLED);
} else if (n.getStatus() == Node.Status.IMPACTING) {
node = node.with(Color.LIGHTBLUE, Style.FILLED);
}
for (Link l : n.getOutgoingLinks()) {
if (!nodeList.contains(l.getTarget())) {
// a sub map may have a link to a node which doesn't exist in the sub map
continue;
}
Style<ForNodeLink> style;
if (l.getReactivityType() == ReactivityType.POSITIVE) {
style = Style.SOLID;
} else if (l.getReactivityType() == ReactivityType.NEGATIVE) {
style = Style.DASHED;
} else {
// UNKNOWN
style = Style.DOTTED;
}
node = node.link(to(node(l.getTarget().getRuleName())).with(style));
}
graph = graph.with(node);
}
return graph;
}
use of org.drools.impact.analysis.graph.Node in project drools by kiegroup.
the class AbstractGraphTest method assertLink.
/**
* Assert that there are exact links with the types between source node and target node.
* If no expectedTypes, it means there is no link
*/
protected void assertLink(Graph graph, String sourceFqdn, String targetFqdn, ReactivityType... expectedTypes) {
Node source = graph.getNodeMap().get(sourceFqdn);
Node target = graph.getNodeMap().get(targetFqdn);
List<Link> outgoingLinks = source.getOutgoingLinks().stream().filter(l -> l.getTarget().equals(target)).collect(Collectors.toList());
List<Link> incomingLinks = target.getIncomingLinks().stream().filter(l -> l.getSource().equals(source)).collect(Collectors.toList());
assertThat(outgoingLinks).hasSameElementsAs(incomingLinks);
List<ReactivityType> outgoingLinkTypelist = outgoingLinks.stream().map(l -> l.getReactivityType()).collect(Collectors.toList());
List<ReactivityType> expectedTypeList = Arrays.asList(expectedTypes);
assertThat(outgoingLinkTypelist).hasSameElementsAs(expectedTypeList);
}
use of org.drools.impact.analysis.graph.Node in project drools by kiegroup.
the class GraphCollapsionTest method testDrlRuleNamePrefix.
@Test
public void testDrlRuleNamePrefix() {
String str = "package mypkg;\n" + "import " + Order.class.getCanonicalName() + ";" + "\n" + "rule CustomerCheck_1\n" + " when\n" + " $o : Order(customerMembershipRank > 5)\n" + " then\n" + " modify($o) {\n" + " setDiscount(1000);\n" + " }\n" + "end\n" + "\n" + "rule CustomerCheck_2\n" + " when\n" + " $o : Order(customerAge > 60)\n" + " then\n" + " modify($o) {\n" + " setDiscount(2000);\n" + " }\n" + "end\n" + "\n" + "rule PriceCheck_1\n" + " when\n" + " $o : Order(itemPrice < 2000, discount >= 2000)\n" + " then\n" + " modify($o) {\n" + " setStatus(\"Too much discount\");\n" + " }\n" + "end\n" + "\n" + "rule PriceCheck_2\n" + " when\n" + " $o : Order(itemPrice > 5000)\n" + " then\n" + " modify($o) {\n" + " setStatus(\"Exclusive order\");\n" + " }\n" + "end\n" + "\n" + "rule StatusCheck_1\n" + " when\n" + " $o : Order(status == \"Too much discount\")\n" + " then\n" + " modify($o) {\n" + " setDiscount(500);\n" + " }\n" + "end\n" + "\n" + "rule StatusCheck_2\n" + " when\n" + " Order(status == \"Exclusive order\")\n" + " then\n" + " // Do some work...\n" + "end";
AnalysisModel analysisModel = new ModelBuilder().build(str);
ModelToGraphConverter converter = new ModelToGraphConverter();
Graph graph = converter.toGraph(analysisModel);
Graph collapsedGraph = new GraphCollapsionHelper().collapseWithRuleNamePrefix(graph);
assertEquals(3, collapsedGraph.getNodeMap().size());
assertLink(collapsedGraph, "mypkg.CustomerCheck", "mypkg.PriceCheck", ReactivityType.POSITIVE, ReactivityType.NEGATIVE);
assertLink(collapsedGraph, "mypkg.PriceCheck", "mypkg.StatusCheck", ReactivityType.POSITIVE, ReactivityType.NEGATIVE);
assertLink(collapsedGraph, "mypkg.StatusCheck", "mypkg.PriceCheck", ReactivityType.NEGATIVE);
// --- impact analysis
// Assuming that "modify" action in PriceCheck_X is changed
// modify action in PriceCheck_X
Node changedNode = collapsedGraph.getNodeMap().get("mypkg.PriceCheck");
ImpactAnalysisHelper impactFilter = new ImpactAnalysisHelper();
Graph impactedSubGraph = impactFilter.filterImpactedNodes(collapsedGraph, changedNode);
assertNull(impactedSubGraph.getNodeMap().get("mypkg.CustomerCheck"));
assertEquals(Status.CHANGED, impactedSubGraph.getNodeMap().get("mypkg.PriceCheck").getStatus());
assertEquals(Status.IMPACTED, impactedSubGraph.getNodeMap().get("mypkg.StatusCheck").getStatus());
}
use of org.drools.impact.analysis.graph.Node in project drools by kiegroup.
the class GraphCollapsionTest method testSpreadsheet.
@Test
public void testSpreadsheet() throws IOException {
KieServices ks = KieServices.Factory.get();
ReleaseId releaseId = ks.newReleaseId("org.drools.impact.analysis.integrationtests", "spreadsheet-test", "1.0.0");
KieFileSystem kfs = createKieFileSystemWithClassPathResourceNames(releaseId, getClass(), "collapsion01.xls", "collapsion02.xls", "collapsion03.xls");
KieBuilder kieBuilder = ks.newKieBuilder(kfs).buildAll(ImpactAnalysisProject.class);
ImpactAnalysisKieModule analysisKieModule = (ImpactAnalysisKieModule) kieBuilder.getKieModule();
AnalysisModel analysisModel = analysisKieModule.getAnalysisModel();
ModelToGraphConverter converter = new ModelToGraphConverter();
Graph graph = converter.toGraph(analysisModel);
Graph collapsedGraph = new GraphCollapsionHelper().collapseWithRuleNamePrefix(graph);
assertEquals(3, collapsedGraph.getNodeMap().size());
assertLink(collapsedGraph, "mypkg2.CustomerCheck", "mypkg2.PriceCheck", ReactivityType.POSITIVE, ReactivityType.NEGATIVE);
assertLink(collapsedGraph, "mypkg2.PriceCheck", "mypkg2.StatusCheck", ReactivityType.POSITIVE, ReactivityType.NEGATIVE);
assertLink(collapsedGraph, "mypkg2.StatusCheck", "mypkg2.PriceCheck", ReactivityType.NEGATIVE);
// --- impact analysis
// Assuming that "modify" action in PriceCheck_X is changed
// modify action in PriceCheck_X
Node changedNode = collapsedGraph.getNodeMap().get("mypkg2.PriceCheck");
ImpactAnalysisHelper impactFilter = new ImpactAnalysisHelper();
Graph impactedSubGraph = impactFilter.filterImpactedNodes(collapsedGraph, changedNode);
assertNull(impactedSubGraph.getNodeMap().get("mypkg2.CustomerCheck"));
assertEquals(Status.CHANGED, impactedSubGraph.getNodeMap().get("mypkg2.PriceCheck").getStatus());
assertEquals(Status.IMPACTED, impactedSubGraph.getNodeMap().get("mypkg2.StatusCheck").getStatus());
}
use of org.drools.impact.analysis.graph.Node in project drools by kiegroup.
the class GraphvizOutputTest method testSimpleGraph.
@Test
public void testSimpleGraph() {
Node node1 = new Node(new Rule("org.example", "rule1", "dummy"));
Node node2 = new Node(new Rule("org.example", "rule2", "dummy"));
Node node3 = new Node(new Rule("org.example", "rule3", "dummy"));
Node node4 = new Node(new Rule("org.example", "rule4", "dummy"));
Node node5 = new Node(new Rule("org.example", "rule5", "dummy"));
Node.linkNodes(node1, node2, ReactivityType.POSITIVE);
Node.linkNodes(node1, node3, ReactivityType.NEGATIVE);
Node.linkNodes(node2, node4, ReactivityType.UNKNOWN);
Node.linkNodes(node3, node5, ReactivityType.POSITIVE);
Map<String, Node> nodeMap = new HashMap<>();
nodeMap.put(node1.getFqdn(), node1);
nodeMap.put(node2.getFqdn(), node2);
nodeMap.put(node3.getFqdn(), node3);
nodeMap.put(node4.getFqdn(), node4);
nodeMap.put(node5.getFqdn(), node5);
Graph graph = new Graph(nodeMap);
GraphImageGenerator generator = new GraphImageGenerator("simple");
generator.generateDot(graph);
generator.generateSvg(graph);
generator.generatePng(graph);
}
Aggregations