我希望创建一个代理来附加到我们的实时 Tomcat 和 Weblogic 服务器,它将拦截对我公司包中声明的所有类的所有方法调用,并记录一些指标,例如执行时间。
我遇到了似乎可以满足此需求的 Byte Buddy 库。但是,我不是 100% 清楚使用 Byte Buddy 创建代理的方法:
- 以下文章建议创建一个自己的代理并且没有提及
byte-buddy-agent
: http: //mydailyjava.blogspot.ie/2015/01/make-agents-not-frameworks.html - 但是,我确实看到有人创建了他/她自己的
byte-buddy-agent
,所以我不确定我是否打算使用它。 https://github.com/raphw/byte-buddy/tree/master/byte-buddy-agent
我采用了创建自己的代理的方法,并使用 Maven 将其打包,以将 Byte Buddy 包含为我从我的catalina.bat
.
编辑:我已经下载了源代码并发现 AgentBuilder 依赖于 byte-buddy-agent 包,因此上述问题无关紧要。
Tomcat 启动正常,我可以看到代理被调用,因为我看到了 "Entered premain" System.out
。
System.out
但是,当我在部署到 Tomcat 的单独战争文件上执行代码时,
我从未看到“拦截” 。编辑:下面的代码根据 Rafael 的响应进行了更新,现在可以正常工作了。
有人可以告诉我我在这里可能做错了什么吗?我在下面包含了代理代码。
另外,有人能告诉我哪个ElementMatchers
最适合包装匹配吗?我试过nameStartsWith
了,但没有效果,API 文档没有说明它是否是完全限定的类名。
*编辑:nameStartsWith 确实检查了包。*
无论如何,提前感谢您的帮助!
package com.mycompany.agent;
import java.lang.instrument.Instrumentation;
import java.util.concurrent.Callable;
import net.bytebuddy.agent.builder.AgentBuilder;
import net.bytebuddy.description.type.TypeDescription;
import net.bytebuddy.dynamic.DynamicType;
import net.bytebuddy.implementation.MethodDelegation;
import net.bytebuddy.implementation.bind.annotation.RuntimeType;
import net.bytebuddy.implementation.bind.annotation.SuperCall;
import net.bytebuddy.matcher.ElementMatchers;
public class MyAgent {
public static void premain(String agentArgument, Instrumentation instrumentation) {
System.out.println("Entered premain");
try{
new AgentBuilder.Default()
.withListener( new AgentBuilder.Listener() {
public void onComplete(String arg0) {
System.out.println("Completed - " + arg0);
}
public void onError(String arg0, Throwable arg1) {
System.out.println("Error - " + arg0+", "+arg1.getMessage());
arg1.printStackTrace();
}
public void onIgnored(String arg0) {
System.out.println("Ignored - " + arg0);
}
public void onTransformation(TypeDescription arg0, DynamicType arg1) {
System.out.println("Transformed - " + arg0+", type = "+arg1);
}
})
.rebase(ElementMatchers.nameStartsWith("com.mycompany"))
.transform(new AgentBuilder.Transformer() {
public DynamicType.Builder transform(DynamicType.Builder builder, TypeDescription typeDescription) {
return builder.method(ElementMatchers.any()).intercept(MethodDelegation.to(new Interceptor()));
}
}).installOn(instrumentation);
}
catch (RuntimeException e) {
System.out.println("Exception instrumenting code : "+e);
e.printStackTrace();
}
}
package com.mycompany.agent;
import java.lang.reflect.Method;
import java.util.concurrent.Callable;
import net.bytebuddy.implementation.bind.annotation.AllArguments;
import net.bytebuddy.implementation.bind.annotation.Origin;
import net.bytebuddy.implementation.bind.annotation.RuntimeType;
import net.bytebuddy.implementation.bind.annotation.SuperCall;
@SuppressWarnings("rawtypes")
public class Interceptor {
@RuntimeType
public Object intercept( @SuperCall Callable<?> callable, @AllArguments Object[] allArguments, @Origin Method method, @Origin Class clazz) throws Exception {
long startTime = System.currentTimeMillis();
Object response;
try{
response = callable.call();
}
catch(Exception e) {
System.out.println("Exception occurred in method call: " + methodName(clazz, method, allArguments) + " Exception = " + e);
throw e;
}
finally{
System.out.println("Method " + methodName(clazz, method) + " completed in " + (System.currentTimeMillis() - startTime) + " miliseconds");
}
return response;
}
private String methodName(Class clazz, Method method){
return methodName(clazz, method, null);
}
private String methodName(Class clazz, Method method, Object[] allArguments){
StringBuilder builder = new StringBuilder();
builder.append(clazz.getName());
builder.append(".");
builder.append(method.getName());
builder.append("(");
for(int i = 0; i < method.getParameters().length; i++) {
builder.append(method.getParameters()[i].getName());
if(allArguments != null) {
Object arg = allArguments[i];
builder.append("=");
builder.append(arg != null ? arg.toString() : "null");
}
if(i < method.getParameters().length - 1) {
builder.append(", ");
}
}
builder.append(")");
return builder.toString();
}