diff options
author | WangTaoTheTonic <barneystinson@aliyun.com> | 2015-01-08 11:45:42 -0800 |
---|---|---|
committer | Andrew Or <andrew@databricks.com> | 2015-01-08 11:46:53 -0800 |
commit | 755f9cc403357da9e5673b5035ebec100b1d83e6 (patch) | |
tree | b4b507c405d874ecc6d3b2379021ff2ca2c2b02a | |
parent | 1770c51b2400ef66e16c8cbca05aa955290c384c (diff) | |
download | spark-755f9cc403357da9e5673b5035ebec100b1d83e6.tar.gz spark-755f9cc403357da9e5673b5035ebec100b1d83e6.tar.bz2 spark-755f9cc403357da9e5673b5035ebec100b1d83e6.zip |
[SPARK-5130][Deploy]Take yarn-cluster as cluster mode in spark-submit
https://issues.apache.org/jira/browse/SPARK-5130
Author: WangTaoTheTonic <barneystinson@aliyun.com>
Closes #3929 from WangTaoTheTonic/SPARK-5130 and squashes the following commits:
c490648 [WangTaoTheTonic] take yarn-cluster as cluster mode in spark-submit
(cherry picked from commit 0760787da885187b0c6dcd5c28753f0ab014d5ed)
Signed-off-by: Andrew Or <andrew@databricks.com>
-rwxr-xr-x | bin/spark-submit | 5 | ||||
-rw-r--r-- | bin/spark-submit2.cmd | 6 |
2 files changed, 11 insertions, 0 deletions
diff --git a/bin/spark-submit b/bin/spark-submit index f92d90c3a6..aefd38a0a2 100755 --- a/bin/spark-submit +++ b/bin/spark-submit @@ -38,11 +38,16 @@ while (($#)); do export SPARK_SUBMIT_CLASSPATH=$2 elif [ "$1" = "--driver-java-options" ]; then export SPARK_SUBMIT_OPTS=$2 + elif [ "$1" = "--master" ]; then + export MASTER=$2 fi shift done DEFAULT_PROPERTIES_FILE="$SPARK_HOME/conf/spark-defaults.conf" +if [ "$MASTER" == "yarn-cluster" ]; then + SPARK_SUBMIT_DEPLOY_MODE=cluster +fi export SPARK_SUBMIT_DEPLOY_MODE=${SPARK_SUBMIT_DEPLOY_MODE:-"client"} export SPARK_SUBMIT_PROPERTIES_FILE=${SPARK_SUBMIT_PROPERTIES_FILE:-"$DEFAULT_PROPERTIES_FILE"} diff --git a/bin/spark-submit2.cmd b/bin/spark-submit2.cmd index cf6046d154..daf0284db9 100644 --- a/bin/spark-submit2.cmd +++ b/bin/spark-submit2.cmd @@ -45,11 +45,17 @@ if [%1] == [] goto continue set SPARK_SUBMIT_CLASSPATH=%2 ) else if [%1] == [--driver-java-options] ( set SPARK_SUBMIT_OPTS=%2 + ) else if [%1] == [--master] ( + set MASTER=%2 ) shift goto loop :continue +if [%MASTER%] == [yarn-cluster] ( + set SPARK_SUBMIT_DEPLOY_MODE=cluster +) + rem For client mode, the driver will be launched in the same JVM that launches rem SparkSubmit, so we may need to read the properties file for any extra class rem paths, library paths, java options and memory early on. Otherwise, it will |