re

Author: codingforkind

mirror-common/src/main/java/cn/com/mirror/analyser/visitor/ControlEdgeVisitor.java

@@ -1,10 +1,13 @@
 package cn.com.mirror.analyser.visitor;
 
 import cn.com.mirror.constant.ControlNodeTypeEnum;
+import cn.com.mirror.project.pair.Vertex;
+import cn.com.mirror.project.pair.VertexTypeEnum;
 import cn.com.mirror.utils.AstUtils;
 import lombok.Getter;
 import lombok.extern.slf4j.Slf4j;
 import org.eclipse.jdt.core.dom.*;
+import org.neo4j.cypher.internal.compiler.v2_3.commands.AstNode;
 
 import java.util.Arrays;
 import java.util.HashMap;
@@ -21,7 +24,14 @@
 @Slf4j
 @Getter
 public class ControlEdgeVisitor extends ASTVisitor {
+    private String targetPath;
+
     private Map<Integer, Integer> controlEdges = new HashMap<>();
+    private Map<Vertex, Vertex> ctrlEdges = new HashMap<>();
+
+    public ControlEdgeVisitor(String targetPath) {
+        this.targetPath = targetPath;
+    }
 
     private ASTNode searchDirectParentControlNode(ASTNode astNode) {
         ASTNode parent = astNode.getParent();
@@ -35,6 +45,8 @@ private ASTNode searchDirectParentControlNode(ASTNode astNode) {
         int currentLine = AstUtils.getSpecificStartLine(astNode);
         int directParentStartLine = AstUtils.getSpecificStartLine(parent);
         markEdges(currentLine, directParentStartLine);
+
+        markCtrlEdge(currentLine, astNode, directParentStartLine, parent);
         return parent;
     }
 
@@ -78,6 +90,35 @@ private boolean isControlType(ASTNode astNode) {
 
     }
 
+    private boolean markCtrlEdge(int curLine,
+                                 ASTNode astNode,
+                                 int parentLine,
+                                 ASTNode parent) {
+        if (-1 == parentLine || parentLine == curLine) {
+            return false;
+        }
+
+        VertexTypeEnum headType = null;
+        if (astNode instanceof MethodDeclaration) {
+            headType = VertexTypeEnum.METHOD;
+        } else {
+            headType = VertexTypeEnum.STATEMENT;
+        }
+
+        Vertex head = new Vertex(this.targetPath, curLine, headType);
+
+        VertexTypeEnum tailType = null;
+        if (parent instanceof MethodDeclaration) {
+            tailType = VertexTypeEnum.METHOD;
+        } else {
+            tailType = VertexTypeEnum.STATEMENT;
+        }
+        Vertex tail = new Vertex(this.targetPath, parentLine, tailType);
+        ctrlEdges.put(head, tail);
+
+        return true;
+    }
+
     private boolean markEdges(int curLine, int parentLine) {
         if (-1 != parentLine &&
                 curLine != parentLine) {

mirror-common/src/main/java/cn/com/mirror/project/pair/Pair.java

@@ -13,8 +13,13 @@
 @Getter
 public class Pair {
     private Map<String, Map<Integer, Integer>> directCtrlEdges = new HashMap<>();
+    private Map<String, Map<Vertex, Vertex>> ctrlEdges = new HashMap<>();
 
     public void addDirectCtrlEdges(String targetPath, Map<Integer, Integer> ctrlEdges) {
         this.directCtrlEdges.put(targetPath, ctrlEdges);
     }
+
+    public void addCtrlEdge(String targetPath, Map<Vertex, Vertex> ctrlEdges) {
+        this.ctrlEdges.put(targetPath, ctrlEdges);
+    }
 }

mirror-common/src/main/java/cn/com/mirror/project/pair/Vertex.java

@@ -0,0 +1,42 @@
+package cn.com.mirror.project.pair;
+
+import lombok.Getter;
+
+import java.io.Serializable;
+import java.util.Objects;
+
+/**
+ * @author piggy
+ * @description
+ * @date 18-8-23
+ */
+@Getter
+public class Vertex implements Serializable {
+
+    private String targetPath;
+    private int lineNum;
+    private VertexTypeEnum vertexType;
+
+    public Vertex(String targetPath, int lineNum, VertexTypeEnum vertexType) {
+        this.targetPath = targetPath;
+        this.lineNum = lineNum;
+        this.vertexType = vertexType;
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+
+        Vertex vertex = (Vertex) o;
+        return lineNum == vertex.lineNum &&
+                Objects.equals(targetPath, vertex.targetPath) &&
+                vertexType == vertex.vertexType;
+    }
+
+    @Override
+    public int hashCode() {
+
+        return Objects.hash(targetPath, lineNum, vertexType);
+    }
+}

mirror-common/src/main/java/cn/com/mirror/project/pair/VertexTypeEnum.java

@@ -0,0 +1,22 @@
+package cn.com.mirror.project.pair;
+
+import lombok.Getter;
+
+/**
+ * @author piggy
+ * @description
+ * @date 18-8-23
+ */
+@Getter
+public enum VertexTypeEnum {
+    STATEMENT(1, "statement"),
+    METHOD(2, "method");
+
+    private Integer key;
+    private String desc;
+
+    VertexTypeEnum(Integer key, String desc) {
+        this.key = key;
+        this.desc = desc;
+    }
+}

mirror-common/src/main/java/cn/com/mirror/project/pair/factory/PairFactory.java

@@ -20,10 +20,11 @@ public Pair newPair(String repositoryUrl) {
 
         Set<String> targetFiles = FileUtils.extractTargetPath(repositoryUrl);
         targetFiles.stream().forEach(targetPath -> {
-            ControlEdgeVisitor controlEdgeVisitor = new ControlEdgeVisitor();
+            ControlEdgeVisitor controlEdgeVisitor = new ControlEdgeVisitor(targetPath);
             CompilationUnit compilationUnit = AstUtils.getCompUnitResolveBinding(targetPath);
             compilationUnit.accept(controlEdgeVisitor);
             pair.addDirectCtrlEdges(targetPath, controlEdgeVisitor.getControlEdges());
+            pair.addCtrlEdge(targetPath, controlEdgeVisitor.getCtrlEdges());
         });
 
         return pair;

mirror-common/src/test/cn/com/mirror/analyser/UnitAnalyserTests.java

@@ -32,7 +32,7 @@ public void testControlDependenceVisitor() {
         init();
 
         unit.getTargets().stream().forEach(targetPah -> {
-            ControlEdgeVisitor controlDependenceVisitor = new ControlEdgeVisitor();
+            ControlEdgeVisitor controlDependenceVisitor = new ControlEdgeVisitor(targetPah);
             CompilationUnit compilationUnit = unit.getCompilationUnits().get(targetPah);
             compilationUnit.accept(controlDependenceVisitor);
             log.info("Target path: {}", targetPah);

mirror-common/src/test/cn/com/mirror/pair/PairAnalyserTests.java

@@ -2,6 +2,7 @@
 
 import cn.com.mirror.analyser.PairAnalyser;
 import cn.com.mirror.project.pair.Pair;
+import cn.com.mirror.project.pair.Vertex;
 import lombok.extern.slf4j.Slf4j;
 import org.junit.Test;
 
@@ -23,14 +24,35 @@ public class PairAnalyserTests {
     public void test1() {
         PairAnalyser pairAnalyser = new PairAnalyser();
         Pair pair = pairAnalyser.analyze();
-        Set<Map.Entry<String, Map<Integer, Integer>>> entries =
-                pair.getDirectCtrlEdges().entrySet();
+        Set<Map.Entry<String, Map<Integer, Integer>>> entries = pair.getDirectCtrlEdges().entrySet();
         entries.stream().forEach(entry -> {
-            if(TEST_FILE.equals(entry.getKey())){
+            if (TEST_FILE.equals(entry.getKey())) {
                 log.info("Target: {}", entry.getKey());
                 log.info("Pair: {}", entry.getValue());
                 return;
             }
         });
     }
+
+
+    @Test
+    public void test2() {
+        PairAnalyser pairAnalyser = new PairAnalyser();
+        Pair pair = pairAnalyser.analyze();
+        for (Map.Entry<String, Map<Vertex, Vertex>> entry : pair.getCtrlEdges().entrySet()) {
+            if (TEST_FILE.equals(entry.getKey())) {
+                log.debug("Target: {}", entry.getKey());
+
+                for (Map.Entry<Vertex, Vertex> ver : entry.getValue().entrySet()) {
+                    log.debug("HEAD: {}, {} -> TAIL: {}, {}",
+                            ver.getKey().getLineNum(),
+                            ver.getKey().getVertexType(),
+                            ver.getValue().getLineNum(),
+                            ver.getValue().getVertexType());
+                }
+
+                return;
+            }
+        }
+    }
 }