diff --git a/hadoop-common-project/hadoop-common/CHANGES.txt b/hadoop-common-project/hadoop-common/CHANGES.txt index afe7ec68a1..3d7291b644 100644 --- a/hadoop-common-project/hadoop-common/CHANGES.txt +++ b/hadoop-common-project/hadoop-common/CHANGES.txt @@ -148,6 +148,9 @@ Trunk (Unreleased) HADOOP-11460. Deprecate shell vars (John Smith via aw) + HADOOP-11346. Rewrite sls/rumen to use new shell framework (John Smith + via aw) + BUG FIXES HADOOP-11473. test-patch says "-1 overall" even when all checks are +1 diff --git a/hadoop-tools/hadoop-sls/src/main/bin/rumen2sls.sh b/hadoop-tools/hadoop-sls/src/main/bin/rumen2sls.sh index 080a715375..c8e38ac633 100644 --- a/hadoop-tools/hadoop-sls/src/main/bin/rumen2sls.sh +++ b/hadoop-tools/hadoop-sls/src/main/bin/rumen2sls.sh @@ -13,94 +13,94 @@ # limitations under the License. See accompanying LICENSE file. # -############################################################################### -printUsage() { +function hadoop_usage() +{ echo "Usage: rumen2sls.sh " echo " --rumen-file=" echo " --output-dir=" echo " [--output-prefix=] (default is sls)" echo } -############################################################################### -parseArgs() { - for i in $* - do + +function parse_args() +{ + for i in "$@"; do case $i in - --rumen-file=*) - rumenfile=${i#*=} + --rumen-file=*) + rumenfile=${i#*=} ;; - --output-dir=*) - outputdir=${i#*=} + --output-dir=*) + outputdir=${i#*=} ;; - --output-prefix=*) - outputprefix=${i#*=} + --output-prefix=*) + outputprefix=${i#*=} ;; - *) - echo "Invalid option" - echo - printUsage - exit 1 + *) + hadoop_error "ERROR: Invalid option ${i}" + hadoop_exit_with_usage 1 ;; esac done - if [[ "${rumenfile}" == "" || "${outputdir}" == "" ]] ; then - echo "Both --rumen-file ${rumenfile} and --output-dir \ - ${outputfdir} must be specified" - echo - printUsage - exit 1 + + if [[ -z "${rumenfile}" ]] ; then + hadoop_error "ERROR: --rumen-file must be specified." + hadoop_exit_with_usage 1 + fi + + if [[ -z "${outputdir}" ]] ; then + hadoop_error "ERROR: --output-dir must be specified." + hadoop_exit_with_usage 1 fi } -############################################################################### -calculateBasedir() { - # resolve links - $0 may be a softlink - PRG="${1}" - while [ -h "${PRG}" ]; do - ls=`ls -ld "${PRG}"` - link=`expr "$ls" : '.*-> \(.*\)$'` - if expr "$link" : '/.*' > /dev/null; then - PRG="$link" - else - PRG=`dirname "${PRG}"`/"$link" - fi - done +function calculate_classpath() +{ + hadoop_debug "Injecting TOOL_PATH into CLASSPATH" + hadoop_add_classpath "${TOOL_PATH}" +} - BASEDIR=`dirname ${PRG}` - BASEDIR=`cd ${BASEDIR}/..;pwd` -} -############################################################################### -calculateClasspath() { - HADOOP_BASE=`which hadoop` - HADOOP_BASE=`dirname $HADOOP_BASE` - DEFAULT_LIBEXEC_DIR=${HADOOP_BASE}/../libexec - HADOOP_LIBEXEC_DIR=${HADOOP_LIBEXEC_DIR:-$DEFAULT_LIBEXEC_DIR} - . $HADOOP_LIBEXEC_DIR/hadoop-config.sh - export HADOOP_CLASSPATH="${HADOOP_CLASSPATH}:${TOOL_PATH}" -} -############################################################################### -runSLSGenerator() { - if [[ "${outputprefix}" == "" ]] ; then +function run_sls_generator() +{ + if [[ -z "${outputprefix}" ]] ; then outputprefix="sls" fi - slsJobs=${outputdir}/${outputprefix}-jobs.json - slsNodes=${outputdir}/${outputprefix}-nodes.json + hadoop_add_param args -input "-input ${rumenfile}" + hadoop_add_param args -outputJobs "-outputJobs ${outputdir}/${outputprefix}-jobs.json" + hadoop_add_param args -outputNodes "-outputNodes ${outputdir}/${outputprefix}-nodes.json" - args="-input ${rumenfile} -outputJobs ${slsJobs}"; - args="${args} -outputNodes ${slsNodes}"; + hadoop_debug "Appending HADOOP_CLIENT_OPTS onto HADOOP_OPTS" + HADOOP_OPTS="${HADOOP_OPTS} ${HADOOP_CLIENT_OPTS}" - hadoop org.apache.hadoop.yarn.sls.RumenToSLSConverter ${args} + hadoop_finalize + # shellcheck disable=SC2086 + hadoop_java_exec rumen2sls org.apache.hadoop.yarn.sls.RumenToSLSConverter ${args} } -############################################################################### -calculateBasedir $0 -calculateClasspath -parseArgs "$@" -runSLSGenerator +# let's locate libexec... +if [[ -n "${HADOOP_PREFIX}" ]]; then + DEFAULT_LIBEXEC_DIR="${HADOOP_PREFIX}/libexec" +else + this="${BASH_SOURCE-$0}" + bin=$(cd -P -- "$(dirname -- "${this}")" >/dev/null && pwd -P) + DEFAULT_LIBEXEC_DIR="${bin}/../../../../../libexec" +fi -echo -echo "SLS simulation files available at: ${outputdir}" -echo +HADOOP_LIBEXEC_DIR="${HADOOP_LIBEXEC_DIR:-$DEFAULT_LIBEXEC_DIR}" +# shellcheck disable=SC2034 +HADOOP_NEW_CONFIG=true +if [[ -f "${HADOOP_LIBEXEC_DIR}/hadoop-config.sh" ]]; then + . "${HADOOP_LIBEXEC_DIR}/hadoop-config.sh" +else + echo "ERROR: Cannot execute ${HADOOP_LIBEXEC_DIR}/hadoop-config.sh." 2>&1 + exit 1 +fi + +if [ $# = 0 ]; then + hadoop_exit_with_usage 1 +fi + +parse_args "${@}" +calculate_classpath +run_sls_generator -exit 0 \ No newline at end of file diff --git a/hadoop-tools/hadoop-sls/src/main/bin/slsrun.sh b/hadoop-tools/hadoop-sls/src/main/bin/slsrun.sh index 1e402df296..463ae23575 100644 --- a/hadoop-tools/hadoop-sls/src/main/bin/slsrun.sh +++ b/hadoop-tools/hadoop-sls/src/main/bin/slsrun.sh @@ -13,100 +13,120 @@ # limitations under the License. See accompanying LICENSE file. # -############################################################################### -printUsage() { - echo "Usage: slsrun.sh " - echo " --input-rumen|--input-sls=" +function hadoop_usage() +{ + echo "Usage: slsrun.sh " + echo " --input-rumen= | --input-sls=" echo " --output-dir=" echo " [--nodes=]" echo " [--track-jobs=]" echo " [--print-simulation]" - echo } -############################################################################### -parseArgs() { - for i in $* - do + +function parse_args() +{ + for i in "$@"; do case $i in - --input-rumen=*) - inputrumen=${i#*=} + --input-rumen=*) + inputrumen=${i#*=} ;; - --input-sls=*) - inputsls=${i#*=} + --input-sls=*) + inputsls=${i#*=} ;; - --output-dir=*) - outputdir=${i#*=} + --output-dir=*) + outputdir=${i#*=} ;; - --nodes=*) - nodes=${i#*=} + --nodes=*) + nodes=${i#*=} ;; - --track-jobs=*) - trackjobs=${i#*=} + --track-jobs=*) + trackjobs=${i#*=} ;; - --print-simulation) - printsimulation="true" + --print-simulation) + printsimulation="true" ;; - *) - echo "Invalid option" - echo - printUsage - exit 1 + *) + hadoop_error "ERROR: Invalid option ${i}" + hadoop_exit_with_usage 1 ;; esac done - if [[ "${inputrumen}" == "" && "${inputsls}" == "" ]] ; then - echo "Either --input-rumen or --input-sls must be specified" - echo - printUsage - exit 1 + if [[ -z "${inputrumen}" && -z "${inputsls}" ]] ; then + hadoop_error "ERROR: Either --input-rumen or --input-sls must be specified." + hadoop_exit_with_usage 1 fi - if [[ "${outputdir}" == "" ]] ; then - echo "The output directory --output-dir must be specified" - echo - printUsage - exit 1 + if [[ -n "${inputrumen}" && -n "${inputsls}" ]] ; then + hadoop_error "ERROR: Only specify one of --input-rumen or --input-sls." + hadoop_exit_with_usage 1 + fi + + if [[ -z "${outputdir}" ]] ; then + hadoop_error "ERROR: The output directory --output-dir must be specified." + hadoop_exit_with_usage 1 fi } -############################################################################### -calculateClasspath() { - HADOOP_BASE=`which hadoop` - HADOOP_BASE=`dirname $HADOOP_BASE` - DEFAULT_LIBEXEC_DIR=${HADOOP_BASE}/../libexec - HADOOP_LIBEXEC_DIR=${HADOOP_LIBEXEC_DIR:-$DEFAULT_LIBEXEC_DIR} - . $HADOOP_LIBEXEC_DIR/hadoop-config.sh - export HADOOP_CLASSPATH="${HADOOP_CLASSPATH}:${TOOL_PATH}:html" +function calculate_classpath() { + hadoop_debug "Injecting TOOL_PATH into CLASSPATH" + hadoop_add_classpath "${TOOL_PATH}" + hadoop_debug "Injecting ${HADOOP_PREFIX}/share/hadoop/tools/sls/html into CLASSPATH" + hadoop_add_classpath "${HADOOP_PREFIX}/share/hadoop/tools/sls/html" } -############################################################################### -runSimulation() { + +function run_simulation() { if [[ "${inputsls}" == "" ]] ; then - args="-inputrumen ${inputrumen}" + hadoop_add_param args -inputrumen "-inputrumen ${inputrumen}" else - args="-inputsls ${inputsls}" + hadoop_add_param args -inputsls "-inputsls ${inputsls}" fi - args="${args} -output ${outputdir}" + hadoop_add_param args -output "-output ${outputdir}" - if [[ "${nodes}" != "" ]] ; then - args="${args} -nodes ${nodes}" + if [[ -n "${nodes}" ]] ; then + hadoop_add_param args -nodes "-nodes ${nodes}" fi - - if [[ "${trackjobs}" != "" ]] ; then - args="${args} -trackjobs ${trackjobs}" + + if [[ -n "${trackjobs}" ]] ; then + hadoop_add_param args -trackjobs "-trackjobs ${trackjobs}" fi - + if [[ "${printsimulation}" == "true" ]] ; then - args="${args} -printsimulation" + hadoop_add_param args -printsimulation "-printsimulation" fi - hadoop org.apache.hadoop.yarn.sls.SLSRunner ${args} + hadoop_debug "Appending HADOOP_CLIENT_OPTS onto HADOOP_OPTS" + HADOOP_OPTS="${HADOOP_OPTS} ${HADOOP_CLIENT_OPTS}" + + hadoop_finalize + # shellcheck disable=SC2086 + hadoop_java_exec sls org.apache.hadoop.yarn.sls.SLSRunner ${args} } -############################################################################### -calculateClasspath -parseArgs "$@" -runSimulation +# let's locate libexec... +if [[ -n "${HADOOP_PREFIX}" ]]; then + DEFAULT_LIBEXEC_DIR="${HADOOP_PREFIX}/libexec" +else + this="${BASH_SOURCE-$0}" + bin=$(cd -P -- "$(dirname -- "${this}")" >/dev/null && pwd -P) + DEFAULT_LIBEXEC_DIR="${bin}/../../../../../libexec" +fi -exit 0 +HADOOP_LIBEXEC_DIR="${HADOOP_LIBEXEC_DIR:-$DEFAULT_LIBEXEC_DIR}" +# shellcheck disable=SC2034 +HADOOP_NEW_CONFIG=true +if [[ -f "${HADOOP_LIBEXEC_DIR}/hadoop-config.sh" ]]; then + . "${HADOOP_LIBEXEC_DIR}/hadoop-config.sh" +else + echo "ERROR: Cannot execute ${HADOOP_LIBEXEC_DIR}/hadoop-config.sh." 2>&1 + exit 1 +fi + +if [[ $# = 0 ]]; then + hadoop_exit_with_usage 1 +fi + +parse_args "${@}" +calculate_classpath +run_simulation