Spring BeanRegistry


import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.config.BeanDefinitionHolder;
import org.springframework.beans.factory.config.BeanFactoryPostProcessor;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
import org.springframework.beans.factory.support.*;
import org.springframework.cglib.core.ReflectUtils;
import org.springframework.stereotype.Component;

import java.util.function.Supplier;

/**
 * @author qbit
 */
@Component
public class ControllerRegistry implements BeanDefinitionRegistryPostProcessor {

    private DynamicCompiler compiler;
    @Override
    public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry beanDefinitionRegistry) throws BeansException {
        SrcGenerator srcGenerator=new SrcGenerator();
        new ControllerScanner().controllers()
                .forEach(clazz->{
                    if(null==compiler){
                        compiler=new DynamicCompiler(clazz.getClassLoader());
                    }
                    final String className=srcGenerator.implClassName(clazz);
                    final String name= StringUtils.substringAfterLast(className,".");
                    final Class implClass=compiler.compile(className,srcGenerator.implSrc(clazz));
                    final Object impl= ReflectUtils.newInstance(implClass);
                    BeanDefinitionBuilder beanDefinitionBuilder=BeanDefinitionBuilder.genericBeanDefinition(implClass, () -> impl);
                    BeanDefinition beanDefinition=beanDefinitionBuilder.getBeanDefinition();
                    beanDefinition.setBeanClassName(className);
                    BeanDefinitionHolder holder=new BeanDefinitionHolder(beanDefinition,name);
                    BeanDefinitionReaderUtils.registerBeanDefinition(holder,beanDefinitionRegistry);
                    System.out.println(beanDefinition);
                });
    }

    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throws BeansException {
        if(null!=compiler){
            configurableListableBeanFactory.setBeanClassLoader(compiler.getClassLoader());
        }
    }
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值