spring 优雅替换bean

方案一：使用 @Primary/@Qualifier 注解（优选）

如果有多个相同类型的 Bean 存在，可以将想要优先使用的 Bean 加上 @Primary 注解。

@Qualifier和@Primary注解的区别：@Primary注解用于标记具有相同类型的多个实例中的主要实例。而@Qualifier注解用于标记具有特定名称的Bean，以便在多个具有相同类型的Bean中选择特定的实例。

例如：接口ContextUserParser有AbsContextUserTokenParser和ContextUserTokenParser两个实现类，现在我想使用AbsContextUserTokenParser。为此配置config类：

package com.psbc.abs.config;import com.psbc.pfpj.yoaf.context.autoconfigure.YoafUserContextProperties;
import com.psbc.pfpj.yoaf.context.parser.AbsContextUserTokenParser;
import com.psbc.pfpj.yoaf.context.parser.ContextUserParser;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;@Configuration
@Slf4j
public class AbsContextUserTokenConfig {@Bean@Primarypublic ContextUserParser absContextUserTokenParser(YoafUserContextProperties properties){return new AbsContextUserTokenParser(properties);}
}

方式二：自定义 Bean 替换逻辑

通过实现BeanDefinitionRegistryPostProcessor 接口，在其方法中根据特定条件来动态替换 Bean 的定义。

以下是一个简单示例：

package com.psbc.abs.support;import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.beans.factory.support.BeanDefinitionRegistryPostProcessor;
import org.springframework.beans.factory.support.GenericBeanDefinition;/*** 替换jar包内默认bean** @author lijunyu* @date 2023/10/23 17:55*/
@Slf4j
public class OverrideExistsBeanProcessor implements BeanDefinitionRegistryPostProcessor {@Overridepublic void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry) throws BeansException {replaceClass(registry,"contextUserParser", AbsContextUserTokenParser.class);}/*** 对容器内现有类进行替换** @param beanName 替换类名* @param clazz    替换类*/public void replaceClass(BeanDefinitionRegistry registry,String beanName, Class clazz) {if (registry.containsBeanDefinition(beanName)) {log.debug("bean:{} 删除", beanName);registry.removeBeanDefinition(beanName);GenericBeanDefinition beanDefinition = new GenericBeanDefinition();beanDefinition.setBeanClass(clazz);registry.registerBeanDefinition(beanName, beanDefinition);log.debug("bean:{} 替换成功", beanName);}}@Overridepublic void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {log.debug("beanFactory:{}", beanFactory);}
}

方式三：利用 Spring 的扩展机制

比如通过自定义 ApplicationContextInitializer 等在特定阶段进行 Bean 的替换操作。