在做大型项目时,我们的系统会有多个运行环境,如开发人员自己的本地环境dev,测试人员的测试环境sit,上线前的预生产环境pre,线上环境prd
在不同环境中我们可能需要不同的配置,如数据库配置,MQ配置等,在不同的环境都有相应的不同的配置,这时候我们需要根据不同的环境来创建不同的配置
强大的Spring为我们提供了条件化装配Bean和profile的bean装配
如下:
TestMain
package com.halfworlders.test;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ActiveProfiles;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import com.halfworlders.dao.DataSource;
import com.halfworlders.dao.DataSourceConfig;
/**
* profile的需要两个参数:spring.profile.active和Spring.profile.default
* 有多种方式配置这两个属性
* 1,作为DispatcherServlet的初始化函数
* 2,作为Web应用的上下文参数
* 3,作为JNDI条目
* 4,作为环境变量
* 5,作为JVM的系统属性
* 6,在集成测试环境上,使用@ActiveProfiles注解
*/
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(classes=DataSourceConfig.class)
@ActiveProfiles("prd")
public class TestMain {
@Autowired(required=false)
private DataSource dataSource;
@Test
public void test(){
System.out.println(dataSource.getClass().getName());
}
}
DataSourceConfig
package com.halfworlders.dao;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import;
@Configuration
@Import({MysqlDataSourceConfig.class,OracleDataSourceConfig.class,DB2DataSourceConfig.class})
public class DataSourceConfig {
}
MysqlDataSourceConfig
package com.halfworlders.dao;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Profile;
@Configuration
public class MysqlDataSourceConfig {
@Bean
@Profile("prd")
public DataSource dataSource(){
return new MysqlDataSource();
}
}
OracleDataSourceConfig
package com.halfworlders.dao;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Profile;
@Configuration
public class OracleDataSourceConfig {
@Bean(destroyMethod="shutdown")
@Profile("dev")
public DataSource dataSource() {
return new OracleDataSource();
}
}
DB2DataSourceConfig
package com.halfworlders.dao;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Conditional;
import org.springframework.context.annotation.Configuration;
import com.halfworlders.condition.MyCondition;
@Configuration
public class DB2DataSourceConfig {
/*
* 基于自定义条件MyCondition决定是否注入此Bean
*/
@Bean
@Conditional(MyCondition.class)
public DataSource dataSource(){
return new DB2DataSource();
}
}
MyCondition
package com.halfworlders.condition;
import org.springframework.context.annotation.Condition;
import org.springframework.context.annotation.ConditionContext;
import org.springframework.core.env.Environment;
import org.springframework.core.type.AnnotatedTypeMetadata;
public class MyCondition implements Condition {
/*
* 条件化Bean
* 如果环境变量中设置了sit,不管这个sit的环境变量值是什么,被@Conditional注解中引用MyCondition的Bean都会被创建
*/
@Override
public boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata) {
Environment evm = context.getEnvironment();
return evm.containsProperty("sit");
}
}
DataSource
package com.halfworlders.dao;
public interface DataSource {
void open();
void shutdown();
}
public class MysqlDataSource implements DataSource {
public void open(){
}
public void shutdown(){
}
}
public class OracleDataSource implements DataSource {
public void open(){
}
public void shutdown(){
}
}
public class DB2DataSource implements DataSource {
public void open(){
}
public void shutdown(){
}
}