re

Author: codingforkind

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

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

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

@@ -5,14 +5,14 @@
  * @description
  * @date 18-8-23
  */
-public class GenerateNodeException extends RuntimeException {
+public class NodeGenerateException extends RuntimeException {
     private static final long serialVersionUID = 1L;
 
-    public GenerateNodeException(String message) {
+    public NodeGenerateException(String message) {
         super(message);
     }
 
-    public GenerateNodeException(String message, Throwable cause) {
+    public NodeGenerateException(String message, Throwable cause) {
         super(message, cause);
     }
 }

mirror-common/src/main/java/cn/com/mirror/exceptions/GenerateVertexException.java renamed to mirror-common/src/main/java/cn/com/mirror/exceptions/VertexGenerateException.java

@@ -5,14 +5,14 @@
  * @description
  * @date 18-8-24
  */
-public class GenerateVertexException extends RuntimeException {
+public class VertexGenerateException extends RuntimeException {
     private static final long serialVersionUID = 1L;
 
-    public GenerateVertexException(String message) {
+    public VertexGenerateException(String message) {
         super(message);
     }
 
-    public GenerateVertexException(String message, Throwable cause) {
+    public VertexGenerateException(String message, Throwable cause) {
         super(message, cause);
     }
 }

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

@@ -1,7 +1,7 @@
 package cn.com.mirror.project.pair.factory;
 
 import cn.com.mirror.constant.VertexTypeEnum;
-import cn.com.mirror.exceptions.GenerateVertexException;
+import cn.com.mirror.exceptions.VertexGenerateException;
 import cn.com.mirror.project.pair.Vertex;
 import lombok.Data;
 import org.eclipse.jdt.core.dom.ASTNode;
@@ -52,11 +52,11 @@ public Vertex genVertex(String targetPath, int curLine, ASTNode node) {
 
     public Vertex genVertex(String targetPath, int curLine, ASTNode node, String ip, int port) {
         if (null != ip && -1 == port) {
-            throw new GenerateVertexException("Port can not be -1.");
+            throw new VertexGenerateException("Port can not be -1.");
         }
 
         if (-1 != port && null == ip) {
-            throw new GenerateVertexException("IP address can not be null");
+            throw new VertexGenerateException("IP address can not be null");
         }
 
         String hostKey = getHostKey(ip, port, targetPath, curLine);

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

@@ -2,6 +2,7 @@
 
 import cn.com.mirror.analyser.PairAnalyser;
 import cn.com.mirror.analyser.UnitAnalyser;
+import cn.com.mirror.exceptions.EdgeTouchException;
 import cn.com.mirror.exceptions.ReflectException;
 import cn.com.mirror.project.pair.Pair;
 import cn.com.mirror.project.pair.Vertex;
@@ -55,13 +56,18 @@ public void construct() {
 
             // basic control edges in the target file
             ctrlEdge.forEach((ctrlKey, ctrlVal) -> {
-                log.debug("TARGET: {}", targetPath);
-                log.debug("FROM: {}:{} \t->\t TO: {}:{}", ctrlKey.getLineNum(),
-                        ctrlKey.getVertexType(), ctrlVal.getLineNum(), ctrlVal.getVertexType());
-
                 Base headB = getBaseElement(ctrlKey);
                 Base tailB = getBaseElement(ctrlVal);
 
+                if (null == headB || null == tailB) {
+                    log.debug("TARGET: {}", targetPath);
+                    log.debug("HEAD: {}:{} \t->\t TAIL: {}:{}", ctrlKey.getLineNum(),
+                            ctrlKey.getVertexType(), ctrlVal.getLineNum(), ctrlVal.getVertexType());
+
+                    throw new EdgeTouchException("Head statement or tail statement is null! \tHEAD: {"
+                            + headB + "} -> TAIL: {" + tailB + "}");
+                }
+
                 // 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);