110 lines
2.3 KiB
Bash
Executable File
110 lines
2.3 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
here=$(dirname "$0")
|
|
# shellcheck source=scripts/gcloud.sh
|
|
source "$here"/../scripts/gcloud.sh
|
|
|
|
command=$1
|
|
prefix=
|
|
num_nodes=
|
|
out_file=
|
|
image_name="ubuntu-16-04-cuda-9-2-new"
|
|
internalNetwork=false
|
|
zone="us-west1-b"
|
|
|
|
shift
|
|
|
|
usage() {
|
|
exitcode=0
|
|
if [[ -n "$1" ]]; then
|
|
exitcode=1
|
|
echo "Error: $*"
|
|
fi
|
|
cat <<EOF
|
|
usage: $0 <create|delete> <-p prefix> <-n num_nodes> <-o file> [-i image-name]
|
|
|
|
Manage a GCE multinode network
|
|
|
|
create|delete - Create or delete the network
|
|
-p prefix - A common prefix for node names, to avoid collision
|
|
-n num_nodes - Number of nodes
|
|
-P - Use IP addresses on GCE internal/private network
|
|
-z - GCP Zone for the nodes (default $zone)
|
|
-o out_file - Used for create option. Outputs an array of IP addresses
|
|
of new nodes to the file
|
|
-i image_name - Existing image on GCE (default $image_name)
|
|
|
|
EOF
|
|
exit $exitcode
|
|
}
|
|
|
|
while getopts "h?p:Pi:n:z:o:" opt; do
|
|
case $opt in
|
|
h | \?)
|
|
usage
|
|
;;
|
|
p)
|
|
prefix=$OPTARG
|
|
;;
|
|
P)
|
|
internalNetwork=true
|
|
;;
|
|
i)
|
|
image_name=$OPTARG
|
|
;;
|
|
o)
|
|
out_file=$OPTARG
|
|
;;
|
|
n)
|
|
num_nodes=$OPTARG
|
|
;;
|
|
z)
|
|
zone=$OPTARG
|
|
;;
|
|
*)
|
|
usage "Error: unhandled option: $opt"
|
|
;;
|
|
esac
|
|
done
|
|
|
|
set -e
|
|
|
|
[[ -n $command ]] || usage "Need a command (create|delete)"
|
|
|
|
[[ -n $prefix ]] || usage "Need a prefix for GCE instance names"
|
|
|
|
|
|
if [[ $command == "create" ]]; then
|
|
[[ -n $num_nodes ]] || usage "Need number of nodes"
|
|
[[ -n $out_file ]] || usage "Need an outfile to store IP Addresses"
|
|
|
|
gcloud_CreateInstances "$prefix" "$num_nodes" "$zone" "$image_name"
|
|
gcloud_FindInstances "name~^$prefix"
|
|
|
|
echo "ip_addr_array=()" > "$out_file"
|
|
recordPublicIp() {
|
|
declare name="$1"
|
|
declare publicIp="$3"
|
|
declare privateIp="$4"
|
|
|
|
if $internalNetwork; then
|
|
echo "ip_addr_array+=($privateIp) # $name" >> "$out_file"
|
|
else
|
|
echo "ip_addr_array+=($publicIp) # $name" >> "$out_file"
|
|
fi
|
|
}
|
|
gcloud_ForEachInstance recordPublicIp
|
|
|
|
echo "Instance ip addresses recorded in $out_file"
|
|
elif [[ $command == "delete" ]]; then
|
|
gcloud_FindInstances "name~^$prefix"
|
|
|
|
if [[ ${#instances[@]} -eq 0 ]]; then
|
|
echo "No instances found matching '^$prefix'"
|
|
exit 0
|
|
fi
|
|
gcloud_DeleteInstances
|
|
else
|
|
usage "Unknown command: $command"
|
|
fi
|