Add support for InstanceTerminationAction.

Author: jwmay2012

api/v1beta1/gcpmachine_types.go

@@ -329,8 +329,21 @@ type GCPMachineSpec struct {
 	// RootDiskEncryptionKey defines the KMS key to be used to encrypt the root disk.
 	// +optional
 	RootDiskEncryptionKey *CustomerEncryptionKey `json:"rootDiskEncryptionKey,omitempty"`
+
+	// InstanceTerminationAction specifies the termination action for the instance upon preemption.
+	//  GCP API defaults to "Unspecified", which defaults the action to "Stop".
+	// +kubebuilder:validation:Enum=Delete;Stop
+	// +optional
+	InstanceTerminationAction *InstanceTerminationAction `json:"instanceTerminationAction,omitempty"`
 }
 
+type InstanceTerminationAction string
+
+const (
+	InstanceTerminationActionDelete InstanceTerminationAction = "Delete"
+	InstanceTerminationActionStop   InstanceTerminationAction = "Stop"
+)
+
 // MetadataItem defines a single piece of metadata associated with an instance.
 type MetadataItem struct {
 	// Key is the identifier for the metadata entry.

cloud/scope/machine.go

@@ -435,6 +435,16 @@ func (m *MachineScope) InstanceSpec(log logr.Logger) *compute.Instance {
 
 		instance.Scheduling.OnHostMaintenance = strings.ToUpper(string(*m.GCPMachine.Spec.OnHostMaintenance))
 	}
+	if m.GCPMachine.Spec.InstanceTerminationAction != nil {
+		switch *m.GCPMachine.Spec.InstanceTerminationAction {
+		case infrav1.InstanceTerminationActionDelete:
+			instance.Scheduling.InstanceTerminationAction = "DELETE"
+		case infrav1.InstanceTerminationActionStop:
+			instance.Scheduling.InstanceTerminationAction = "STOP"
+		default:
+			log.Error(errors.New("Invalid value"), "Unknown InstanceTerminationAction value", "Spec.InstanceTerminationAction", *m.GCPMachine.Spec.InstanceTerminationAction)
+		}
+	}
 	if m.GCPMachine.Spec.ConfidentialCompute != nil {
 		enabled := *m.GCPMachine.Spec.ConfidentialCompute == infrav1.ConfidentialComputePolicyEnabled
 		instance.ConfidentialInstanceConfig = &compute.ConfidentialInstanceConfig{

config/crd/bases/infrastructure.cluster.x-k8s.io_gcpmachines.yaml

@@ -422,6 +422,14 @@ spec:
                   Subnet is a reference to the subnetwork to use for this instance. If not specified,
                   the first subnetwork retrieved from the Cluster Region and Network is picked.
                 type: string
+              instanceTerminationAction:
+                description: |-
+                  InstanceTerminationAction specifies the termination action for the instance upon preemption.
+                  The default action is "Stop".
+                enum:
+                - Delete
+                - Stop
+                type: string
             required:
             - instanceType
             type: object

config/crd/bases/infrastructure.cluster.x-k8s.io_gcpmachinetemplates.yaml

@@ -439,6 +439,14 @@ spec:
                           Subnet is a reference to the subnetwork to use for this instance. If not specified,
                           the first subnetwork retrieved from the Cluster Region and Network is picked.
                         type: string
+                      instanceTerminationAction:
+                        description: |-
+                          InstanceTerminationAction specifies the termination action for the instance upon preemption.
+                          The default action is "Stop".
+                        enum:
+                        - Delete
+                        - Stop
+                        type: string
                     required:
                     - instanceType
                     type: object