我目前正在实现一些代码,对于类的每个方法,应该在 .class 文件上运行几个访问者,以检测其字节码。目前,我刚刚实现了一个 MethodRenamerVisitor,但这已经变得很尴尬:
ClassReader classReader = null;
try {
classReader = new ClassReader(monitoringFile.getCannonicalName());
} catch (IOException e1) {
e1.printStackTrace();
}
ClassWriter classWriter = null;
for (BytecodeMethod bytecodeMethod : bytecodeClass.bytecodeMethods) {
System.out.println("\t" + bytecodeMethod.getName());
classWriter = new ClassWriter(ClassWriter.COMPUTE_FRAMES);
MethodRenamerVisitor methodRenamerVisitor = new MethodRenamerVisitor(classWriter, bytecodeMethod);
classReader.accept(methodRenamerVisitor, 0);
String outputFile = monitoringFile.getCannonicalName();
outputFile = outputFile.replace('.', File.separatorChar)+".class";
classReader = new ClassReader(classWriter.toByteArray());
}
try {
fileSystem.writeToDisk(
classFilename,
classWriter.toByteArray()
);
} catch (FileNotFoundException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
我现在希望将其他访问者应用于每个 BytecodeMethods。我是否过于复杂了?如果我能有以下形式的东西,那将是理想的:
MultiVisitors multiVisitors = new MultiVisitors();
visitors.add(new AVisitor(...));
visitors.add(new BVisitor(...));
...
visitors.run();
所以我可以轻松地堆叠我想要运行的所有访问者,然后只有最后我必须将它们保存在磁盘上。请记住,每个访问者都会添加/删除/修改它正在访问的方法!