YARN-5538. Apply SLIDER-875 to yarn-native-services. Contributed by Billie Rinaldi

This commit is contained in:
Jian He 2016-08-22 13:58:50 +08:00
parent 02e2740bdf
commit 3279baecb9
16 changed files with 669 additions and 115 deletions

View File

@ -34,6 +34,11 @@ public interface RoleKeys {
*/ */
String ROLE_GROUP = "role.group"; String ROLE_GROUP = "role.group";
/**
* The prefix of a role: {@value}
*/
String ROLE_PREFIX = "role.prefix";
/** /**
* Status report: number actually granted : {@value} * Status report: number actually granted : {@value}
*/ */

View File

@ -151,6 +151,7 @@ import org.apache.slider.core.registry.YarnAppListClient;
import org.apache.slider.core.registry.docstore.ConfigFormat; import org.apache.slider.core.registry.docstore.ConfigFormat;
import org.apache.slider.core.registry.docstore.PublishedConfigSet; import org.apache.slider.core.registry.docstore.PublishedConfigSet;
import org.apache.slider.core.registry.docstore.PublishedConfiguration; import org.apache.slider.core.registry.docstore.PublishedConfiguration;
import org.apache.slider.core.registry.docstore.PublishedConfigurationOutputter;
import org.apache.slider.core.registry.docstore.PublishedExports; import org.apache.slider.core.registry.docstore.PublishedExports;
import org.apache.slider.core.registry.docstore.PublishedExportsOutputter; import org.apache.slider.core.registry.docstore.PublishedExportsOutputter;
import org.apache.slider.core.registry.docstore.PublishedExportsSet; import org.apache.slider.core.registry.docstore.PublishedExportsSet;
@ -724,7 +725,8 @@ public class SliderClient extends AbstractSliderLaunchedService implements RunSe
AggregateConf instanceDefinition = loadInstanceDefinitionUnresolved( AggregateConf instanceDefinition = loadInstanceDefinitionUnresolved(
clustername, clusterDirectory); clustername, clusterDirectory);
try { try {
checkForCredentials(getConfig(), instanceDefinition.getAppConf()); checkForCredentials(getConfig(), instanceDefinition.getAppConf(),
clustername);
} catch (IOException e) { } catch (IOException e) {
sliderFileSystem.getFileSystem().delete(clusterDirectory, true); sliderFileSystem.getFileSystem().delete(clusterDirectory, true);
throw e; throw e;
@ -906,7 +908,7 @@ public class SliderClient extends AbstractSliderLaunchedService implements RunSe
} }
protected static void checkForCredentials(Configuration conf, protected static void checkForCredentials(Configuration conf,
ConfTree tree) throws IOException { ConfTree tree, String clusterName) throws IOException {
if (tree.credentials == null || tree.credentials.isEmpty()) { if (tree.credentials == null || tree.credentials.isEmpty()) {
log.info("No credentials requested"); log.info("No credentials requested");
return; return;
@ -915,7 +917,9 @@ public class SliderClient extends AbstractSliderLaunchedService implements RunSe
BufferedReader br = null; BufferedReader br = null;
try { try {
for (Entry<String, List<String>> cred : tree.credentials.entrySet()) { for (Entry<String, List<String>> cred : tree.credentials.entrySet()) {
String provider = cred.getKey(); String provider = cred.getKey()
.replaceAll(Pattern.quote("${CLUSTER_NAME}"), clusterName)
.replaceAll(Pattern.quote("${CLUSTER}"), clusterName);
List<String> aliases = cred.getValue(); List<String> aliases = cred.getValue();
if (aliases == null || aliases.isEmpty()) { if (aliases == null || aliases.isEmpty()) {
continue; continue;
@ -1727,6 +1731,7 @@ public class SliderClient extends AbstractSliderLaunchedService implements RunSe
resources.mergeComponents(buildInfo.getResourceCompOptionMap()); resources.mergeComponents(buildInfo.getResourceCompOptionMap());
builder.init(providerName, instanceDefinition); builder.init(providerName, instanceDefinition);
builder.resolve();
builder.propagateFilename(); builder.propagateFilename();
builder.propagatePrincipals(); builder.propagatePrincipals();
builder.setImageDetailsIfAvailable(buildInfo.getImage(), builder.setImageDetailsIfAvailable(buildInfo.getImage(),
@ -1917,8 +1922,7 @@ public class SliderClient extends AbstractSliderLaunchedService implements RunSe
private static String replaceTokens(String s, String userName, private static String replaceTokens(String s, String userName,
String clusterName) throws IOException { String clusterName) throws IOException {
return s.replaceAll(Pattern.quote("${USER}"), userName) return s.replaceAll(Pattern.quote("${USER}"), userName)
.replaceAll(Pattern.quote("${USER_NAME}"), userName) .replaceAll(Pattern.quote("${USER_NAME}"), userName);
.replaceAll(Pattern.quote("${CLUSTER_NAME}"), clusterName);
} }
public FsPermission getClusterDirectoryPermissions(Configuration conf) { public FsPermission getClusterDirectoryPermissions(Configuration conf) {

View File

@ -69,6 +69,19 @@ public interface SliderKeys extends SliderXmlConfKeys {
*/ */
String APP_TYPE = "org-apache-slider"; String APP_TYPE = "org-apache-slider";
/**
* Key for component type. This MUST NOT be set in app_config/global {@value}
*/
String COMPONENT_TYPE_KEY = "site.global.component_type";
/**
* A component type for an external app that has been predefined using the
* slider build command
*/
String COMPONENT_TYPE_EXTERNAL_APP = "external_app";
String COMPONENT_SEPARATOR = "-";
String[] COMPONENT_KEYS_TO_SKIP = {"zookeeper.", "env.MALLOC_ARENA_MAX",
"site.fs.", "site.dfs."};
/** /**
* Key for application version. This must be set in app_config/global {@value} * Key for application version. This must be set in app_config/global {@value}
*/ */

View File

@ -69,6 +69,7 @@ import org.apache.slider.core.exceptions.SliderException;
import org.apache.slider.core.launch.ClasspathConstructor; import org.apache.slider.core.launch.ClasspathConstructor;
import org.apache.slider.core.main.LauncherExitCodes; import org.apache.slider.core.main.LauncherExitCodes;
import org.apache.slider.providers.agent.AgentKeys; import org.apache.slider.providers.agent.AgentKeys;
import org.apache.slider.providers.agent.application.metadata.Component;
import org.apache.slider.server.services.utility.PatternValidator; import org.apache.slider.server.services.utility.PatternValidator;
import org.apache.slider.server.services.workflow.ForkedProcessService; import org.apache.slider.server.services.workflow.ForkedProcessService;
import org.apache.zookeeper.server.util.KerberosUtil; import org.apache.zookeeper.server.util.KerberosUtil;
@ -122,6 +123,8 @@ import java.util.zip.GZIPOutputStream;
import java.util.zip.ZipEntry; import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream; import java.util.zip.ZipOutputStream;
import static org.apache.slider.common.SliderKeys.COMPONENT_SEPARATOR;
/** /**
* These are slider-specific Util methods * These are slider-specific Util methods
*/ */
@ -475,6 +478,32 @@ public final class SliderUtils {
return srcFileCount; return srcFileCount;
} }
/**
* Copy a file to a new FS -both paths must be qualified.
* @param conf conf file
* @param srcFile src file
* @param destFile dest file
*/
public static void copy(Configuration conf,
Path srcFile,
Path destFile) throws
IOException,
BadClusterStateException {
FileSystem srcFS = FileSystem.get(srcFile.toUri(), conf);
//list all paths in the src.
if (!srcFS.exists(srcFile)) {
throw new FileNotFoundException("Source file not found " + srcFile);
}
if (!srcFS.isFile(srcFile)) {
throw new FileNotFoundException(
"Source file not a file " + srcFile);
}
FileSystem destFS = FileSystem.get(destFile.toUri(), conf);
if (destFS.exists(destFile)) {
throw new IOException("Dest file already exists " + destFile);
}
FileUtil.copy(srcFS, srcFile, destFS, destFile, false, true, conf);
}
public static String stringify(Throwable t) { public static String stringify(Throwable t) {
StringWriter sw = new StringWriter(); StringWriter sw = new StringWriter();
@ -926,6 +955,38 @@ public final class SliderUtils {
return first; return first;
} }
/**
* Merge string maps excluding prefixes
* @param first first map
* @param second second map
* @param prefixes prefixes to ignore
* @return 'first' merged with the second
*/
public static Map<String, String> mergeMapsIgnoreDuplicateKeysAndPrefixes(
Map<String, String> first, Map<String, String> second,
String... prefixes) {
Preconditions.checkArgument(first != null, "Null 'first' value");
Preconditions.checkArgument(second != null, "Null 'second' value");
Preconditions.checkArgument(prefixes != null, "Null 'prefixes' value");
for (Map.Entry<String, String> entry : second.entrySet()) {
String key = entry.getKey();
boolean hasPrefix = false;
for (String prefix : prefixes) {
if (key.startsWith(prefix)) {
hasPrefix = true;
break;
}
}
if (hasPrefix) {
continue;
}
if (!first.containsKey(key)) {
first.put(key, entry.getValue());
}
}
return first;
}
/** /**
* Convert a map to a multi-line string for printing * Convert a map to a multi-line string for printing
* @param map map to stringify * @param map map to stringify
@ -2352,8 +2413,28 @@ public final class SliderUtils {
*/ */
public static String getApplicationDefinitionPath(ConfTreeOperations conf) public static String getApplicationDefinitionPath(ConfTreeOperations conf)
throws BadConfigException { throws BadConfigException {
return getApplicationDefinitionPath(conf, null);
}
/**
* return the HDFS path where the application package has been uploaded
* manually or by using slider client (install package command)
*
* @param conf configuration
* @param roleGroup name of component
* @return
*/
public static String getApplicationDefinitionPath(ConfTreeOperations conf,
String roleGroup)
throws BadConfigException {
String appDefPath = conf.getGlobalOptions().getMandatoryOption( String appDefPath = conf.getGlobalOptions().getMandatoryOption(
AgentKeys.APP_DEF); AgentKeys.APP_DEF);
if (roleGroup != null) {
MapOperations component = conf.getComponent(roleGroup);
if (component != null) {
appDefPath = component.getOption(AgentKeys.APP_DEF, appDefPath);
}
}
return appDefPath; return appDefPath;
} }
@ -2545,4 +2626,11 @@ public final class SliderUtils {
} }
return buffer.toString(); return buffer.toString();
} }
public static String trimPrefix(String prefix) {
if (prefix != null && prefix.endsWith(COMPONENT_SEPARATOR)) {
return prefix.substring(0, prefix.length()-1);
}
return prefix;
}
} }

View File

@ -25,7 +25,9 @@ import org.apache.hadoop.fs.Path;
import org.apache.hadoop.fs.permission.FsPermission; import org.apache.hadoop.fs.permission.FsPermission;
import org.apache.slider.api.InternalKeys; import org.apache.slider.api.InternalKeys;
import org.apache.slider.api.OptionKeys; import org.apache.slider.api.OptionKeys;
import org.apache.slider.api.ResourceKeys;
import org.apache.slider.api.StatusKeys; import org.apache.slider.api.StatusKeys;
import org.apache.slider.common.SliderKeys;
import org.apache.slider.common.SliderXmlConfKeys; import org.apache.slider.common.SliderXmlConfKeys;
import org.apache.slider.common.tools.CoreFileSystem; import org.apache.slider.common.tools.CoreFileSystem;
import org.apache.slider.common.tools.SliderUtils; import org.apache.slider.common.tools.SliderUtils;
@ -42,11 +44,17 @@ import org.apache.slider.core.persist.LockAcquireFailedException;
import org.apache.slider.core.persist.LockHeldAction; import org.apache.slider.core.persist.LockHeldAction;
import org.apache.slider.core.zk.ZKPathBuilder; import org.apache.slider.core.zk.ZKPathBuilder;
import org.apache.slider.core.zk.ZookeeperUtils; import org.apache.slider.core.zk.ZookeeperUtils;
import org.apache.slider.providers.agent.AgentKeys;
import org.slf4j.Logger; import org.slf4j.Logger;
import org.slf4j.LoggerFactory; import org.slf4j.LoggerFactory;
import java.io.IOException; import java.io.IOException;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map; import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import java.util.TreeSet;
import static org.apache.slider.api.InternalKeys.INTERNAL_ADDONS_DIR_PATH; import static org.apache.slider.api.InternalKeys.INTERNAL_ADDONS_DIR_PATH;
import static org.apache.slider.api.InternalKeys.INTERNAL_APPDEF_DIR_PATH; import static org.apache.slider.api.InternalKeys.INTERNAL_APPDEF_DIR_PATH;
@ -61,6 +69,12 @@ import static org.apache.slider.api.OptionKeys.INTERNAL_SNAPSHOT_CONF_PATH;
import static org.apache.slider.api.OptionKeys.ZOOKEEPER_HOSTS; import static org.apache.slider.api.OptionKeys.ZOOKEEPER_HOSTS;
import static org.apache.slider.api.OptionKeys.ZOOKEEPER_PATH; import static org.apache.slider.api.OptionKeys.ZOOKEEPER_PATH;
import static org.apache.slider.api.OptionKeys.ZOOKEEPER_QUORUM; import static org.apache.slider.api.OptionKeys.ZOOKEEPER_QUORUM;
import static org.apache.slider.api.RoleKeys.ROLE_PREFIX;
import static org.apache.slider.common.SliderKeys.COMPONENT_AM;
import static org.apache.slider.common.SliderKeys.COMPONENT_SEPARATOR;
import static org.apache.slider.common.SliderKeys.COMPONENT_TYPE_EXTERNAL_APP;
import static org.apache.slider.common.SliderKeys.COMPONENT_TYPE_KEY;
import static org.apache.slider.common.tools.SliderUtils.isClusternameValid;
/** /**
* Build up the instance of a cluster. * Build up the instance of a cluster.
@ -72,6 +86,8 @@ public class InstanceBuilder {
private final CoreFileSystem coreFS; private final CoreFileSystem coreFS;
private final InstancePaths instancePaths; private final InstancePaths instancePaths;
private AggregateConf instanceDescription; private AggregateConf instanceDescription;
private Map<String, Path> externalAppDefs = new HashMap<>();
private TreeSet<Integer> priorities = new TreeSet<>();
private static final Logger log = private static final Logger log =
LoggerFactory.getLogger(InstanceBuilder.class); LoggerFactory.getLogger(InstanceBuilder.class);
@ -244,6 +260,192 @@ public class InstanceBuilder {
} }
private Set<String> getExternalComponents(ConfTreeOperations ops)
throws BadConfigException {
Set<String> externalComponents = new HashSet<>();
if (ops.getGlobalOptions().containsKey(COMPONENT_TYPE_KEY)) {
throw new BadConfigException(COMPONENT_TYPE_KEY + " must be " +
"specified per-component, not in global");
}
for (Entry<String, Map<String, String>> entry : ops.getComponents()
.entrySet()) {
if (COMPONENT_AM.equals(entry.getKey())) {
continue;
}
Map<String, String> options = entry.getValue();
if (COMPONENT_TYPE_EXTERNAL_APP.equals(options.get(COMPONENT_TYPE_KEY))) {
externalComponents.add(entry.getKey());
}
}
return externalComponents;
}
private void mergeExternalComponent(ConfTreeOperations ops,
ConfTreeOperations externalOps, String externalComponent,
Integer priority) throws BadConfigException {
for (String subComponent : externalOps.getComponentNames()) {
if (COMPONENT_AM.equals(subComponent)) {
continue;
}
String prefix = externalComponent + COMPONENT_SEPARATOR;
log.debug("Merging options for {} into {}", subComponent,
prefix + subComponent);
MapOperations subComponentOps = ops.getOrAddComponent(
prefix + subComponent);
if (priority == null) {
SliderUtils.mergeMaps(subComponentOps,
ops.getComponent(externalComponent).options);
subComponentOps.remove(COMPONENT_TYPE_KEY);
}
SliderUtils.mergeMapsIgnoreDuplicateKeysAndPrefixes(subComponentOps,
externalOps.getComponent(subComponent),
SliderKeys.COMPONENT_KEYS_TO_SKIP);
// add prefix to existing prefix
String existingPrefix = subComponentOps.get(ROLE_PREFIX);
if (existingPrefix != null) {
if (!subComponent.startsWith(existingPrefix)) {
throw new BadConfigException("Bad prefix " + existingPrefix +
" for subcomponent " + subComponent + " of " + externalComponent);
}
prefix = prefix + existingPrefix;
}
subComponentOps.set(ROLE_PREFIX, prefix);
// adjust priority
if (priority != null) {
subComponentOps.put(ResourceKeys.COMPONENT_PRIORITY,
Integer.toString(priority));
priorities.add(priority);
priority++;
}
}
}
private int getNextPriority() {
if (priorities.isEmpty()) {
return 1;
} else {
return priorities.last() + 1;
}
}
public void resolve()
throws BadConfigException, IOException, BadClusterStateException {
ConfTreeOperations appConf = instanceDescription.getAppConfOperations();
ConfTreeOperations resources = instanceDescription.getResourceOperations();
for (Entry<String, Map<String, String>> entry : resources.getComponents()
.entrySet()) {
if (COMPONENT_AM.equals(entry.getKey())) {
continue;
}
if (entry.getValue().containsKey(ResourceKeys.COMPONENT_PRIORITY)) {
priorities.add(Integer.parseInt(entry.getValue().get(
ResourceKeys.COMPONENT_PRIORITY)));
}
}
Set<String> externalComponents = getExternalComponents(appConf);
if (!externalComponents.isEmpty()) {
log.info("Found external components {}", externalComponents);
}
for (String component : externalComponents) {
if (!isClusternameValid(component)) {
throw new BadConfigException(component + " is not a valid external " +
"component");
}
Path componentClusterDir = coreFS.buildClusterDirPath(component);
try {
coreFS.verifyPathExists(componentClusterDir);
} catch (IOException e) {
throw new BadConfigException("external component " + component +
" doesn't exist");
}
AggregateConf componentConf = new AggregateConf();
ConfPersister persister = new ConfPersister(coreFS, componentClusterDir);
try {
persister.load(componentConf);
} catch (Exception e) {
throw new BadConfigException("Couldn't read configuration for " +
"external component " + component);
}
ConfTreeOperations componentAppConf = componentConf.getAppConfOperations();
String externalAppDef = componentAppConf.get(AgentKeys.APP_DEF);
if (SliderUtils.isSet(externalAppDef)) {
Path newAppDef = new Path(coreFS.buildAppDefDirPath(clustername),
component + "_" + SliderKeys.DEFAULT_APP_PKG);
componentAppConf.set(AgentKeys.APP_DEF, newAppDef);
componentAppConf.append(AgentKeys.APP_DEF_ORIGINAL, externalAppDef);
log.info("Copying external appdef {} to {} for {}", externalAppDef,
newAppDef, component);
externalAppDefs.put(externalAppDef, newAppDef);
externalAppDef = newAppDef.toString();
}
for (String rcomp : componentConf.getResourceOperations()
.getComponentNames()) {
if (COMPONENT_AM.equals(rcomp)) {
continue;
}
log.debug("Adding component {} to appConf for {}", rcomp, component);
componentAppConf.getOrAddComponent(rcomp);
}
componentConf.resolve();
for (String rcomp : componentConf.getResourceOperations()
.getComponentNames()) {
if (COMPONENT_AM.equals(rcomp)) {
continue;
}
String componentAppDef = componentAppConf.getComponentOpt(
rcomp, AgentKeys.APP_DEF, null);
if (SliderUtils.isUnset(componentAppDef) ||
componentAppDef.equals(externalAppDef)) {
continue;
}
if (externalAppDefs.containsKey(componentAppDef)) {
log.info("Using external appdef {} for {}",
externalAppDefs.get(componentAppDef), rcomp);
} else {
String existingPrefix = componentAppConf.getComponentOpt(rcomp,
ROLE_PREFIX, null);
if (SliderUtils.isUnset(existingPrefix)) {
existingPrefix = "";
} else {
existingPrefix = COMPONENT_SEPARATOR + SliderUtils.trimPrefix(
existingPrefix);
}
Path newAppDef = new Path(coreFS.buildAppDefDirPath(clustername),
component + existingPrefix + "_" + SliderKeys.DEFAULT_APP_PKG);
externalAppDefs.put(componentAppDef, newAppDef);
log.info("Copying external appdef {} to {} for {}", componentAppDef,
newAppDef, component + COMPONENT_SEPARATOR + rcomp);
}
componentAppConf.setComponentOpt(rcomp, AgentKeys.APP_DEF,
externalAppDefs.get(componentAppDef).toString());
componentAppConf.appendComponentOpt(rcomp,
AgentKeys.APP_DEF_ORIGINAL, componentAppDef);
}
Set<Path> newAppDefs = new HashSet<>();
newAppDefs.addAll(externalAppDefs.values());
if (newAppDefs.size() != externalAppDefs.size()) {
throw new IllegalStateException("Values repeat in external appdefs "
+ externalAppDefs);
}
log.info("External appdefs after {}: {}", component, externalAppDefs);
mergeExternalComponent(appConf, componentAppConf, component, null);
mergeExternalComponent(resources, componentConf.getResourceOperations(),
component, getNextPriority());
}
}
/** /**
* Persist this * Persist this
* @param appconfdir conf dir * @param appconfdir conf dir
@ -266,6 +468,9 @@ public class InstanceBuilder {
action = new ConfDirSnapshotAction(appconfdir); action = new ConfDirSnapshotAction(appconfdir);
} }
persister.save(instanceDescription, action); persister.save(instanceDescription, action);
for (Entry<String, Path> appDef : externalAppDefs.entrySet()) {
SliderUtils.copy(conf, new Path(appDef.getKey()), appDef.getValue());
}
} }
/** /**

View File

@ -214,6 +214,23 @@ public class ConfTreeOperations {
public String get(String key) { public String get(String key) {
return globalOptions.get(key); return globalOptions.get(key);
} }
/**
* append to a global option
* @param key key
* @return value
*
*/
public String append(String key, String value) {
if (SliderUtils.isUnset(value)) {
return null;
}
if (globalOptions.containsKey(key)) {
globalOptions.put(key, globalOptions.get(key) + "," + value);
} else {
globalOptions.put(key, value);
}
return globalOptions.get(key);
}
/** /**
* Propagate all global keys matching a prefix * Propagate all global keys matching a prefix
@ -257,6 +274,17 @@ public class ConfTreeOperations {
Map<String, String> map, Map<String, String> map,
String prefix, String prefix,
boolean overwrite) { boolean overwrite) {
boolean needsMerge = false;
for (Map.Entry<String, String> entry : map.entrySet()) {
String key = entry.getKey();
if (key.startsWith(prefix)) {
needsMerge = true;
break;
}
}
if (!needsMerge) {
return;
}
MapOperations comp = getOrAddComponent(component); MapOperations comp = getOrAddComponent(component);
comp.mergeMapPrefixedKeys(map,prefix, overwrite); comp.mergeMapPrefixedKeys(map,prefix, overwrite);
} }
@ -474,4 +502,26 @@ public class ConfTreeOperations {
setComponentOpt(role, option, Long.toString(val)); setComponentOpt(role, option, Long.toString(val));
} }
/**
* append to a component option
* @param key key
* @return value
*
*/
public String appendComponentOpt(String role, String key, String value) {
if (SliderUtils.isUnset(value)) {
return null;
}
MapOperations roleopts = getComponent(role);
if (roleopts == null) {
return null;
}
if (roleopts.containsKey(key)) {
roleopts.put(key, roleopts.get(key) + "," + value);
} else {
roleopts.put(key, value);
}
return roleopts.get(key);
}
} }

View File

@ -169,7 +169,8 @@ public abstract class AbstractProviderService
*/ */
@Override @Override
public void initializeApplicationConfiguration( public void initializeApplicationConfiguration(
AggregateConf instanceDefinition, SliderFileSystem fileSystem) AggregateConf instanceDefinition, SliderFileSystem fileSystem,
String roleGroup)
throws IOException, SliderException { throws IOException, SliderException {
} }

View File

@ -118,11 +118,13 @@ public interface ProviderService extends ProviderCore,
* *
* @param instanceDefinition * @param instanceDefinition
* @param fileSystem * @param fileSystem
* @param roleGroup
* @throws IOException * @throws IOException
* @throws SliderException * @throws SliderException
*/ */
void initializeApplicationConfiguration(AggregateConf instanceDefinition, void initializeApplicationConfiguration(AggregateConf instanceDefinition,
SliderFileSystem fileSystem) throws IOException, SliderException; SliderFileSystem fileSystem, String roleGroup) throws IOException,
SliderException;
/** /**
* This is a validation of the application configuration on the AM. * This is a validation of the application configuration on the AM.

View File

@ -176,13 +176,22 @@ public class AgentClientProvider extends AbstractClientProvider
names.remove(SliderKeys.COMPONENT_AM); names.remove(SliderKeys.COMPONENT_AM);
Map<Integer, String> priorityMap = new HashMap<Integer, String>(); Map<Integer, String> priorityMap = new HashMap<Integer, String>();
Metainfo metaInfo = getMetainfo(fs, appDef);
for (String name : names) { for (String name : names) {
try {
// Validate the app definition
appDef = SliderUtils.getApplicationDefinitionPath(instanceDefinition
.getAppConfOperations(), name);
} catch (BadConfigException bce) {
throw new BadConfigException("Application definition must be provided. " + bce.getMessage());
}
Metainfo metaInfo = getMetainfo(fs, appDef);
MapOperations component = resources.getMandatoryComponent(name); MapOperations component = resources.getMandatoryComponent(name);
if (metaInfo != null) { if (metaInfo != null) {
Component componentDef = metaInfo.getApplicationComponent(name); Component componentDef = metaInfo.getApplicationComponent(
AgentUtils.getMetainfoComponentName(name,
instanceDefinition.getAppConfOperations()));
if (componentDef == null) { if (componentDef == null) {
throw new BadConfigException( throw new BadConfigException(
"Component %s is not a member of application.", name); "Component %s is not a member of application.", name);
@ -208,16 +217,12 @@ public class AgentClientProvider extends AbstractClientProvider
existing); existing);
} }
priorityMap.put(priority, name); priorityMap.put(priority, name);
}
// fileSystem may be null for tests // fileSystem may be null for tests
if (metaInfo != null) { if (metaInfo != null) {
for (String name : names) { Component componentDef = metaInfo.getApplicationComponent(
Component componentDef = metaInfo.getApplicationComponent(name); AgentUtils.getMetainfoComponentName(name,
if (componentDef == null) { instanceDefinition.getAppConfOperations()));
throw new BadConfigException(
"Component %s is not a member of application.", name);
}
// ensure that intance count is 0 for client components // ensure that intance count is 0 for client components
if ("CLIENT".equals(componentDef.getCategory())) { if ("CLIENT".equals(componentDef.getCategory())) {

View File

@ -69,6 +69,7 @@ public interface AgentKeys {
String AGENT_MAIN_SCRIPT = "agent/main.py"; String AGENT_MAIN_SCRIPT = "agent/main.py";
String APP_DEF = "application.def"; String APP_DEF = "application.def";
String APP_DEF_ORIGINAL = "application.def.original";
String ADDON_PREFIX = "application.addon."; String ADDON_PREFIX = "application.addon.";
String ADDONS = "application.addons"; String ADDONS = "application.addons";
String AGENT_VERSION = "agent.version"; String AGENT_VERSION = "agent.version";
@ -104,6 +105,8 @@ public interface AgentKeys {
String KEY_CONTAINER_LAUNCH_DELAY = "container.launch.delay.sec"; String KEY_CONTAINER_LAUNCH_DELAY = "container.launch.delay.sec";
String TEST_RELAX_VERIFICATION = "test.relax.validation"; String TEST_RELAX_VERIFICATION = "test.relax.validation";
String AM_CONFIG_GENERATION = "am.config.generation"; String AM_CONFIG_GENERATION = "am.config.generation";
String DEFAULT_METAINFO_MAP_KEY = "DEFAULT_KEY";
} }

View File

@ -75,6 +75,7 @@ import org.apache.slider.providers.ProviderRole;
import org.apache.slider.providers.ProviderUtils; import org.apache.slider.providers.ProviderUtils;
import org.apache.slider.providers.agent.application.metadata.AbstractComponent; import org.apache.slider.providers.agent.application.metadata.AbstractComponent;
import org.apache.slider.providers.agent.application.metadata.Application; import org.apache.slider.providers.agent.application.metadata.Application;
import org.apache.slider.providers.agent.application.metadata.CommandOrder;
import org.apache.slider.providers.agent.application.metadata.CommandScript; import org.apache.slider.providers.agent.application.metadata.CommandScript;
import org.apache.slider.providers.agent.application.metadata.Component; import org.apache.slider.providers.agent.application.metadata.Component;
import org.apache.slider.providers.agent.application.metadata.ComponentCommand; import org.apache.slider.providers.agent.application.metadata.ComponentCommand;
@ -128,6 +129,7 @@ import java.util.LinkedHashMap;
import java.util.List; import java.util.List;
import java.util.Locale; import java.util.Locale;
import java.util.Map; import java.util.Map;
import java.util.Map.Entry;
import java.util.Scanner; import java.util.Scanner;
import java.util.Set; import java.util.Set;
import java.util.TreeMap; import java.util.TreeMap;
@ -135,7 +137,9 @@ import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CopyOnWriteArrayList; import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger; import java.util.concurrent.atomic.AtomicInteger;
import java.util.regex.Pattern;
import static org.apache.slider.api.RoleKeys.ROLE_PREFIX;
import static org.apache.slider.server.appmaster.web.rest.RestPaths.SLIDER_PATH_AGENTS; import static org.apache.slider.server.appmaster.web.rest.RestPaths.SLIDER_PATH_AGENTS;
/** /**
@ -170,10 +174,10 @@ public class AgentProviderService extends AbstractProviderService implements
private int heartbeatMonitorInterval = 0; private int heartbeatMonitorInterval = 0;
private AgentClientProvider clientProvider; private AgentClientProvider clientProvider;
private AtomicInteger taskId = new AtomicInteger(0); private AtomicInteger taskId = new AtomicInteger(0);
private volatile Metainfo metaInfo = null; private volatile Map<String, MetainfoHolder> metaInfoMap = new HashMap<>();
private SliderFileSystem fileSystem = null; private SliderFileSystem fileSystem = null;
private Map<String, DefaultConfig> defaultConfigs = null; private Map<String, DefaultConfig> defaultConfigs = null;
private ComponentCommandOrder commandOrder = null; private ComponentCommandOrder commandOrder = new ComponentCommandOrder();
private HeartbeatMonitor monitor; private HeartbeatMonitor monitor;
private Boolean canAnyMasterPublish = null; private Boolean canAnyMasterPublish = null;
private AgentLaunchParameter agentLaunchParameter = null; private AgentLaunchParameter agentLaunchParameter = null;
@ -208,6 +212,17 @@ public class AgentProviderService extends AbstractProviderService implements
private final Map<String, Set<String>> containerExportsMap = private final Map<String, Set<String>> containerExportsMap =
new HashMap<String, Set<String>>(); new HashMap<String, Set<String>>();
private static class MetainfoHolder {
Metainfo metaInfo;
private Map<String, DefaultConfig> defaultConfigs = null;
public MetainfoHolder(Metainfo metaInfo,
Map<String, DefaultConfig> defaultConfigs) {
this.metaInfo = metaInfo;
this.defaultConfigs = defaultConfigs;
}
}
/** /**
* Create an instance of AgentProviderService * Create an instance of AgentProviderService
*/ */
@ -252,10 +267,11 @@ public class AgentProviderService extends AbstractProviderService implements
Set<String> names = resources.getComponentNames(); Set<String> names = resources.getComponentNames();
names.remove(SliderKeys.COMPONENT_AM); names.remove(SliderKeys.COMPONENT_AM);
for (String name : names) { for (String name : names) {
Component componentDef = getMetaInfo().getApplicationComponent(name); Component componentDef = getApplicationComponent(name);
if (componentDef == null) { if (componentDef == null) {
throw new BadConfigException( // component member is validated elsewhere, so we don't need to throw
"Component %s is not a member of application.", name); // an exception here
continue;
} }
MapOperations componentConfig = resources.getMandatoryComponent(name); MapOperations componentConfig = resources.getMandatoryComponent(name);
@ -277,32 +293,67 @@ public class AgentProviderService extends AbstractProviderService implements
// Reads the metainfo.xml in the application package and loads it // Reads the metainfo.xml in the application package and loads it
private void buildMetainfo(AggregateConf instanceDefinition, private void buildMetainfo(AggregateConf instanceDefinition,
SliderFileSystem fileSystem) throws IOException, SliderException { SliderFileSystem fileSystem,
String appDef = SliderUtils.getApplicationDefinitionPath(instanceDefinition String roleGroup)
.getAppConfOperations()); throws IOException, SliderException {
String mapKey = instanceDefinition.getAppConfOperations()
.getComponentOpt(roleGroup, ROLE_PREFIX, DEFAULT_METAINFO_MAP_KEY);
String appDef = SliderUtils.getApplicationDefinitionPath(
instanceDefinition.getAppConfOperations(), roleGroup);
MapOperations component = null;
if (roleGroup != null) {
component = instanceDefinition.getAppConfOperations().getComponent(roleGroup);
}
if (metaInfo == null) { MetainfoHolder metaInfoHolder = metaInfoMap.get(mapKey);
if (metaInfoHolder == null) {
synchronized (syncLock) { synchronized (syncLock) {
if (metaInfo == null) { if (this.fileSystem == null) {
this.fileSystem = fileSystem; this.fileSystem = fileSystem;
}
metaInfoHolder = metaInfoMap.get(mapKey);
if (metaInfoHolder == null) {
readAndSetHeartbeatMonitoringInterval(instanceDefinition); readAndSetHeartbeatMonitoringInterval(instanceDefinition);
initializeAgentDebugCommands(instanceDefinition); initializeAgentDebugCommands(instanceDefinition);
metaInfo = getApplicationMetainfo(fileSystem, appDef, false); Metainfo metaInfo = getApplicationMetainfo(fileSystem, appDef, false);
log.info("Master package metainfo: {}", metaInfo.toString()); log.info("Master package metainfo: {}", metaInfo.toString());
if (metaInfo == null || metaInfo.getApplication() == null) { if (metaInfo == null || metaInfo.getApplication() == null) {
log.error("metainfo.xml is unavailable or malformed at {}.", appDef); log.error("metainfo.xml is unavailable or malformed at {}.", appDef);
throw new SliderException( throw new SliderException(
"metainfo.xml is required in app package."); "metainfo.xml is required in app package.");
} }
commandOrder = new ComponentCommandOrder(metaInfo.getApplication().getCommandOrders()); List<CommandOrder> commandOrders = metaInfo.getApplication()
defaultConfigs = initializeDefaultConfigs(fileSystem, appDef, metaInfo); .getCommandOrders();
if (!DEFAULT_METAINFO_MAP_KEY.equals(mapKey)) {
for (Component comp : metaInfo.getApplication().getComponents()) {
comp.setName(mapKey + comp.getName());
log.info("Modifying external metainfo component name to {}",
comp.getName());
}
for (CommandOrder co : commandOrders) {
log.info("Adding prefix {} to command order {}",
mapKey, co);
co.setCommand(mapKey + co.getCommand());
co.setRequires(mapKey + co.getRequires());
}
}
log.debug("Merging command orders {} for {}", commandOrders,
roleGroup);
commandOrder.mergeCommandOrders(commandOrders,
instanceDefinition.getResourceOperations());
Map<String, DefaultConfig> defaultConfigs =
initializeDefaultConfigs(fileSystem, appDef, metaInfo);
metaInfoMap.put(mapKey, new MetainfoHolder(metaInfo, defaultConfigs));
monitor = new HeartbeatMonitor(this, getHeartbeatMonitorInterval()); monitor = new HeartbeatMonitor(this, getHeartbeatMonitorInterval());
monitor.start(); monitor.start();
// build a map from component to metainfo // build a map from component to metainfo
String addonAppDefString = instanceDefinition.getAppConfOperations() String addonAppDefString = instanceDefinition.getAppConfOperations()
.getGlobalOptions().getOption(AgentKeys.ADDONS, null); .getGlobalOptions().getOption(AgentKeys.ADDONS, null);
if (component != null) {
addonAppDefString = component.getOption(AgentKeys.ADDONS, addonAppDefString);
}
log.debug("All addon appdefs: {}", addonAppDefString); log.debug("All addon appdefs: {}", addonAppDefString);
if (addonAppDefString != null) { if (addonAppDefString != null) {
Scanner scanner = new Scanner(addonAppDefString).useDelimiter(","); Scanner scanner = new Scanner(addonAppDefString).useDelimiter(",");
@ -310,6 +361,9 @@ public class AgentProviderService extends AbstractProviderService implements
String addonAppDef = scanner.next(); String addonAppDef = scanner.next();
String addonAppDefPath = instanceDefinition String addonAppDefPath = instanceDefinition
.getAppConfOperations().getGlobalOptions().get(addonAppDef); .getAppConfOperations().getGlobalOptions().get(addonAppDef);
if (component != null) {
addonAppDefPath = component.getOption(addonAppDef, addonAppDefPath);
}
log.debug("Addon package {} is stored at: {}", addonAppDef log.debug("Addon package {} is stored at: {}", addonAppDef
+ addonAppDefPath); + addonAppDefPath);
Metainfo addonMetaInfo = getApplicationMetainfo(fileSystem, Metainfo addonMetaInfo = getApplicationMetainfo(fileSystem,
@ -328,9 +382,10 @@ public class AgentProviderService extends AbstractProviderService implements
@Override @Override
public void initializeApplicationConfiguration( public void initializeApplicationConfiguration(
AggregateConf instanceDefinition, SliderFileSystem fileSystem) AggregateConf instanceDefinition, SliderFileSystem fileSystem,
String roleGroup)
throws IOException, SliderException { throws IOException, SliderException {
buildMetainfo(instanceDefinition, fileSystem); buildMetainfo(instanceDefinition, fileSystem, roleGroup);
} }
@Override @Override
@ -349,9 +404,9 @@ public class AgentProviderService extends AbstractProviderService implements
String roleName = providerRole.name; String roleName = providerRole.name;
String roleGroup = providerRole.group; String roleGroup = providerRole.group;
String appDef = SliderUtils.getApplicationDefinitionPath(instanceDefinition String appDef = SliderUtils.getApplicationDefinitionPath(instanceDefinition
.getAppConfOperations()); .getAppConfOperations(), roleGroup);
initializeApplicationConfiguration(instanceDefinition, fileSystem); initializeApplicationConfiguration(instanceDefinition, fileSystem, roleGroup);
log.info("Build launch context for Agent"); log.info("Build launch context for Agent");
log.debug(instanceDefinition.toString()); log.debug(instanceDefinition.toString());
@ -439,7 +494,7 @@ public class AgentProviderService extends AbstractProviderService implements
LocalResourceType.ARCHIVE); LocalResourceType.ARCHIVE);
launcher.addLocalResource(AgentKeys.APP_DEFINITION_DIR, appDefRes); launcher.addLocalResource(AgentKeys.APP_DEFINITION_DIR, appDefRes);
for (Package pkg : getMetaInfo().getApplication().getPackages()) { for (Package pkg : getMetaInfo(roleGroup).getApplication().getPackages()) {
Path pkgPath = fileSystem.buildResourcePath(pkg.getName()); Path pkgPath = fileSystem.buildResourcePath(pkg.getName());
if (!fileSystem.isFile(pkgPath)) { if (!fileSystem.isFile(pkgPath)) {
pkgPath = fileSystem.buildResourcePath(getClusterName(), pkgPath = fileSystem.buildResourcePath(getClusterName(),
@ -505,7 +560,7 @@ public class AgentProviderService extends AbstractProviderService implements
Map<String, Map<String, String>> configurations = Map<String, Map<String, String>> configurations =
buildCommandConfigurations(instanceDefinition.getAppConfOperations(), buildCommandConfigurations(instanceDefinition.getAppConfOperations(),
container.getId().toString(), roleName, roleGroup); container.getId().toString(), roleName, roleGroup);
localizeConfigFiles(launcher, roleName, roleGroup, getMetaInfo(), localizeConfigFiles(launcher, roleName, roleGroup, getMetaInfo(roleGroup),
configurations, launcher.getEnv(), fileSystem); configurations, launcher.getEnv(), fileSystem);
} }
@ -597,7 +652,7 @@ public class AgentProviderService extends AbstractProviderService implements
// initialize the component instance state // initialize the component instance state
getComponentStatuses().put(label, getComponentStatuses().put(label,
new ComponentInstanceState( new ComponentInstanceState(
roleName, roleGroup,
container.getId(), container.getId(),
getClusterInfoPropertyValue(OptionKeys.APPLICATION_NAME), getClusterInfoPropertyValue(OptionKeys.APPLICATION_NAME),
pkgStatuses)); pkgStatuses));
@ -610,6 +665,22 @@ public class AgentProviderService extends AbstractProviderService implements
AggregateConf instanceDefinition, AggregateConf instanceDefinition,
MapOperations compOps) MapOperations compOps)
throws SliderException, IOException { throws SliderException, IOException {
// substitute CLUSTER_NAME into credentials
Map<String,List<String>> newcred = new HashMap<>();
for (Entry<String,List<String>> entry : instanceDefinition.getAppConf().credentials.entrySet()) {
List<String> resultList = new ArrayList<>();
for (String v : entry.getValue()) {
resultList.add(v.replaceAll(Pattern.quote("${CLUSTER_NAME}"),
clusterName).replaceAll(Pattern.quote("${CLUSTER}"),
clusterName));
}
newcred.put(entry.getKey().replaceAll(Pattern.quote("${CLUSTER_NAME}"),
clusterName).replaceAll(Pattern.quote("${CLUSTER}"),
clusterName),
resultList);
}
instanceDefinition.getAppConf().credentials = newcred;
// generate and localize security stores // generate and localize security stores
SecurityStore[] stores = generateSecurityStores(container, role, SecurityStore[] stores = generateSecurityStores(container, role,
instanceDefinition, compOps); instanceDefinition, compOps);
@ -858,11 +929,12 @@ public class AgentProviderService extends AbstractProviderService implements
.extractRole(container)); .extractRole(container));
if (role != null) { if (role != null) {
String roleName = role.name; String roleName = role.name;
String label = getContainerLabel(container, roleName, role.group); String roleGroup = role.group;
String label = getContainerLabel(container, roleName, roleGroup);
log.info("Rebuilding in-memory: container {} in role {} in cluster {}", log.info("Rebuilding in-memory: container {} in role {} in cluster {}",
container.getId(), roleName, applicationId); container.getId(), roleName, applicationId);
getComponentStatuses().put(label, getComponentStatuses().put(label,
new ComponentInstanceState(roleName, container.getId(), new ComponentInstanceState(roleGroup, container.getId(),
applicationId)); applicationId));
} else { } else {
log.warn("Role not found for container {} in cluster {}", log.warn("Role not found for container {} in cluster {}",
@ -983,7 +1055,7 @@ public class AgentProviderService extends AbstractProviderService implements
StateAccessForProviders accessor = getAmState(); StateAccessForProviders accessor = getAmState();
CommandScript cmdScript = getScriptPathForMasterPackage(roleGroup); CommandScript cmdScript = getScriptPathForMasterPackage(roleGroup);
List<ComponentCommand> commands = getMetaInfo().getApplicationComponent(roleGroup).getCommands(); List<ComponentCommand> commands = getApplicationComponent(roleGroup).getCommands();
if (!isDockerContainer(roleGroup) && !isYarnDockerContainer(roleGroup) if (!isDockerContainer(roleGroup) && !isYarnDockerContainer(roleGroup)
&& (cmdScript == null || cmdScript.getScript() == null) && (cmdScript == null || cmdScript.getScript() == null)
@ -1261,7 +1333,7 @@ public class AgentProviderService extends AbstractProviderService implements
} }
private boolean isDockerContainer(String roleGroup) { private boolean isDockerContainer(String roleGroup) {
String type = getMetaInfo().getApplicationComponent(roleGroup).getType(); String type = getApplicationComponent(roleGroup).getType();
if (SliderUtils.isSet(type)) { if (SliderUtils.isSet(type)) {
return type.toLowerCase().equals(SliderUtils.DOCKER) || type.toLowerCase().equals(SliderUtils.DOCKER_YARN); return type.toLowerCase().equals(SliderUtils.DOCKER) || type.toLowerCase().equals(SliderUtils.DOCKER_YARN);
} }
@ -1269,7 +1341,7 @@ public class AgentProviderService extends AbstractProviderService implements
} }
private boolean isYarnDockerContainer(String roleGroup) { private boolean isYarnDockerContainer(String roleGroup) {
String type = getMetaInfo().getApplicationComponent(roleGroup).getType(); String type = getApplicationComponent(roleGroup).getType();
if (SliderUtils.isSet(type)) { if (SliderUtils.isSet(type)) {
return type.toLowerCase().equals(SliderUtils.DOCKER_YARN); return type.toLowerCase().equals(SliderUtils.DOCKER_YARN);
} }
@ -1393,23 +1465,21 @@ public class AgentProviderService extends AbstractProviderService implements
throw new IOException(e); throw new IOException(e);
} }
for (ConfigFile configFile : getMetaInfo() for (ConfigFile configFile : getMetaInfo().getComponentConfigFiles(client.getName())) {
.getComponentConfigFiles(client.getName())) {
addNamedConfiguration(configFile.getDictionaryName(), addNamedConfiguration(configFile.getDictionaryName(),
appConf.getGlobalOptions().options, configurations, tokens, null, appConf.getGlobalOptions().options, configurations, tokens, null,
client.getName()); client.getName(), client.getName());
if (appConf.getComponent(client.getName()) != null) { if (appConf.getComponent(client.getName()) != null) {
addNamedConfiguration(configFile.getDictionaryName(), addNamedConfiguration(configFile.getDictionaryName(),
appConf.getComponent(client.getName()).options, configurations, appConf.getComponent(client.getName()).options, configurations,
tokens, null, client.getName()); tokens, null, client.getName(), client.getName());
} }
} }
//do a final replacement of re-used configs //do a final replacement of re-used configs
dereferenceAllConfigs(configurations); dereferenceAllConfigs(configurations);
for (ConfigFile configFile : getMetaInfo() for (ConfigFile configFile : getMetaInfo().getComponentConfigFiles(client.getName())) {
.getComponentConfigFiles(client.getName())) {
ConfigFormat configFormat = ConfigFormat.resolve(configFile.getType()); ConfigFormat configFormat = ConfigFormat.resolve(configFile.getType());
Map<String, String> config = configurations.get(configFile.getDictionaryName()); Map<String, String> config = configurations.get(configFile.getDictionaryName());
@ -1525,9 +1595,23 @@ public class AgentProviderService extends AbstractProviderService implements
return workFolderExports; return workFolderExports;
} }
@VisibleForTesting
protected Metainfo getMetaInfo() { protected Metainfo getMetaInfo() {
return this.metaInfo; return getMetaInfo(null);
}
@VisibleForTesting
protected Metainfo getMetaInfo(String roleGroup) {
String mapKey = DEFAULT_METAINFO_MAP_KEY;
if (roleGroup != null) {
ConfTreeOperations appConf = getAmState().getAppConfSnapshot();
mapKey = appConf.getComponentOpt(roleGroup, ROLE_PREFIX,
DEFAULT_METAINFO_MAP_KEY);
}
MetainfoHolder mh = this.metaInfoMap.get(mapKey);
if (mh == null) {
return null;
}
return mh.metaInfo;
} }
@VisibleForTesting @VisibleForTesting
@ -1597,8 +1681,11 @@ public class AgentProviderService extends AbstractProviderService implements
return defaultConfigMap; return defaultConfigMap;
} }
protected Map<String, DefaultConfig> getDefaultConfigs() { protected Map<String, DefaultConfig> getDefaultConfigs(String roleGroup) {
return defaultConfigs; ConfTreeOperations appConf = getAmState().getAppConfSnapshot();
String mapKey = appConf.getComponentOpt(roleGroup, ROLE_PREFIX,
DEFAULT_METAINFO_MAP_KEY);
return metaInfoMap.get(mapKey).defaultConfigs;
} }
private int getHeartbeatMonitorInterval() { private int getHeartbeatMonitorInterval() {
@ -1764,9 +1851,9 @@ public class AgentProviderService extends AbstractProviderService implements
log.info("Status report: {}", status.toString()); log.info("Status report: {}", status.toString());
if (status.getConfigs() != null) { if (status.getConfigs() != null) {
Application application = getMetaInfo().getApplication(); Application application = getMetaInfo(componentGroup).getApplication();
if ((!canAnyMasterPublishConfig() || canPublishConfig(componentGroup)) && if ((!canAnyMasterPublishConfig(componentGroup) || canPublishConfig(componentGroup)) &&
!getAmState().getAppConfSnapshot().getComponentOptBool( !getAmState().getAppConfSnapshot().getComponentOptBool(
componentGroup, AgentKeys.AM_CONFIG_GENERATION, false)) { componentGroup, AgentKeys.AM_CONFIG_GENERATION, false)) {
// If no Master can explicitly publish then publish if its a master // If no Master can explicitly publish then publish if its a master
@ -1914,7 +2001,7 @@ public class AgentProviderService extends AbstractProviderService implements
String hostNamePattern = "${THIS_HOST}"; String hostNamePattern = "${THIS_HOST}";
Map<String, String> toPublish = new HashMap<String, String>(); Map<String, String> toPublish = new HashMap<String, String>();
Application application = getMetaInfo().getApplication(); Application application = getMetaInfo(componentGroup).getApplication();
for (Component component : application.getComponents()) { for (Component component : application.getComponents()) {
if (component.getName().equals(componentGroup)) { if (component.getName().equals(componentGroup)) {
if (component.getComponentExports().size() > 0) { if (component.getComponentExports().size() > 0) {
@ -1965,8 +2052,8 @@ public class AgentProviderService extends AbstractProviderService implements
String portVarFormat = "${site.%s}"; String portVarFormat = "${site.%s}";
String hostNamePattern = "${" + compGroup + "_HOST}"; String hostNamePattern = "${" + compGroup + "_HOST}";
List<ExportGroup> appExportGroups = getMetaInfo().getApplication().getExportGroups(); List<ExportGroup> appExportGroups = getMetaInfo(compGroup).getApplication().getExportGroups();
Component component = getMetaInfo().getApplicationComponent(compGroup); Component component = getApplicationComponent(compGroup);
if (component != null && SliderUtils.isSet(component.getCompExports()) if (component != null && SliderUtils.isSet(component.getCompExports())
&& SliderUtils.isNotEmpty(appExportGroups)) { && SliderUtils.isNotEmpty(appExportGroups)) {
@ -2068,7 +2155,11 @@ public class AgentProviderService extends AbstractProviderService implements
* @return the component entry or null for no match * @return the component entry or null for no match
*/ */
protected Component getApplicationComponent(String roleGroup) { protected Component getApplicationComponent(String roleGroup) {
return getMetaInfo().getApplicationComponent(roleGroup); Metainfo metainfo = getMetaInfo(roleGroup);
if (metainfo == null) {
return null;
}
return metainfo.getApplicationComponent(roleGroup);
} }
/** /**
@ -2137,9 +2228,9 @@ public class AgentProviderService extends AbstractProviderService implements
* *
* @return true if the condition holds * @return true if the condition holds
*/ */
protected boolean canAnyMasterPublishConfig() { protected boolean canAnyMasterPublishConfig(String roleGroup) {
if (canAnyMasterPublish == null) { if (canAnyMasterPublish == null) {
Application application = getMetaInfo().getApplication(); Application application = getMetaInfo(roleGroup).getApplication();
if (application == null) { if (application == null) {
log.error("Malformed app definition: Expect application as root element in the metainfo.xml"); log.error("Malformed app definition: Expect application as root element in the metainfo.xml");
} else { } else {
@ -2214,7 +2305,7 @@ public class AgentProviderService extends AbstractProviderService implements
cmd.setPkg(pkg); cmd.setPkg(pkg);
Map<String, String> hostLevelParams = new TreeMap<String, String>(); Map<String, String> hostLevelParams = new TreeMap<String, String>();
hostLevelParams.put(JAVA_HOME, appConf.getGlobalOptions().getOption(JAVA_HOME, getJDKDir())); hostLevelParams.put(JAVA_HOME, appConf.getGlobalOptions().getOption(JAVA_HOME, getJDKDir()));
hostLevelParams.put(PACKAGE_LIST, getPackageList()); hostLevelParams.put(PACKAGE_LIST, getPackageList(roleGroup));
hostLevelParams.put(CONTAINER_ID, containerId); hostLevelParams.put(CONTAINER_ID, containerId);
cmd.setHostLevelParams(hostLevelParams); cmd.setHostLevelParams(hostLevelParams);
@ -2263,7 +2354,7 @@ public class AgentProviderService extends AbstractProviderService implements
cmd.setComponentName(roleName); cmd.setComponentName(roleName);
cmd.setRole(roleName); cmd.setRole(roleName);
Map<String, String> hostLevelParams = new TreeMap<String, String>(); Map<String, String> hostLevelParams = new TreeMap<String, String>();
hostLevelParams.put(PACKAGE_LIST, getPackageList()); hostLevelParams.put(PACKAGE_LIST, getPackageList(roleGroup));
hostLevelParams.put(CONTAINER_ID, containerId); hostLevelParams.put(CONTAINER_ID, containerId);
cmd.setHostLevelParams(hostLevelParams); cmd.setHostLevelParams(hostLevelParams);
@ -2283,7 +2374,7 @@ public class AgentProviderService extends AbstractProviderService implements
configurations.get("global").put("exec_cmd", effectiveCommand.getExec()); configurations.get("global").put("exec_cmd", effectiveCommand.getExec());
cmd.setHostname(getClusterInfoPropertyValue(StatusKeys.INFO_AM_HOSTNAME)); cmd.setHostname(getClusterInfoPropertyValue(StatusKeys.INFO_AM_HOSTNAME));
cmd.addContainerDetails(roleGroup, getMetaInfo()); cmd.addContainerDetails(roleGroup, getMetaInfo(roleGroup));
Map<String, String> dockerConfig = new HashMap<String, String>(); Map<String, String> dockerConfig = new HashMap<String, String>();
if(isYarnDockerContainer(roleGroup)){ if(isYarnDockerContainer(roleGroup)){
@ -2366,8 +2457,8 @@ public class AgentProviderService extends AbstractProviderService implements
} }
} }
private String getPackageList() { private String getPackageList(String roleGroup) {
return getPackageListFromApplication(getMetaInfo().getApplication()); return getPackageListFromApplication(getMetaInfo(roleGroup).getApplication());
} }
private void prepareExecutionCommand(ExecutionCommand cmd) { private void prepareExecutionCommand(ExecutionCommand cmd) {
@ -2532,7 +2623,7 @@ public class AgentProviderService extends AbstractProviderService implements
private String getConfigFromMetaInfoWithAppConfigOverriding(String roleGroup, private String getConfigFromMetaInfoWithAppConfigOverriding(String roleGroup,
String configName){ String configName){
ConfTreeOperations appConf = getAmState().getAppConfSnapshot(); ConfTreeOperations appConf = getAmState().getAppConfSnapshot();
String containerName = getMetaInfo().getApplicationComponent(roleGroup) String containerName = getApplicationComponent(roleGroup)
.getDockerContainers().get(0).getName(); .getDockerContainers().get(0).getName();
String composedConfigName = null; String composedConfigName = null;
String appConfigValue = null; String appConfigValue = null;
@ -2673,7 +2764,7 @@ public class AgentProviderService extends AbstractProviderService implements
cmd.setConfigurations(configurations); cmd.setConfigurations(configurations);
// configurations.get("global").put("exec_cmd", startCommand.getExec()); // configurations.get("global").put("exec_cmd", startCommand.getExec());
cmd.addContainerDetails(roleGroup, getMetaInfo()); cmd.addContainerDetails(roleGroup, getMetaInfo(roleGroup));
log.info("Docker- command: {}", cmd.toString()); log.info("Docker- command: {}", cmd.toString());
@ -2683,7 +2774,7 @@ public class AgentProviderService extends AbstractProviderService implements
private String getConfigFromMetaInfo(String roleGroup, String configName) { private String getConfigFromMetaInfo(String roleGroup, String configName) {
String result = null; String result = null;
List<DockerContainer> containers = getMetaInfo().getApplicationComponent( List<DockerContainer> containers = getApplicationComponent(
roleGroup).getDockerContainers();// to support multi container per roleGroup).getDockerContainers();// to support multi container per
// component later // component later
log.debug("Docker- containers metainfo: {}", containers.toString()); log.debug("Docker- containers metainfo: {}", containers.toString());
@ -2985,10 +3076,11 @@ public class AgentProviderService extends AbstractProviderService implements
for (String configType : configs) { for (String configType : configs) {
addNamedConfiguration(configType, appConf.getGlobalOptions().options, addNamedConfiguration(configType, appConf.getGlobalOptions().options,
configurations, tokens, containerId, roleName); configurations, tokens, containerId, roleName,
roleGroup);
if (appConf.getComponent(roleGroup) != null) { if (appConf.getComponent(roleGroup) != null) {
addNamedConfiguration(configType, appConf.getComponent(roleGroup).options, addNamedConfiguration(configType, appConf.getComponent(roleGroup).options,
configurations, tokens, containerId, roleName); configurations, tokens, containerId, roleName, roleGroup);
} }
} }
@ -3058,15 +3150,32 @@ public class AgentProviderService extends AbstractProviderService implements
tokens.put("${NN_HOST}", URI.create(nnuri).getHost()); tokens.put("${NN_HOST}", URI.create(nnuri).getHost());
tokens.put("${ZK_HOST}", appConf.get(OptionKeys.ZOOKEEPER_HOSTS)); tokens.put("${ZK_HOST}", appConf.get(OptionKeys.ZOOKEEPER_HOSTS));
tokens.put("${DEFAULT_ZK_PATH}", appConf.get(OptionKeys.ZOOKEEPER_PATH)); tokens.put("${DEFAULT_ZK_PATH}", appConf.get(OptionKeys.ZOOKEEPER_PATH));
String prefix = appConf.getComponentOpt(componentGroup, ROLE_PREFIX,
null);
String dataDirSuffix = "";
if (prefix == null) {
prefix = "";
} else {
dataDirSuffix = "_" + SliderUtils.trimPrefix(prefix);
}
tokens.put("${DEFAULT_DATA_DIR}", getAmState() tokens.put("${DEFAULT_DATA_DIR}", getAmState()
.getInternalsSnapshot() .getInternalsSnapshot()
.getGlobalOptions() .getGlobalOptions()
.getMandatoryOption(InternalKeys.INTERNAL_DATA_DIR_PATH)); .getMandatoryOption(InternalKeys.INTERNAL_DATA_DIR_PATH) + dataDirSuffix);
tokens.put("${JAVA_HOME}", appConf.get(AgentKeys.JAVA_HOME)); tokens.put("${JAVA_HOME}", appConf.get(AgentKeys.JAVA_HOME));
tokens.put("${COMPONENT_NAME}", componentName); tokens.put("${COMPONENT_NAME}", componentName);
tokens.put("${COMPONENT_NAME.lc}", componentName.toLowerCase());
tokens.put("${COMPONENT_PREFIX}", prefix);
tokens.put("${COMPONENT_PREFIX.lc}", prefix.toLowerCase());
if (!componentName.equals(componentGroup) && componentName.startsWith(componentGroup)) { if (!componentName.equals(componentGroup) && componentName.startsWith(componentGroup)) {
tokens.put("${COMPONENT_ID}", componentName.substring(componentGroup.length())); tokens.put("${COMPONENT_ID}", componentName.substring(componentGroup.length()));
} }
tokens.put("${CLUSTER_NAME}", getClusterName());
tokens.put("${CLUSTER_NAME.lc}", getClusterName().toLowerCase());
tokens.put("${APP_NAME}", getClusterName());
tokens.put("${APP_NAME.lc}", getClusterName().toLowerCase());
tokens.put("${APP_COMPONENT_NAME}", componentName);
tokens.put("${APP_COMPONENT_NAME.lc}", componentName.toLowerCase());
return tokens; return tokens;
} }
@ -3091,12 +3200,12 @@ public class AgentProviderService extends AbstractProviderService implements
List<String> configList = new ArrayList<String>(); List<String> configList = new ArrayList<String>();
configList.add(GLOBAL_CONFIG_TAG); configList.add(GLOBAL_CONFIG_TAG);
List<ConfigFile> configFiles = getMetaInfo().getApplication().getConfigFiles(); List<ConfigFile> configFiles = getMetaInfo(roleGroup).getApplication().getConfigFiles();
for (ConfigFile configFile : configFiles) { for (ConfigFile configFile : configFiles) {
log.info("Expecting config type {}.", configFile.getDictionaryName()); log.info("Expecting config type {}.", configFile.getDictionaryName());
configList.add(configFile.getDictionaryName()); configList.add(configFile.getDictionaryName());
} }
for (Component component : getMetaInfo().getApplication().getComponents()) { for (Component component : getMetaInfo(roleGroup).getApplication().getComponents()) {
if (!component.getName().equals(roleGroup)) { if (!component.getName().equals(roleGroup)) {
continue; continue;
} }
@ -3121,7 +3230,7 @@ public class AgentProviderService extends AbstractProviderService implements
private void addNamedConfiguration(String configName, Map<String, String> sourceConfig, private void addNamedConfiguration(String configName, Map<String, String> sourceConfig,
Map<String, Map<String, String>> configurations, Map<String, Map<String, String>> configurations,
Map<String, String> tokens, String containerId, Map<String, String> tokens, String containerId,
String roleName) { String roleName, String roleGroup) {
Map<String, String> config = new HashMap<String, String>(); Map<String, String> config = new HashMap<String, String>();
if (configName.equals(GLOBAL_CONFIG_TAG)) { if (configName.equals(GLOBAL_CONFIG_TAG)) {
addDefaultGlobalConfig(config, containerId, roleName); addDefaultGlobalConfig(config, containerId, roleName);
@ -3150,9 +3259,9 @@ public class AgentProviderService extends AbstractProviderService implements
} }
//apply defaults only if the key is not present and value is not empty //apply defaults only if the key is not present and value is not empty
if (getDefaultConfigs().containsKey(configName)) { if (getDefaultConfigs(roleGroup).containsKey(configName)) {
log.info("Adding default configs for type {}.", configName); log.info("Adding default configs for type {}.", configName);
for (PropertyInfo defaultConfigProp : getDefaultConfigs().get(configName).getPropertyInfos()) { for (PropertyInfo defaultConfigProp : getDefaultConfigs(roleGroup).get(configName).getPropertyInfos()) {
if (!config.containsKey(defaultConfigProp.getName())) { if (!config.containsKey(defaultConfigProp.getName())) {
if (!defaultConfigProp.getName().isEmpty() && if (!defaultConfigProp.getName().isEmpty() &&
defaultConfigProp.getValue() != null && defaultConfigProp.getValue() != null &&

View File

@ -21,6 +21,7 @@ import org.apache.hadoop.fs.FileSystem;
import org.apache.hadoop.fs.Path; import org.apache.hadoop.fs.Path;
import org.apache.slider.common.tools.SliderFileSystem; import org.apache.slider.common.tools.SliderFileSystem;
import org.apache.slider.common.tools.SliderUtils; import org.apache.slider.common.tools.SliderUtils;
import org.apache.slider.core.conf.ConfTreeOperations;
import org.apache.slider.core.exceptions.BadConfigException; import org.apache.slider.core.exceptions.BadConfigException;
import org.apache.slider.providers.agent.application.metadata.AbstractMetainfoParser; import org.apache.slider.providers.agent.application.metadata.AbstractMetainfoParser;
import org.apache.slider.providers.agent.application.metadata.AddonPackageMetainfoParser; import org.apache.slider.providers.agent.application.metadata.AddonPackageMetainfoParser;
@ -35,6 +36,8 @@ import java.io.FileNotFoundException;
import java.io.IOException; import java.io.IOException;
import java.io.InputStream; import java.io.InputStream;
import static org.apache.slider.api.RoleKeys.ROLE_PREFIX;
/** /**
* *
*/ */
@ -131,4 +134,17 @@ public class AgentUtils {
return new DefaultConfigParser().parse(configStream); return new DefaultConfigParser().parse(configStream);
} }
static String getMetainfoComponentName(String roleGroup,
ConfTreeOperations appConf) throws BadConfigException {
String prefix = appConf.getComponentOpt(roleGroup, ROLE_PREFIX, null);
if (prefix == null) {
return roleGroup;
}
if (!roleGroup.startsWith(prefix)) {
throw new BadConfigException("Component " + roleGroup + " doesn't start" +
" with prefix " + prefix);
}
return roleGroup.substring(prefix.length());
}
} }

View File

@ -18,6 +18,8 @@
package org.apache.slider.providers.agent; package org.apache.slider.providers.agent;
import org.apache.slider.common.tools.SliderUtils;
import org.apache.slider.core.conf.ConfTreeOperations;
import org.apache.slider.providers.agent.application.metadata.CommandOrder; import org.apache.slider.providers.agent.application.metadata.CommandOrder;
import org.slf4j.Logger; import org.slf4j.Logger;
import org.slf4j.LoggerFactory; import org.slf4j.LoggerFactory;
@ -25,9 +27,12 @@ import org.slf4j.LoggerFactory;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Collection; import java.util.Collection;
import java.util.HashMap; import java.util.HashMap;
import java.util.HashSet;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import static org.apache.slider.api.RoleKeys.ROLE_PREFIX;
/** /**
* Stores the command dependency order for all components in a service. <commandOrder> * Stores the command dependency order for all components in a service. <commandOrder>
* <command>SUPERVISOR-START</command> <requires>NIMBUS-STARTED</requires> </commandOrder> Means, SUPERVISOR START * <command>SUPERVISOR-START</command> <requires>NIMBUS-STARTED</requires> </commandOrder> Means, SUPERVISOR START
@ -39,13 +44,36 @@ public class ComponentCommandOrder {
private static char SPLIT_CHAR = '-'; private static char SPLIT_CHAR = '-';
Map<Command, Map<String, List<ComponentState>>> dependencies = Map<Command, Map<String, List<ComponentState>>> dependencies =
new HashMap<Command, Map<String, List<ComponentState>>>(); new HashMap<Command, Map<String, List<ComponentState>>>();
Map<String, Collection<String>> prefixRoleMap = new HashMap<>();
Map<String, String> rolePrefixMap = new HashMap<>();
public ComponentCommandOrder(List<CommandOrder> commandOrders) { public ComponentCommandOrder() {}
public ComponentCommandOrder(List<CommandOrder> commandOrders,
ConfTreeOperations resources) {
mergeCommandOrders(commandOrders, resources);
}
void mergeCommandOrders(List<CommandOrder> commandOrders,
ConfTreeOperations resources) {
for (String component : resources.getComponentNames()) {
String prefix = SliderUtils.trimPrefix(
resources.getComponentOpt(component, ROLE_PREFIX, null));
if (prefix != null) {
rolePrefixMap.put(component, prefix);
if (!prefixRoleMap.containsKey(prefix)) {
prefixRoleMap.put(prefix, new HashSet<String>());
}
prefixRoleMap.get(prefix).add(component);
}
}
if (commandOrders != null && commandOrders.size() > 0) { if (commandOrders != null && commandOrders.size() > 0) {
for (CommandOrder commandOrder : commandOrders) { for (CommandOrder commandOrder : commandOrders) {
ComponentCommand componentCmd = getComponentCommand(commandOrder.getCommand()); ComponentCommand componentCmd = getComponentCommand(
commandOrder.getCommand(), resources);
String requires = commandOrder.getRequires(); String requires = commandOrder.getRequires();
List<ComponentState> requiredStates = parseRequiredStates(requires); List<ComponentState> requiredStates = parseRequiredStates(requires,
resources);
if (requiredStates.size() > 0) { if (requiredStates.size() > 0) {
Map<String, List<ComponentState>> compDep = dependencies.get(componentCmd.command); Map<String, List<ComponentState>> compDep = dependencies.get(componentCmd.command);
if (compDep == null) { if (compDep == null) {
@ -65,7 +93,8 @@ public class ComponentCommandOrder {
} }
} }
private List<ComponentState> parseRequiredStates(String requires) { private List<ComponentState> parseRequiredStates(String requires,
ConfTreeOperations resources) {
if (requires == null || requires.length() < 2) { if (requires == null || requires.length() < 2) {
throw new IllegalArgumentException("Input cannot be null and must contain component and state."); throw new IllegalArgumentException("Input cannot be null and must contain component and state.");
} }
@ -73,13 +102,14 @@ public class ComponentCommandOrder {
String[] componentStates = requires.split(","); String[] componentStates = requires.split(",");
List<ComponentState> retList = new ArrayList<ComponentState>(); List<ComponentState> retList = new ArrayList<ComponentState>();
for (String componentStateStr : componentStates) { for (String componentStateStr : componentStates) {
retList.add(getComponentState(componentStateStr)); retList.add(getComponentState(componentStateStr, resources));
} }
return retList; return retList;
} }
private ComponentCommand getComponentCommand(String compCmdStr) { private ComponentCommand getComponentCommand(String compCmdStr,
ConfTreeOperations resources) {
if (compCmdStr == null || compCmdStr.trim().length() < 2) { if (compCmdStr == null || compCmdStr.trim().length() < 2) {
throw new IllegalArgumentException("Input cannot be null and must contain component and command."); throw new IllegalArgumentException("Input cannot be null and must contain component and command.");
} }
@ -92,6 +122,11 @@ public class ComponentCommandOrder {
String compStr = compCmdStr.substring(0, splitIndex); String compStr = compCmdStr.substring(0, splitIndex);
String cmdStr = compCmdStr.substring(splitIndex + 1); String cmdStr = compCmdStr.substring(splitIndex + 1);
if (resources.getComponent(compStr) == null && !prefixRoleMap.containsKey(compStr)) {
throw new IllegalArgumentException("Component " + compStr + " specified" +
" in command order does not exist");
}
Command cmd = Command.valueOf(cmdStr); Command cmd = Command.valueOf(cmdStr);
if (cmd != Command.START) { if (cmd != Command.START) {
@ -100,7 +135,8 @@ public class ComponentCommandOrder {
return new ComponentCommand(compStr, cmd); return new ComponentCommand(compStr, cmd);
} }
private ComponentState getComponentState(String compStStr) { private ComponentState getComponentState(String compStStr,
ConfTreeOperations resources) {
if (compStStr == null || compStStr.trim().length() < 2) { if (compStStr == null || compStStr.trim().length() < 2) {
throw new IllegalArgumentException("Input cannot be null."); throw new IllegalArgumentException("Input cannot be null.");
} }
@ -113,6 +149,11 @@ public class ComponentCommandOrder {
String compStr = compStStr.substring(0, splitIndex); String compStr = compStStr.substring(0, splitIndex);
String stateStr = compStStr.substring(splitIndex + 1); String stateStr = compStStr.substring(splitIndex + 1);
if (resources.getComponent(compStr) == null && !prefixRoleMap.containsKey(compStr)) {
throw new IllegalArgumentException("Component " + compStr + " specified" +
" in command order does not exist");
}
State state = State.valueOf(stateStr); State state = State.valueOf(stateStr);
if (state != State.STARTED && state != State.INSTALLED) { if (state != State.STARTED && state != State.INSTALLED) {
throw new IllegalArgumentException("Dependency order can only be specified against STARTED/INSTALLED."); throw new IllegalArgumentException("Dependency order can only be specified against STARTED/INSTALLED.");
@ -123,40 +164,43 @@ public class ComponentCommandOrder {
// dependency is still on component level, but not package level // dependency is still on component level, but not package level
// so use component name to check dependency, not component-package // so use component name to check dependency, not component-package
public boolean canExecute(String component, Command command, Collection<ComponentInstanceState> currentStates) { public boolean canExecute(String component, Command command, Collection<ComponentInstanceState> currentStates) {
boolean canExecute = true; if (!dependencies.containsKey(command)) {
if (dependencies.containsKey(command) && dependencies.get(command).containsKey(component)) { return true;
List<ComponentState> required = dependencies.get(command).get(component); }
for (ComponentState stateToMatch : required) { List<ComponentState> required = new ArrayList<>();
for (ComponentInstanceState currState : currentStates) { if (dependencies.get(command).containsKey(component)) {
log.debug("Checking schedule {} {} against dependency {} is {}", required.addAll(dependencies.get(command).get(component));
component, command, currState.getComponentName(), currState.getState()); }
if (currState.getComponentName().equals(stateToMatch.componentName)) { String prefix = rolePrefixMap.get(component);
if (currState.getState() != stateToMatch.state) { if (prefix != null && dependencies.get(command).containsKey(prefix)) {
if (stateToMatch.state == State.STARTED) { required.addAll(dependencies.get(command).get(prefix));
}
for (ComponentState stateToMatch : required) {
for (ComponentInstanceState currState : currentStates) {
log.debug("Checking schedule {} {} against dependency {} is {}",
component, command, currState.getComponentName(), currState.getState());
if (currState.getComponentName().equals(stateToMatch.componentName) ||
(prefixRoleMap.containsKey(stateToMatch.componentName) &&
prefixRoleMap.get(stateToMatch.componentName).contains(currState.getComponentName()))) {
if (currState.getState() != stateToMatch.state) {
if (stateToMatch.state == State.STARTED) {
log.info("Cannot schedule {} {} as dependency {} is {}",
component, command, currState.getComponentName(), currState.getState());
return false;
} else {
//state is INSTALLED
if (currState.getState() != State.STARTING && currState.getState() != State.STARTED) {
log.info("Cannot schedule {} {} as dependency {} is {}", log.info("Cannot schedule {} {} as dependency {} is {}",
component, command, currState.getComponentName(), currState.getState()); component, command, currState.getComponentName(), currState.getState());
canExecute = false; return false;
} else {
//state is INSTALLED
if (currState.getState() != State.STARTING && currState.getState() != State.STARTED) {
log.info("Cannot schedule {} {} as dependency {} is {}",
component, command, currState.getComponentName(), currState.getState());
canExecute = false;
}
} }
} }
} }
if (!canExecute) {
break;
}
}
if (!canExecute) {
break;
} }
} }
} }
return true;
return canExecute;
} }
static class ComponentState { static class ComponentState {

View File

@ -20,6 +20,9 @@ package org.apache.slider.server.appmaster;
import com.codahale.metrics.MetricRegistry; import com.codahale.metrics.MetricRegistry;
import com.codahale.metrics.health.HealthCheckRegistry; import com.codahale.metrics.health.HealthCheckRegistry;
import com.codahale.metrics.jvm.GarbageCollectorMetricSet;
import com.codahale.metrics.jvm.MemoryUsageGaugeSet;
import com.codahale.metrics.jvm.ThreadStatesGaugeSet;
import com.google.common.base.Preconditions; import com.google.common.base.Preconditions;
import com.google.protobuf.BlockingService; import com.google.protobuf.BlockingService;
@ -879,7 +882,8 @@ public class SliderAppMaster extends AbstractSliderLaunchedService
Configuration providerConf = Configuration providerConf =
providerService.loadProviderConfigurationInformation(confDir); providerService.loadProviderConfigurationInformation(confDir);
providerService.initializeApplicationConfiguration(instanceDefinition, fs); providerService.initializeApplicationConfiguration(instanceDefinition,
fs, null);
providerService.validateApplicationConfiguration(instanceDefinition, providerService.validateApplicationConfiguration(instanceDefinition,
confDir, confDir,

View File

@ -1818,6 +1818,11 @@ public class AppState {
SliderUtils.mergeMapsIgnoreDuplicateKeys(cd.getRole(rolename), SliderUtils.mergeMapsIgnoreDuplicateKeys(cd.getRole(rolename),
groupOptions.options); groupOptions.options);
} }
String prefix = instanceDefinition.getAppConfOperations()
.getComponentOpt(role.getGroup(), ROLE_PREFIX, null);
if (SliderUtils.isSet(prefix)) {
cd.setRoleOpt(rolename, ROLE_PREFIX, SliderUtils.trimPrefix(prefix));
}
List<String> instances = instanceMap.get(rolename); List<String> instances = instanceMap.get(rolename);
int nodeCount = instances != null ? instances.size(): 0; int nodeCount = instances != null ? instances.size(): 0;
cd.setRoleOpt(rolename, COMPONENT_INSTANCES, cd.setRoleOpt(rolename, COMPONENT_INSTANCES,

View File

@ -73,7 +73,7 @@ public class AgentResource extends AbstractSliderResource {
} }
@POST @POST
@Path("/{agent_name: [a-zA-Z][a-zA-Z_0-9]*}/register") @Path("/{agent_name: [a-zA-Z][a-zA-Z0-9_-]*}/register")
@Consumes({MediaType.APPLICATION_JSON}) @Consumes({MediaType.APPLICATION_JSON})
@Produces({MediaType.APPLICATION_JSON}) @Produces({MediaType.APPLICATION_JSON})
public RegistrationResponse register(Register registration, public RegistrationResponse register(Register registration,
@ -87,7 +87,7 @@ public class AgentResource extends AbstractSliderResource {
} }
@POST @POST
@Path("/{agent_name: [a-zA-Z][a-zA-Z_0-9]*}/heartbeat") @Path("/{agent_name: [a-zA-Z][a-zA-Z0-9_-]*}/heartbeat")
@Consumes(MediaType.APPLICATION_JSON) @Consumes(MediaType.APPLICATION_JSON)
@Produces({MediaType.APPLICATION_JSON}) @Produces({MediaType.APPLICATION_JSON})
public HeartBeatResponse heartbeat(HeartBeat message, public HeartBeatResponse heartbeat(HeartBeat message,