if(log.isDebugEnabled()) {
log.debug("Intercept launch context: " + context);
}
StepExecution stepExecution = containerToStepMap.get(container.getId());
String jobName = remoteStepNames.get(stepExecution);
if(service != null) {
int port = service.getPort();
String address = service.getHost();
Map<String, String> env = new HashMap<String, String>(context.getEnvironment());
env.put(YarnSystemConstants.FS_ADDRESS, getConfiguration().get(CommonConfigurationKeysPublic.FS_DEFAULT_NAME_KEY));
env.put(YarnSystemConstants.AMSERVICE_PORT, Integer.toString(port));
env.put(YarnSystemConstants.AMSERVICE_HOST, address);
env.put(YarnSystemConstants.AMSERVICE_BATCH_STEPNAME, jobName);
env.put(YarnSystemConstants.AMSERVICE_BATCH_STEPNAME, jobName);
env.put(YarnSystemConstants.AMSERVICE_BATCH_STEPEXECUTIONNAME, stepExecution.getStepName());
env.put(YarnSystemConstants.AMSERVICE_BATCH_JOBEXECUTIONID, Long.toString(stepExecution.getJobExecutionId()));
env.put(YarnSystemConstants.AMSERVICE_BATCH_STEPEXECUTIONID, Long.toString(stepExecution.getId()));
context.setEnvironment(env);
return context;
} else {
return context;
}