org.iartisan.runtime.env.EnvPropertiesLoader Maven / Gradle / Ivy
package org.iartisan.runtime.env;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.*;
import java.util.Properties;
import java.util.ResourceBundle;
import java.util.Set;
/**
*
* env 文件加载
*
* @author King
* @since 2017/6/14
*/
public class EnvPropertiesLoader {
private static Logger logger = LoggerFactory.getLogger("env");
private static String DEFAULT_CONF_FILE_NAME = "env.properties";
private static String DEFAULT_CONF_FILE = File.separator + "wls" + File.separator + "envconfig" + File.separator + DEFAULT_CONF_FILE_NAME;
private static Properties loadLocalFile() {
Properties filePropIn = new Properties();
InputStream fis = null;
InputStreamReader reader = null;
try {
fis = Thread.currentThread().getContextClassLoader().getResourceAsStream(DEFAULT_CONF_FILE_NAME);
if (fis == null) {
ResourceBundle res = ResourceBundle.getBundle("env");
Set keys = res.keySet();
for (String key : keys) {
filePropIn.put(key, new String(res.getString(key).getBytes("ISO-8859-1"), "GBK"));
}
} else {
reader = new InputStreamReader(fis, "GBK");
filePropIn.load(reader);
}
} catch (Exception e) {
logger.error(e.getMessage(), e);
} finally {
closeReader(reader);
closeStream(fis);
}
return filePropIn;
}
private static Properties loadOutFile(String filePath) {
Properties filePropOut = new Properties();
InputStream fis = null;
InputStreamReader reader = null;
try {
fis = new FileInputStream(filePath);
reader = new InputStreamReader(fis, "utf-8");
filePropOut.load(reader);
logger.info("外置配置文件{}加载成功", filePath);
} catch (Exception e) {
logger.error("外置配置文件" + filePath + "加载失败！{}", e.getMessage());
} finally {
closeReader(reader);
closeStream(fis);
}
return filePropOut;
}
private static void closeStream(InputStream is) {
if (is != null) {
try {
is.close();
} catch (IOException e) {
logger.error(e.getMessage(), e);
}
}
}
private static void closeReader(InputStreamReader is) {
if (is != null) {
try {
is.close();
} catch (IOException e) {
logger.error(e.getMessage(), e);
}
}
}
public static Properties loadFile() {
logger.info("<<<======开始加载配置文件======>>>");
Properties properties = new Properties();
properties.putAll(loadLocalFile());
if (properties.containsKey("app.name")) {
String appName = properties.getProperty("app.name");
DEFAULT_CONF_FILE = File.separator + "wls" + File.separator + "envconfig" + File.separator + appName + File.separator + DEFAULT_CONF_FILE_NAME;
}
properties.putAll(loadOutFile(DEFAULT_CONF_FILE));
properties.entrySet().forEach(value -> {
logger.info(value.getKey().toString() + "=" + value.getValue().toString());
EnvContextConfig.put(value.getKey().toString(), value.getValue().toString());
}
);
return properties;
}
}