parent
5e3697807c
commit
d085c8626f
|
@ -291,6 +291,7 @@ if ! $skipCreate; then
|
|||
-c "$clientNodeCount"
|
||||
-n "$additionalValidatorCount"
|
||||
--dedicated
|
||||
--self-destruct-hours 0
|
||||
)
|
||||
|
||||
if [[ -n $bootstrapValidatorAddress ]]; then
|
||||
|
|
49
net/gce.sh
49
net/gce.sh
|
@ -78,6 +78,7 @@ letsEncryptDomainName=
|
|||
enableGpu=false
|
||||
customMachineType=
|
||||
customAddress=
|
||||
selfDestructHours=8
|
||||
zones=()
|
||||
|
||||
containsZone() {
|
||||
|
@ -151,6 +152,9 @@ Manage testnet instances
|
|||
(by default preemptible instances are used to reduce
|
||||
cost). Note that the bootstrap leader, archiver,
|
||||
blockstreamer and client nodes are always dedicated.
|
||||
--self-destruct-hours [number]
|
||||
- Specify lifetime of the allocated instances in hours. 0 to
|
||||
disable. Only supported on GCE. (default: $selfDestructHours)
|
||||
|
||||
config-specific options:
|
||||
-P - Use public network IP addresses (default: $publicNetwork)
|
||||
|
@ -200,6 +204,15 @@ while [[ -n $1 ]]; do
|
|||
elif [[ $1 = --custom-machine-type ]]; then
|
||||
customMachineType="$2"
|
||||
shift 2
|
||||
elif [[ $1 == --self-destruct-hours ]]; then
|
||||
maybeTimeout=$2
|
||||
if [[ $maybeTimeout =~ ^[0-9]+$ ]]; then
|
||||
selfDestructHours=$maybeTimeout
|
||||
else
|
||||
echo " Invalid parameter ($maybeTimeout) to $1"
|
||||
usage 1
|
||||
fi
|
||||
shift 2
|
||||
else
|
||||
usage "Unknown long option: $1"
|
||||
fi
|
||||
|
@ -737,6 +750,42 @@ $(
|
|||
if [[ -n $validatorAdditionalDiskSizeInGb ]]; then
|
||||
cat mount-additional-disk.sh
|
||||
fi
|
||||
|
||||
if [[ $selfDestructHours -gt 0 ]]; then
|
||||
cat <<EOSD
|
||||
|
||||
# Setup GCE self-destruct
|
||||
cat >/solana-scratch/gce-self-destruct.sh <<'EOS'
|
||||
$(cat gce-self-destruct.sh)
|
||||
EOS
|
||||
EOSD
|
||||
cat <<'EOSD'
|
||||
|
||||
# Populate terminal prompt update script
|
||||
cat >/solana-scratch/gce-self-destruct-ps1.sh <<'EOS'
|
||||
#!/usr/bin/env bash
|
||||
source "$(dirname "$0")/gce-self-destruct.sh"
|
||||
gce_self_destruct_ps1
|
||||
EOS
|
||||
chmod +x /solana-scratch/gce-self-destruct-ps1.sh
|
||||
|
||||
# Append MOTD and PS1 replacement to .profile
|
||||
cat >>~solana/.profile <<'EOS'
|
||||
|
||||
# Print self-destruct countdown on login
|
||||
source "/solana-scratch/gce-self-destruct.sh"
|
||||
gce_self_destruct_motd
|
||||
|
||||
# Add self-destruct countdown to terminal prompt
|
||||
export PS1='\[\e]0;\u@\h: \w\a\]${debian_chroot:+($debian_chroot)}\[\033[01;32m\]\u@\h\[\033[00m\]$(/solana-scratch/gce-self-destruct-ps1.sh):\[\033[01;34m\]\w\[\033[00m\]\$ '
|
||||
EOS
|
||||
EOSD
|
||||
cat <<EOSD
|
||||
|
||||
source /solana-scratch/gce-self-destruct.sh
|
||||
gce_self_destruct_setup $selfDestructHours
|
||||
EOSD
|
||||
fi
|
||||
)
|
||||
|
||||
cat > /etc/motd <<EOM
|
||||
|
|
|
@ -197,6 +197,7 @@ cloud_CreateInstances() {
|
|||
--image "$imageName"
|
||||
--maintenance-policy TERMINATE
|
||||
--restart-on-failure
|
||||
--scopes compute-rw
|
||||
)
|
||||
|
||||
# shellcheck disable=SC2206 # Do not want to quote $imageName as it may contain extra args
|
||||
|
|
|
@ -0,0 +1,151 @@
|
|||
#!/usr/bin/env bash
|
||||
|
||||
__gce_sd_here="$(dirname "${BASH_SOURCE[0]}")"
|
||||
# shellcheck disable=SC1091
|
||||
__gce_sd_conf="${__gce_sd_here}/gce-self-destruct.conf"
|
||||
|
||||
gce_metadata_req() {
|
||||
declare endpoint url
|
||||
endpoint="$1"
|
||||
url="http://metadata.google.internal/computeMetadata/v1/$endpoint"
|
||||
curl -sf -H Metadata-Flavor:Google "$url"
|
||||
}
|
||||
|
||||
unix_to_at_time() {
|
||||
declare unix="$1"
|
||||
date --date="@$unix" "+%Y%m%d%H%M.%S"
|
||||
}
|
||||
|
||||
timeout_to_destruct() {
|
||||
declare timeout_sec now_unix
|
||||
declare timeout_hrs=$1
|
||||
timeout_sec=$((timeout_hrs * 60 * 60))
|
||||
now_unix=$(date +%s)
|
||||
echo $((now_unix + timeout_sec))
|
||||
}
|
||||
|
||||
relative_timespan()
|
||||
{
|
||||
declare timeSpan="$1"
|
||||
declare -a units divs
|
||||
units+=( s ); divs+=( 60 )
|
||||
units+=( m ); divs+=( 60 )
|
||||
units+=( h ); divs+=( 24 )
|
||||
units+=( d ); divs+=( 7 )
|
||||
units+=( w ); divs+=( 52 )
|
||||
numUnits="${#units[@]}"
|
||||
units+=( y ); divs+=( 100 )
|
||||
|
||||
declare -a outs
|
||||
declare i div remain
|
||||
for (( i=0; i < "$numUnits"; i++ )); do
|
||||
div="${divs[$i]}"
|
||||
[[ "$timeSpan" -lt "$div" ]] && break
|
||||
remain="$((timeSpan % div))"
|
||||
timeSpan="$((timeSpan / div))"
|
||||
outs+=( "$remain" )
|
||||
done
|
||||
outs+=( "$timeSpan" )
|
||||
|
||||
numOut="${#outs[@]}"
|
||||
out1="$((numOut-1))"
|
||||
out2="$((numOut-2))"
|
||||
|
||||
if [[ "$numOut" -eq 1 ]] || \
|
||||
[[ "$numOut" -ge "$numUnits" && \
|
||||
"${outs[out1]}" -ge "${divs[out1]}" ]]; then
|
||||
printf "%d%s" "${outs[out1]}" "${units[out1]}"
|
||||
else
|
||||
printf "%d%s%02d%s" "${outs[out1]}" \
|
||||
"${units[out1]}" "${outs[out2]}" "${units[out2]}"
|
||||
fi
|
||||
}
|
||||
|
||||
gce_self_destruct_ttl() {
|
||||
declare colorize="${1:-true}"
|
||||
declare prefix="${2}"
|
||||
declare suffix="${3}"
|
||||
declare output=0
|
||||
|
||||
if [[ -f "${__gce_sd_conf}" ]]; then
|
||||
# shellcheck disable=SC1090
|
||||
source "${__gce_sd_conf}"
|
||||
declare ttl pttl color
|
||||
ttl="$((destruct - $(date +%s)))"
|
||||
if [[ "$ttl" -lt 0 ]]; then
|
||||
ttl=0
|
||||
fi
|
||||
pttl="$(relative_timespan "$ttl")"
|
||||
color=
|
||||
if [[ ttl -lt 3600 ]]; then
|
||||
color="\033[01;31m"
|
||||
fi
|
||||
output="${prefix}${pttl}${suffix}"
|
||||
if $colorize; then
|
||||
output="${color}${output}\033[01;00m"
|
||||
fi
|
||||
fi
|
||||
echo -e "$output"
|
||||
}
|
||||
|
||||
gce_self_destruct_setup() {
|
||||
declare destruct at_time zone
|
||||
destruct="$(timeout_to_destruct "$1")"
|
||||
at_time="$(unix_to_at_time "$destruct")"
|
||||
zone=$(gce_metadata_req "instance/zone")
|
||||
zone=$(basename "$zone")
|
||||
|
||||
cat >"${__gce_sd_conf}" <<EOF
|
||||
export destruct=$destruct
|
||||
export zone=$zone
|
||||
EOF
|
||||
|
||||
at -t "$at_time" <<EOF
|
||||
bash -i <<'EOF2'
|
||||
source /solana-scratch/gce-self-destruct.sh
|
||||
gce_self_destruct_check
|
||||
EOF2
|
||||
EOF
|
||||
}
|
||||
|
||||
gce_self_destruct_check() {
|
||||
if [[ -f "${__gce_sd_conf}" ]]; then
|
||||
# shellcheck disable=SC1090
|
||||
source "${__gce_sd_conf}"
|
||||
declare now gcloudBin
|
||||
now=$(date +%s)
|
||||
if [[ "$now" -ge "$destruct" ]]; then
|
||||
# gcloud is installed in /snap/bin, but /snap/bin isn't in root's PATH...
|
||||
gcloudBin="$(command -v gcloud)"
|
||||
gcloudBin="${gcloudBin:-/snap/bin/gcloud}"
|
||||
"$gcloudBin" compute instances delete --quiet "$(hostname)" --zone "$zone"
|
||||
else
|
||||
at -t "$(unix_to_at_time "$destruct")" <<EOF
|
||||
bash -i <<'OEF2'
|
||||
source /solana-scratch/gce-self-destruct.sh
|
||||
gce_self_destruct_check
|
||||
EOF2
|
||||
EOF
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
gce_self_destruct_motd() {
|
||||
declare ttl
|
||||
ttl="$(gce_self_destruct_ttl)"
|
||||
echo -e '\n~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~\n'
|
||||
if [[ -n "${ttl}" ]]; then
|
||||
echo -e "\tThis instance will self-destruct in ${ttl}!"
|
||||
else
|
||||
echo -e "\tThis instance will NOT self-destruct. YOU are responsible for deleting it!"
|
||||
fi
|
||||
echo -e '\n~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~\n'
|
||||
}
|
||||
|
||||
gce_self_destruct_ps1() {
|
||||
declare ttl
|
||||
ttl="$(gce_self_destruct_ttl true "[T-" "]")"
|
||||
ttl="${ttl:-"[T-~~~~~]"}"
|
||||
echo "!${ttl}"
|
||||
}
|
||||
|
Loading…
Reference in New Issue