/*
* This file is part of rockframework.
*
* rockframework is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* rockframework is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>;.
*/
package br.net.woodstock.rockframework.domain.spring;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.context.annotation.ClassPathScanningCandidateComponentProvider;
import org.springframework.core.type.filter.TypeFilter;
import br.net.woodstock.rockframework.util.Assert;
public class AbstractEntityDetector implements EntityDetector {
private ClassPathScanningCandidateComponentProvider provider;
private String basePackage;
private List<Class<?>> classes;
public AbstractEntityDetector(final TypeFilter filter) {
this(filter, false);
}
public AbstractEntityDetector(final TypeFilter filter, final boolean useDefaultFilters) {
super();
Assert.notNull(filter, "filter");
this.provider = new ClassPathScanningCandidateComponentProvider(useDefaultFilters);
this.provider.addIncludeFilter(filter);
}
public void setBasePackage(final String basePackage) {
this.basePackage = basePackage;
}
@Override
public List<Class<?>> getClasses() throws ClassNotFoundException {
if (this.classes == null) {
this.classes = new ArrayList<Class<? extends Object>>();
Set<BeanDefinition> set = this.provider.findCandidateComponents(this.basePackage);
for (BeanDefinition beanDefinition : set) {
this.classes.add(Class.forName(beanDefinition.getBeanClassName()));
}
}
return this.classes;
}
}