/*
* Copyright 2014 NAVER Corp.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.navercorp.pinpoint.profiler.interceptor;
import java.io.IOException;
import java.lang.reflect.Method;
import java.util.Arrays;
import com.navercorp.pinpoint.profiler.util.LoaderUtils;
import javassist.CannotCompileException;
import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtMethod;
import javassist.Loader;
import javassist.NotFoundException;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.navercorp.pinpoint.bootstrap.interceptor.AroundInterceptor;
import com.navercorp.pinpoint.bootstrap.interceptor.registry.DefaultInterceptorRegistryAdaptor;
import com.navercorp.pinpoint.bootstrap.interceptor.registry.InterceptorRegistry;
/**
* @author emeroad
*/
public class InterceptorTest {
private final Logger logger = LoggerFactory.getLogger(InterceptorTest.class.getName());
private static DefaultInterceptorRegistryAdaptor INTERCEPTOR_REGISTRY_ADAPTOR;
@Before
public void setUp() throws Exception {
INTERCEPTOR_REGISTRY_ADAPTOR = new DefaultInterceptorRegistryAdaptor();
InterceptorRegistry.bind(INTERCEPTOR_REGISTRY_ADAPTOR, null);
}
@After
public void tearDown() throws Exception {
InterceptorRegistry.unbind(null);
INTERCEPTOR_REGISTRY_ADAPTOR = null;
}
// @Deprecated
@Test
public void interceptor() throws NotFoundException, CannotCompileException, IllegalAccessException, InstantiationException, IOException, ClassNotFoundException, NoSuchMethodException {
AroundInterceptor aroundInterceptor = new AroundInterceptor() {
@Override
public void before(Object target, Object[] args) {
logger.info("BEFORE target:" + target + " args:" + Arrays.toString(args));
}
@Override
public void after(Object target, Object[] args, Object result, Throwable throwable) {
logger.info("AFTER target: " + target + " args:" + Arrays.toString(args) + " result:" + result + " throwable:" + throwable);
}
};
int interceptorId = INTERCEPTOR_REGISTRY_ADAPTOR.addInterceptor(aroundInterceptor);
final ClassPool classPool = new ClassPool(true);
CtClass throwable = classPool.get(Throwable.class.getName());
CtClass ctClass = classPool.get("com.navercorp.pinpoint.profiler.interceptor.JavaAssistTestObject");
final CtMethod hello = ctClass.getMethod("hello", "(Ljava/lang/String;)Ljava/lang/String;");
logger.debug("longName:{}", hello.getLongName());
logger.debug("name:{}", hello.getName());
String interceptorClassName = AroundInterceptor.class.getName();
CtClass interceptor = classPool.get(interceptorClassName);
hello.addLocalVariable("interceptor", interceptor);
CtClass object = classPool.get(Object.class.getName());
hello.addLocalVariable("result", object);
// hello.insertBefore("{ System.out.println(\"BEFORE\"); }");
hello.insertBefore("{" +
"interceptor = (" + interceptorClassName + ") " + InterceptorRegistry.class.getName() + ".getInterceptor(" + interceptorId + ");" +
"interceptor.before(this, $args);" +
"}");
// hello.addCatch("{" +
//// " interceptor.after(ctx);"+
//// " AroundInterceptor a = (AroundInterceptor) " + InterceptorRegistry.class.getName() + ".getStaticInterceptor(\"a\");"+
// " throw $e;" +
// "}", throwable);
// hello.insertAfter("{" +
// "interceptor.after(this, $args, ($w)$_, null); " +
// "}");
// hello.setBody(generatedAroundInterceptor("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("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");
Loader loader = LoaderUtils.createLoader(classPool);
loader.delegateLoadingOf("com.navercorp.pinpoint.bootstrap.");
Class aClass = loader.loadClass(ctClass.getName());
Object testObject = aClass.newInstance();
Method helloMethod = testObject.getClass().getDeclaredMethod("hello", String.class);
try {
helloMethod.invoke(testObject, "hello~~");
} catch (Exception e) {
Assert.fail(e.getMessage());
}
// o.hello();
}
public void println(StringBuilder sb, String out) {
sb.append("System.out.println(\"" + out.replace("\"", "\\\"") + "\");");
}
}