Created
March 30, 2022 11:48
-
-
Save ravwojdyla/05249805680f5cd8b54b6f01cd0c383b to your computer and use it in GitHub Desktop.
Auto suspend GCP/GCE VM
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # FILE: /etc/systemd/system/auto-suspend.service | |
| [Unit] | |
| Description=Auto-suspend VM | |
| After=network-online.target | |
| [Service] | |
| # NOTE: adjust this to whatever you need | |
| ExecStart=/usr/bin/auto_suspend_vm 1.0 60 60 | |
| [Install] | |
| WantedBy=multi-user.target |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/bin/bash | |
| # FILE: /usr/bin/auto_suspend_vm | |
| threshold=${1:-1} | |
| intervals=${2:-60} | |
| sleep_time=${3:-60} | |
| function require() { | |
| if ! which $1 >/dev/null; then | |
| echo "This script requires $1, aborting ..." >&2 | |
| exit 1 | |
| fi | |
| } | |
| require gcloud | |
| require python3 | |
| require curl | |
| if ! curl -s -i metadata.google.internal | grep "Metadata-Flavor: Google" >/dev/null; then | |
| echo "This script only works on GCE VMs, aborting ..." >&2 | |
| exit 1 | |
| fi | |
| COMPUTE_METADATA_URL="http://metadata.google.internal/computeMetadata/v1" | |
| VM_PROJECT=$(curl -s "${COMPUTE_METADATA_URL}/project/project-id" -H "Metadata-Flavor: Google" || true) | |
| VM_NAME=$(curl -s "${COMPUTE_METADATA_URL}/instance/hostname" -H "Metadata-Flavor: Google" | cut -d '.' -f 1) | |
| VM_ZONE=$(curl -s "${COMPUTE_METADATA_URL}/instance/zone" -H "Metadata-Flavor: Google" | sed 's/.*zones\///') | |
| count=0 | |
| while true; do | |
| load=$(uptime | sed -e 's/.*load average: //g' | awk '{ print $3 }') | |
| if python3 -c "exit(0) if $load >= $threshold else exit(1)"; then | |
| echo "Resetting count ..." >&2 | |
| count=0 | |
| else | |
| ((count+=1)) | |
| echo "Idle #${count} at $load ..." >&2 | |
| fi | |
| if ((count>intervals)); then | |
| if who | grep -v tmux 1>&2; then | |
| echo "Someone is logged in, won't shut down, resetting count ..." >&2 | |
| else | |
| echo "Suspending ${VM_NAME} ..." >&2 | |
| gcloud beta compute instances suspend ${VM_NAME} --project ${VM_PROJECT} --zone ${VM_ZONE} | |
| fi | |
| count=0 | |
| fi | |
| sleep $sleep_time | |
| done |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
For the script to work, the VM must support suspend operation, see https://cloud.google.com/compute/docs/instances/suspend-resume-instance#limitations and https://cloud.google.com/compute/docs/images/os-details#user-space-features. Note suspend is not supported on Debian 10, but works on Debian 11.
To ssh into the VM, resume the VM via
gcloud beta compute instances resumeand ssh in (you can write a simple util to combine both actions).