adding fields to specify that we should autoscale a cluster.

Azure · Jul 10, 2018 · 0b17d3f · 0b17d3f
1 parent acac8b0
commit 0b17d3f
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 0 deletions.
diff --git a/pkg/api/agentPoolOnlyApi/v20180331/types.go b/pkg/api/agentPoolOnlyApi/v20180331/types.go
@@ -183,6 +183,9 @@ type AgentPoolProfile struct {
 	StorageProfile string `json:"storageProfile" validate:"eq=ManagedDisks|len=0"`
 	VnetSubnetID   string `json:"vnetSubnetID,omitempty"`
 	MaxPods        *int   `json:"maxPods,omitempty"`
+	MaxCount       *int   `json:"maxCount,omitempty"`
+	MinCount       *int   `json:"minCount,omitempty"`
+	AutoscalePool  bool   `json:"autoscalePool,omitempty"`
 
 	// OSType is the operating system type for agents
 	// Set as nullable to support backward compat because

diff --git a/pkg/api/agentPoolOnlyApi/vlabs/types.go b/pkg/api/agentPoolOnlyApi/vlabs/types.go
@@ -129,6 +129,9 @@ type AgentPoolProfile struct {
 	AvailabilityProfile string `json:"availabilityProfile"`
 	StorageProfile      string `json:"storageProfile" validate:"eq=StorageAccount|eq=ManagedDisks|len=0"`
 	VnetSubnetID        string `json:"vnetSubnetID,omitempty"`
+	MaxCount            *int   `json:"maxCount,omitempty"`
+	MinCount            *int   `json:"minCount,omitempty"`
+	AutoscalePool       bool   `json:"autoscalePool,omitempty"`
 
 	// OSType is the operating system type for agents
 	// Set as nullable to support backward compat because

diff --git a/pkg/api/converterfromagentpoolonlyapi.go b/pkg/api/converterfromagentpoolonlyapi.go
@@ -251,6 +251,9 @@ func convertAgentPoolProfileToV20180331AgentPoolOnly(api *AgentPoolProfile, p *v
 			p.MaxPods = &agentPoolMaxPods
 		}
 	}
+	p.AutoscalePool = api.AutoscalePool
+	p.MaxCount = api.MaxCount
+	p.MinCount = api.MinCount
 }
 
 func convertServicePrincipalProfileToV20180331AgentPoolOnly(api *ServicePrincipalProfile, v20180331 *v20180331.ServicePrincipalProfile) {

diff --git a/pkg/api/convertertoagentpoolonlyapi.go b/pkg/api/convertertoagentpoolonlyapi.go
@@ -270,6 +270,9 @@ func convertVLabsAgentPoolOnlyAgentPoolProfile(vlabs *vlabs.AgentPoolProfile, ap
 	api.AvailabilityProfile = vlabs.AvailabilityProfile
 	api.VnetSubnetID = vlabs.VnetSubnetID
 	api.Subnet = vlabs.GetSubnet()
+	api.AutoscalePool = vlabs.AutoscalePool
+	api.MinCount = vlabs.MinCount
+	api.MaxCount = vlabs.MaxCount
 }
 
 func convertVLabsAgentPoolOnlyServicePrincipalProfile(vlabs *vlabs.ServicePrincipalProfile, api *ServicePrincipalProfile) {
@@ -502,6 +505,9 @@ func convertV20180331AgentPoolOnlyAgentPoolProfile(agentPoolProfile *v20180331.A
 	api.KubernetesConfig = kubernetesConfig
 	api.Subnet = agentPoolProfile.GetSubnet()
 	api.AvailabilityProfile = availabilityProfile
+	api.MaxCount = agentPoolProfile.MaxCount
+	api.MinCount = agentPoolProfile.MinCount
+	api.AutoscalePool = agentPoolProfile.AutoscalePool
 	return api
 }
 

diff --git a/pkg/api/types.go b/pkg/api/types.go
@@ -449,6 +449,9 @@ type AgentPoolProfile struct {
 	Extensions                   []Extension          `json:"extensions"`
 	KubernetesConfig             *KubernetesConfig    `json:"kubernetesConfig,omitempty"`
 	ImageRef                     *ImageReference      `json:"imageReference,omitempty"`
+	MaxCount                     *int                 `json:"maxCount,omitempty"`
+	MinCount                     *int                 `json:"minCount,omitempty"`
+	AutoscalePool                bool                 `json:"autoscalePool,omitempty"`
 }
 
 // AgentPoolProfileRole represents an agent role