YARN-8923. Cleanup references to ENV file type in YARN service code.
Contributed by Suma Shivaprasad
This commit is contained in:
parent
c58811c77d
commit
e98a506227
@ -31,10 +31,9 @@ public enum ConfigFormat {
|
||||
PROPERTIES("properties"),
|
||||
XML("xml"),
|
||||
HADOOP_XML("hadoop_xml"),
|
||||
ENV("env"),
|
||||
TEMPLATE("template"),
|
||||
YAML("yaml"),
|
||||
;
|
||||
YAML("yaml");
|
||||
|
||||
ConfigFormat(String suffix) {
|
||||
this.suffix = suffix;
|
||||
}
|
||||
|
@ -17,11 +17,6 @@
|
||||
*/
|
||||
package org.apache.hadoop.yarn.service.utils;
|
||||
|
||||
import org.apache.hadoop.fs.Path;
|
||||
import org.apache.hadoop.yarn.service.api.records.ConfigFormat;
|
||||
import org.apache.hadoop.yarn.service.utils.SliderFileSystem;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
import java.util.Map.Entry;
|
||||
@ -29,7 +24,6 @@
|
||||
import java.util.regex.Pattern;
|
||||
|
||||
public class ConfigUtils {
|
||||
public static final String TEMPLATE_FILE = "template.file";
|
||||
|
||||
public static String replaceProps(Map<String, String> config, String content) {
|
||||
Map<String, String> tokens = new HashMap<>();
|
||||
@ -44,54 +38,4 @@ public static String replaceProps(Map<String, String> config, String content) {
|
||||
}
|
||||
return value;
|
||||
}
|
||||
|
||||
public static Map<String, String> replacePropsInConfig(
|
||||
Map<String, String> config, Map<String, String> env) {
|
||||
Map<String, String> tokens = new HashMap<>();
|
||||
for (Entry<String, String> entry : env.entrySet()) {
|
||||
tokens.put("${" + entry.getKey() + "}", entry.getValue());
|
||||
}
|
||||
Map<String, String> newConfig = new HashMap<>();
|
||||
for (Entry<String, String> entry : config.entrySet()) {
|
||||
String value = entry.getValue();
|
||||
for (Map.Entry<String,String> token : tokens.entrySet()) {
|
||||
value = value.replaceAll(Pattern.quote(token.getKey()),
|
||||
Matcher.quoteReplacement(token.getValue()));
|
||||
}
|
||||
newConfig.put(entry.getKey(), entry.getValue());
|
||||
}
|
||||
return newConfig;
|
||||
}
|
||||
|
||||
public static void prepConfigForTemplateOutputter(ConfigFormat configFormat,
|
||||
Map<String, String> config, SliderFileSystem fileSystem,
|
||||
String clusterName, String fileName) throws IOException {
|
||||
if (!configFormat.equals(ConfigFormat.TEMPLATE)) {
|
||||
return;
|
||||
}
|
||||
Path templateFile = null;
|
||||
if (config.containsKey(TEMPLATE_FILE)) {
|
||||
templateFile = fileSystem.buildResourcePath(config.get(TEMPLATE_FILE));
|
||||
if (!fileSystem.isFile(templateFile)) {
|
||||
templateFile = fileSystem.buildResourcePath(clusterName,
|
||||
config.get(TEMPLATE_FILE));
|
||||
}
|
||||
if (!fileSystem.isFile(templateFile)) {
|
||||
throw new IOException("config specified template file " + config
|
||||
.get(TEMPLATE_FILE) + " but " + templateFile + " doesn't exist");
|
||||
}
|
||||
}
|
||||
if (templateFile == null && fileName != null) {
|
||||
templateFile = fileSystem.buildResourcePath(fileName);
|
||||
if (!fileSystem.isFile(templateFile)) {
|
||||
templateFile = fileSystem.buildResourcePath(clusterName,
|
||||
fileName);
|
||||
}
|
||||
}
|
||||
if (fileSystem.isFile(templateFile)) {
|
||||
config.put("content", fileSystem.cat(templateFile));
|
||||
} else {
|
||||
config.put("content", "");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -19,7 +19,6 @@
|
||||
package org.apache.hadoop.yarn.service.utils;
|
||||
|
||||
import org.apache.hadoop.conf.Configuration;
|
||||
import org.apache.hadoop.yarn.service.api.records.ConfigFormat;
|
||||
import org.apache.hadoop.yarn.service.exceptions.BadConfigException;
|
||||
import org.codehaus.jackson.annotate.JsonIgnoreProperties;
|
||||
import org.codehaus.jackson.map.ObjectMapper;
|
||||
@ -185,12 +184,4 @@ public String toString() {
|
||||
return sb.toString();
|
||||
}
|
||||
|
||||
/**
|
||||
* Create an outputter for a given format
|
||||
* @param format format to use
|
||||
* @return an instance of output
|
||||
*/
|
||||
public PublishedConfigurationOutputter createOutputter(ConfigFormat format) {
|
||||
return PublishedConfigurationOutputter.createOutputter(format, this);
|
||||
}
|
||||
}
|
||||
|
@ -99,8 +99,6 @@ public static PublishedConfigurationOutputter createOutputter(ConfigFormat forma
|
||||
return new PropertiesOutputter(owner);
|
||||
case JSON:
|
||||
return new JsonOutputter(owner);
|
||||
case ENV:
|
||||
return new EnvOutputter(owner);
|
||||
case TEMPLATE:
|
||||
return new TemplateOutputter(owner);
|
||||
case YAML:
|
||||
|
Loading…
Reference in New Issue
Block a user