Question

我必须在运行时修改以下类来打印实例变量的值＆＃39; count＆＃39;在每种方法的最后。

package test.hib.javaassist;

import java.io.IOException;

import javassist.CannotCompileException;
import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtMethod;
import javassist.NotFoundException;

public class JavaAssistTest {

    int count;


    public void doSomething1(){
        count++;
    }

    public void doSomething2(){
        count++;
    }

    public void doSomething3(){
        count++;
    }

    public void doSomething4(){
        count++;
    }

}

以下是主类，我试图在javaassist的帮助下改变字节码。

package test.hib.javaassist;

import java.io.IOException;

import javassist.CannotCompileException;
import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtMethod;
import javassist.NotFoundException;

public class Main {

    public static void main(String[] args) throws NotFoundException, CannotCompileException, IOException {

        ClassPool pool = ClassPool.getDefault();
        CtClass cc = pool.get("test.hib.javaassist.JavaAssistTest");
        CtMethod[] methods = cc.getDeclaredMethods();

        for(CtMethod method : methods){
            if(! (method.getName().equals("main"))){
                method.insertAfter("{System.out.println(count);}");
                //method.insertAfter("System.out.println($type);");
            }
        }

        cc.writeFile();

        System.out.println("Completed editting");

        JavaAssistTest test = new JavaAssistTest();
        test.doSomething1();
        test.doSomething2();
        test.doSomething3();
        test.doSomething4();

        System.out.println("Finished");

    }

}

目前正在打印

Completed editting
Finished

我希望它打印

Completed editting
1
2
3
4
Finished

你可以在我的代码中指出错误吗？

Answer 1

Javassist API在这里有点令人困惑。看看writeFile的实现。它是writeFile(".")的快捷方式，它将类保存在程序目录中而不是类路径上。您希望它覆盖原始类文件，此方法不执行此操作。您的系统类加载器将无法找到更新的类文件。

您可以使用writeFile(String)手动将类文件保存在正确的位置，或者通过调用cc.toClass()强制加载修改后的类，这会强制在类加载器查看之前加载修改后的类手动。

使用javasssist打印实例变量

1 个答案: