diff --git a/sbin/start-connect-server.sh b/sbin/start-connect-server.sh index a347f43db8b16..fecda717eb341 100755 --- a/sbin/start-connect-server.sh +++ b/sbin/start-connect-server.sh @@ -38,4 +38,10 @@ fi . "${SPARK_HOME}/bin/load-spark-env.sh" -exec "${SPARK_HOME}"/sbin/spark-daemon.sh submit $CLASS 1 --name "Spark Connect server" "$@" +if [ "$1" == "--wait" ]; then + shift + exec "${SPARK_HOME}"/bin/spark-submit --class $CLASS 1 --name "Spark Connect Server" "$@" +else + exec "${SPARK_HOME}"/sbin/spark-daemon.sh submit $CLASS 1 --name "Spark Connect server" "$@" +fi +