Add support for InstanceTerminationAction for instance preemption.

api/v1beta1/gcpmachine_types.go

@@ -329,8 +329,24 @@ 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"`
 }
 
+// InstanceTerminationAction is a type for specifying to Delete or Stop a VM upon preemption.
+type InstanceTerminationAction string
+
+const (
+	// InstanceTerminationActionDelete means to delete an instance upon preemption termination.
+	InstanceTerminationActionDelete InstanceTerminationAction = "Delete"
+	// InstanceTerminationActionStop means to stop an instance upon preemption termination.
+	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

@@ -207,6 +207,14 @@ spec:
                 description: ImageFamily is the full reference to a valid image family
                   to be used for this machine.
                 type: string
+              instanceTerminationAction:
+                description: |-
+                  InstanceTerminationAction specifies the termination action for the instance upon preemption.
+                   GCP API defaults to "Unspecified", which defaults the action to "Stop".
+                enum:
+                - Delete
+                - Stop
+                type: string
               instanceType:
                 description: 'InstanceType is the type of instance to create. Example:
                   n1.standard-2'

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

@@ -222,6 +222,14 @@ spec:
                         description: ImageFamily is the full reference to a valid
                           image family to be used for this machine.
                         type: string
+                      instanceTerminationAction:
+                        description: |-
+                          InstanceTerminationAction specifies the termination action for the instance upon preemption.
+                           GCP API defaults to "Unspecified", which defaults the action to "Stop".
+                        enum:
+                        - Delete
+                        - Stop
+                        type: string
                       instanceType:
                         description: 'InstanceType is the type of instance to create.
                           Example: n1.standard-2'