re

Author: codingforkind

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

@@ -1,8 +1,8 @@
 package cn.com.mirror.analyser.visitor;
 
 import cn.com.mirror.constant.ControlNodeTypeEnum;
-import cn.com.mirror.project.pair.Vertex;
 import cn.com.mirror.constant.VertexTypeEnum;
+import cn.com.mirror.project.pair.Vertex;
 import cn.com.mirror.utils.AstUtils;
 import lombok.Getter;
 import lombok.extern.slf4j.Slf4j;
@@ -94,6 +94,7 @@ private boolean markCtrlEdge(int curLine,
             return false;
         }
 
+        // TODO xyz vertex might not need to create all the time, use a cache and generate from it
         Vertex head = new Vertex(this.targetPath, curLine, checkVertexType(astNode), null, -1);
         Vertex tail = new Vertex(this.targetPath, parentLine, checkVertexType(parent), null, -1);
         ctrlEdges.put(head, tail);

mirror-common/src/main/java/cn/com/mirror/exceptions/GenerateNodeException.java

@@ -0,0 +1,18 @@
+package cn.com.mirror.exceptions;
+
+/**
+ * @author piggy
+ * @description
+ * @date 18-8-23
+ */
+public class GenerateNodeException extends RuntimeException {
+    private static final long serialVersionUID = 1L;
+
+    public GenerateNodeException(String message) {
+        super(message);
+    }
+
+    public GenerateNodeException(String message, Throwable cause) {
+        super(message, cause);
+    }
+}

mirror-common/src/main/java/cn/com/mirror/reflect/EdgeConstructor.java

@@ -55,14 +55,14 @@ public void construct() {
 
             // basic control edges in the target file
             ctrlEdge.forEach((ctrlKey, ctrlVal) -> {
-                // TODO xyz generate nodes and construct control edges based on the work before
                 log.debug("TARGET: {}", targetPath);
                 log.debug("FROM: {}:{} \t->\t TO: {}:{}", ctrlKey.getLineNum(),
                         ctrlKey.getVertexType(), ctrlVal.getLineNum(), ctrlVal.getVertexType());
 
-                Base headB = genBase(ctrlKey);
-                Base tailB = genBase(ctrlVal);
+                Base headB = getBaseElement(ctrlKey);
+                Base tailB = getBaseElement(ctrlVal);
 
+                // TODO xyz generate start and end node for this target and write it into the graph db.
                 BaseNode newGraphNodeTail = touch(headB, tailB);
                 graphEngine.write(newGraphNodeTail);
 
@@ -104,7 +104,7 @@ public BaseNode touch(Base headB, Base tailB) {
         return headNode;
     }
 
-    private Base genBase(Vertex vertex) {
+    private Base getBaseElement(Vertex vertex) {
         switch (vertex.getVertexType()) {
             case CLASS: {
                 Set<Class> classes = unit.getClasses().get(vertex.getTargetPath());

mirror-common/src/main/java/cn/com/mirror/repository/neo4j/node/BaseNode.java

@@ -30,7 +30,7 @@ public class BaseNode implements Serializable {
     @Property(name = "target path")
     private String targetPath;
 
-    @Property(name = "details in this line")
+    //    @Property(name = "details in this line")
     private String content;
 
     @Property(name = "package name")