Sdk Support for ami auto update

spotinst · Jan 16, 2024 · 6ca7b82 · 6ca7b82
1 parent 762a4c3
commit 6ca7b82
Show file tree

Hide file tree

Showing 2 changed files with 133 additions and 3 deletions.
diff --git a/examples/service/ocean/providers/aws/create/main.go b/examples/service/ocean/providers/aws/create/main.go
@@ -103,6 +103,27 @@ func main() {
 					},
 				},
 			},
+			Scheduling: &aws.Scheduling{
+				Tasks: []*aws.Task{
+					{
+						IsEnabled: spotinst.Bool(true),
+						Parameter: &aws.Parameter{
+							AmiAutoUpdate: &aws.AmiAutoUpdate{
+								ApplyRoll: spotinst.Bool(false),
+								ClusterRoll: &aws.ClusterRoll{
+									BatchMinHealthyPercentage: spotinst.Int(50),
+									BatchSizePercentage:       spotinst.Int(1),
+									Comment:                   spotinst.String("Test Description"),
+									RespectPdb:                spotinst.Bool(false),
+								},
+								MinorVersion: spotinst.Bool(true),
+								Patch:        spotinst.Bool(true),
+							},
+						},
+						Type: spotinst.String("amiAutoUpdate"),
+					},
+				},
+			},
 		},
 	})
 	if err != nil {

diff --git a/service/ocean/providers/aws/cluster.go b/service/ocean/providers/aws/cluster.go
@@ -119,9 +119,38 @@ type ShutdownHours struct {
 }
 
 type Task struct {
-	IsEnabled      *bool   `json:"isEnabled,omitempty"`
-	Type           *string `json:"taskType,omitempty"`
-	CronExpression *string `json:"cronExpression,omitempty"`
+	IsEnabled      *bool          `json:"isEnabled,omitempty"`
+	Type           *string        `json:"taskType,omitempty"`
+	CronExpression *string        `json:"cronExpression,omitempty"`
+	Parameter      *Parameter     `json:"parameters,omitempty"`
+	AmiAutoUpdate  *AmiAutoUpdate `json:"amiAutoUpdate,omitempty"`
+
+	forceSendFields []string
+	nullFields      []string
+}
+
+type Parameter struct {
+	AmiAutoUpdate *AmiAutoUpdate `json:"amiAutoUpdate,omitempty"`
+
+	forceSendFields []string
+	nullFields      []string
+}
+
+type AmiAutoUpdate struct {
+	ApplyRoll    *bool        `json:"applyRoll,omitempty"`
+	ClusterRoll  *ClusterRoll `json:"clusterRoll,omitempty"`
+	MinorVersion *bool        `json:"minorVersion,omitempty"`
+	Patch        *bool        `json:"patch,omitempty"`
+
+	forceSendFields []string
+	nullFields      []string
+}
+
+type ClusterRoll struct {
+	BatchMinHealthyPercentage *int    `json:"batchMinHealthyPercentage,omitempty"`
+	BatchSizePercentage       *int    `json:"batchSizePercentage,omitempty"`
+	Comment                   *string `json:"comment,omitempty"`
+	RespectPdb                *bool   `json:"respectPdb,omitempty"`
 
 	forceSendFields []string
 	nullFields      []string
@@ -1340,6 +1369,13 @@ func (o *Task) SetType(v *string) *Task {
 	return o
 }
 
+func (o *Parameter) SetAmiAutoUpdate(v *AmiAutoUpdate) *Parameter {
+	if o.AmiAutoUpdate = v; o.AmiAutoUpdate == nil {
+		o.nullFields = append(o.nullFields, "AmiAutoUpdate")
+	}
+	return o
+}
+
 func (o *Task) SetCronExpression(v *string) *Task {
 	if o.CronExpression = v; o.CronExpression == nil {
 		o.nullFields = append(o.nullFields, "CronExpression")
@@ -1349,6 +1385,79 @@ func (o *Task) SetCronExpression(v *string) *Task {
 
 // endregion
 
+// region Parameter
+
+func (o *Task) SetParameter(v *Parameter) *Task {
+	if o.Parameter = v; o.Parameter == nil {
+		o.nullFields = append(o.nullFields, "Parameter")
+	}
+	return o
+}
+
+// region AmiAutoUpdate
+
+func (o *AmiAutoUpdate) SetApplyRoll(v *bool) *AmiAutoUpdate {
+	if o.ApplyRoll = v; o.ApplyRoll == nil {
+		o.nullFields = append(o.nullFields, "ApplyRoll")
+	}
+	return o
+}
+
+func (o *AmiAutoUpdate) SetClusterRoll(v *ClusterRoll) *AmiAutoUpdate {
+	if o.ClusterRoll = v; o.ClusterRoll == nil {
+		o.nullFields = append(o.nullFields, "ClusterRoll")
+	}
+	return o
+}
+
+func (o *AmiAutoUpdate) SetMinorVersion(v *bool) *AmiAutoUpdate {
+	if o.MinorVersion = v; o.MinorVersion == nil {
+		o.nullFields = append(o.nullFields, "MinorVersion")
+	}
+	return o
+}
+
+func (o *AmiAutoUpdate) SetPatch(v *bool) *AmiAutoUpdate {
+	if o.Patch = v; o.Patch == nil {
+		o.nullFields = append(o.nullFields, "Patch")
+	}
+	return o
+}
+
+// endregion
+
+// region ClusterRoll
+
+func (o *ClusterRoll) SetBatchMinHealthyPercentage(v *int) *ClusterRoll {
+	if o.BatchMinHealthyPercentage = v; o.BatchMinHealthyPercentage == nil {
+		o.nullFields = append(o.nullFields, "BatchMinHealthyPercentage")
+	}
+	return o
+}
+
+func (o *ClusterRoll) SetBatchSizePercentage(v *int) *ClusterRoll {
+	if o.BatchSizePercentage = v; o.BatchSizePercentage == nil {
+		o.nullFields = append(o.nullFields, "BatchSizePercentage")
+	}
+	return o
+}
+
+func (o *ClusterRoll) SetComment(v *string) *ClusterRoll {
+	if o.Comment = v; o.Comment == nil {
+		o.nullFields = append(o.nullFields, "Comment")
+	}
+	return o
+}
+
+func (o *ClusterRoll) SetRespectPdb(v *bool) *ClusterRoll {
+	if o.RespectPdb = v; o.RespectPdb == nil {
+		o.nullFields = append(o.nullFields, "RespectPdb")
+	}
+	return o
+}
+
+// endregion
+
 // region ShutdownHours
 
 func (o ShutdownHours) MarshalJSON() ([]byte, error) {