Search in sources :

Example 1 with Node

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;
}
Also used : ForNodeLink(guru.nidi.graphviz.attribute.ForNodeLink) Node(org.drools.impact.analysis.graph.Node) ForNodeLink(guru.nidi.graphviz.attribute.ForNodeLink) Link(org.drools.impact.analysis.graph.Link)

Example 2 with Node

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);
}
Also used : KieFileSystem(org.kie.api.builder.KieFileSystem) DefaultAgendaEventListener(org.drools.core.event.DefaultAgendaEventListener) Arrays(java.util.Arrays) GraphImageGenerator(org.drools.impact.analysis.graph.graphviz.GraphImageGenerator) Logger(org.slf4j.Logger) Assertions.assertThat(org.assertj.core.api.Assertions.assertThat) Graph(org.drools.impact.analysis.graph.Graph) LoggerFactory(org.slf4j.LoggerFactory) IOException(java.io.IOException) ReactivityType(org.drools.impact.analysis.graph.ReactivityType) Collectors(java.util.stream.Collectors) KieResources(org.kie.api.io.KieResources) ArrayList(java.util.ArrayList) ReleaseId(org.kie.api.builder.ReleaseId) Resource(org.kie.api.io.Resource) List(java.util.List) Rule(org.junit.Rule) Node(org.drools.impact.analysis.graph.Node) TestName(org.junit.rules.TestName) AfterMatchFiredEvent(org.kie.api.event.rule.AfterMatchFiredEvent) KieServices(org.kie.api.KieServices) KieSession(org.kie.api.runtime.KieSession) Link(org.drools.impact.analysis.graph.Link) Node(org.drools.impact.analysis.graph.Node) ReactivityType(org.drools.impact.analysis.graph.ReactivityType) Link(org.drools.impact.analysis.graph.Link)

Example 3 with Node

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());
}
Also used : ModelBuilder(org.drools.impact.analysis.parser.ModelBuilder) Graph(org.drools.impact.analysis.graph.Graph) AnalysisModel(org.drools.impact.analysis.model.AnalysisModel) Node(org.drools.impact.analysis.graph.Node) GraphCollapsionHelper(org.drools.impact.analysis.graph.GraphCollapsionHelper) ImpactAnalysisHelper(org.drools.impact.analysis.graph.ImpactAnalysisHelper) ModelToGraphConverter(org.drools.impact.analysis.graph.ModelToGraphConverter) Test(org.junit.Test)

Example 4 with Node

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());
}
Also used : Graph(org.drools.impact.analysis.graph.Graph) KieFileSystem(org.kie.api.builder.KieFileSystem) AnalysisModel(org.drools.impact.analysis.model.AnalysisModel) Node(org.drools.impact.analysis.graph.Node) ImpactAnalysisKieModule(org.drools.impact.analysis.parser.internal.ImpactAnalysisKieModule) GraphCollapsionHelper(org.drools.impact.analysis.graph.GraphCollapsionHelper) KieServices(org.kie.api.KieServices) ImpactAnalysisHelper(org.drools.impact.analysis.graph.ImpactAnalysisHelper) ReleaseId(org.kie.api.builder.ReleaseId) ModelToGraphConverter(org.drools.impact.analysis.graph.ModelToGraphConverter) KieBuilder(org.kie.api.builder.KieBuilder) Test(org.junit.Test)

Example 5 with Node

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);
}
Also used : Graph(org.drools.impact.analysis.graph.Graph) HashMap(java.util.HashMap) Node(org.drools.impact.analysis.graph.Node) Rule(org.drools.impact.analysis.model.Rule) Test(org.junit.Test)

Aggregations

Node (org.drools.impact.analysis.graph.Node)5 Graph (org.drools.impact.analysis.graph.Graph)4 Test (org.junit.Test)3 GraphCollapsionHelper (org.drools.impact.analysis.graph.GraphCollapsionHelper)2 ImpactAnalysisHelper (org.drools.impact.analysis.graph.ImpactAnalysisHelper)2 Link (org.drools.impact.analysis.graph.Link)2 ModelToGraphConverter (org.drools.impact.analysis.graph.ModelToGraphConverter)2 AnalysisModel (org.drools.impact.analysis.model.AnalysisModel)2 KieServices (org.kie.api.KieServices)2 KieFileSystem (org.kie.api.builder.KieFileSystem)2 ReleaseId (org.kie.api.builder.ReleaseId)2 ForNodeLink (guru.nidi.graphviz.attribute.ForNodeLink)1 IOException (java.io.IOException)1 ArrayList (java.util.ArrayList)1 Arrays (java.util.Arrays)1 HashMap (java.util.HashMap)1 List (java.util.List)1 Collectors (java.util.stream.Collectors)1 Assertions.assertThat (org.assertj.core.api.Assertions.assertThat)1 DefaultAgendaEventListener (org.drools.core.event.DefaultAgendaEventListener)1