Java Class-File API 怎么转换方法代码属性

在 Java 24 中，Class-File API 已经作为正式 API 放进 java.lang.classfile。如果任务是“修改方法体中的指令，同时保留类、字段和方法的其他结构”，最短路径不是手工重建整个 Code 属性，而是先定义 CodeTransform，再通过 ClassTransform.transformingMethodBodies 把它应用到方法体。

下面用一个可控的小任务说明完整写法：读取现有 class 文件，删除所有 nop 指令，再写出新的 class 字节。nop 没有可见状态，适合用来观察“丢弃一个代码元素”时转换器如何工作。

先明确模型：Code 属性不是一段裸字节

Class-File API 把 class 文件拆成可遍历的模型。ClassModel 包含方法模型，方法体由 CodeModel 表示；实际 class 文件中的 Code 属性由 CodeAttribute 建模，而 CodeAttribute 本身也是 CodeModel。再往下，字节码指令、标签、行号和局部变量信息会作为 CodeElement 出现在遍历结果中。

因此，转换方法代码属性时通常处理的是 CodeElement 流，而不是自己计算指令偏移、异常表范围或属性长度。转换器对每个元素有三种基本选择：传递、丢弃，或者向构建器写入替代元素。

最小 CodeTransform：删除 nop 指令

CodeTransform 是函数式接口，回调参数分别是目标 CodeBuilder 和当前 CodeElement。要保留元素就调用 builder.with(element)；要删除元素则不写入构建器。

import java.lang.classfile.CodeTransform;
import java.lang.classfile.instruction.NopInstruction;

// 删除 nop，其余代码元素原样传递给新的 CodeBuilder
CodeTransform removeNops = (builder, element) -> {
    if (!(element instanceof NopInstruction)) {
        builder.with(element);
    }
};

这里判断的是 NopInstruction，它是 CodeElement 的一种。标签、异常处理边界、行号等其他元素不会命中条件，所以仍会进入新方法体。这个写法也展示了 Class-File API 的核心语义：转换本质上是对元素序列做选择性映射。

把代码转换提升到整个类

单独的 CodeTransform 只描述“遇到代码元素时怎么办”。要处理现有类中每个带方法体的方法，可用 ClassTransform.transformingMethodBodies(removeNops) 把它提升为类转换器。完整程序如下：

import java.lang.classfile.ClassFile;
import java.lang.classfile.ClassModel;
import java.lang.classfile.ClassTransform;
import java.lang.classfile.CodeTransform;
import java.lang.classfile.instruction.NopInstruction;
import java.nio.file.Files;
import java.nio.file.Path;

public class RemoveNop {
    public static void main(String[] args) throws Exception {
        // 第一个参数是输入 class，第二个参数是输出 class
        Path input = Path.of(args[0]);
        Path output = Path.of(args[1]);

        ClassFile classFile = ClassFile.of();
        byte[] sourceBytes = Files.readAllBytes(input);
        ClassModel model = classFile.parse(sourceBytes);

        // 不向 builder 写入 NopInstruction，即表示删除该元素
        CodeTransform removeNops = (builder, element) -> {
            if (!(element instanceof NopInstruction)) {
                builder.with(element);
            }
        };

        // 把代码级转换应用到类中所有存在 CodeModel 的方法
        ClassTransform transform =
                ClassTransform.transformingMethodBodies(removeNops);

        // transformClass 返回重新生成的完整 class 字节
        byte[] result = classFile.transformClass(model, transform);
        Files.write(output, result);
    }
}

编译和运行程序时需要 JDK 24 或更高版本。示例程序本身接收一个待处理的 .class 文件，因此可以把输入与输出分开保存，避免直接覆盖原文件。

# 使用 JDK 24 编译转换程序
javac RemoveNop.java

# 读取 Sample.class，并把转换结果写到 out/Sample.class
mkdir -p out
java RemoveNop Sample.class out/Sample.class

transformingMethodBodies 会处理方法模型中的代码部分，并把其他类元素、方法元素继续传给对应构建器。也就是说，调用方不必为了改一条指令而重新声明字段、方法签名、注解或常量池。

只转换指定方法

如果只想处理名为 compute 的方法，可使用带 Predicate 的重载。方法名是常量池中的 UTF-8 项，使用 equalsString 比较可以直接表达目标名称。

ClassTransform onlyCompute =
        ClassTransform.transformingMethodBodies(
                // 只有方法名为 compute 时才进入 removeNops
                method -> method.methodName().equalsString("compute"),
                removeNops
        );

// 其余方法体和其他类元素保持传递
byte[] result = classFile.transformClass(model, onlyCompute);

只按名称过滤时需要注意重载方法：同名的多个 compute 都会匹配。如果目标是某个确定签名，还应同时比较方法类型描述符。对于已经处于方法级转换逻辑中的代码，也可以用 MethodTransform.transformingCode(removeNops)，语义相同，只是组合层级更低。

为什么不要直接替换整个 CodeAttribute

手工构造新的 CodeAttribute 往往意味着调用方要同时照顾指令顺序、标签绑定、异常处理范围、局部变量信息和栈映射等细节。使用 CodeTransform 时，未修改的元素可以直接传递，新的代码结构交给 CodeBuilder 生成，更适合局部修改。

不过，“使用转换器”不等于所有属性都必然无条件保留。官方文档特别提醒，自定义或未知属性在结构变化后可能无法证明仍然有效，具体保留策略受 ClassFile 选项影响。若项目依赖代理框架、覆盖率工具或编译器私有属性，应在真实产物上做回归测试。

边界情况速查

结论

转换方法代码属性时，可以把思路固定成三层：CodeTransform 决定如何处理代码元素，ClassTransform.transformingMethodBodies 把它应用到方法体，ClassFile.transformClass 负责生成新的 class 字节。这样既能进行局部字节码修改，也能让未涉及的类和方法结构按模型传递。

官方参考：JEP 484：Class-File API、CodeTransform 文档、ClassTransform 文档。