spring实战-条件装配bean

时间:2022-10-26 20:33:52

在做大型项目时,我们的系统会有多个运行环境,如开发人员自己的本地环境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(){			}}