com.centurylink.mdw.config.JavaPropertyManager Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mdw-common Show documentation
Show all versions of mdw-common Show documentation
MDW is a microservices based workflow framework
/*
* Copyright (C) 2017 CenturyLink, Inc.
*
* 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.centurylink.mdw.config;
import com.centurylink.mdw.app.ApplicationContext;
import com.centurylink.mdw.cli.Decrypt;
import com.centurylink.mdw.startup.StartupException;
import com.centurylink.mdw.util.MiniCrypter;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;
public class JavaPropertyManager extends PropertyManager {
protected Properties properties = new Properties();
private String mainPropertyFileName;
public static final String APP_CONFIG_NAME = "mdw.application.config.name";
/**
* This method loads mdw.yaml, any app yaml files defined for {@link #APP_CONFIG_NAME}
* The property mdw.application.config.name takes property file names seperated by , without the .properties extension
*/
public JavaPropertyManager() throws StartupException {
mainPropertyFileName = getMainPropertyFileName();
if (mainPropertyFileName == null) {
if ("standalone".equals(ApplicationContext.getRuntimeEnvironment()))
return;
else
throw new StartupException("Cannot find mdw.yaml or mdw.properties");
}
loadPropertiesFromFile(null, mainPropertyFileName, true, true);
loadPropertiesFromFile(null, APPLICATION_PROPERTIES_FILE_NAME, true, false);
// Application config files should be in same place as mdw.yaml.
String appPropertyFiles = this.getStringProperty(APP_CONFIG_NAME);
if (appPropertyFiles != null) {
for (String fileName : appPropertyFiles.split(",")) {
loadPropertiesFromFile(null, fileName + ".properties", true, false);
}
}
}
public void loadPropertiesFromFile(Properties tempProperties, String fileName,
boolean verboseSuccess, boolean verboseFailure) {
try {
loadFromFile(tempProperties == null ? properties : tempProperties, fileName);
if (verboseSuccess)
System.out.println("Loaded properties from file " + fileName);
}
catch (PropertyException e) {
if (verboseFailure) {
System.out.println("Cannot load properties from " + fileName);
e.printStackTrace();
}
}
}
public void refreshCache() throws PropertyException {
refreshProperties(true);
}
public void clearCache() {
properties.clear();
}
/**
* Null means not found.
*/
private String getMainPropertyFileName() throws StartupException {
String configLoc = getConfigLocation();
File file = new File(configLoc == null ? MDW_PROPERTIES_FILE_NAME : (configLoc+MDW_PROPERTIES_FILE_NAME));
if (file.exists())
return MDW_PROPERTIES_FILE_NAME;
URL url = this.getClass().getClassLoader().getResource(MDW_PROPERTIES_FILE_NAME);
if (url != null)
return MDW_PROPERTIES_FILE_NAME;
return null;
}
private void refreshProperties(boolean printStackTraceWhenError) throws PropertyException {
Properties tempProperties = new Properties();
getSources().clear();
// 1. load properties from mdw.yaml or mdw.properties from configuration directory
loadPropertiesFromFile(tempProperties, mainPropertyFileName, true, true);
// 2. load properties from application yaml configs
loadPropertiesFromFile(tempProperties, APPLICATION_PROPERTIES_FILE_NAME, true, false);
// app config files should be in same place as mdw.yaml.
String appPropertyFiles = this.getStringProperty(APP_CONFIG_NAME);
if (appPropertyFiles != null) {
for (String fileName : appPropertyFiles.split(",")) {
loadPropertiesFromFile(tempProperties, fileName + ".properties", true, false);
}
}
// update the cache
updateCache(tempProperties);
}
final protected void updateCache(Properties tempProperties) {
synchronized (properties) {
properties.clear();
properties.putAll(tempProperties);
}
}
/**
* returns the properties for group.
* When the group is specified with ApplicationProperties.xml
* (old style), the names of the properties returned
* will not contain the group names and '/', for backward
* compatibility. When the group is in the new style,
* where a group is any property name prefix terminated by a '.',
* the property names returned will contain the
* whole property names including the group names.
* @param group
* @return Properties for the group
*/
public Properties getProperties(String group) {
Properties props = new Properties();
int k = group.length();
for (Object key : properties.keySet()) {
String propname = (String)key;
int l = propname.length();
char ch = l>k?propname.charAt(k):' ';
if ((ch=='.'||ch=='/') && propname.startsWith(group)) {
if (ch=='.') props.put(propname, properties.get(propname));
else props.put(propname.substring(k+1), properties.get(propname));
}
}
return props;
}
public String getStringProperty(String name) {
String value = properties.getProperty(name);
if (value != null && value.startsWith("~[") && value.endsWith("]")) {
try {
Decrypt decrypt = new Decrypt();
decrypt.setInput(value.substring(2, value.length() - 1));
value = decrypt.decrypt();
}
catch (IOException ex) {
ex.printStackTrace();
}
}
return value;
}
public boolean isEncrypted(String propName) {
String value = properties.getProperty(propName);
return value != null && value.startsWith("~[") && value.endsWith("]");
}
final protected void loadFromFile(Properties properties, String filename)
throws PropertyException {
String configLoc = getConfigLocation();
InputStream stream = null;
try {
File file = new File(configLoc==null?filename:(configLoc+filename));
if (file.exists()) {
stream = new FileInputStream(file);
}
else {
stream = this.getClass().getClassLoader().getResourceAsStream(filename);
if (stream == null) // try context classloader
stream = Thread.currentThread().getContextClassLoader().getResourceAsStream(filename);
}
if (stream==null)
throw new PropertyException("Property file does not exist: " + filename);
else
loadFromStream(properties, stream, filename);
// handle encrypted values
Map namesOfEncrypted = new HashMap<>();
for (Object key : properties.keySet()) {
String name = key.toString();
if (name.startsWith("[") && name.endsWith("]")) {
namesOfEncrypted.put(name, name.substring(1, name.length() - 1));
}
}
for (String nameOfEncrypted : namesOfEncrypted.keySet()) {
String encValue = (String) properties.remove(nameOfEncrypted);
String value = MiniCrypter.decrypt(encValue);
properties.setProperty(namesOfEncrypted.get(nameOfEncrypted), value);
}
}
catch (Exception ex) {
throw new PropertyException("Failed to load properties from " + filename, ex);
}
}
public Properties getAllProperties() {
return properties;
}
public void setStringProperty(String name, String value) {
if (value==null || value.length() == 0)
properties.remove(name);
else
properties.put(name, value);
}
} © 2015 - 2025 Weber Informatics LLC | Privacy Policy