/***
* Copyright (c) 2009 Caelum - www.caelum.com.br/opensource
* All rights reserved.
*
* 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 br.com.caelum.vraptor.ioc.spring;
import java.util.Arrays;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.support.AbstractBeanDefinition;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.beans.factory.support.BeanNameGenerator;
import org.springframework.beans.factory.support.DefaultListableBeanFactory;
import org.springframework.context.annotation.AnnotationBeanNameGenerator;
import org.springframework.context.annotation.ClassPathBeanDefinitionScanner;
import br.com.caelum.vraptor.ioc.ComponentFactory;
import br.com.caelum.vraptor.ioc.Container;
/**
* @author Fabio Kung
*/
class ComponentScanner extends ClassPathBeanDefinitionScanner {
private static final Logger logger = LoggerFactory.getLogger(ComponentScanner.class);
private final DefaultListableBeanFactory registry;
private final Container container;
public ComponentScanner(DefaultListableBeanFactory registry, Container container) {
super(registry, false);
this.registry = registry;
this.container = container;
addIncludeFilter(new ComponentTypeFilter());
setScopeMetadataResolver(new VRaptorScopeResolver());
setBeanNameGenerator(new UniqueBeanNameGenerator(new AnnotationBeanNameGenerator()));
}
public static class UniqueBeanNameGenerator implements BeanNameGenerator {
private final BeanNameGenerator delegate;
public UniqueBeanNameGenerator(BeanNameGenerator delegate) {
this.delegate = delegate;
}
public String generateBeanName(BeanDefinition definition, BeanDefinitionRegistry registry) {
String name = delegate.generateBeanName(definition, registry);
while (registry.containsBeanDefinition(name) &&
!registry.getBeanDefinition(name).getBeanClassName().equals(definition.getBeanClassName())) {
name = name + "$";
}
return name;
}
}
@SuppressWarnings("unchecked")
@Override
protected void postProcessBeanDefinition(AbstractBeanDefinition beanDefinition, String beanName) {
super.postProcessBeanDefinition(beanDefinition, beanName);
beanDefinition.setPrimary(true);
try {
Class<?> componentType = Class.forName(beanDefinition.getBeanClassName());
if (ComponentFactory.class.isAssignableFrom(componentType) && checkCandidate(beanName, beanDefinition)) {
registry.registerSingleton(beanDefinition.getBeanClassName(), new ComponentFactoryBean(container,
componentType));
}
} catch (ClassNotFoundException e) {
logger.warn("Class " + beanDefinition.getBeanClassName()
+ " was not found during bean definition proccess");
}
catch (ExceptionInInitializerError e) {
// log and rethrow antipattern is needed, this is rally important
logger.warn("Class " + beanDefinition.getBeanClassName()
+ " has problems during initialization", e.getCause());
throw e;
}
}
@Override
public int scan(String... basePackages) {
logger.debug("scanning " + Arrays.toString(basePackages));
return super.scan(basePackages);
}
@Override
protected boolean checkCandidate(String beanName, BeanDefinition beanDefinition) throws IllegalStateException {
if (registry.containsBeanDefinition(beanName) && registry.getBeanDefinition(beanName).getBeanClassName().equals(beanDefinition.getBeanClassName())) {
logger
.warn("bean already found previously, there is probably no need to declare its package in web.xml:"
+ beanDefinition.getBeanClassName());
return false;
}
return super.checkCandidate(beanName, beanDefinition);
}
}