blob: 77726009ae8bc42cd34114c8bf0015b148961ebe [file] [log] [blame]
#!/usr/bin/env bash
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements. See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# The Hama command script
#
# Environment Variables
#
# JAVA_HOME The java implementation to use. Overrides JAVA_HOME.
#
# HAMA_CLASSPATH Extra Java CLASSPATH entries.
#
# HAMA_HEAPSIZE The maximum amount of heap to use, in MB.
# Default is 1000.
#
# HAMA_OPTS Extra Java runtime options.
#
# HAMA_NAMENODE_OPTS These options are added to HAMA_OPTS
# HAMA_CLIENT_OPTS when the respective command is run.
# HAMA_{COMMAND}_OPTS etc HAMA_JT_OPTS applies to JobTracker
# for e.g. HAMA_CLIENT_OPTS applies to
# more than one command (fs, dfs, fsck,
# dfsadmin etc)
#
# HAMA_CONF_DIR Alternate conf dir. Default is ${HAMA_HOME}/conf.
#
# HAMA_ROOT_LOGGER The root appender. Default is INFO,console
#
bin=`dirname "$0"`
bin=`cd "$bin"; pwd`
. "$bin"/hama-config.sh
cygwin=false
case "`uname`" in
CYGWIN*) cygwin=true;;
esac
# if no args specified, show usage
if [ $# = 0 ]; then
echo "Usage: hama [--config confdir] COMMAND"
echo "where COMMAND is one of:"
echo " bspmaster run the BSP Master node"
echo " groom run the Groom node"
echo " zookeeper run a Zookeeper server"
echo " job manipulate BSP jobs"
echo " jar <jar> run a jar file"
echo " pipes run a pipe job"
echo " seqdumper run SequenceFileDumper"
echo " or"
echo " CLASSNAME run the class named CLASSNAME"
echo "Most commands print help when invoked w/o parameters."
exit 1
fi
# get arguments
COMMAND=$1
shift
if [ -f "${HAMA_CONF_DIR}/hama-env.sh" ]; then
. "${HAMA_CONF_DIR}/hama-env.sh"
fi
# some Java parameters
if [ "$JAVA_HOME" != "" ]; then
#echo "run java in $JAVA_HOME"
JAVA_HOME=$JAVA_HOME
fi
if [ "$JAVA_HOME" = "" ]; then
echo "Error: JAVA_HOME is not set."
exit 1
fi
JAVA=$JAVA_HOME/bin/java
JAVA_HEAP_MAX=-Xmx1000m
# check envvars which might override default args
if [ "$HAMA_HEAPSIZE" != "" ]; then
#echo "run with heapsize $HAMA_HEAPSIZE"
JAVA_HEAP_MAX="-Xmx""$HAMA_HEAPSIZE""m"
#echo $JAVA_HEAP_MAX
fi
# CLASSPATH initially contains $HAMA_CONF_DIR
CLASSPATH="${HAMA_CONF_DIR}"
CLASSPATH=${CLASSPATH}:$JAVA_HOME/lib/tools.jar
# for developers, add Hama classes to CLASSPATH
if [ -d "$HAMA_HOME/core/target/classes" ]; then
CLASSPATH=${CLASSPATH}:$HAMA_HOME/core/target/classes
fi
if [ -d "$HAMA_HOME/core/target/test-classes/classes" ]; then
CLASSPATH=${CLASSPATH}:$HAMA_HOME/core/target/test-classes
fi
# for developers, add Commons classes to CLASSPATH
if [ -d "$HAMA_HOME/commons/target/classes" ]; then
CLASSPATH=${CLASSPATH}:$HAMA_HOME/commons/target/classes
fi
if [ -d "$HAMA_HOME/commons/target/test-classes/classes" ]; then
CLASSPATH=${CLASSPATH}:$HAMA_HOME/commons/target/test-classes
fi
# for developers, add Graph classes to CLASSPATH
if [ -d "$HAMA_HOME/graph/target/classes" ]; then
CLASSPATH=${CLASSPATH}:$HAMA_HOME/graph/target/classes
fi
if [ -d "$HAMA_HOME/graph/target/test-classes/classes" ]; then
CLASSPATH=${CLASSPATH}:$HAMA_HOME/graph/target/test-classes
fi
# for developers, add ML classes to CLASSPATH
if [ -d "$HAMA_HOME/ml/target/classes" ]; then
CLASSPATH=${CLASSPATH}:$HAMA_HOME/ml/target/classes
fi
if [ -d "$HAMA_HOME/ml/target/test-classes/classes" ]; then
CLASSPATH=${CLASSPATH}:$HAMA_HOME/ml/target/test-classes
fi
# add mesos classes to CLASSPATH
if [ -d "$HAMA_HOME/mesos/target/classes" ]; then
CLASSPATH=${CLASSPATH}:$HAMA_HOME/mesos/target/classes
fi
if [ -d "$HAMA_HOME/mesos/target/test-classes/classes" ]; then
CLASSPATH=${CLASSPATH}:$HAMA_HOME/mesos/target/test-classes
fi
# so that filenames w/ spaces are handled correctly in loops below
IFS=
# for releases, add core hama jar to CLASSPATH
for f in $HAMA_HOME/hama-**.jar; do
CLASSPATH=${CLASSPATH}:$f;
done
# add libs to CLASSPATH
for f in $HAMA_HOME/lib/*.jar; do
CLASSPATH=${CLASSPATH}:$f;
done
# add user-specified CLASSPATH last
if [ "$HAMA_CLASSPATH" != "" ]; then
CLASSPATH=${CLASSPATH}:${HAMA_CLASSPATH}
fi
# default log directory & file
if [ "$HAMA_LOG_DIR" = "" ]; then
HAMA_LOG_DIR="$HAMA_HOME/logs"
fi
if [ "$HAMA_LOGFILE" = "" ]; then
HAMA_LOGFILE='hama.log'
fi
# default policy file for service-level authorization
if [ "$HAMA_POLICYFILE" = "" ]; then
HAMA_POLICYFILE="hama-policy.xml"
fi
# restore ordinary behaviour
unset IFS
# figure out which class to run
if [ "$COMMAND" = "bspmaster" ] ; then
CLASS='org.apache.hama.BSPMasterRunner'
BSP_OPTS="$BSP_OPTS $BSP_BSPMASTER_OPTS"
elif [ "$COMMAND" = "groom" ] ; then
CLASS='org.apache.hama.GroomServerRunner'
BSP_OPTS="$BSP_OPTS $BSP_GROOMSERVER_OPTS"
elif [ "$COMMAND" = "zookeeper" ] ; then
CLASS='org.apache.hama.ZooKeeperRunner'
elif [ "$COMMAND" = "job" ] ; then
CLASS='org.apache.hama.bsp.BSPJobClient'
elif [ "$COMMAND" = "pipes" ] ; then
CLASS='org.apache.hama.pipes.Submitter'
elif [ "$COMMAND" = "seqdumper" ] ; then
CLASS='org.apache.hama.pipes.util.SequenceFileDumper'
elif [ "$COMMAND" = "jar" ] ; then
CLASS=org.apache.hama.util.RunJar
BSP_OPTS="$BSP_OPTS"
else
CLASS=$COMMAND
fi
# cygwin path translation
if $cygwin; then
CLASSPATH=`cygpath -p -w "$CLASSPATH"`
HAMA_HOME=`cygpath -w "$HAMA_HOME"`
HAMA_LOG_DIR=`cygpath -w "$HAMA_LOG_DIR"`
TOOL_PATH=`cygpath -p -w "$TOOL_PATH"`
fi
# cygwin path translation
if $cygwin; then
JAVA_LIBRARY_PATH=`cygpath -p "$JAVA_LIBRARY_PATH"`
fi
HAMA_OPTS="$HAMA_OPTS -Dhama.log.dir=$HAMA_LOG_DIR"
HAMA_OPTS="$HAMA_OPTS -Dhama.log.file=$HAMA_LOGFILE"
HAMA_OPTS="$HAMA_OPTS -Dhama.home.dir=$HAMA_HOME"
HAMA_OPTS="$HAMA_OPTS -Dhama.id.str=$HAMA_IDENT_STRING"
HAMA_OPTS="$HAMA_OPTS -Dhama.root.logger=${HAMA_ROOT_LOGGER:-INFO,console}"
if [ "x$JAVA_LIBRARY_PATH" != "x" ]; then
HAMA_OPTS="$HAMA_OPTS -Djava.library.path=$JAVA_LIBRARY_PATH"
fi
HAMA_OPTS="$HAMA_OPTS -Dhama.policy.file=$HAMA_POLICYFILE"
# run it
exec "$JAVA" $JAVA_HEAP_MAX $HAMA_OPTS -classpath "$CLASSPATH" $CLASS "$@"