使用ByteBuddy生成带注解类时，注解无法被识别的问题

我之前踩过ByteBuddy注解识别的坑，你的情况大概率是这几个细节没处理对，给你梳理下：

1. 检查类加载器是否一致 #

这是最容易踩的坑！ByteBuddy默认的类加载策略可能会生成一个独立的类加载器，导致你检查注解时用的Entity.class和生成类上的注解类不是同一个类实例（类加载器不同，即使全类名相同也会被视为不同类）。

解决方法：生成类时指定当前上下文的类加载器，比如用你当前测试类的类加载器，加载策略用ClassLoadingStrategy.Default.WRAPPER：

Class<?> generatedClass = builder.make()
    .load(YourTestClass.class.getClassLoader(), ClassLoadingStrategy.Default.WRAPPER)
    .getLoaded();

2. 确认注解的添加方式正确 #

ByteBuddy添加类注解和字段注解的方法是分开的，要确保用对方法：

• 类级注解用annotateType()
• 字段级注解用annotateField()

示例代码片段：

// 构建@Entity注解实例
AnnotationDescription entityAnn = AnnotationDescription.Builder.ofType(Entity.class).build();
// 构建@Table注解实例（指定表名）
AnnotationDescription tableAnn = AnnotationDescription.Builder.ofType(Table.class)
        .define("name", "dynamic_test_table")
        .build();

// 生成类时添加类注解
DynamicType.Builder<?> builder = new ByteBuddy()
    .subclass(Object.class)
    .implement(Serializable.class)
    .annotateType(entityAnn)
    .annotateType(tableAnn);

// 给字段添加@Column注解
builder = builder.defineField("id", Long.class, Modifier.PRIVATE)
    .annotateField(AnnotationDescription.Builder.ofType(Column.class)
        .define("name", "id")
        .define("nullable", false)
        .build());

3. 验证AnnotationDescription的构建是否正确 #

有时候会因为注解属性定义错误导致注解无法被正常识别，比如遗漏了必填属性（不过JPA的注解大多属性可选），或者构建时引用了错误的注解类。

可以在生成类后打印注解信息来排查：

if (generatedClass.isAnnotationPresent(Entity.class)) {
    System.out.println("找到@Entity注解");
} else {
    // 打印所有类注解看看实际有什么
    for (Annotation ann : generatedClass.getAnnotations()) {
        System.out.println("实际存在的注解：" + ann.annotationType().getName());
    }
}

完整可运行示例 #

import net.bytebuddy.ByteBuddy;
import net.bytebuddy.description.annotation.AnnotationDescription;
import net.bytebuddy.dynamic.loading.ClassLoadingStrategy;
import net.bytebuddy.modifier.Modifier;

import javax.persistence.Column;
import javax.persistence.Entity;
import javax.persistence.Table;
import java.io.Serializable;

public class DynamicEntityGenerator {
    public static void main(String[] args) {
        // 构建类注解
        AnnotationDescription entityAnnotation = AnnotationDescription.Builder.ofType(Entity.class).build();
        AnnotationDescription tableAnnotation = AnnotationDescription.Builder.ofType(Table.class)
                .define("name", "dynamic_entity")
                .build();

        // 构建字段注解
        AnnotationDescription idColumn = AnnotationDescription.Builder.ofType(Column.class)
                .define("name", "id")
                .define("nullable", false)
                .build();

        // 生成并加载类
        Class<?> dynamicClass = new ByteBuddy()
                .subclass(Object.class)
                .implement(Serializable.class)
                .annotateType(entityAnnotation)
                .annotateType(tableAnnotation)
                .defineField("id", Long.class, Modifier.PRIVATE)
                .annotateField(idColumn)
                .make()
                .load(DynamicEntityGenerator.class.getClassLoader(), ClassLoadingStrategy.Default.WRAPPER)
                .getLoaded();

        // 测试断言
        assert dynamicClass.isAnnotationPresent(Entity.class) : "未识别到@Entity注解";
        assert dynamicClass.isAnnotationPresent(Table.class) : "未识别到@Table注解";

        System.out.println("注解识别成功！");
        System.out.println("表名：" + dynamicClass.getAnnotation(Table.class).name());
    }
}

按照上面的方法调整后，你的断言应该就能正常通过了。

内容的提问来源于stack exchange，提问作者khmarbaise