diff options
Diffstat (limited to 'scripts/04-dataproc-create-cluster.sh')
| -rwxr-xr-x | scripts/04-dataproc-create-cluster.sh | 24 |
1 files changed, 11 insertions, 13 deletions
diff --git a/scripts/04-dataproc-create-cluster.sh b/scripts/04-dataproc-create-cluster.sh index d2dd080..8bdeccb 100755 --- a/scripts/04-dataproc-create-cluster.sh +++ b/scripts/04-dataproc-create-cluster.sh @@ -2,29 +2,28 @@ set -eu -if [ "$#" -ne 1 ]; then - echo "Usage: 'sh ${PWD}/$0 <num-workers>'" +if [ "$#" -ne 3 ]; then + echo "Usage: 'sh ${PWD}/$0 <num-workers> <master-machine> <worker-machine>'" exit 1 fi NUM_WORKERS="$1" +MASTER_MACHINE="$2" +WORKER_MACHINE="$3" + if [ "$NUM_WORKERS" -lt 1 ] || [ "$NUM_WORKERS" -gt 4 ]; then echo "<num-workers> must be 1, 2, 3, or 4" exit 1 fi -read -p "Enter worker-machine-type: " worker_machine - -if [ -z "$worker_machine" ]; then - echo "Error: Worker machine type cannot be empty." +if [ -z "$MASTER_MACHINE" ]; then + echo "Error: Master machine type cannot be empty." exit 1 fi -read -p "Enter master-machine-type: " master_machine - -if [ -z "$master_machine" ]; then - echo "Error: Master machine type cannot be empty." +if [ -z "$WORKER_MACHINE" ]; then + echo "Error: Worker machine type cannot be empty." exit 1 fi @@ -34,9 +33,8 @@ COMMON_PARAMS="\ --service-account=${SERVICE_ACCOUNT}@${PROJECT}.iam.gserviceaccount.com \ --master-boot-disk-size=240 \ --worker-boot-disk-size=240 \ - --worker-machine-type=${worker_machine} \ - --master-machine-type=${master_machine}" - + --worker-machine-type=${WORKER_MACHINE} \ + --master-machine-type=${MASTER_MACHINE}" if [ "$NUM_WORKERS" -eq 1 ]; then echo ">>>> Creating a single-node cluster..." |