实现spring mvc 的java config 加载不同配置文件

需要配置一个spring 启动时的事件监听器.

配置类

/**
* spring 根容器启动时的监听事件
*
* @return ApplicationContextInitializer<?>[]
*/
@Override
protected ApplicationContextInitializer<?>[] getRootApplicationContextInitializers() {
SpringRootAppInitListener springRootAppInitListener = new SpringRootAppInitListener();
return new ApplicationContextInitializer[]{springRootAppInitListener};
}

实现方式

public class SpringRootAppInitListener implements ApplicationContextInitializer {

private static final Logger logger = LoggerFactory.getLogger(SpringRootAppInitListener.class);

private static final String DEFAULT_CONFIG_PREFIX = "application";

private static final String DEFAULT_CONFIG_SUFFIX = ".properties";

private static final String DEFAULT_CONFIG_DELIMITE = "-";

private static final String DEFAULT_CONFIG_KEY = "spring.active.profile";

@Override
public void initialize(ConfigurableApplicationContext applicationContext) {
MutablePropertySources sources = applicationContext.getEnvironment().getPropertySources();
//加载默认的配置文件
PropertiesLoader propertiesLoader = new PropertiesLoader(DEFAULT_CONFIG_PREFIX + DEFAULT_CONFIG_SUFFIX);
Properties defaultProperties = propertiesLoader.getProperties();
//获取要加载的配置文件的后缀名称
String suFix = defaultProperties.getProperty(DEFAULT_CONFIG_KEY);

//加载配置文件
Properties properties = null;
if (!StringUtil.isEmpty(suFix)) {
properties = propertiesLoader.loadClassPathProperties(DEFAULT_CONFIG_PREFIX + DEFAULT_CONFIG_DELIMITE + suFix + DEFAULT_CONFIG_SUFFIX);
}
//两个配置文件合并
propertiesLoader.mergerProperties(defaultProperties, properties);
//加载到spring 的配置中去
sources.addFirst(new PropertiesPropertySource(DEFAULT_CONFIG_PREFIX, properties));
logger.info("spring root init!");
}
}

PropertiesLoader.java

public class PropertiesLoader {
private static Logger logger = LoggerFactory.getLogger(PropertiesLoader.class);

private static ResourceLoader resourceLoader = new DefaultResourceLoader();

private final Properties properties;

public PropertiesLoader(String... resourcesPaths) {
properties = loadProperties(resourcesPaths);
}

public Properties getProperties() {
return properties;
}

/**
* 取出Property，但以System的Property优先,取不到返回空字符串.
*/
private String getValue(String key) {
String systemProperty = System.getProperty(key);
if (systemProperty != null) {
return systemProperty;
}
if (properties.containsKey(key)) {
return properties.getProperty(key);
}
return "";
}

/**
* 取出String类型的Property，但以System的Property优先,如果都为Null则抛出异常.
*/
public String getProperty(String key) {
String value = getValue(key);
if (value == null) {
throw new NoSuchElementException();
}
return value;
}

/**
* 取出String类型的Property，但以System的Property优先.如果都为Null则返回Default值.
*/
public String getProperty(String key, String defaultValue) {
String value = getValue(key);
return value != null ? value : defaultValue;
}

/**
* 取出Integer类型的Property，但以System的Property优先.如果都为Null或内容错误则抛出异常.
*/
public Integer getInteger(String key) {
String value = getValue(key);
if (value == null) {
throw new NoSuchElementException();
}
return Integer.valueOf(value);
}

/**
* 取出Integer类型的Property，但以System的Property优先.如果都为Null则返回Default值，如果内容错误则抛出异常
*/
public Integer getInteger(String key, Integer defaultValue) {
String value = getValue(key);
return value != null ? Integer.valueOf(value) : defaultValue;
}

/**
* 取出Double类型的Property，但以System的Property优先.如果都为Null或内容错误则抛出异常.
*/
public Double getDouble(String key) {
String value = getValue(key);
if (value == null) {
throw new NoSuchElementException();
}
return Double.valueOf(value);
}

/**
* 取出Double类型的Property，但以System的Property优先.如果都为Null则返回Default值，如果内容错误则抛出异常
*/
public Double getDouble(String key, Integer defaultValue) {
String value = getValue(key);
return value != null ? Double.valueOf(value) : defaultValue;
}

/**
* 取出Boolean类型的Property，但以System的Property优先.如果都为Null抛出异常,如果内容不是true/false则返回false.
*/
public Boolean getBoolean(String key) {
String value = getValue(key);
if (value == null) {
throw new NoSuchElementException();
}
return Boolean.valueOf(value);
}

/**
* 取出Boolean类型的Property，但以System的Property优先.如果都为Null则返回Default值,如果内容不为true/false则返回false.
*/
public Boolean getBoolean(String key, boolean defaultValue) {
String value = getValue(key);
return value != null ? Boolean.valueOf(value) : defaultValue;
}

/**
* 载入多个文件, 文件路径使用Spring Resource格式.
*/
private Properties loadProperties(String... resourcesPaths) {
Properties props = new Properties();

for (String location : resourcesPaths) {

InputStream is = null;
try {
Resource resource = resourceLoader.getResource(location);
is = resource.getInputStream();
props.load(is);
} catch (IOException ex) {
logger.info("Could not load properties from path:" + location + ", " + ex.getMessage());
} finally {
IOUtils.closeQuietly(is);
}
}
return props;
}

/**
* 载入多个文件 classpath 文件
*/
public Properties loadClassPathProperties(String resourcesPaths) {
Properties props = new Properties();
Resource resource = new ClassPathResource(resourcesPaths);
InputStream is = null;
try {
is = resource.getInputStream();
props.load(is);
} catch (IOException ex) {
logger.info("Could not load properties from path:" + resourcesPaths + ", " + ex.getMessage());
} finally {
IOUtils.closeQuietly(is);
}
return props;
}

/**
* 合并两个配置
*
* @param source source
* @param dist   dist
*/
public void mergerProperties(Properties source, Properties dist) {
Enumeration enum1 = source.propertyNames();//得到配置文件的名字
while (enum1.hasMoreElements()) {
Object strKey = enum1.nextElement();
Object value = source.get(strKey);
dist.put(strKey, value);
}
}
}

配置方式

resources
application.properties
application-dev.properties
application-prod.properties
application-test.properties

配置方式

在application.properties 配置
spring.active.profile=dev // 激活不同的环境