抛异常追踪

caipeichao · caipeichao · commit c54154c4fbdd · 2016-12-10T16:41:28.000+08:00
diff --git a/src/main/java/gr/gousiosg/javacg/dyn/Instrumenter.java b/src/main/java/gr/gousiosg/javacg/dyn/Instrumenter.java
@@ -43,6 +43,8 @@
 import javassist.CtBehavior;
 import javassist.CtClass;
 import javassist.NotFoundException;
+import javassist.expr.ExprEditor;
+import javassist.expr.MethodCall;
 
 public class Instrumenter implements ClassFileTransformer {
 
@@ -169,9 +171,8 @@ private void enhanceMethod(CtBehavior method, String className)
 
         if (method.getName().equals(name))
             methodName = "<init>";
-
         method.insertBefore(String.format("gr.gousiosg.javacg.dyn.CallTrace.push(\"%s\",\"%s\");", className, methodName));
-        method.insertAfter("gr.gousiosg.javacg.dyn.CallTrace.pop();");
+        method.insertAfter("gr.gousiosg.javacg.dyn.CallTrace.pop();", true);
     }
 
     private static void err(String msg) {
diff --git a/src/main/java/gr/gousiosg/javacg/test/TestException.java b/src/main/java/gr/gousiosg/javacg/test/TestException.java
@@ -0,0 +1,26 @@
+package gr.gousiosg.javacg.test;
+
+/**
+ * Created by caipeichao on 16/12/10.
+ */
+public class TestException {
+
+    public static void main(String[] argv) {
+        new TestException().run();
+    }
+
+    public void run() {
+        System.out.println("Hello world");
+        foo();
+        bar();
+    }
+
+    public void foo() {
+        System.out.println("Foo");
+        throw new RuntimeException("Bad times");
+    }
+
+    public void bar() {
+        System.out.println("Bar");
+    }
+}
diff --git a/test/test.sh b/test/test.sh
@@ -1,9 +1,18 @@
+# check args
 cd "`dirname "$0"`"
+c="$1"
+if [ "$c" == "" ] ; then
+  echo try:
+  ls ../src/main/java/gr/gousiosg/javacg/test | sed 's/\.java$//g' | xargs -n 1 echo 'bash' "$0"
+  exit 1
+fi
+
+# run test
 cd ..
 mvn package -DskipTests
 cd target
 export CLASSPATH=javacg-0.1-SNAPSHOT-test.jar
-java -javaagent:javacg-0.1-SNAPSHOT-dycg-agent.jar="incl=gr.gousiosg.javacg.test.*;" gr.gousiosg.javacg.test.TestOneThread
+java -javaagent:javacg-0.1-SNAPSHOT-dycg-agent.jar="incl=gr.gousiosg.javacg.test.*;" gr.gousiosg.javacg.test.$1
 echo ------------------------
 echo Call trace is:
 cat calltrace.json
