如何通过JBoss可移植扩展在运行时为注解类的接口添加注解？

需求概述 #

实现运行时为带有@DomainDTO注解的类所实现的接口添加@RegisteredDTO注解，示例如下：

修改前代码 #

@DomainDTO
public class SomeDto implements CheckedDto { ... }

public interface CheckedDto { ... }

修改后预期效果 #

@RegisteredDTO(dto = {SomeDto.class})
public interface CheckedDto { ... }

现有尝试与问题 #

使用JBoss可移植扩展的ProcessAnnotatedType观察者方法处理，但仅知晓如何为被观察的类（如SomeDto）添加注解，不清楚如何修改其实现的接口（如CheckedDto）。尝试过Javaassist和ByteBuddyAgent，但在JBoss环境下无法生效。

现有代码片段：

<T> void processAnnotatedType( @Observes ProcessAnnotatedType<T> event, BeanManager beanManager )
{
    Class<T> javaClass = event.getAnnotatedType()
        .getJavaClass();
    if( javaClass.isAnnotationPresent( DomainDTO.class ) )
    {
        // 获取SomeDto实现的接口

        // 创建RegisteredDTO注解实例

    }

}

解决方案 #

步骤1：通过CDI扩展收集并修改元数据 #

利用CDI扩展的生命周期事件，先收集目标类与接口的映射，再在接口的元数据处理阶段添加注解，这是JBoss环境下的可移植方案，无需直接修改字节码：

import javax.enterprise.context.spi.AnnotatedType;
import javax.enterprise.event.Observes;
import javax.enterprise.inject.spi.Extension;
import javax.enterprise.inject.spi.ProcessAnnotatedType;
import javax.interceptor.Interceptor;
import java.util.*;
import java.util.stream.Collectors;

@Priority(Interceptor.Priority.PLATFORM_BEFORE)
public class DtoRegistrationExtension implements Extension {
    // 存储带@DomainDTO的类与对应实现接口的映射
    private final Map<Class<?>, Set<Class<?>>> dtoInterfaceMap = new HashMap<>();

    // 第一阶段：收集所有带@DomainDTO的类及其实现的目标接口
    <T> void collectDtoClasses(@Observes ProcessAnnotatedType<T> patEvent) {
        Class<T> dtoClass = patEvent.getAnnotatedType().getJavaClass();
        if (dtoClass.isAnnotationPresent(DomainDTO.class)) {
            // 过滤出需要处理的接口（这里以CheckedDto及其子类为例）
            Set<Class<?>> targetInterfaces = Arrays.stream(dtoClass.getInterfaces())
                    .filter(iface -> CheckedDto.class.isAssignableFrom(iface))
                    .collect(Collectors.toSet());
            if (!targetInterfaces.isEmpty()) {
                dtoInterfaceMap.put(dtoClass, targetInterfaces);
            }
        }
    }

    // 第二阶段：处理接口的元数据，添加@RegisteredDTO注解
    <T> void processInterfaceMetadata(@Observes ProcessAnnotatedType<T> patEvent) {
        Class<T> interfaceClass = patEvent.getAnnotatedType().getJavaClass();
        // 仅处理目标接口类型
        if (interfaceClass.isInterface() && CheckedDto.class.isAssignableFrom(interfaceClass)) {
            // 找出所有实现该接口的带@DomainDTO的类
            List<Class<?>> matchedDtos = dtoInterfaceMap.entrySet().stream()
                    .filter(entry -> entry.getValue().contains(interfaceClass))
                    .map(Map.Entry::getKey)
                    .collect(Collectors.toList());

            if (!matchedDtos.isEmpty()) {
                // 构建新的AnnotatedType并添加注解
                AnnotatedTypeBuilder<T> typeBuilder = new AnnotatedTypeBuilder<>();
                typeBuilder.readFromType(patEvent.getAnnotatedType());

                // 创建@RegisteredDTO注解实例（需实现注解的所有方法）
                RegisteredDTO registeredAnnotation = new RegisteredDTO() {
                    @Override
                    public Class<? extends Annotation> annotationType() {
                        return RegisteredDTO.class;
                    }

                    @Override
                    public Class<?>[] dto() {
                        return matchedDtos.toArray(new Class[0]);
                    }

                    // 若注解有其他属性，需在此处补充实现
                };

                typeBuilder.addAnnotation(registeredAnnotation);
                // 替换原有的AnnotatedType，让CDI容器感知到修改
                patEvent.setAnnotatedType(typeBuilder.create());
            }
        }
    }
}

步骤2：关键注意事项 #

• 注解兼容性：确保@RegisteredDTO的dto参数支持数组类型，若多个类实现同一接口，可自动合并到数组中；若注解不支持重复，需提前设计好参数结构。
• 扩展优先级：通过@Priority注解设置扩展的执行顺序，确保在其他可能处理接口元数据的扩展之前执行，避免覆盖修改。
• 避免直接修改字节码：JBoss/WildFly的模块化类加载机制会隔离类资源，直接修改字节码无法被CDI容器的元数据扫描流程感知，而通过ProcessAnnotatedType修改元数据是官方推荐的可移植方式。

内容的提问来源于stack exchange，提问作者tomasz-mer