зеркало из https://github.com/Azure/azurehpc.git
104 строки
2.6 KiB
Bash
Executable File
104 строки
2.6 KiB
Bash
Executable File
#!/bin/bash
|
|
azhpc_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )/.." && pwd )"
|
|
source "$azhpc_dir/libexec/common.sh"
|
|
|
|
config_file="config.json"
|
|
|
|
DEBUG_ON=0
|
|
COLOR_ON=1
|
|
|
|
function usage() {
|
|
echo "Command:"
|
|
echo " $0 [options]"
|
|
echo
|
|
echo "Arguments"
|
|
echo " -h --help : diplay this help"
|
|
echo " -c --config: config file to use"
|
|
echo " default: config.json"
|
|
echo
|
|
}
|
|
|
|
while true; do
|
|
case $1 in
|
|
-h|--help)
|
|
usage
|
|
exit 0
|
|
;;
|
|
-c|--config)
|
|
config_path="$2"
|
|
shift
|
|
shift
|
|
;;
|
|
*)
|
|
break
|
|
esac
|
|
done
|
|
|
|
read_value location ".location"
|
|
read_value resource_group ".resource_group"
|
|
read_value vnet_name ".vnet.name"
|
|
read_value address_prefix ".vnet.address_prefix"
|
|
read_value admin_user ".admin_user"
|
|
read_value install_node ".install_from"
|
|
|
|
ssh_private_key=${admin_user}_id_rsa
|
|
ssh_public_key=${admin_user}_id_rsa.pub
|
|
if [ ! -e "$ssh_private_key" ]; then
|
|
error "keys not found"
|
|
fi
|
|
|
|
fqdn=$(
|
|
az network public-ip show \
|
|
--resource-group $resource_group \
|
|
--name ${install_node}pip --query dnsSettings.fqdn \
|
|
--output tsv \
|
|
)
|
|
|
|
if [ "$fqdn" = "" ]; then
|
|
status "The install node does not have a public IP. Using hostname - $install_node - and must be on this node must be on the same vnet"
|
|
fi
|
|
|
|
if [ "$COLOR_ON" -ne "1" ]; then
|
|
printf "%%-20s %s\n" Hostname Status
|
|
else
|
|
printf "\e[1m%-20s %s\e[0m\n" Hostname Status
|
|
fi
|
|
|
|
nodes=()
|
|
|
|
for resource_name in $(jq -r ".resources | keys | @tsv" $config_file); do
|
|
read_value resource_type ".resources.$resource_name.type"
|
|
|
|
read_value password_required ".resources.$resource_name.password" "<no-password>"
|
|
|
|
if [ "$password_required" != "<no-password>" ]; then
|
|
continue
|
|
fi
|
|
|
|
if [ "$resource_type" = "vm" ]; then
|
|
|
|
read_value resource_instances ".resources.$resource_name.instances" 1
|
|
|
|
if [ "$resource_instances" = 1 ]; then
|
|
nodes+=($resource_name)
|
|
else
|
|
for i in $(seq -w $resource_instances); do
|
|
nodes+=(${resource_name}${i})
|
|
done
|
|
fi
|
|
|
|
elif [ "$resource_type" = "vmss" ]; then
|
|
nodes+=($(az vmss list-instances \
|
|
--resource-group $resource_group \
|
|
--name $resource_name \
|
|
--query [].osProfile.computerName \
|
|
--output tsv))
|
|
|
|
else
|
|
error "unknown resource type ($resource_type) for $resource_name"
|
|
fi
|
|
|
|
done
|
|
|
|
ssh $SSH_ARGS -q -i $ssh_private_key $admin_user@$fqdn 'pssh -H "'${nodes[@]}'" -i "printf \"%-20s%s\n\" \"\$(hostname)\" \"\$(uptime)\""' | grep -v SUCCESS
|