com.mayabot.nlp.Settings Maven / Gradle / Ivy
Show all versions of mynlp Show documentation
/*
* Copyright 2018 mayabot.com authors. All rights reserved.
*
* 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.mayabot.nlp;
import com.google.common.base.Splitter;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.io.Files;
import com.google.common.io.Resources;
import com.mayabot.nlp.logging.InternalLogger;
import com.mayabot.nlp.logging.InternalLoggerFactory;
import java.io.File;
import java.io.InputStream;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.function.Supplier;
/**
* 系统属性 mynlp.conf 可以配置配置文件加载的目录,默认当前工作目录的conf
*
* TODO 读取过的属性,就不能再次修改了。
*
* @author jimichan
*/
public class Settings {
private static final InternalLogger logger = InternalLoggerFactory.getInstance(Settings.class);
private Map settings;
public final static Settings EMPTY = new Settings(Maps.newHashMap());
Settings(Map settings) {
this.settings = Maps.newHashMap(settings);
}
Settings(Settings settings) {
this.settings = Maps.newHashMap(settings.settings);
}
public T get(SettingItem setting) {
String value = get(setting.getKey(), setting.getDefaultValue());
return setting.getParse().apply(value);
}
public Settings put(SettingItem setting, String value) {
settings.put(setting.getKey(), value);
return this;
}
public Settings put(String key, String value) {
settings.put(key, value);
return this;
}
public Settings put(String prefix, SettingItem setting, String value) {
settings.put(prefix + "." + setting.getKey(), value);
return this;
}
/**
* 后面的覆盖前面的设置
*
* @param settings
* @return Settings
*/
public static Settings merge(Settings... settings) {
Map all = Maps.newHashMap();
for (Settings setting : settings) {
if (setting == null) {
continue;
}
all.putAll(setting.settings);
}
return new Settings(ImmutableMap.copyOf(all));
}
public static Settings createEmpty() {
return new Settings(ImmutableMap.of());
}
public static Settings create(Map settings) {
return new Settings(settings);
}
public static Settings create(Properties properties) {
Map map = new HashMap<>(16);
for (String key : properties.stringPropertyNames()) {
String value = properties.getProperty(key);
map.put(key, value);
}
return new Settings(map);
}
/**
* Returns the setting value associated with the setting key.
*
* @param setting The setting key
* @return The setting value, null if it does not exists.
*/
public String get(String setting) {
return settings.get(setting);
}
/**
* Returns the setting value associated with the setting key. If it does not exists,
* returns the default value provided.
*/
public String get(String setting, String defaultValue) {
return settings.getOrDefault(setting, defaultValue);
}
public List getAsList(SettingItem setting) {
return getAsList(setting.getKey(), setting.getDefaultValue());
}
public List getAsList(String setting, String default_) {
String obj = get(setting, default_);
if (obj == null) {
return null;
}
return Splitter.on(',').omitEmptyStrings().trimResults().splitToList(obj);
}
/**
* A settings that are filtered (and key is removed) with the specified prefix.
*/
public Settings getByPrefix(String prefix) {
if (!prefix.endsWith(".")) {
prefix += ".";
}
String _prefix = prefix;
Map sub = Maps.filterKeys(settings, key -> key.startsWith(_prefix));
HashMap result = Maps.newHashMap();
for (Map.Entry entry : sub.entrySet()) {
result.put(entry.getKey().substring(_prefix.length()), entry.getValue());
}
return new Settings(ImmutableMap.copyOf(result));
}
/**
* Returns the setting value (as float) associated with the setting key. If it does not exists,
* returns the default value provided.
*/
public Float getAsFloat(String setting, Float defaultValue) {
String sValue = get(setting);
if (sValue == null) {
return defaultValue;
}
try {
return Float.parseFloat(sValue);
} catch (NumberFormatException e) {
throw new SettingsException("Failed to parse float setting [" + setting + "] with value [" + sValue + "]", e);
}
}
/**
* Returns the setting value (as double) associated with the setting key. If it does not exists,
* returns the default value provided.
*/
public Double getAsDouble(String setting, Double defaultValue) {
String sValue = get(setting);
if (sValue == null) {
return defaultValue;
}
try {
return Double.parseDouble(sValue);
} catch (NumberFormatException e) {
throw new SettingsException("Failed to parse double setting [" + setting + "] with value [" + sValue + "]", e);
}
}
/**
* Returns the setting value (as int) associated with the setting key. If it does not exists,
* returns the default value provided.
*/
public Integer getAsInt(String setting, Integer defaultValue) {
String sValue = get(setting);
if (sValue == null) {
return defaultValue;
}
try {
return Integer.parseInt(sValue);
} catch (NumberFormatException e) {
throw new SettingsException("Failed to parse int setting [" + setting + "] with value [" + sValue + "]", e);
}
}
/**
* Returns the setting value (as long) associated with the setting key. If it does not exists,
* returns the default value provided.
*/
public Long getAsLong(String setting, Long defaultValue) {
String sValue = get(setting);
if (sValue == null) {
return defaultValue;
}
try {
return Long.parseLong(sValue);
} catch (NumberFormatException e) {
throw new SettingsException("Failed to parse long setting [" + setting + "] with value [" + sValue + "]", e);
}
}
/**
* Returns the setting value (as boolean) associated with the setting key. If it does not exists,
* returns the default value provided.
*/
public Boolean getAsBoolean(String setting, Boolean defaultValue) {
String value = get(setting, defaultValue.toString());
return Boolean.valueOf(value);
}
/**
* A generic failure to handle settings.
*/
public static class SettingsException extends RuntimeException {
public SettingsException(String message) {
super(message);
}
public SettingsException(String message, Throwable cause) {
super(message, cause);
}
}
/**
* 在下面的位置搜索文件 mynlp.properties，并且合并。
* 排在前面的优先级高.
*
* 1. ${work dir}/mynlp.properties
* 2. classpath:mynlp.properties
* 3. ~/mynlp.properties
*
* @return Settings
*/
public static Settings defaultSystemSettings() {
@SuppressWarnings("unchecked")
List> list = Lists.newArrayList(
() -> {
try {
File file = new File("mynlp.properties");
if (file.exists() && file.canRead()) {
logger.info("read settings from " + file);
return Files.asByteSource(file).openBufferedStream();
}
} catch (Exception e) {
return null;
}
return null;
},
() -> {
try {
InputStream inputStream = Resources.getResource("mynlp.properties").openStream();
logger.info("read settings from classpath://mynlp.properties");
return inputStream;
} catch (Exception e) {
return null;
}
},
() -> {
try {
String userHome = System.getProperty("user.home");
File file = new File(userHome, "mynlp.properties");
if (file.exists() && file.canRead()) {
logger.info("read settings from " + file);
return Files.asByteSource(file).openBufferedStream();
}
} catch (Exception e) {
return null;
}
return null;
}
);
Map map = Maps.newHashMap();
for (Supplier supplier : Lists.reverse(list)) {
InputStream in = supplier.get();
if (in != null) {
Properties properties = new Properties();
try {
properties.load(in);
for (String key : properties.stringPropertyNames()) {
String value = properties.getProperty(key);
map.put(key, value);
}
} catch (Exception e) {
}
}
}
return new Settings(map);
}
}