Feat (Ocean/AWS) : Added support for cluster_orientation field in str…

…ategy (#218)
spotinst · Jan 16, 2023 · dbbcb3f · dbbcb3f
1 parent a9f7b82
commit dbbcb3f
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 8 deletions.
diff --git a/examples/service/ocean/providers/aws/create/main.go b/examples/service/ocean/providers/aws/create/main.go
@@ -39,6 +39,9 @@ func main() {
 			Strategy: &aws.Strategy{
 				SpotPercentage:     spotinst.Float64(100),
 				FallbackToOnDemand: spotinst.Bool(true),
+				ClusterOrientation: &aws.ClusterOrientation{
+					AvailabilityVsCost: spotinst.String("cheapest"),
+				},
 			},
 			Compute: &aws.Compute{
 				InstanceTypes: &aws.InstanceTypes{

diff --git a/service/ocean/providers/aws/cluster.go b/service/ocean/providers/aws/cluster.go
@@ -48,15 +48,39 @@ type Cluster struct {
 }
 
 type Strategy struct {
-	SpotPercentage           *float64 `json:"spotPercentage,omitempty"`
-	UtilizeReservedInstances *bool    `json:"utilizeReservedInstances,omitempty"`
-	FallbackToOnDemand       *bool    `json:"fallbackToOd,omitempty"`
-	DrainingTimeout          *int     `json:"drainingTimeout,omitempty"`
-	GracePeriod              *int     `json:"gracePeriod,omitempty"`
-	UtilizeCommitments       *bool    `json:"utilizeCommitments,omitempty"`
+	SpotPercentage           *float64            `json:"spotPercentage,omitempty"`
+	UtilizeReservedInstances *bool               `json:"utilizeReservedInstances,omitempty"`
+	FallbackToOnDemand       *bool               `json:"fallbackToOd,omitempty"`
+	DrainingTimeout          *int                `json:"drainingTimeout,omitempty"`
+	GracePeriod              *int                `json:"gracePeriod,omitempty"`
+	UtilizeCommitments       *bool               `json:"utilizeCommitments,omitempty"`
+	ClusterOrientation       *ClusterOrientation `json:"clusterOrientation,omitempty"`
+	forceSendFields          []string
+	nullFields               []string
+}
+type ClusterOrientation struct {
+	AvailabilityVsCost *string `json:"availabilityVsCost,omitempty"`
+	forceSendFields    []string
+	nullFields         []string
+}
 
-	forceSendFields []string
-	nullFields      []string
+func (o *ClusterOrientation) SetAvailabilityVsCost(v *string) *ClusterOrientation {
+	if o.AvailabilityVsCost = v; o.AvailabilityVsCost == nil {
+		o.nullFields = append(o.nullFields, "AvailabilityVsCost")
+	}
+	return o
+}
+
+func (o *Strategy) SetClusterOrientation(v *ClusterOrientation) *Strategy {
+	if o.ClusterOrientation = v; o.ClusterOrientation == nil {
+		o.nullFields = append(o.nullFields, "ClusterOrientation")
+	}
+	return o
+}
+func (o ClusterOrientation) MarshalJSON() ([]byte, error) {
+	type noMethod ClusterOrientation
+	raw := noMethod(o)
+	return jsonutil.MarshalJSON(raw, o.forceSendFields, o.nullFields)
 }
 
 type Capacity struct {