package com.vincent.rsf.common.utils;
|
|
import com.vincent.rsf.framework.common.SpringUtils;
|
import org.springframework.context.ApplicationContext;
|
import org.springframework.core.env.Environment;
|
import org.yaml.snakeyaml.Yaml;
|
|
import java.io.BufferedInputStream;
|
import java.io.InputStream;
|
import java.nio.charset.StandardCharsets;
|
import java.util.HashMap;
|
import java.util.Map;
|
import java.util.Properties;
|
|
public class ConfigHelper {
|
|
public static Properties m_props = new Properties();
|
public static String m_charset = "GBK";
|
public static Yaml m_yaml = new Yaml();
|
public static Map<String, Object> m_root = new HashMap<>();
|
public static Environment environment;
|
|
static {
|
try {
|
if (ClassLoader.class.getResource("/application.properties") != null) {
|
m_charset = getCharset(ClassLoader.class.getResourceAsStream("/application.properties"));
|
m_props.load(ClassLoader.class.getResourceAsStream("/application.properties"));
|
} else if ( Thread.currentThread().getContextClassLoader().getResource("/application.properties")!=null) {
|
m_charset = getCharset(Thread.currentThread().getContextClassLoader().getResourceAsStream("/application.properties"));
|
m_props.load(Thread.currentThread().getContextClassLoader().getResourceAsStream("/application.properties"));
|
} else if (ConfigHelper.class.getResource("/application.properties")!=null){
|
m_charset=getCharset(ConfigHelper.class.getResourceAsStream("/application.properties"));
|
m_props.load(ConfigHelper.class.getResourceAsStream("/application.properties"));
|
}
|
} catch (Exception ex){
|
System.err.println("ERR - undefined application.properties");
|
}
|
try {
|
if (ClassLoader.class.getResource("/application.yml") != null) {
|
InputStream as = ClassLoader.class.getResourceAsStream("/application.yml");
|
if(as!=null){
|
m_root = (Map) m_yaml.load(as);
|
}
|
} else if(Thread.currentThread().getContextClassLoader().getResource("/application.yml")!=null) {
|
InputStream as = Thread.currentThread().getContextClassLoader().getResourceAsStream("/application.yml");
|
if(as!=null){
|
m_root = (Map) m_yaml.load(as);
|
}
|
} else if(ConfigHelper.class.getResource("/application.yml")!=null){
|
InputStream as = ConfigHelper.class.getResourceAsStream("/application.yml");
|
if(as!=null){
|
m_root = (Map) m_yaml.load(as);
|
}
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
System.err.println("ERR - undefined application.yml");
|
}
|
}
|
|
public static void main(String...strings){
|
System.out.println(getString("redis.session.index"));
|
}
|
|
public static Boolean contains(String key){
|
return getString(key)!=null;
|
}
|
|
public static String getString(String key) {
|
if (environment==null) {
|
ApplicationContext applicationContext = null;
|
try {
|
applicationContext = SpringUtils.getApplicationContext();
|
} catch (Exception ignore) {}
|
if ( applicationContext != null) {
|
environment = SpringUtils.getApplicationContext().getEnvironment();
|
}
|
}
|
if (environment!=null) {
|
String value = environment.getProperty(key);
|
if(value!=null){
|
return value;
|
}
|
}
|
String value = m_props.getProperty(key);
|
try {
|
if (value!=null){
|
value = new String(value.getBytes(StandardCharsets.ISO_8859_1),m_charset);
|
} else if(m_root!=null) {
|
String[] items = key.split("\\.");
|
Map map = m_root;
|
for(int i=0;i<items.length-1;i++){
|
String item = items[i];
|
if(map.get(item)!=null && map.get(item) instanceof Map){
|
map = (Map) map.get(item);
|
}else{
|
map = null;
|
break;
|
}
|
}
|
if(map!=null){
|
Object v = map.get(items[items.length-1]);
|
if(v!=null){
|
return v.toString();
|
}
|
}
|
}
|
} catch(Exception ex) {
|
System.err.println("获取配置文件属性["+key+"]时出错 - "+ ex.getMessage());
|
}
|
return value;
|
}
|
|
public static Long getLong(String key) {
|
Long l=null;
|
try {
|
l = Long.parseLong(getString(key));
|
} catch (Exception ex) {
|
System.err.println("获取配置文件属性["+key+"]时出错 - "+ ex.getMessage());
|
}
|
return l;
|
}
|
|
public static Integer getInteger(String key) {
|
Integer i=null;
|
try{
|
i=Integer.parseInt(getString(key));
|
}catch(Exception ex){
|
//ex.printStackTrace();
|
System.out.println("获取配置文件属性["+key+"]时出错 - "+ ex.getMessage());
|
}
|
return i;
|
}
|
|
public static Boolean getBoolean(String key) {
|
boolean b=false;
|
try{
|
String s=getString(key);
|
b=Boolean.parseBoolean(s)||s.equals("1");
|
}catch(Exception ex){
|
//ex.printStackTrace();
|
System.out.println("获取配置文件属性["+key+"]时出错 - "+ ex.getMessage());
|
}
|
return b;
|
}
|
|
public static Environment getEnvironment() {
|
return environment;
|
}
|
|
public static void setEnvironment(Environment environment) {
|
ConfigHelper.environment = environment;
|
}
|
|
/**
|
* 获取文件编码
|
* ANSI：无格式定义；
|
* Unicode： 前两个字节为0xFFFE；
|
* Unicode big endian：前两字节为0xFEFF；
|
* UTF-8：前两字节为0xEFBB；
|
*/
|
public static String getCharset( InputStream is ) {
|
String charset = "GBK";
|
byte[] first3Bytes = new byte[3];
|
try {
|
boolean checked=false;
|
BufferedInputStream bis = new BufferedInputStream(is);
|
bis.mark( 0 );
|
int read = bis.read( first3Bytes, 0, 3 );
|
if ( read == -1 ) return charset;
|
if ( first3Bytes[0] == (byte) 0xFF && first3Bytes[1] == (byte) 0xFE ) {
|
charset = "UTF-16LE";
|
checked = true;
|
}
|
else if ( first3Bytes[0] == (byte) 0xFE && first3Bytes[1] == (byte) 0xFF ) {
|
charset = "UTF-16BE";
|
checked = true;
|
}
|
else if ( first3Bytes[0] == (byte) 0xEF && first3Bytes[1] == (byte) 0xBB && first3Bytes[2] == (byte) 0xBF ) {
|
charset = "UTF-8";
|
checked = true;
|
}
|
bis.reset();
|
if ( !checked ) {
|
int loc = 0;
|
while ( (read = bis.read()) != -1 ) {
|
loc++;
|
if ( read >= 0xF0 ) break;
|
if ( 0x80 <= read && read <= 0xBF ) // 单独出现BF以下的，也算是GBK
|
break;
|
if ( 0xC0 <= read && read <= 0xDF ) {
|
read = bis.read();
|
if ( !(0x80 <= read && read <= 0xBF) ) // 双字节 (0xC0 - 0xDF) (0x80
|
// - 0xBF),也可能在GB编码内
|
{
|
break;
|
}
|
}
|
else if (0xE0 <= read) {// 也有可能出错，但是几率较小
|
read = bis.read();
|
if ( 0x80 <= read && read <= 0xBF ) {
|
read = bis.read();
|
if ( 0x80 <= read && read <= 0xBF ) {
|
charset = "UTF-8";
|
break;
|
}
|
else break;
|
}
|
else break;
|
}
|
}
|
bis.close();
|
}
|
return charset;
|
} catch ( Exception e ) {
|
e.printStackTrace();
|
return null;
|
}
|
}
|
|
}
|
