【云+社区年度征文】java agent及字节码技术得到DNS时间流程尝试

public static void main(String[] args) throws UnknownHostException {
                                            
    InetAddress ip = InetAddress.getByName("www.changyou.com");  // 你可以试一下baidu www.baidu.com
    InetAddress[] ips = InetAddress.getAllByName("www.changyou.com");
 
    System.out.println(ip);
    System.out.println();
    for (InetAddress ip1:ips) {
        System.out.println(ip1);
    }
 
}
  
sout :    
  
www.changyou.com/211.93.241.55
 
www.changyou.com/211.93.241.55
www.changyou.com/60.28.100.37

@Pointcut("execution(* java.net.InetAddress.getByName(String))") // java.net. InetAddress.getByname(String)
public void pointcut2() {
}

@Around("pointcut2()")
public Object around(ProceedingJoinPoint proceedingJoinPoint) throws Throwable {
long startTime = System.currentTimeMillis();

 Object result = proceedingJoinPoint.proceed();

long endTime = System.currentTimeMillis();
 System.out.println(endTime-startTime);

return result;
}

 

@Pointcut("execution(* java.util.HashMap.put(..))") // java.net.. InetAddress.getByname(String)
public void pointcut() {
}

@Around("pointcut2()")
public Object around(ProceedingJoinPoint proceedingJoinPoint) throws Throwable {
long startTime = System.currentTimeMillis();

 Object result = proceedingJoinPoint.proceed();
 

while (System.currentTimeMillis()-startTime<2000){
// 为了结果明显，在这里加了一个停止2 s的操作
}

long endTime = System.currentTimeMillis();
 System.out.println(endTime-startTime);

return result;
}

 

Controller中的执行putMap 操作

@GetMapping("/putMap")
@ResponseBody
public String putMap () throws Exception {

UUID temp = UUID.randomUUID();
 hashMap1.put(temp, UUID.randomUUID());

return temp.toString();
}

在每次请求 putMap 的url 时 会停止两秒中 再会显示结果。控制台输出时间为2000ms
2000
2000
2000
2000
不暂停2s 的话 记录的时间为 ：侧面说明如果我aop 成功的话，那hashmap 的put 操作速度是真的快啊，只有第一次put时因为要进行 初始化map 时间为 5ms 其余的操作都是0ms，，，，
5
0
0
0
0
0
0
0

public class PerformMonitorTransformer implements ClassFileTransformer {
 
    private static final String PACKAGE_PREFIX = "java.net.InetAddress";  //要增强的前缀
 
    // 通过 transform 来修改类文件
    @Override
    public byte[] transform(ClassLoader loader,
                            String className,
                            Class<?> classBeingRedefined,
                            ProtectionDomain protectionDomain,
                            byte[] classfileBuffer) throws IllegalClassFormatException {
        try {
            if (className == null){
                return classfileBuffer;
            }
            String currentClassName = className.replaceAll("/", ".");
            // 仅仅提升这个包中的类
            if (!currentClassName.equals(PACKAGE_PREFIX)) {
                return classfileBuffer;
            }
            System.out.println("now transform: [" + currentClassName + "]");
 
            // 得到类  -》》   决定对方法的 是否要增强。
            CtClass ctClass = ClassPool.getDefault().get(currentClassName);
            CtBehavior[] methods = ctClass.getDeclaredBehaviors();
         
            for (CtBehavior method : methods) {
                if (method.getName().equals("getAllByName"))//getByName
                    enhanceMethod(method); // 提升方法
            }
 
            return ctClass.toBytecode();  //类的字节数组
        } catch (Exception e) {
            e.printStackTrace();
        }
        return classfileBuffer;
 
    }
 
 
  
 /* 提升方法    javassist */
    private void enhanceMethod(CtBehavior method) throws Exception {
        if (method.isEmpty()) {
            return;
        }
        final String methodName = method.getName();
 
        ExprEditor editor = new ExprEditor() {
            @Override
            public void edit(MethodCall methodCall) throws CannotCompileException {
                //
                try {
                    methodCall.replace(genSource(methodName));
                } catch (ClassNotFoundException e) {
                    e.printStackTrace();
                }
            }
        };
        method.instrument(editor);
    }
  
     
/*  具体增强的实现 打入新的代码    关键实现逻辑 */
private String genSource(String methodName) throws ClassNotFoundException {
    StringBuilder source = new StringBuilder();
    source.append("{")
            .append("long start = System.currentTimeMillis();\n") // 前置增强: 打入时间戳
            .append("$_ = $proceed($$);\n") // 保留原有的代码处理逻辑
            .append("long spendTime =System.currentTimeMillis() -start;")
           // .append("this.getClass().getClassLoader().loadClass(\"instrument.HttpContext\");")     本来想采用 threadlocal 来存放记录的值 结果无法使用
            .append("if (spendTime!=0){")
            .append("System.setProperty(\"spendTime\", String.valueOf(spendTime));")  // 通过向系统变量中设置 dns 花费的时间来解决这问题，通过取出 System.getProperty("spendTime");         
           // .append("HttpContext.get().put(\"spendTime\",spendTime);")
            .append("System.out.print(\"method:[" + methodName + "]\");").append("\n")
            .append("System.out.println(\" cost:[\" +(spendTime)+ \"ms]\");}") // 后置增强
            .append("}");
    return source.toString();
 
}

System.getProperty("spendTime"); // 完后要  清除掉设置的这个时间
System.clearProperty("spendTime"); 

@GetMapping("/getDnsTime3")
@ResponseBody
public String getDnsTime3 () throws Exception {
InetAddress inetAddress = InetAddress.getByName("www.nowcoder.com");  // 对牛客网
 String spendTime = System.getProperty("spendTime");     //可取到这个值

 

System.out.println(spendTime);

 System.clearProperty("spendTime");                                    // 清理这个key



return inetAddress.toString()+" "+spendTime;    // 响应体 
}

对牛客网的域名测试
method:[getAllByName] cost:[17ms]
method:[getAllByName] cost:[17ms]
对b站的 测试时间
method:[getAllByName] cost:[2ms]
查看响应体 
www.nowcoder.com/114.55.207.244 17 
www.bilibili.com/103.41.165.46 2

AgentBuilder agentBuilder = new AgentBuilder.Default().ignore(none());

public static void premain (String arg , Instrumentation instrumentation) {

new AgentBuilder.Default()
.with(AgentBuilder.Listener.StreamWriting.toSystemError())
.ignore(none()) // 设置忽略
.disableClassFormatChanges()
.with(AgentBuilder.RedefinitionStrategy.REDEFINITION)
.with(AgentBuilder.TypeStrategy.Default.REDEFINE)
.type(named("java.lang.String"))
.transform((builder, typeDescription, classLoader) -> builder
.visit(Advice
.to(GetTimeAdvice.class) // 交给 addvice 类进行增强
.on(nameEndsWith("toString")) //.and(takesArguments(String.class)) 对toString()方法进行增强
 )
)
.installOn(instrumentation);
}

修改使用简单的 Advice 类

 

public class GetTimeAdvice {

    @Advice.OnMethodEnter
 public static long enter(@Advice.Origin Method method , @Advice.This Object thiz){


        long l = System.currentTimeMillis();
  System.out.println("进入的时间为 "+ l );
 return l;
 }


 @Advice.OnMethodExit
 public static void out( @Advice.Origin java.lang.reflect.Method method,
 @Advice.Enter long start
                             ){
        System.out.println("开始的时间为 "+start);

  System.out.println(" 花费的时间为： "+ String.valueOf(System.currentTimeMillis() - start)  );

 }


}

进入的时间为 1601346851430
开始的时间为 1601346851430
花费的时间为：0

进入的时间为 1601346851434
开始的时间为 1601346851434
花费的时间为: 0
进入的时间为 1601346851436
开始的时间为 1601346851436
花费的时间为：0
 
进入的时间为 1601346851437
 开始的时间为 1601346851437
花费的时间为：0
尝试对 String 的mathes（） 进行增强，只有一处进行了输出，说明增强成功
进入的时间为 1601347498553
开始的时间为 1601347498553
花费的时间为： 1
正式对 java.net.InetAddress 的增强
对www.newcoder.com 进行解析
[Byte Buddy] TRANSFORM java.net.InetAddress [null, null, loaded=true]
public static java.net.InetAddress[] java.net.InetAddress.getAllByName(java.lang.String) throws java.net.UnknownHostException took 72 ms
www.newcoder.com/121.199.77.57
对www.changyou.com/ 进行解析
[Byte Buddy] TRANSFORM java.net.InetAddress [null, null, loaded=true]
public static java.net.InetAddress[] java.net.InetAddress.getAllByName(java.lang.String) throws java.net.UnknownHostException took 8 ms
www.changyou.com/125.39.1.138

开始尝试时的代码（未成功）

public static void premain(String arguments, Instrumentation instrumentation) {


new AgentBuilder.Default()
.type(nameMatches("java.net.InetAddress")) // 指定类
.transform(new AgentBuilder.Transformer() {
@Override
 public DynamicType.Builder transform(DynamicType.Builder builder,
 TypeDescription typeDescription,
 ClassLoader classloader) {
return builder.method(named("getByName")) // 指定方法
.intercept(MethodDelegation.to(TimeInterceptor.class)); //增加委派方法

  }
}).installOn(instrumentation); //加入到 instrumenation中




}

 

对 hashmap 的改造

public static void premain(String argument, Instrumentation inst) {

System.out.println("start premain)");
 // HashMap<Object, Object> map = new HashMap<>();
 new AgentBuilder.Default()// bytebuddy.demo
 .type(named("java.util.HashMap").and(not(isInterface())).and(not(isStatic())))
.transform((builder, typeDescription, classLoader) -> builder
.method(ElementMatchers.named("put"))
.intercept(MethodDelegation.to(TimeInterceptor.class)
)
).with(new AgentBuilder.Listener() {
@Override
 public void onTransformation(TypeDescription typeDescription, ClassLoader classLoader, JavaModule javaModule, DynamicType dynamicType) {

}

@Override
 public void onIgnored(TypeDescription typeDescription, ClassLoader classLoader, JavaModule javaModule) {

}

@Override
 public void onError(String s, ClassLoader classLoader, JavaModule javaModule, Throwable throwable) {
throwable.printStackTrace();
 }

@Override
 public void onComplete(String s, ClassLoader classLoader, JavaModule javaModule) {

}
})
.installOn(inst);
}

 

 

//TimeInterceptor.class 的方法：

@RuntimeType
public static Object intercept(@Origin Method method, @SuperCall Callable<?> callable)
throws Exception {
// aop
 long start = System.currentTimeMillis();
 // 原有函数执行
 Object call = callable.call();

 System.out.println(method + ": took " + (System.currentTimeMillis() - start) + "ms");

return call;

}

经过对这两种方式的改造 ，但是都没有生效。

public static void main(String[] args) throws UnknownHostException {


      //比如说 调用这个方法 ，或者你可以写一个 httpclient 的get请求
     InetAddress byName2 = InetAddress.getByName("www.changyou.com");

 // 可以在System.getProperty("spendTime")  取出这个值 ，用到比如说 打日志的地方 
  System.out.println("dns的时间"+ System.getProperty("spendTime"));

}

this is an perform monitor agent.
now transform: [java.net.InetAddress]
method:[getAllByName] cost:[1ms]
method:[getAllByName] cost:[1ms]
method:[getAllByName] cost:[39ms]
method:[getAllByName] cost:[40ms]
dns的时间40