Added support for RestartReplicas, `ExcludePreliminaryRecommendatio…

…ns` and `RecommendationApplicationHPA` (#319)
spotinst · Aug 27, 2024 · 11a0d18 · 11a0d18
1 parent be777e5
commit 11a0d18
Show file tree

Hide file tree

Showing 3 changed files with 49 additions and 9 deletions.
diff --git a/examples/service/ocean/right_sizing/create/main.go b/examples/service/ocean/right_sizing/create/main.go
@@ -30,9 +30,13 @@ func main() {
 	// Create a new right sizing rule.
 	out, err := svc.RightSizing().CreateRightsizingRule(ctx, &right_sizing.CreateRightsizingRuleInput{
 		RightsizingRule: &right_sizing.RightsizingRule{
-			RuleName:    spotinst.String("test-rule1"),
-			OceanId:     spotinst.String("o-12ab34"),
-			RestartPods: spotinst.Bool(true),
+			RuleName:                          spotinst.String("test-rule1"),
+			OceanId:                           spotinst.String("o-12ab34"),
+			ExcludePreliminaryRecommendations: spotinst.Bool(true),
+			RestartReplicas:                   spotinst.String("MORE_THAN_ONE_REPLICA"),
+			RecommendationApplicationHPA: &right_sizing.RecommendationApplicationHPA{
+				AllowHPARecommendations: spotinst.Bool(true),
+			},
 			RecommendationApplicationIntervals: []*right_sizing.RecommendationApplicationIntervals{
 				{
 					RepetitionBasis: spotinst.String("MONTHLY"),

diff --git a/examples/service/ocean/right_sizing/update/main.go b/examples/service/ocean/right_sizing/update/main.go
@@ -31,8 +31,7 @@ func main() {
 	out, err := svc.RightSizing().UpdateRightsizingRule(ctx, &right_sizing.UpdateRightsizingRuleInput{
 		RuleName: spotinst.String("test-rule1"),
 		RightsizingRule: &right_sizing.RightsizingRule{
-			OceanId:     spotinst.String("o-12ab34"),
-			RestartPods: spotinst.Bool(true),
+			OceanId: spotinst.String("o-12ab34"),
 			RecommendationApplicationIntervals: []*right_sizing.RecommendationApplicationIntervals{
 				{
 					RepetitionBasis: spotinst.String("MONTHLY"),

diff --git a/service/ocean/right_sizing/right_sizing.go b/service/ocean/right_sizing/right_sizing.go
@@ -14,11 +14,13 @@ import (
 type RightsizingRule struct {
 	RuleName                                *string                                  `json:"ruleName,omitempty"`
 	OceanId                                 *string                                  `json:"oceanId,omitempty"`
-	RestartPods                             *bool                                    `json:"restartPods,omitempty"`
+	RestartReplicas                         *string                                  `json:"restartReplicas,omitempty"`
+	ExcludePreliminaryRecommendations       *bool                                    `json:"excludePreliminaryRecommendations,omitempty"`
 	RecommendationApplicationIntervals      []*RecommendationApplicationIntervals    `json:"recommendationApplicationIntervals,omitempty"`
 	RecommendationApplicationMinThreshold   *RecommendationApplicationMinThreshold   `json:"recommendationApplicationMinThreshold,omitempty"`
 	RecommendationApplicationBoundaries     *RecommendationApplicationBoundaries     `json:"recommendationApplicationBoundaries,omitempty"`
 	RecommendationApplicationOverheadValues *RecommendationApplicationOverheadValues `json:"recommendationApplicationOverheadValues,omitempty"`
+	RecommendationApplicationHPA            *RecommendationApplicationHPA            `json:"recommendationApplicationHPA,omitempty"`
 
 	forceSendFields []string
 	nullFields      []string
@@ -123,6 +125,13 @@ type RecommendationApplicationOverheadValues struct {
 	nullFields      []string
 }
 
+type RecommendationApplicationHPA struct {
+	AllowHPARecommendations *bool `json:"allowHPARecommendations,omitempty"`
+
+	forceSendFields []string
+	nullFields      []string
+}
+
 type ListRightsizingRulesInput struct {
 	OceanId *string `json:"oceanId,omitempty"`
 }
@@ -414,9 +423,16 @@ func (o *RightsizingRule) SetOceanId(v *string) *RightsizingRule {
 	return o
 }
 
-func (o *RightsizingRule) SetRestartPods(v *bool) *RightsizingRule {
-	if o.RestartPods = v; o.RestartPods == nil {
-		o.nullFields = append(o.nullFields, "RestartPods")
+func (o *RightsizingRule) SetRestartReplicas(v *string) *RightsizingRule {
+	if o.RestartReplicas = v; o.RestartReplicas == nil {
+		o.nullFields = append(o.nullFields, "RestartReplicas")
+	}
+	return o
+}
+
+func (o *RightsizingRule) SetExcludePreliminaryRecommendations(v *bool) *RightsizingRule {
+	if o.ExcludePreliminaryRecommendations = v; o.ExcludePreliminaryRecommendations == nil {
+		o.nullFields = append(o.nullFields, "ExcludePreliminaryRecommendations")
 	}
 	return o
 }
@@ -449,6 +465,13 @@ func (o *RightsizingRule) SetRecommendationApplicationOverheadValues(v *Recommen
 	return o
 }
 
+func (o *RightsizingRule) SetRecommendationApplicationHPA(v *RecommendationApplicationHPA) *RightsizingRule {
+	if o.RecommendationApplicationHPA = v; o.RecommendationApplicationHPA == nil {
+		o.nullFields = append(o.nullFields, "RecommendationApplicationHPA")
+	}
+	return o
+}
+
 // region RecommendationApplicationInterval
 
 func (o RecommendationApplicationIntervals) MarshalJSON() ([]byte, error) {
@@ -636,3 +659,17 @@ func (o *RecommendationApplicationOverheadValues) SetOverheadMemoryPercentage(v
 	}
 	return o
 }
+
+// region RecommendationApplicationHPA
+func (o RecommendationApplicationHPA) MarshalJSON() ([]byte, error) {
+	type noMethod RecommendationApplicationHPA
+	raw := noMethod(o)
+	return jsonutil.MarshalJSON(raw, o.forceSendFields, o.nullFields)
+}
+
+func (o *RecommendationApplicationHPA) SetAllowHPARecommendations(v *bool) *RecommendationApplicationHPA {
+	if o.AllowHPARecommendations = v; o.AllowHPARecommendations == nil {
+		o.nullFields = append(o.nullFields, "AllowHPARecommendations")
+	}
+	return o
+}