AOP или APT для переопределения методов из суперклассов - PullRequest
3 голосов
/ 06 сентября 2010

У меня есть большая библиотека компонентов калитки, аннотированная пользовательской аннотацией @ReferencedResource или другой аннотацией @ReferencedResources, в которой есть параметр ReferencedResouce[] value(), разрешающий несколько аннотаций.

Вот пример кода:

@ReferencedResources({
    @ReferencedResource(value = Libraries.MOO_TOOLS, type = ResourceType.JAVASCRIPT),
    @ReferencedResource(value = "behaviors/promoteSelectOptions", type = ResourceType.JAVASCRIPT) })
public class PromoteSelectOptionsBehavior extends AbstractBehavior{
 ...
}

Пока что я использую apt , чтобы проверить, что ссылочные ресурсы действительно существуют. Например.

@ReferencedResource(value = "behaviors/promoteSelectOptions",
                     type = ResourceType.JAVASCRIPT)

приведет к ошибке компиляции, если только файл js/behaviors/promoteSelectOptions.js не найден в пути к классам. Эта часть прекрасно работает.

Теперь я также фанат DRY и хотел бы использовать ту же аннотацию для фактического внедрения ресурсов в Объекты при их создании. Используя AspectJ, я реализовал часть этого.

Аннотированные Объекты всегда являются экземплярами Компонент или AbstractBehavior .

Для компонентов все просто, просто совпадают после конструктора. Вот совет, который делает это:

pointcut singleAnnotation() : @within(ReferencedResource);

pointcut multiAnnotation() : @within(ReferencedResources);

after() : execution(Component+.new(..)) && (singleAnnotation() || multiAnnotation()){
    final Component component = (Component) thisJoinPoint.getTarget();
    final Collection<ReferencedResource> resourceAnnotations =
        // gather annotations from cache
        this.getResourceAnnotations(component.getClass());
    for(final ReferencedResource annotation : resourceAnnotations){
        // helper utility that handles the creation of statements like
        // component.add(JavascriptPackageResource.getHeaderContribution(path))
        this.resourceInjector.inject(component, annotation);
    }
}

Однако для поведения мне нужно прикрепить ресурсы к ответу, а не к самому поведению. Вот я использую pointcuts:

pointcut renderHead(IHeaderResponse response) :
    execution(* org.apache.wicket.behavior.AbstractBehavior+.renderHead(*))
        && args(response);

А вот и совет:

before(final IHeaderResponse response) : 
    renderHead(response) && (multiAnnotation() || singleAnnotation()) {
    final Collection<ReferencedResource> resourceAnnotations =
        this.getResourceAnnotations(thisJoinPoint.getTarget().getClass());
    for(final ReferencedResource resource : resourceAnnotations){
        this.resourceInjector.inject(response, resource);
    }
}

Это также хорошо работает, если класс переопределяет метод renderHead (response) , но во многих случаях это просто не нужно, поскольку суперкласс уже реализует базовую функциональность, в то время как дочерний класс добавляет только некоторую конфигурацию. Таким образом, одним из решений было бы позволить этим классам определять метод следующим образом:

@Override
public void renderHead(IHeaderResponse response){
    super.renderHead(response);
}

Я бы ненавидел это, потому что это мертвый код, но в настоящее время это единственная рабочая опция, которую я вижу, поэтому я ищу другие решения.

EDIT:

Я создал рабочее решение, используя вызовы APT и Sun Javac. Однако это приводит к следующей проблеме: Запуск APT и AspectJ в одном проекте с использованием maven .

В любом случае, как только у меня будет свободное время, я опубликую ответ на этот вопрос (или его части).

1 Ответ

4 голосов
/ 10 сентября 2010

Отвечая на мой собственный вопрос:

Вот соответствующий бит кода для вставки супер-вызова:

все эти поля инициализируются в init (env) или process (annotations, roundEnv) :

private static Filer filer;
private static JavacProcessingEnvironment environment;
private static Messager messager;
private static Types types;
private static JavacElements elementUtils;
private Trees trees;
private TreeMaker treeMaker;
private IdentityHashMap<JCCompilationUnit, Void> compilationUnits;
private Map<String, JCCompilationUnit> typeMap;

А вот логика, которая вызывается, если подтип AbstractBehavior, имеющий аннотацию, не переопределяет метод renderHead(response):

private void addMissingSuperCall(final TypeElement element){
    final String className = element.getQualifiedName().toString();
    final JCClassDecl classDeclaration =
        // look up class declaration from a local map 
        this.findClassDeclarationForName(className);
    if(classDeclaration == null){
        this.error(element, "Can't find class declaration for " + className);
    } else{
        this.info(element, "Creating renderHead(response) method");
        final JCTree extending = classDeclaration.extending;
        if(extending != null){
            final String p = extending.toString();
            if(p.startsWith("com.myclient")){
                // leave it alone, we'll edit the super class instead, if
                // necessary
                return;
            } else{
                // @formatter:off (turns off eclipse formatter if configured)

                // define method parameter name
                final com.sun.tools.javac.util.Name paramName =
                    elementUtils.getName("response");
                // Create @Override annotation
                final JCAnnotation overrideAnnotation =
                    this.treeMaker.Annotation(
                        Processor.buildTypeExpressionForClass(
                            this.treeMaker,
                            elementUtils,
                            Override.class
                        ),
                        // with no annotation parameters
                        List.<JCExpression> nil()
                    );
                // public
                final JCModifiers mods =
                    this.treeMaker.Modifiers(Flags.PUBLIC,
                        List.of(overrideAnnotation));
                // parameters:(final IHeaderResponse response)
                final List<JCVariableDecl> params =
                    List.of(this.treeMaker.VarDef(this.treeMaker.Modifiers(Flags.FINAL),
                        paramName,
                        Processor.buildTypeExpressionForClass(this.treeMaker,
                            elementUtils,
                            IHeaderResponse.class),
                        null));

                //method return type: void
                final JCExpression returnType =
                    this.treeMaker.TypeIdent(TypeTags.VOID);


                // super.renderHead(response);
                final List<JCStatement> statements =
                    List.<JCStatement> of(
                        // Execute this:
                        this.treeMaker.Exec(
                            // Create a Method call:
                            this.treeMaker.Apply(
                                // (no generic type arguments)
                                List.<JCExpression> nil(),
                                // super.renderHead
                                this.treeMaker.Select(
                                    this.treeMaker.Ident(
                                        elementUtils.getName("super")
                                    ),
                                    elementUtils.getName("renderHead")
                                ),
                                // (response)
                                List.<JCExpression> of(this.treeMaker.Ident(paramName)))
                            )
                     );
                // build code block from statements
                final JCBlock body = this.treeMaker.Block(0, statements);
                // build method
                final JCMethodDecl methodDef =
                    this.treeMaker.MethodDef(
                        // public
                        mods,
                        // renderHead
                        elementUtils.getName("renderHead"),
                        // void
                        returnType,
                        // <no generic parameters>
                        List.<JCTypeParameter> nil(),
                        // (final IHeaderResponse response)
                        params,
                        // <no declared exceptions>
                        List.<JCExpression> nil(),
                        // super.renderHead(response);
                        body,
                        // <no default value>
                        null);

                // add this method to the class tree
                classDeclaration.defs =
                    classDeclaration.defs.append(methodDef);

                // @formatter:on turn eclipse formatter on again
                this.info(element,
                    "Created renderHead(response) method successfully");

            }
        }

    }
}
...