You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
1.5 KiB
Bash
57 lines
1.5 KiB
Bash
#!/usr/bin/env bash
|
|
# vim:ts=4:sts=4:sw=4:et
|
|
#
|
|
# Author: Hari Sekhon
|
|
# Date: 2019-11-21 16:02:54 +0000 (Thu, 21 Nov 2019)
|
|
#
|
|
# https://github.com/harisekhon/bash-tools
|
|
#
|
|
# License: see accompanying Hari Sekhon LICENSE file
|
|
#
|
|
# If you're using my code you're welcome to connect with me on LinkedIn and optionally send me feedback to help steer this or other code I publish
|
|
#
|
|
# https://www.linkedin.com/in/harisekhon
|
|
#
|
|
|
|
set -euo pipefail
|
|
[ -n "${DEBUG:-}" ] && set -x
|
|
|
|
usage(){
|
|
echo "
|
|
Checks GCE Metadata API for the preemption status and returns preempted / not preempted and 0 or 1 exit code
|
|
|
|
Can combine in a shutdown script to do something different if preempted vs a normal shutdown
|
|
|
|
https://cloud.google.com/compute/docs/instances/create-start-preemptible-instance
|
|
|
|
https://cloud.google.com/compute/docs/shutdownscript
|
|
|
|
usage:
|
|
|
|
${0##*/}
|
|
"
|
|
exit 3
|
|
}
|
|
|
|
if [ $# -ne 0 ]; then
|
|
usage
|
|
fi
|
|
|
|
#if ! curl -sS --connect-timeout 2 -H "Metadata-Flavor: Google" "http://metadata.google.internal/computeMetadata/v1/" &>/dev/null; then
|
|
# echo "This script must be run from within a GCE instance as that is the only place the GCP GCE Metadata API is available"
|
|
# exit 2
|
|
#fi
|
|
|
|
if output="$(curl -sS -H "Metadata-Flavor: Google" "http://metadata.google.internal/computeMetadata/v1/instance/preempted")"; then
|
|
if grep -q TRUE <<< "$output"; then
|
|
echo "preempted"
|
|
exit 0
|
|
else
|
|
echo "not preempted"
|
|
exit 1
|
|
fi
|
|
else
|
|
echo "FAILED to query GCE Metadata API, not running inside GCE?"
|
|
exit 2
|
|
fi
|