Apache Hadoop YARN: Moving beyond MapReduce and Batch Processing with Apache Hadoop 2 (2014)

B. YARN Installation Scripts

The following is a listing of the installation scripts discussed in Chapter 5, “Installing Apache Hadoop YARN.” They can be used to help follow the installation discussion. All of the scripts are available from the download page listed in Appendix A.

install-hadoop2.sh

#!/bin/bash
#
# Install Hadoop 2 using pdsh/pdcp where possible.
#
# Command can be interactive or file-based. This script sets up
# a Hadoop 2 cluster with basic configuration. Modify data, log, and pid
# directories as desired. Further configure your cluster with ./conf-hadoop2.sh
# after running this installation script.
#

# Basic environment variables. Edit as necessary
HADOOP_VERSION=2.2.0
HADOOP_HOME="/opt/hadoop-${HADOOP_VERSION}"
NN_DATA_DIR=/var/data/hadoop/hdfs/nn
SNN_DATA_DIR=/var/data/hadoop/hdfs/snn
DN_DATA_DIR=/var/data/hadoop/hdfs/dn
YARN_LOG_DIR=/var/log/hadoop/yarn
HADOOP_LOG_DIR=/var/log/hadoop/hdfs
HADOOP_MAPRED_LOG_DIR=/var/log/hadoop/mapred
YARN_PID_DIR=/var/run/hadoop/yarn
HADOOP_PID_DIR=/var/run/hadoop/hdfs
HADOOP_MAPRED_PID_DIR=/var/run/hadoop/mapred
HTTP_STATIC_USER=hdfs
YARN_PROXY_PORT=8081

source hadoop-xml-conf.sh
CMD_OPTIONS=$(getopt -n "$0" -o hif --long "help,interactive,file" -- "$@")

# Take care of bad options in the command
if [ $? -ne 0 ];
then
exit 1
fi
eval set -- "$CMD_OPTIONS"

all_hosts="all_hosts"
nn_host="nn_host"
snn_host="snn_host"
dn_hosts="dn_hosts"
rm_host="rm_host"
nm_hosts="nm_hosts"
mr_history_host="mr_history_host"
yarn_proxy_host="yarn_proxy_host"

install()
{
echo "Copying Hadoop $HADOOP_VERSION to all hosts..."
pdcp -w ^all_hosts hadoop-"$HADOOP_VERSION".tar.gz /opt

echo "Copying JDK 1.6.0_31 to all hosts..."
pdcp -w ^all_hosts jdk-6u31-linux-x64-rpm.bin /opt

echo "Installing JDK 1.6.0_31 on all hosts..."
pdsh -w ^all_hosts chmod a+x /opt/jdk-6u31-linux-x64-rpm.bin
pdsh -w ^all_hosts /opt/jdk-6u31-linux-x64-rpm.bin -noregister 1>&- 2>&-

echo "Setting JAVA_HOME and HADOOP_HOME environment variables on all hosts..."
pdsh -w ^all_hosts 'echo export JAVA_HOME=/usr/java/jdk1.6.0_31 > /etc/profile.d/java.sh'
pdsh -w ^all_hosts "source /etc/profile.d/java.sh"
pdsh -w ^all_hosts "echo export HADOOP_HOME=$HADOOP_HOME > /etc/profile.d/hadoop.sh"
pdsh -w ^all_hosts 'echo export HADOOP_PREFIX=$HADOOP_HOME >> /etc/profile.d/hadoop.sh'
pdsh -w ^all_hosts "source /etc/profile.d/hadoop.sh"

echo "Extracting Hadoop $HADOOP_VERSION distribution on all hosts..."
pdsh -w ^all_hosts tar -zxf /opt/hadoop-"$HADOOP_VERSION".tar.gz -C /opt

echo "Creating system accounts and groups on all hosts..."
pdsh -w ^all_hosts groupadd hadoop
pdsh -w ^all_hosts useradd -g hadoop yarn
pdsh -w ^all_hosts useradd -g hadoop hdfs
pdsh -w ^all_hosts useradd -g hadoop mapred

echo "Creating HDFS data directories on NameNode host, Secondary NameNode host, and DataNode hosts..."
pdsh -w ^nn_host "mkdir -p $NN_DATA_DIR && chown hdfs:hadoop $NN_DATA_DIR"
pdsh -w ^snn_host "mkdir -p $SNN_DATA_DIR && chown hdfs:hadoop $SNN_DATA_DIR"
pdsh -w ^dn_hosts "mkdir -p $DN_DATA_DIR && chown hdfs:hadoop $DN_DATA_DIR"

echo "Creating log directories on all hosts..."
pdsh -w ^all_hosts "mkdir -p $YARN_LOG_DIR && chown yarn:hadoop $YARN_LOG_DIR"
pdsh -w ^all_hosts "mkdir -p $HADOOP_LOG_DIR && chown hdfs:hadoop $HADOOP_LOG_DIR"
pdsh -w ^all_hosts "mkdir -p $HADOOP_MAPRED_LOG_DIR && chown mapred:hadoop $HADOOP_MAPRED_LOG_DIR"

echo "Creating pid directories on all hosts..."
pdsh -w ^all_hosts "mkdir -p $YARN_PID_DIR && chown yarn:hadoop $YARN_PID_DIR"
pdsh -w ^all_hosts "mkdir -p $HADOOP_PID_DIR && chown hdfs:hadoop $HADOOP_PID_DIR"
pdsh -w ^all_hosts "mkdir -p $HADOOP_MAPRED_PID_DIR && chown mapred:hadoop $HADOOP_MAPRED_PID_DIR"

echo "Editing Hadoop environment scripts for log directories on all hosts..."
pdsh -w ^all_hosts echo "export HADOOP_LOG_DIR=$HADOOP_LOG_DIR >> $HADOOP_HOME/etc/hadoop/hadoop-env.sh"
pdsh -w ^all_hosts echo "export YARN_LOG_DIR=$YARN_LOG_DIR >> $HADOOP_HOME/etc/hadoop/yarn-env.sh"
pdsh -w ^all_hosts echo "export HADOOP_MAPRED_LOG_DIR=$HADOOP_MAPRED_LOG_DIR >> $HADOOP_HOME/etc/hadoop/mapred-env.sh"

echo "Editing Hadoop environment scripts for pid directories on all hosts..."
pdsh -w ^all_hosts echo "export HADOOP_PID_DIR=$HADOOP_PID_DIR >> $HADOOP_HOME/etc/hadoop/hadoop-env.sh"
pdsh -w ^all_hosts echo "export YARN_PID_DIR=$YARN_PID_DIR >> $HADOOP_HOME/etc/hadoop/yarn-env.sh"
pdsh -w ^all_hosts echo "export HADOOP_MAPRED_PID_DIR=$HADOOP_MAPRED_PID_DIR >> $HADOOP_HOME/etc/hadoop/mapred-env.sh"

echo "Creating base Hadoop XML config files..."
create_config --file core-site.xml
put_config --file core-site.xml --property fs.default.name --value "hdfs://$nn:9000"
put_config --file core-site.xml --property hadoop.http.staticuser.user --value "$HTTP_STATIC_USER"

create_config --file hdfs-site.xml
put_config --file hdfs-site.xml --property dfs.namenode.name.dir --value "$NN_DATA_DIR"
put_config --file hdfs-site.xml --property fs.checkpoint.dir --value "$SNN_DATA_DIR"
put_config --file hdfs-site.xml --property fs.checkpoint.edits.dir --value "$SNN_DATA_DIR"
put_config --file hdfs-site.xml --property dfs.datanode.data.dir --value "$DN_DATA_DIR"
put_config --file hdfs-site.xml --property dfs.namenode.http-address --value "$nn:50070"
put_config --file hdfs-site.xml --property dfs.namenode.secondary.http-address --value "$snn:50090"

create_config --file mapred-site.xml
put_config --file mapred-site.xml --property mapreduce.framework.name --value yarn
put_config --file mapred-site.xml --property mapreduce.jobhistory.address --value "$mr_hist:10020"
put_config --file mapred-site.xml --property mapreduce.jobhistory.webapp.address --value "$mr_hist:19888"
put_config --file mapred-site.xml --property yarn.app.mapreduce.am.staging-dir --value /mapred

create_config --file yarn-site.xml
put_config --file yarn-site.xml --property yarn.nodemanager.aux-services --value mapreduce.shuffle
put_config --file yarn-site.xml --property yarn.nodemanager.aux-services.mapreduce.shuffle.class --value org.apache.hadoop.mapred.ShuffleHandler
put_config --file yarn-site.xml --property yarn.web-proxy.address --value "$yarn_proxy:$YARN_PROXY_PORT"
put_config --file yarn-site.xml -–property yarn.resourcemanager.scheduler.address --value "$rmgr:8030"
put_config --file yarn-site.xml --property yarn.resourcemanager.resource-tracker.address --value "$rmgr:8031"
put_config --file yarn-site.xml --property yarn.resourcemanager.address --value "$rmgr:8032"
put_config --file yarn-site.xml --property yarn.resourcemanager.admin.address --value "$rmgr:8033"
put_config --file yarn-site.xml --property yarn.resourcemanager.webapp.address --value "$rmgr:8088"

echo "Copying base Hadoop XML config files to all hosts..."
pdcp -w ^all_hosts core-site.xml hdfs-site.xml mapred-site.xml yarn-site.xml $HADOOP_HOME/etc/hadoop/

echo "Creating configuration, command, and script links on all hosts..."
pdsh -w ^all_hosts "ln -s $HADOOP_HOME/etc/hadoop /etc/hadoop"
pdsh -w ^all_hosts "ln -s $HADOOP_HOME/bin/* /usr/bin"
pdsh -w ^all_hosts "ln -s $HADOOP_HOME/libexec/* /usr/libexec"

echo "Formatting the NameNode..."
pdsh -w ^nn_host "su - hdfs -c '$HADOOP_HOME/bin/hdfs namenode -format'"

echo "Copying startup scripts to all hosts..."
pdcp -w ^nn_host hadoop-namenode /etc/init.d/
pdcp -w ^snn_host hadoop-secondarynamenode /etc/init.d/
pdcp -w ^dn_hosts hadoop-datanode /etc/init.d/
pdcp -w ^rm_host hadoop-resourcemanager /etc/init.d/
pdcp -w ^nm_hosts hadoop-nodemanager /etc/init.d/
pdcp -w ^mr_history_host hadoop-historyserver /etc/init.d/
pdcp -w ^yarn_proxy_host hadoop-proxyserver /etc/init.d/

echo "Starting Hadoop $HADOOP_VERSION services on all hosts..."
pdsh -w ^nn_host "chmod 755 /etc/init.d/hadoop-namenode && chkconfig hadoop-namenode on && service hadoop-namenode start"
pdsh -w ^snn_host "chmod 755 /etc/init.d/hadoop-secondarynamenode && chkconfig hadoop-secondarynamenode on && service hadoop-secondarynamenode start"
pdsh -w ^dn_hosts "chmod 755 /etc/init.d/hadoop-datanode && chkconfig hadoop-datanode on && service hadoop-datanode start"
pdsh -w ^rm_host "chmod 755 /etc/init.d/hadoop-resourcemanager && chkconfig hadoop-resourcemanager on && service hadoop-resourcemanager start"
pdsh -w ^nm_hosts "chmod 755 /etc/init.d/hadoop-nodemanager && chkconfig hadoop-nodemanager on && service hadoop-nodemanager start"

pdsh -w ^yarn_proxy_host "chmod 755 /etc/init.d/hadoop-proxyserver && chkconfig hadoop-proxyserver on && service hadoop-proxyserver start"

echo "Creating MapReduce Job History directories..."
su - hdfs -c "hadoop fs -mkdir -p /mapred/history/done_intermediate"
su - hdfs -c "hadoop fs -chown -R mapred:hadoop /mapred"
su - hdfs -c "hadoop fs -chmod -R g+rwx /mapred"

pdsh -w ^mr_history_host "chmod 755 /etc/init.d/hadoop-historyserver && chkconfig hadoop-historyserver on && service hadoop-historyserver start"

echo "Running YARN smoke test..."
pdsh -w ^all_hosts "usermod -a -G hadoop $(whoami)"
su - hdfs -c "hadoop fs -mkdir -p /user/$(whoami)"
su - hdfs -c "hadoop fs -chown $(whoami):$(whoami) /user/$(whoami)"
source /etc/profile.d/java.sh
source /etc/profile.d/hadoop.sh
source /etc/hadoop/hadoop-env.sh
source /etc/hadoop/yarn-env.sh
hadoop jar $HADOOP_HOME/share/hadoop/mapreduce/hadoop-mapreduce-examples-$HADOOP_VERSION.jar pi -Dmapreduce.clientfactory.class.name=org.apache.hadoop.mapred.YarnClientFactory -libjars $HADOOP_HOME/share/hadoop/mapreduce/hadoop-mapreduce-client-jobclient-$HADOOP_VERSION.jar 16 10000
}

interactive()
{
echo -n "Enter NameNode hostname: "
read nn
echo -n "Enter Secondary NameNode hostname: "
read snn
echo -n "Enter ResourceManager hostname: "
read rmgr
echo -n "Enter Job History Server hostname: "
read mr_hist
echo -n "Enter YARN Proxy hostname: "
read yarn_proxy
echo -n "Enter DataNode hostnames (comma-separated or hostlist syntax): "
read dns
echo -n "Enter NodeManager hostnames (comma-separated or hostlist syntax): "
read nms

echo "$nn" > "$nn_host"
echo "$snn" > "$snn_host"
echo "$rmgr" > "$rm_host"
echo "$mr_hist" > "$mr_history_host"
echo "$yarn_proxy" > "$yarn_proxy_host"
dn_hosts_var=$(sed 's/\,/\n/g' <<< $dns)
nm_hosts_var=$(sed 's/\,/\n/g' <<< $nms)
echo "$dn_hosts_var" > "$dn_hosts"
echo "$nm_hosts_var" > "$nm_hosts"
echo "$(echo "$nn $snn $rmgr $mr_hist $yarn_proxy $dn_hosts_var $nm_hosts_var" | tr ' ' '\n' | sort -u)" > "$all_hosts"
}

file()
{
nn=$(cat nn_host)
snn=$(cat snn_host)
rmgr=$(cat rm_host)
mr_hist=$(cat mr_history_host)
yarn_proxy=$(cat yarn_proxy_host)
dns=$(cat dn_hosts)
nms=$(cat nm_hosts)

echo "$(echo "$nn $snn $rmgr $mr_hist $dns $nms" | tr ' ' '\n' | sort -u)" > "$all_hosts"
}

help()
{
cat << EOF
install-hadoop2.sh

This script installs Hadoop 2 with basic data, log, and pid directories.

USAGE: install-hadoop2.sh [options]

OPTIONS:
-i, --interactive Prompt for fully qualified domain names (FQDN) of the NameNode,
Secondary NameNode, DataNodes, ResourceManager, NodeManagers,
MapReduce Job History Server, and YARN Proxy server. Values
entered are stored in files in the same directory as this command.

-f, --file Use files with fully qualified domain names (FQDN), newline
separated. Place files in the same directory as this script.
Services and file name are as follows:
NameNode = nn_host
Secondary NameNode = snn_host
DataNodes = dn_hosts
ResourceManager = rm_host
NodeManagers = nm_hosts
MapReduce Job History Server = mr_history_host
YARN Proxy Server = yarn_proxy_host

-h, --help Show this message.

EXAMPLES:
Prompt for host names:
install-hadoop2.sh -i
install-hadoop2.sh --interactive

Use values from files in the same directory:
install-hadoop2.sh -f
install-hadoop2.sh --file

EOF
}

while true;
do
case "$1" in

-h|--help)
help
exit 0
;;
-i|--interactive)
interactive
install
shift
;;
-f|--file)
file
install
shift
;;
--)
shift
break
;;
esac
done

uninstall-hadoop2.sh

#!/bin/bash

HADOOP_VERSION=2.0.5-alpha
HADOOP_HOME="/opt/hadoop-${HADOOP_VERSION}"
NN_DATA_DIR=/var/data/hadoop/hdfs/nn
SNN_DATA_DIR=/var/data/hadoop/hdfs/snn
DN_DATA_DIR=/var/data/hadoop/hdfs/dn
YARN_LOG_DIR=/var/log/hadoop/yarn
HADOOP_LOG_DIR=/var/log/hadoop/hdfs
HADOOP_MAPRED_LOG_DIR=/var/log/hadoop/mapred

echo "Stopping Hadoop 2 services..."
pdsh -w ^dn_hosts "service hadoop-datanode stop"
pdsh -w ^snn_host "service hadoop-secondarynamenode stop"
pdsh -w ^nn_host "service hadoop-namenode stop"
pdsh -w ^mr_history_host "service hadoop-historyserver stop"
pdsh -w ^yarn_proxy_host "service hadoop-proxyserver stop"
pdsh -w ^nm_hosts "service hadoop-nodemanager stop"
pdsh -w ^rm_host "service hadoop-resourcemanager stop"

echo "Removing Hadoop 2 services from run levels..."
pdsh -w ^dn_hosts "chkconfig --del hadoop-datanode"
pdsh -w ^snn_host "chkconfig --del hadoop-secondarynamenode"
pdsh -w ^nn_host "chkconfig --del hadoop-namenode"
pdsh -w ^mr_history_host "chkconfig --del hadoop-historyserver"
pdsh -w ^yarn_proxy_host "chkconfig --del hadoop-proxyserver"
pdsh -w ^nm_hosts "chkconfig --del hadoop-nodemanager"
pdsh -w ^rm_host "chkconfig --del hadoop-resourcemanager"

echo "Removing Hadoop 2 startup scripts..."
pdsh -w ^all_hosts "rm -f /etc/init.d/hadoop-*"

echo "Removing Hadoop 2 distribution tarball..."
pdsh -w ^all_hosts "rm -f /opt/hadoop-2*.tar.gz"

echo "Removing JDK 1.6.0_31 distribution..."
pdsh -w ^all_hosts "rm -f /opt/jdk*"

echo "Removing JDK 1.6.0_31 artifacts..."
pdsh -w ^all_hosts "rm -f sun-java*"
pdsh -w ^all_hosts "rm -f jdk*"

echo "Removing Hadoop 2 home directory..."
pdsh -w ^all_hosts "rm -Rf $HADOOP_HOME"

echo "Removing Hadoop 2 bash environment setting..."
pdsh -w ^all_hosts "rm -f /etc/profile.d/hadoop.sh"

echo "Removing Java bash environment setting..."
pdsh -w ^all_hosts "rm -f /etc/profile.d/java.sh"

echo "Removing /etc/hadoop link..."
pdsh -w ^all_hosts "unlink /etc/hadoop"

echo "Removing Hadoop 2 command links..."
pdsh -w ^all_hosts "unlink /usr/bin/container-executor"
pdsh -w ^all_hosts "unlink /usr/bin/hadoop"
pdsh -w ^all_hosts "unlink /usr/bin/hdfs"
pdsh -w ^all_hosts "unlink /usr/bin/mapred"
pdsh -w ^all_hosts "unlink /usr/bin/rcc"
pdsh -w ^all_hosts "unlink /usr/bin/test-container-executor"
pdsh -w ^all_hosts "unlink /usr/bin/yarn"

echo "Removing Hadoop 2 script links..."
pdsh -w ^all_hosts "unlink /usr/libexec/hadoop-config.sh"
pdsh -w ^all_hosts "unlink /usr/libexec/hdfs-config.sh"
pdsh -w ^all_hosts "unlink /usr/libexec/httpfs-config.sh"
pdsh -w ^all_hosts "unlink /usr/libexec/mapred-config.sh"
pdsh -w ^all_hosts "unlink /usr/libexec/yarn-config.sh"

echo "Uninstalling JDK 1.6.0_31 RPM..."
pdsh -w ^all_hosts "rpm -ev jdk-1.6.0_31-fcs.x86_64"

echo "Removing NameNode data directory..."
pdsh -w ^nn_host "rm -Rf $NN_DATA_DIR"

echo "Removing Secondary NameNode data directory..."
pdsh -w ^snn_host "rm -Rf $SNN_DATA_DIR"

echo "Removing DataNode data directories..."
pdsh -w ^dn_hosts "rm -Rf $DN_DATA_DIR"

echo "Removing YARN log directories..."
pdsh -w ^all_hosts "rm -Rf $YARN_LOG_DIR"

echo "Removing HDFS log directories..."
pdsh -w ^all_hosts "rm -Rf $HADOOP_LOG_DIR"

echo "Removing MapReduce log directories..."
pdsh -w ^all_hosts "rm -Rf $HADOOP_MAPRED_LOG_DIR"

echo "Removing HDFS account..."
pdsh -w ^all_hosts "userdel -r hdfs"

echo "Removing MapReduce system account..."
pdsh -w ^all_hosts "userdel -r mapred"

echo "Removing YARN system account..."
pdsh -w ^all_hosts "userdel -r yarn"

echo "Removing Hadoop system group..."
pdsh -w ^all_hosts "groupdel hadoop"

hadoop-xml-conf.sh

#!/bin/bash
#
# Utility functions for processing Hadoop 2 XML configuration files.
#
# Depends on Python built-in XML processing and libxml2 for formatting.
#

installed=false
if [ -f /etc/profile.d/hadoop.sh ]; then
source /etc/profile.d/hadoop.sh
source $HADOOP_HOME/etc/hadoop/hadoop-env.sh
installed=true
fi


create_config()
{
local filename=

case $1 in
'') echo $"$0: Usage: create_config --file"
return 1;;
--file)
filename=$2
;;
esac

python - <<END
from xml.etree import ElementTree
from xml.etree.ElementTree import Element

conf = Element('configuration')

conf_file = open("$filename",'w')
conf_file.write(ElementTree.tostring(conf))
conf_file.close()
END
write_file $filename
}

put_config()
{
local filename= property= value=

while [ "$1" != "" ]; do
case $1 in
'') echo $"$0: Usage: put_config --file --property --value"
return 1;;
--file)
filename=$2
shift 2
;;
--property)
property=$2
shift 2
;;
--value)
value=$2
shift 2
;;
esac
done

python - <<END
from xml.etree import ElementTree
from xml.etree.ElementTree import Element
from xml.etree.ElementTree import SubElement

def putconfig(root, name, value):
for existing_prop in root.getchildren():
if existing_prop.find('name').text == name:
root.remove(existing_prop)
break
property = SubElement(root, 'property')
name_elem = SubElement(property, 'name')
name_elem.text = name
value_elem = SubElement(property, 'value')
value_elem.text = value

path = ''
if "$installed" == 'true':
path = "$HADOOP_CONF_DIR" + '/'

conf = ElementTree.parse(path + "$filename").getroot()
putconfig(root = conf, name = "$property", value = "$value")

conf_file = open("$filename",'w')
conf_file.write(ElementTree.tostring(conf))
conf_file.close()
END
write_file $filename
}

del_config()
{
local filename= property=

while [ "$1" != "" ]; do
case $1 in
'') echo $"$0: Usage: del_config --file --property"
return 1;;
--file)
filename=$2
shift 2
;;
--property)
property=$2
shift 2
;;
esac
done

python - <<END
from xml.etree import ElementTree
from xml.etree.ElementTree import Element
from xml.etree.ElementTree import SubElement

def delconfig(root, name):
for existing_prop in root.getchildren():
if existing_prop.find('name').text == name:
root.remove(existing_prop)
break

path = ''
if "$installed" == 'true':
path = "$HADOOP_CONF_DIR" + '/'

conf = ElementTree.parse(path + "$filename").getroot()
delconfig(root = conf, name = "$property")

conf_file = open("$filename",'w')
conf_file.write(ElementTree.tostring(conf))
conf_file.close()
END
write_file $filename
}

write_file()
{
local file=$1
xmllint --format "$file" > "$file".pp && mv "$file".pp "$file"
}