src/pkg/gce/delete_old_vm.go

// Copyright 2022 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// Package gce contains high-level functionality for manipulating GCE resources.
package gce

import (
	"fmt"
	"time"

	"google.golang.org/api/compute/v1"
)

const timeLayout = "2006-01-02T15:04:05.999-07:00"

// For test overwriting.
var timeNow = time.Now

// DeleteOldVmWithLabel deletes all old VMs in the target project in the target zone with
// the given label. If the value of the label is empty, all VMs with the provided key of
// the label will be deleted. ttl must be at least 1 hour.
func DeleteOldVMWithLabel(gceService *compute.Service, project, zone, labelKey, labelValue string, ttl time.Duration) error {
	if project == "" || zone == "" || labelKey == "" {
		return fmt.Errorf("project name, zone, and labelKey cannot be empty. project: %s, zone: %s, labelKey: %s", project, zone, labelKey)
	}
	if ttl < time.Hour {
		return fmt.Errorf("ttl must be at least 1 hour, ttl: %v", ttl)
	}
	instancesListCall := gceService.Instances.List(project, zone)
	instancesList, err := instancesListCall.Do()
	if err != nil {
		return fmt.Errorf("failed to list instances in project %q in zone %q, err: %v", project, zone, err)
	}
	for _, instance := range instancesList.Items {
		if value, found := instance.Labels[labelKey]; found {
			if labelValue != "" && value != labelValue {
				continue
			}
			creationTime, err := time.Parse(timeLayout, instance.CreationTimestamp)
			if err != nil {
				return fmt.Errorf("failed to parse instanceCreationTimestamp %q, err: %v", instance.CreationTimestamp, err)
			}
			if timeNow().Before(creationTime.Add(ttl)) {
				continue
			}
			instancesDeleteCall := gceService.Instances.Delete(project, zone, instance.Name)
			if _, err := instancesDeleteCall.Do(); err != nil {
				return fmt.Errorf("failed to delete instance %q in project %q in zone %q, err: %v", instance.Name, project, zone, err)
			}
		}
	}
	return nil
}