spring实战-条件装配bean

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