springboot中如何根据字符串替换符替换 springboot替换bean

使用SpringBoot的启动事件监听机制，通过修改注册的Bean，这样业务方可以无感知的实用

实现方式

这里简化了实现的业务，提供一个简单demo，在方法执行前执行一段代码（不是通过判断参数是否时BasicBean来做了）

思路

• 监听器创建
  通过实现接口ApplicationListener来创建监听器

import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.support.DefaultListableBeanFactory;
import org.springframework.cglib.proxy.Enhancer;
import org.springframework.context.ApplicationEvent;
import org.springframework.context.ApplicationListener;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.core.Ordered;

import java.util.Map;

public class DemoApplicationListener implements ApplicationListener<ApplicationEvent>, Ordered {

    /\*\*
 \* The default order for the LoggingApplicationListener.
 \*/
    public static final int DEFAULT_ORDER = Ordered.HIGHEST_PRECEDENCE + 20;

    @Override
    public void onApplicationEvent(ApplicationEvent event) {
        if (event instanceof ContextRefreshedEvent) {
            System.out.println("==========ContextRefreshedEvent==========");
            //处理逻辑
           test((ContextRefreshedEvent)event);
        }

    }

    @Override
    public int getOrder() {
        return DEFAULT_ORDER;
    }

其中test方法就是具体的处理逻辑实现

• 指定Bean获取

ConfigurableApplicationContext applicationContext = (ConfigurableApplicationContext)e.getApplicationContext();

Map<String, Object> beansWithAnnotation = applicationContext.getBeansWithAnnotation(DemoAnnotaion.class);

DefaultListableBeanFactory beanFactory = (DefaultListableBeanFactory)applicationContext.getBeanFactory();

• 新对象创建

Object obj = beansWithAnnotation.get(key);

            System.out.println(beansWithAnnotation.keySet());

            // 创建一个代理类
            Enhancer enhancer = new Enhancer();
            enhancer.setSuperclass(obj.getClass());
            // 设置回调方法
            enhancer.setCallback(new DemoMethodInterceptor());
            
            Object o = enhancer.create();

这里动态代理使用cglib来实现

• 替换老对象

// 消除原先的bean
            BeanDefinition beanDefinition = beanFactory.getBeanDefinition(key);

            beanFactory.removeBeanDefinition(key);

            beanDefinition.setBeanClassName(o.getClass().getCanonicalName());

            beanFactory.registerBeanDefinition(key, beanDefinition);

            beanFactory.registerSingleton(key, o);

• 监听器注册
  在resources目录下创建META-INF/spring.factories
  
  spring.factories

# Application Listeners
org.springframework.context.ApplicationListener=\
com.example.demo.frame.DemoApplicationListener

具体代码

目录结构

DemoAnnotaion

import java.lang.annotation.*;

@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DemoAnnotaion {
    String[] value();
}

DemoApplicationListener

import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.support.DefaultListableBeanFactory;
import org.springframework.cglib.proxy.Enhancer;
import org.springframework.context.ApplicationEvent;
import org.springframework.context.ApplicationListener;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.core.Ordered;

import java.util.Map;

public class DemoApplicationListener implements ApplicationListener<ApplicationEvent>, Ordered {

    /\*\*
 \* The default order for the LoggingApplicationListener.
 \*/
    public static final int DEFAULT_ORDER = Ordered.HIGHEST_PRECEDENCE + 20;

    @Override
    public void onApplicationEvent(ApplicationEvent event) {
        if (event instanceof ContextRefreshedEvent) {
            System.out.println("==========ContextRefreshedEvent==========");
           test((ContextRefreshedEvent)event);
        }

    }

    @Override
    public int getOrder() {
        return DEFAULT_ORDER;
    }

    private void test(ContextRefreshedEvent e){
        ConfigurableApplicationContext applicationContext = (ConfigurableApplicationContext)e.getApplicationContext();

        Map<String, Object> beansWithAnnotation = applicationContext.getBeansWithAnnotation(DemoAnnotaion.class);

        DefaultListableBeanFactory beanFactory = (DefaultListableBeanFactory)applicationContext.getBeanFactory();

        for(String key : beansWithAnnotation.keySet()) {

            Object obj = beansWithAnnotation.get(key);

            System.out.println(beansWithAnnotation.keySet());

            // 创建一个代理类
            Enhancer enhancer = new Enhancer();
            enhancer.setSuperclass(obj.getClass());
            // 设置回调方法
            enhancer.setCallback(new DemoMethodInterceptor());

            Object o = enhancer.create();

            // 消除原先的bean
            BeanDefinition beanDefinition = beanFactory.getBeanDefinition(key);

            beanFactory.removeBeanDefinition(key);

            beanDefinition.setBeanClassName(o.getClass().getCanonicalName());

            beanFactory.registerBeanDefinition(key, beanDefinition);

            beanFactory.registerSingleton(key, o);
        }
    }
}

DemoMethodInterceptor

import org.springframework.cglib.proxy.MethodInterceptor;
import org.springframework.cglib.proxy.MethodProxy;

import java.lang.reflect.Method;


public class DemoMethodInterceptor implements MethodInterceptor {
    @Override
    public Object intercept(Object o, Method method, Object[] args, MethodProxy methodProxy) throws Throwable {
        System.out.println("\*\*\*\*\*\*cglib======");
        Object result = methodProxy.invokeSuper(o, args);
        return result;
    }
}

TestService

import org.springframework.stereotype.Component;

@Component
@DemoAnnotaion("save")
public class TestService {

    public void test(){
        System.out.println("test");
    }