Добавление аннотации к сгенерированному во время выполнения методу / классу с использованием Javassist - PullRequest
Новая
       32

Добавление аннотации к сгенерированному во время выполнения методу / классу с использованием Javassist

20 голосов
/ 03 июня 2010

Я использую Javassist для создания класса foo с методом bar, но я не могу найти способ добавить аннотацию (сама аннотация не является средой выполнения генерируется) к методу. Код, который я пробовал, выглядит следующим образом: 

ClassPool pool = ClassPool.getDefault();

// create the class
CtClass cc = pool.makeClass("foo");

// create the method
CtMethod mthd = CtNewMethod.make("public Integer getInteger() { return null; }", cc);
cc.addMethod(mthd);

ClassFile ccFile = cc.getClassFile();
ConstPool constpool = ccFile.getConstPool();

// create the annotation
AnnotationsAttribute attr = new AnnotationsAttribute(constpool, AnnotationsAttribute.visibleTag);
Annotation annot = new Annotation("MyAnnotation", constpool);
annot.addMemberValue("value", new IntegerMemberValue(ccFile.getConstPool(), 0));
attr.addAnnotation(annot);
ccFile.addAttribute(attr);

// generate the class
clazz = cc.toClass();

// length is zero
java.lang.annotation.Annotation[] annots = clazz.getAnnotations();

И, очевидно, я делаю что-то не так, поскольку annots - пустой массив.

Вот так выглядит аннотация: 

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface MyAnnotation {
    int value();
}

1 Ответ

25 голосов
/ 03 июня 2010

Решил это в итоге, я добавлял аннотацию не туда. Я хотел добавить его в метод, но я добавлял его в класс.

Вот так выглядит фиксированный код: 

// wrong
ccFile.addAttribute(attr);

// right
mthd.getMethodInfo().addAttribute(attr);
...