org.ff4j.aop.FeatureAutoProxy Maven / Gradle / Ivy

Go to download
package org.ff4j.aop;

/*
 * #%L ff4j-aop %% Copyright (C) 2013 Ff4J %% Licensed under the Apache License, Version 2.0 (the "License"); you may not use this
 * file except in compliance with the License. You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS"
 * BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language
 * governing permissions and limitations under the License. #L%
 */

import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.Map;

import org.springframework.aop.TargetSource;
import org.springframework.aop.framework.autoproxy.AbstractAutoProxyCreator;
import org.springframework.stereotype.Component;

/**
 * This bean will automatically process every in applicationContext.
 * 
 * @author Cedrick LUNVEN
 */
@Component("ff.autoproxy")
public class FeatureAutoProxy extends AbstractAutoProxyCreator {

    /** serial number. */
    private static final long serialVersionUID = -364406999854610869L;

    /** Processed Interfaces. */
    private final Map processedInterface = new HashMap();

    /**
     * Constructor invoked through IoC per Spring
     */
    public FeatureAutoProxy() {
        setInterceptorNames(new String[] {FeatureAdvisor.class.getAnnotation(Component.class).value()});
    }

    /** {@inheritDoc} */
    @Override
    protected Object[] getAdvicesAndAdvisorsForBean(Class beanClass, String beanName, TargetSource targetSource) {
        // Scan interface only once.
        if (!beanClass.isInterface() && beanClass.getInterfaces() != null) {
            // Get Interface
            for (Class currentInterface : beanClass.getInterfaces()) {
                Object[] r = addAnnotedInterface(currentInterface);
                if (r != null) {
                    return r;
                }
            }
        }
        return DO_NOT_PROXY;
    }

    private Object[] addAnnotedInterface(Class currentInterface) {
        String currentInterfaceName = currentInterface.getCanonicalName();
        if (!currentInterfaceName.startsWith("java.")) {
            // Avoid process same interface several times
            Boolean isInterfaceFlipped = processedInterface.get(currentInterfaceName);
            if (isInterfaceFlipped != null) {
                if (isInterfaceFlipped) {
                    return PROXY_WITHOUT_ADDITIONAL_INTERCEPTORS;
                }
            } else {
               if (currentInterface.isAnnotationPresent(Flip.class)) {
                   // If annotation is registered on Interface class 
                   processedInterface.put(currentInterfaceName, true);
                    return PROXY_WITHOUT_ADDITIONAL_INTERCEPTORS;
                } else {
                    for (Method method : currentInterface.getDeclaredMethods()) {
                        if (method.isAnnotationPresent(Flip.class)) {
                            processedInterface.put(currentInterfaceName, true);
                            return PROXY_WITHOUT_ADDITIONAL_INTERCEPTORS;
                        }
                    }
                    processedInterface.put(currentInterfaceName, false);
                }
            }
        }
        return null;
    }
}