org.constretto.spring.internal.ConstrettoAutowireCandidateResolver Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of constretto-spring Show documentation

There is a newer version: 3.0.0-BETA1

/*
 * Copyright 2008 the original author or authors. 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.
 */
package org.constretto.spring.internal;

import org.constretto.spring.EnvironmentAnnotationConfigurer;
import org.constretto.spring.annotation.Environment;
import org.springframework.beans.factory.annotation.QualifierAnnotationAutowireCandidateResolver;
import org.springframework.beans.factory.config.BeanDefinitionHolder;
import org.springframework.beans.factory.config.DependencyDescriptor;
import org.springframework.beans.factory.support.AutowireCandidateResolver;
import org.springframework.core.MethodParameter;

import static org.constretto.spring.EnvironmentAnnotationConfigurer.findEnvironmentAnnotation;

/**
 * Internal class that reads @Environment annotations on classes and removes all classes that are not annotated
 * with the current environment as autowire candidates.
 *
 * @author Kaare Nilsen
 */
public class ConstrettoAutowireCandidateResolver extends QualifierAnnotationAutowireCandidateResolver {

    @SuppressWarnings("unchecked")
    public boolean isAutowireCandidate(BeanDefinitionHolder bdHolder, DependencyDescriptor descriptor) {
        if (bdHolder.getBeanDefinition().getAttribute(EnvironmentAnnotationConfigurer.INCLUDE_IN_COLLECTIONS) != null && descriptor.getCollectionType() != null) {
            Class[] interfaces = (Class[]) bdHolder.getBeanDefinition().getAttribute(EnvironmentAnnotationConfigurer.INCLUDE_IN_COLLECTIONS);
            for (Class anInterface : interfaces) {
                if (descriptor.getCollectionType().isAssignableFrom(anInterface)) {
                    return true;
                }
            }
        }

        if (descriptor.getDependencyType().isInterface()){
            return super.isAutowireCandidate(bdHolder, descriptor);
        } else {
            return findEnvironmentAnnotation(descriptor.getDependencyType()) == null || super.isAutowireCandidate(bdHolder, descriptor);
        }

    }    
}