Spring注解系列二十三：自动装配-@Profile

清疚 2022-01-23 04:51 243阅读 0赞

Profile：Spring为我们提供的可以根据当前环境，动态的激活和切换一系列组件的功能。如：开发环境、测试环境、生产环境。  
 @Profile：指定组件在哪个环境的情况下才能被注册到容器中，不指定，任何环境下都能注册这个组件。  
  1）、加了环境标识的bean，只有这个环境被激活的时候才能注册到容器中。默认是default环境。  
  2）、写在配置类上，只有是指定的环境的时候，整个配置类里面的所有配置才能开始生效。  
  3）、没有标注环境标识的bean在，任何环境下都是加载的。  
1、配置文件dbconfig.properties

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

2、MainConfigOfProfile

@PropertySource("classpath:/dbconfig.properties")
    @Configuration
    public class MainConfigOfProfile implements EmbeddedValueResolverAware{
    	
    	@Value("${db.user}")
    	private String user;
    	private StringValueResolver valueResolver;
    	private String  driverClass;
    
    
    	@Profile("test")
    	@Bean("testDataSource")
    	public DataSource dataSourceTest(@Value("${db.password}")String pwd) throws Exception{
    		ComboPooledDataSource dataSource = new ComboPooledDataSource();
    		dataSource.setUser(user);
    		dataSource.setPassword(pwd);
    		dataSource.setJdbcUrl("jdbc:mysql://localhost:3306/test");
    		dataSource.setDriverClass(driverClass);
    		return dataSource;
    	}
    	
    	
    	@Profile("dev")
    	@Bean("devDataSource")
    	public DataSource dataSourceDev(@Value("${db.password}")String pwd) throws Exception{
    		ComboPooledDataSource dataSource = new ComboPooledDataSource();
    		dataSource.setUser(user);
    		dataSource.setPassword(pwd);
    		dataSource.setJdbcUrl("jdbc:mysql://localhost:3306/dev");
    		dataSource.setDriverClass(driverClass);
    		return dataSource;
    	}
    	
    	@Profile("prod")
    	@Bean("prodDataSource")
    	public DataSource dataSourceProd(@Value("${db.password}")String pwd) throws Exception{
    		ComboPooledDataSource dataSource = new ComboPooledDataSource();
    		dataSource.setUser(user);
    		dataSource.setPassword(pwd);
    		dataSource.setJdbcUrl("jdbc:mysql://localhost:3306/prod");
    		dataSource.setDriverClass(driverClass);
    		return dataSource;
    	}
    
    	@Override
    	public void setEmbeddedValueResolver(StringValueResolver resolver) {
    		this.valueResolver = resolver;
    		driverClass = valueResolver.resolveStringValue("${db.driverClass}");
    	}
    
    }

测试：

//1、使用命令行动态参数: 在虚拟机参数位置加载 -Dspring.profiles.active=test
    //2、代码的方式激活某种环境
    @Test
    public void test01(){
    	AnnotationConfigApplicationContext applicationContext = 
    			new AnnotationConfigApplicationContext();
    	//1、创建一个applicationContext
    	//2、设置需要激活的环境
    	applicationContext.getEnvironment().setActiveProfiles("test","dev","prod");
    	//3、注册主配置类
    	applicationContext.register(MainConfigOfProfile.class);
    	//4、启动刷新容器
    	applicationContext.refresh();
    
    	String[] namesForType = applicationContext.getBeanNamesForType(DataSource.class);
    	for (String string : namesForType) {
    		System.out.println(string);
    	}
    	
    	applicationContext.close();
    }

![在这里插入图片描述][20190602150307139.png]

[20190602150307139.png]: /images/20220123/b233e0715e944318bb4a75b24b3ed233.png