kazusa
/
skywalking-java
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修改afterMethod接口，以及相关的依赖实现。允许afterMethod复写方法的返回值，用于进行特定的类包装。#36

This commit is contained in:
wusheng 2016-03-14 17:32:12 +08:00
parent 904d0cb205
commit 8c70efcc6c
4 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
skywalking-api/src/main/java/com/ai/cloud/skywalking/plugin/interceptor/ClassMethodInterceptor.java
View File

@ -45,15 +45,17 @@ public class ClassMethodInterceptor {
logger.error("class[{}] before method[{}] intercept failue:{}",
obj.getClass(), method.getName(), t.getMessage(), t);
}
Object ret = null;
try {
return zuper.call();
ret = zuper.call();
} finally {
try {
interceptor.afterMethod(instanceContext, interceptorContext);
ret = interceptor.afterMethod(instanceContext, interceptorContext, ret);
} catch (Throwable t) {
logger.error("class[{}] after method[{}] intercept failue:{}",
obj.getClass(), method.getName(), t.getMessage(), t);
}
}
return ret;
}
}

2
skywalking-api/src/main/java/com/ai/cloud/skywalking/plugin/interceptor/IAroundInterceptor.java
View File

@ -5,7 +5,7 @@ public interface IAroundInterceptor {
public void beforeMethod(EnhancedClassInstanceContext context, InterceptorContext interceptorContext);
public void afterMethod(EnhancedClassInstanceContext context, InterceptorContext interceptorContext);
public Object afterMethod(EnhancedClassInstanceContext context, InterceptorContext interceptorContext, Object ret);
}

15
skywalking-api/src/test/java/test/ai/cloud/plugin/PluginMainTest.java
View File

@ -8,12 +8,19 @@ import com.ai.cloud.skywalking.plugin.TracingBootstrap;
public class PluginMainTest {
@Test
public void testMain() throws IllegalAccessException, IllegalArgumentException, InvocationTargetException, NoSuchMethodException, SecurityException, ClassNotFoundException{
TracingBootstrap.main(new String[]{"test.ai.cloud.plugin.PluginMainTest"});
public void testMain() throws IllegalAccessException,
IllegalArgumentException, InvocationTargetException,
NoSuchMethodException, SecurityException, ClassNotFoundException {
TracingBootstrap
.main(new String[] { "test.ai.cloud.plugin.PluginMainTest" });
}
public static void main(String[] args){
public static void main(String[] args) {
long start = System.currentTimeMillis();
BeInterceptedClass inst = new BeInterceptedClass();
inst.printabc();
long end = System.currentTimeMillis();
System.out.println(end - start + "ms");
}
}

3
skywalking-api/src/test/java/test/ai/cloud/plugin/TestAroundInterceptor.java
View File

@ -19,8 +19,9 @@ public class TestAroundInterceptor implements IAroundInterceptor {
}
@Override
public void afterMethod(EnhancedClassInstanceContext context, InterceptorContext interceptorContext) {
public Object afterMethod(EnhancedClassInstanceContext context, InterceptorContext interceptorContext, Object ret) {
System.out.println("afterMethod: " + context.get("test.key", String.class));
return ret;
}
}