com.blade.kit.Environment Maven / Gradle / Ivy
/**
* Copyright (c) 2016, biezhi 王爵 ([email protected])
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.blade.kit;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Properties;
import java.util.Set;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* Environment
*
* @author biezhi
* @since 1.0
*/
public class Environment {
private static final Logger LOGGER = LoggerFactory.getLogger(Environment.class);
private Map envMap = new HashMap(20);
public Environment(String confPath){
this.envMap = loadMap(confPath);
}
private Map loadMap(String confPath){
Map envMap = new HashMap(20);
Properties config = new Properties();
confPath = confPath.startsWith("/") ? confPath : "/" + confPath;
InputStream in = this.getClass().getResourceAsStream(confPath);
try {
if(null != in){
config.load(in);
// parse properties file
Set> set = config.entrySet();
if(CollectionKit.isNotEmpty(set)){
Iterator> it = set.iterator();
while (it.hasNext()) {
Entry entry = it.next();
String key = entry.getKey().toString();
String value = entry.getValue().toString();
String fuKey = getWildcard(value);
if(null != fuKey && null != config.get(fuKey)){
String fuValue = config.get(fuKey).toString();
value = value.replaceAll("\\$\\{" + fuKey + "\\}", fuValue);
}
envMap.put(key, value);
}
LOGGER.info("load environment config [classpath:" + confPath + "]");
}
}
} catch (IOException e) {
LOGGER.error("load environment config error", e);
} finally {
IOKit.closeQuietly(in);
}
return envMap;
}
public static Environment load(String confPath) {
return new Environment(confPath);
}
public Environment add(String confPath){
this.envMap.putAll(loadMap(confPath));
return this;
}
private static String getWildcard(String str) {
if (null != str && str.indexOf("${") != -1) {
int start = str.indexOf("${");
int end = str.indexOf("}");
if (start != -1 && end != -1) {
return str.substring(start + 2, end);
}
}
return null;
}
public Map getEnvMap(){
return envMap;
}
public String getString(String key) {
return envMap.get(key);
}
public String getString(String key, String defaultValue) {
return null != envMap.get(key) ? envMap.get(key) : defaultValue;
}
public Integer getInt(String key) {
String value = getString(key);
if (StringKit.isNotBlank(value)) {
return Integer.valueOf(value);
}
return null;
}
public Integer getInt(String key, Integer defaultValue) {
return null != getInt(key) ? getInt(key) : defaultValue;
}
public Long getLong(String key) {
String value = getString(key);
if (StringKit.isNotBlank(value)) {
return Long.valueOf(value);
}
return null;
}
public Long getLong(String key, Long defaultValue) {
return null != getLong(key) ? getLong(key) : defaultValue;
}
public Boolean getBoolean(String key) {
String value = getString(key);
if (StringKit.isNotBlank(value)) {
return Boolean.valueOf(value);
}
return null;
}
public Boolean getBoolean(String key, boolean defaultValue) {
return null != getBoolean(key) ? getBoolean(key) : defaultValue;
}
}