com.github.nezha.httpfetch.spring.HttpApiClassPathBeanDefinitionScanner Maven / Gradle / Ivy
package com.github.nezha.httpfetch.spring;
import com.github.nezha.httpfetch.HttpApiService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.AnnotatedBeanDefinition;
import org.springframework.beans.factory.config.BeanDefinitionHolder;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.beans.factory.support.GenericBeanDefinition;
import org.springframework.context.annotation.ClassPathBeanDefinitionScanner;
import org.springframework.core.type.classreading.MetadataReader;
import org.springframework.core.type.classreading.MetadataReaderFactory;
import org.springframework.core.type.filter.TypeFilter;
import java.io.IOException;
import java.util.Set;
/**
* Created by daiqiang on 17/7/26.
*/
public class HttpApiClassPathBeanDefinitionScanner extends ClassPathBeanDefinitionScanner {
private static final Logger LOGGER = LoggerFactory.getLogger(HttpApiClassPathBeanDefinitionScanner.class);
public HttpApiClassPathBeanDefinitionScanner(BeanDefinitionRegistry registry) {
super(registry);
}
public void register(){
//仅加载HttpApi 注解的class
addIncludeFilter(new TypeFilter(){
public boolean match(MetadataReader metadataReader, MetadataReaderFactory metadataReaderFactory) throws IOException {
//只支持接口
return true;
}
});
}
protected Set doScan(String... basePackages) {
Set beanDefinitionHolders = super.doScan(basePackages);
for(BeanDefinitionHolder holder : beanDefinitionHolders){
GenericBeanDefinition definition = (GenericBeanDefinition) holder.getBeanDefinition();
try {
Class clazz = Class.forName(definition.getBeanClassName());
definition.getPropertyValues().add("targetClass", clazz);
definition.setBeanClass(HttpApiFactoryBean.class);
} catch (ClassNotFoundException e) {
LOGGER.error("class not found! className [{}]", definition.getBeanClassName(), e);
throw new RuntimeException("class not found!");
}
}
return beanDefinitionHolders;
}
/**
* {@inheritDoc}
*/
@Override
protected boolean isCandidateComponent(AnnotatedBeanDefinition beanDefinition) {
return (beanDefinition.getMetadata().isInterface() && beanDefinition.getMetadata().isIndependent());
}
}