home/pinpoint
2
0
Fork 0
mirror of https://github.com/wahyd4/pinpoint.git synced 2026-08-13 23:06:54 +10:00
Code Issues Packages Projects Releases Wiki Activity

[강운덕] [LUCYSUS-1744] interceptor test.

Browse Source 
git-svn-id: http://svn.bds.nhncorp.com/pe/hippo-tomcat-profiler/trunk@528 84d0f5b1-2673-498c-a247-62c4ff18d310
This commit is contained in:
Woonduk Kang
2012-08-22 06:42:30 +00:00
parent 7164dee610
commit 546520ceb2
2 changed files with 134 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/test/java/com/profiler/interceptor/InterceptorRegistryTest.java
+128 -30
View File
@@ -1,14 +1,15 @@
package com.profiler.interceptor;
import javassist.*;
import javassist.compiler.ast.Expr;
import javassist.bytecode.Descriptor;
import javassist.expr.ExprEditor;
import javassist.expr.MethodCall;
import org.apache.commons.lang.StringEscapeUtils;
import org.apache.commons.lang.StringUtils;
import org.junit.Test;
import org.omg.PortableInterceptor.*;
import java.io.IOException;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.Map;
public class InterceptorRegistryTest {
@@ -24,54 +25,151 @@ public class InterceptorRegistryTest {
}
@Test
public void intercet() throws NotFoundException, CannotCompileException, IllegalAccessException, InstantiationException {
public void interceptor() throws NotFoundException, CannotCompileException, IllegalAccessException, InstantiationException, IOException {
InterceptorRegistry.addInterceptor("TargetGetClass", new Interceptor() {
@Override
public void before(InterceptorContext ctx) {
System.out.println("before");
}
InterceptorRegistry.addInterceptor("a", new AroundInterceptor() {
@Override
public void before(InterceptorContext ctx) {
System.out.println("before ctx:" + ctx );
}
@Override
public void after(InterceptorContext ctx) {
System.out.println("after");
}
});
@Override
public void after(InterceptorContext ctx) {
System.out.println("after ctx:" + ctx);
}
});
ClassPool p = ClassPool.getDefault();
CtClass throwable = p.get(Throwable.class.getName());
CtClass ctx = p.get(InterceptorContext.class.getName());
CtClass ctClass = p.get("com.profiler.interceptor.TestObject");
// System.out.println(ctClass);
CtMethod hello = ctClass.getMethod("hello", "()V");
System.out.println(ctClass);
final CtMethod hello = ctClass.getMethod("hello", "(Ljava/lang/String;)Ljava/lang/String;");
CtClass ctx = p.get(InterceptorContext.class.getName());
hello.addLocalVariable("ctx", ctx);
CtClass interceptor = p.get(AroundInterceptor.class.getName());
hello.addLocalVariable("interceptor", interceptor);
CtClass object = p.get(Object.class.getName());
hello.addLocalVariable("result", object);
hello.insertBefore("{" +
"ctx = new com.profiler.interceptor.InterceptorContext();" +
"ctx.setParameter($args);" +
"System.out.println(ctx); " +
InterceptorRegistry.class.getName() + ".getInterceptor(\"a\").before(null);" +
// InterceptorRegistry.class.getName() + ".getInterceptor(\"a\").before(ctx);" +
"interceptor = (com.profiler.interceptor.AroundInterceptor) " + InterceptorRegistry.class.getName() + ".getInterceptor(\"a\");"+
"interceptor.before(ctx);" +
"}");
hello.insertAfter("{" +
"System.out.println(ctx); " +
InterceptorRegistry.class.getName() + ".getInterceptor(\"a\").after(null);" +
"}");
hello.addCatch("{" +
InterceptorRegistry.class.getName() + ".getInterceptor(\"a\").after(null);" +
hello.addCatch("{" +
// " interceptor.after(ctx);"+
// " com.profiler.interceptor.AroundInterceptor a = (com.profiler.interceptor.AroundInterceptor) " + InterceptorRegistry.class.getName() + ".getInterceptor(\"a\");"+
" throw $e;" +
"}", throwable);
hello.insertAfter("{" +
"interceptor.after(ctx); " +
"}");
//
// hello.setBody(generatedAroundInterceptor("com.profiler.interceptor.TestObject", "hello"));
// hello.setBody("{ System.out.println(\"ddd\"); }", ClassMap map );
hello.insertBefore(" System.out.println(\" before + \");");
hello.insertAfter(" System.out.println($_);");
// hello.insertAfter(" System.out.println($r);");
// hello.insertAfter(" System.out.println($w);");
hello.insertAfter(" System.out.println($sig);");
hello.insertAfter(" System.out.println($type);");
hello.insertAfter(" System.out.println($class);");
// hello.instrument(new ExprEditor() {
// public void edit(MethodCall m)
// throws CannotCompileException
// {
// try {
// System.out.println("method call" + m.getMethod().getName());
// } catch (NotFoundException e) {
// e.printStackTrace(); //To change body of catch statement use File | Settings | File Templates.
// }
// String code = generatedAroundInterceptor("com.profiler.interceptor.TestObject", "hello");
// m.replace(code);
// }
// });
// hello.addCatch("System.out.println(\"catch\"); throw $e;", throwable);
// hello.setName("__hello");
// CtMethod method = CtNewMethod.make("public void hello() { try {__hello(); } catch(Throwable th){throw th;}}", ctClass);
// CtMethod method = CtNewMethod.make("public void hello() { System.out.println(\"ddd\"); } catch(Throwable th){throw th;}}", ctClass);
// ctClass.addMethod(method);
ctClass.freeze();
// ctClass.writeFile("./debug");
ctClass.debugWriteFile("./debug");
Class aClass = ctClass.toClass();
TestObject o = (TestObject) aClass.newInstance();
o.hello();
// ctClass.getMethod("toString", null);
// ctClass.getDeclaredMethod("toString", null);
o.hello();
try {
o.hello("aaaaaa");
} catch (Exception e) {
e.printStackTrace(); //To change body of catch statement use File | Settings | File Templates.
}
// o.hello();
}
private String generatedAroundInterceptor(String className, String methodName) {
StringBuilder sb = new StringBuilder();
sb.append("{");
sb.append(" ctx = new com.profiler.interceptor.InterceptorContext();");
sb.append(" ctx.setParameter($args);");
sb.append(" ctx.setTarget(this);");
sb.append(" ");
// sb.append(" ctx.setMethodName(\"" + methodName + "\");");
// sb.append(" System.out.println(\"args check : \" + $args );");
// sb.append(" System.out.println(\"0 check : \" + $0 );");
// sb.append(" System.out.println(\"1 check : \" + $1 );");
// sb.append(" System.out.println(\"sig check : \" + $sig );");
// sb.append(" System.out.println(\"class check : \" + $class );");
// sb.append(" System.out.println(\" r check : \" + $r);");
sb.append("}");
sb.append("{");
sb.append(" interceptor = (com.profiler.interceptor.AroundInterceptor) " + InterceptorRegistry.class.getName() + ".getInterceptor(\"a\");");
sb.append(" interceptor.before(ctx);");
sb.append(" result = null;");
// println(sb, "before systemout \"ttt\"");
sb.append("}");
sb.append("try {");
sb.append(" $_ = $proceed($$);");
sb.append(" result = $_;");
sb.append("}");
// sb.append("catch(Throwable th) {");
// sb.append(" System.out.println(\"test11\" + th);");
// sb.append(" ctx.setException(th);");
// sb.append(" System.out.println(\"catch\");");
// sb.append("}");
sb.append("finally {");
// sb.append(" System.out.println(\"finally\");");
sb.append(" ctx.setReturnValue(result);");
sb.append(" interceptor.after(ctx);");
sb.append("}");
// System.out.println(sb);
return sb.toString();
}
public void println(StringBuilder sb, String out) {
sb.append("System.out.println(\"" + out.replace("\"", "\\\"") + "\");");
}
}
;
src/test/java/com/profiler/interceptor/TestObject.java
+6 -3
View File
@@ -2,8 +2,11 @@ package com.profiler.interceptor;
public class TestObject{
public void hello() {
System.out.println("hello");
throw new RuntimeException();
public String hello(String a) {
System.out.println("a:" + a);
System.out.println("test");
// throw new RuntimeException("test");
return "a";
}
}