@Profile 根据环境变量注册Bean

最新推荐文章于 2023-11-03 17:21:47 发布

Bee.F

最新推荐文章于 2023-11-03 17:21:47 发布

阅读量200

点赞数

分类专栏： spring 文章标签： spring 注解 spring bean

本文链接：https://blog.csdn.net/qq_33732195/article/details/110913239

版权

spring 专栏收录该内容

20 篇文章 0 订阅

订阅专栏

本文详细介绍了Spring框架中@Profile注解的使用，如何根据开发环境、测试环境和生产环境动态加载不同的数据源配置。通过示例展示了如何在配置类中定义不同环境的@Bean，并通过设置spring.profiles.active属性来激活特定环境，从而实现不同环境下数据源的切换。同时，提供了通过命令行参数和代码方式激活环境的测试用例。

摘要由CSDN通过智能技术生成

@Profile：提供可以根据当前环境，动态激活和切换一系列组件的功能；

开发环境、测试环境、生产环境；
数据源：A/B/C；

@Profile：指定组件在哪个环境的情况下才能被注册到容器中，不指定，任何环境下都能注册这个组件；

加了环境标识的bean，只有这个环境被激活的时候才能注册到容器中。默认是default环境；
写在配置类上，只有是指定的环境的时候，整个配置类里面的所有配置才能开始生效；
没有标注环境标识的bean在，任何环境下都是加载的；

根据环境加载数据源

配置文件：dbconfig.properties

db.user=root
db.password=123456
db.driverClass=com.mysql.jdbc.Driver

@PropertySource("classpath:/dbconfig.properties")
@Configuration
public class MainConfigOfProfile {

	@Value("${db.user}")
	private String user;
	@Value("${db.password}")
	private String password;
	@Value("${db.driverClass}")
	private String driverClass;

	//@Profile("default")
	@Profile("test")
	@Bean("dataSourceTest")
	public DataSource dataSourceTest() throws PropertyVetoException {
		ComboPooledDataSource dataSource = new ComboPooledDataSource();
		dataSource.setUser(user);
		dataSource.setPassword(password);
		dataSource.setJdbcUrl("jdbc:mysql://localhost:3306/clouddb01");
		dataSource.setDriverClass(driverClass);
		return dataSource;
	}

	@Profile("dev")
	@Bean("dataSourceDev")
	public DataSource dataSourceDev() throws PropertyVetoException {
		ComboPooledDataSource dataSource = new ComboPooledDataSource();
		dataSource.setUser(user);
		dataSource.setPassword(password);
		dataSource.setJdbcUrl("jdbc:mysql://localhost:3306/clouddb02");
		dataSource.setDriverClass(driverClass);
		return dataSource;
	}

	@Profile("prod")
	@Bean("dataSourceProd")
	public DataSource dataSourceProd() throws PropertyVetoException {
		ComboPooledDataSource dataSource = new ComboPooledDataSource();
		dataSource.setUser(user);
		dataSource.setPassword(password);
		dataSource.setJdbcUrl("jdbc:mysql://localhost:3306/clouddb02");
		dataSource.setDriverClass(driverClass);
		return dataSource;
	}
}

测试：

激活环境的两种方式：

1、使用命令行动态参数: 在虚拟机参数位置加载 -Dspring.profiles.active=test

2、代码的方式激活某种环境；

第一种：

public class IOCTest_Profile {

	//1、使用命令行动态参数: 在虚拟机参数位置加载 -Dspring.profiles.active=test
	@Test
	public void test01(){
		ApplicationContext applicationContext = new AnnotationConfigApplicationContext(MainConfigOfProfile.class);
		System.out.println("创建容器完成！");
		String[] names = applicationContext.getBeanDefinitionNames();
		for (String name : names) {
			System.out.println(name);
		}
	}
}

//打印
dataSourceTest

第二种：

@Test
public void test02(){
    //1、创建一个applicationContext
    AnnotationConfigApplicationContext  applicationContext = new AnnotationConfigApplicationContext();
    //2、设置需要激活的环境
    applicationContext.getEnvironment().setActiveProfiles("dev");
    //3、注册主配置类
    applicationContext.register(MainConfigOfProfile.class);
    //4、启动刷新容器
    applicationContext.refresh();
    System.out.println("创建容器完成！");

    String[] names = applicationContext.getBeanDefinitionNames();
    for (String name : names) {
        System.out.println(name);
    }
}

//打印
dataSourceDev