providers/aws: add autoscalinggroup
This commit is contained in:
parent
16485ba3ca
commit
62a311a341
|
@ -0,0 +1,217 @@
|
||||||
|
package aws
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/hashicorp/terraform/flatmap"
|
||||||
|
"github.com/hashicorp/terraform/helper/diff"
|
||||||
|
"github.com/hashicorp/terraform/terraform"
|
||||||
|
"github.com/mitchellh/goamz/autoscaling"
|
||||||
|
)
|
||||||
|
|
||||||
|
func resource_aws_autoscaling_group_create(
|
||||||
|
s *terraform.ResourceState,
|
||||||
|
d *terraform.ResourceDiff,
|
||||||
|
meta interface{}) (*terraform.ResourceState, error) {
|
||||||
|
p := meta.(*ResourceProvider)
|
||||||
|
autoscalingconn := p.autoscalingconn
|
||||||
|
|
||||||
|
// Merge the diff into the state so that we have all the attributes
|
||||||
|
// properly.
|
||||||
|
rs := s.MergeDiff(d)
|
||||||
|
log.Println(rs.Attributes["availability_zones"])
|
||||||
|
|
||||||
|
var err error
|
||||||
|
autoScalingGroupOpts := autoscaling.CreateAutoScalingGroup{}
|
||||||
|
|
||||||
|
if rs.Attributes["min_size"] != "" {
|
||||||
|
autoScalingGroupOpts.MinSize, err = strconv.Atoi(rs.Attributes["min_size"])
|
||||||
|
}
|
||||||
|
|
||||||
|
if rs.Attributes["max_size"] != "" {
|
||||||
|
autoScalingGroupOpts.MaxSize, err = strconv.Atoi(rs.Attributes["max_size"])
|
||||||
|
}
|
||||||
|
|
||||||
|
if rs.Attributes["default_cooldown"] != "" {
|
||||||
|
autoScalingGroupOpts.DefaultCooldown, err = strconv.Atoi(rs.Attributes["default_cooldown"])
|
||||||
|
}
|
||||||
|
|
||||||
|
if rs.Attributes["desired_capicity"] != "" {
|
||||||
|
autoScalingGroupOpts.DesiredCapacity, err = strconv.Atoi(rs.Attributes["desired_capicity"])
|
||||||
|
}
|
||||||
|
|
||||||
|
if rs.Attributes["healthcheck_grace_period"] != "" {
|
||||||
|
autoScalingGroupOpts.HealthCheckGracePeriod, err = strconv.Atoi(rs.Attributes["healthcheck_grace_period"])
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("Error parsing configuration: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if rs.Attributes["availability_zones"] != "" {
|
||||||
|
autoScalingGroupOpts.AvailZone = expandStringList(flatmap.Expand(
|
||||||
|
rs.Attributes, "availability_zones").([]interface{}))
|
||||||
|
}
|
||||||
|
|
||||||
|
if rs.Attributes["load_balancers"] != "" {
|
||||||
|
autoScalingGroupOpts.LoadBalancerNames = expandStringList(flatmap.Expand(
|
||||||
|
rs.Attributes, "load_balancers").([]interface{}))
|
||||||
|
}
|
||||||
|
|
||||||
|
if rs.Attributes["vpc_identifier"] != "" {
|
||||||
|
autoScalingGroupOpts.VPCZoneIdentifier = expandStringList(flatmap.Expand(
|
||||||
|
rs.Attributes, "vpc_identifier").([]interface{}))
|
||||||
|
}
|
||||||
|
|
||||||
|
autoScalingGroupOpts.Name = rs.Attributes["name"]
|
||||||
|
autoScalingGroupOpts.HealthCheckType = rs.Attributes["health_check_type"]
|
||||||
|
autoScalingGroupOpts.LaunchConfigurationName = rs.Attributes["launch_configuration"]
|
||||||
|
|
||||||
|
log.Printf("[DEBUG] autoscaling Group create configuration: %#v", autoScalingGroupOpts)
|
||||||
|
_, err = autoscalingconn.CreateAutoScalingGroup(&autoScalingGroupOpts)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("Error creating autoscaling Group: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
rs.ID = rs.Attributes["name"]
|
||||||
|
|
||||||
|
log.Printf("[INFO] autoscaling Group ID: %s", rs.ID)
|
||||||
|
|
||||||
|
g, err := resource_aws_autoscaling_group_retrieve(rs.ID, autoscalingconn)
|
||||||
|
if err != nil {
|
||||||
|
return rs, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return resource_aws_autoscaling_group_update_state(rs, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
func resource_aws_autoscaling_group_update(
|
||||||
|
s *terraform.ResourceState,
|
||||||
|
d *terraform.ResourceDiff,
|
||||||
|
meta interface{}) (*terraform.ResourceState, error) {
|
||||||
|
|
||||||
|
rs := s.MergeDiff(d)
|
||||||
|
log.Printf("ResourceDiff: %s", d)
|
||||||
|
log.Printf("ResourceState: %s", s)
|
||||||
|
log.Printf("Merged: %s", rs)
|
||||||
|
|
||||||
|
return nil, fmt.Errorf("Did not update")
|
||||||
|
}
|
||||||
|
|
||||||
|
func resource_aws_autoscaling_group_destroy(
|
||||||
|
s *terraform.ResourceState,
|
||||||
|
meta interface{}) error {
|
||||||
|
// p := meta.(*ResourceProvider)
|
||||||
|
// autoscalingconn := p.autoscalingconn
|
||||||
|
|
||||||
|
log.Printf("[DEBUG] autoscaling Group destroy: %v", s.ID)
|
||||||
|
|
||||||
|
// err := nil
|
||||||
|
|
||||||
|
// _, err := autoscalingconn.DeleteAutoScalingGroup(autoscaling.autoscalingGroup{Id: s.ID})
|
||||||
|
|
||||||
|
// if err != nil {
|
||||||
|
// autoscalingerr, ok := err.(*autoscaling.Error)
|
||||||
|
// if ok && autoscalingerr.Code == "InvalidGroup.NotFound" {
|
||||||
|
// return nil
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func resource_aws_autoscaling_group_refresh(
|
||||||
|
s *terraform.ResourceState,
|
||||||
|
meta interface{}) (*terraform.ResourceState, error) {
|
||||||
|
p := meta.(*ResourceProvider)
|
||||||
|
autoscalingconn := p.autoscalingconn
|
||||||
|
|
||||||
|
g, err := resource_aws_autoscaling_group_retrieve(s.ID, autoscalingconn)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return s, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return resource_aws_autoscaling_group_update_state(s, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
func resource_aws_autoscaling_group_diff(
|
||||||
|
s *terraform.ResourceState,
|
||||||
|
c *terraform.ResourceConfig,
|
||||||
|
meta interface{}) (*terraform.ResourceDiff, error) {
|
||||||
|
|
||||||
|
b := &diff.ResourceBuilder{
|
||||||
|
Attrs: map[string]diff.AttrType{
|
||||||
|
"min_size": diff.AttrTypeCreate,
|
||||||
|
"max_size": diff.AttrTypeCreate,
|
||||||
|
"default_cooldown": diff.AttrTypeCreate,
|
||||||
|
"name": diff.AttrTypeCreate,
|
||||||
|
"desired_capicity": diff.AttrTypeCreate,
|
||||||
|
"health_check_grace_period": diff.AttrTypeCreate,
|
||||||
|
"health_check_type": diff.AttrTypeCreate,
|
||||||
|
"launch_configuration": diff.AttrTypeCreate,
|
||||||
|
"vpc_zone_identifier": diff.AttrTypeCreate,
|
||||||
|
"load_balancers": diff.AttrTypeCreate,
|
||||||
|
"availability_zones": diff.AttrTypeCreate,
|
||||||
|
},
|
||||||
|
|
||||||
|
ComputedAttrs: []string{},
|
||||||
|
}
|
||||||
|
|
||||||
|
return b.Diff(s, c)
|
||||||
|
}
|
||||||
|
|
||||||
|
func resource_aws_autoscaling_group_update_state(
|
||||||
|
s *terraform.ResourceState,
|
||||||
|
g *autoscaling.AutoScalingGroup) (*terraform.ResourceState, error) {
|
||||||
|
|
||||||
|
s.Attributes["min_size"] = strconv.Itoa(g.MinSize)
|
||||||
|
s.Attributes["max_size"] = strconv.Itoa(g.MaxSize)
|
||||||
|
s.Attributes["default_cooldown"] = strconv.Itoa(g.DefaultCooldown)
|
||||||
|
s.Attributes["name"] = g.Name
|
||||||
|
s.Attributes["desired_capacity"] = strconv.Itoa(g.DesiredCapacity)
|
||||||
|
s.Attributes["health_check_grace_period"] = strconv.Itoa(g.HealthCheckGracePeriod)
|
||||||
|
s.Attributes["health_check_type"] = g.HealthCheckType
|
||||||
|
s.Attributes["launch_configuration"] = g.LaunchConfigurationName
|
||||||
|
s.Attributes["vpc_zone_identifier"] = g.VPCZoneIdentifier
|
||||||
|
|
||||||
|
// Flatten our sg values
|
||||||
|
toFlatten := make(map[string]interface{})
|
||||||
|
toFlatten["load_balancers"] = flattenLoadBalancers(g.LoadBalancerNames)
|
||||||
|
toFlatten["availability_zones"] = flattenAvailabilityZones(g.AvailabilityZones)
|
||||||
|
|
||||||
|
for k, v := range flatmap.Flatten(toFlatten) {
|
||||||
|
s.Attributes[k] = v
|
||||||
|
}
|
||||||
|
|
||||||
|
return s, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns a single group by it's ID
|
||||||
|
func resource_aws_autoscaling_group_retrieve(id string, autoscalingconn *autoscaling.AutoScaling) (*autoscaling.AutoScalingGroup, error) {
|
||||||
|
describeOpts := autoscaling.DescribeAutoScalingGroups{
|
||||||
|
Names: []string{id},
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Printf("[DEBUG] autoscaling Group describe configuration: %#v", describeOpts)
|
||||||
|
|
||||||
|
describeGroups, err := autoscalingconn.DescribeAutoScalingGroups(&describeOpts)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("Error retrieving autoscaling groups: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify AWS returned our sg
|
||||||
|
if len(describeGroups.AutoScalingGroups) != 1 ||
|
||||||
|
describeGroups.AutoScalingGroups[0].Name != id {
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("Unable to find autoscaling group: %#v", describeGroups.AutoScalingGroups)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
g := describeGroups.AutoScalingGroups[0]
|
||||||
|
|
||||||
|
return &g, nil
|
||||||
|
}
|
|
@ -6,6 +6,7 @@ import (
|
||||||
"github.com/hashicorp/terraform/helper/config"
|
"github.com/hashicorp/terraform/helper/config"
|
||||||
"github.com/hashicorp/terraform/helper/multierror"
|
"github.com/hashicorp/terraform/helper/multierror"
|
||||||
"github.com/hashicorp/terraform/terraform"
|
"github.com/hashicorp/terraform/terraform"
|
||||||
|
"github.com/mitchellh/goamz/autoscaling"
|
||||||
"github.com/mitchellh/goamz/ec2"
|
"github.com/mitchellh/goamz/ec2"
|
||||||
"github.com/mitchellh/goamz/elb"
|
"github.com/mitchellh/goamz/elb"
|
||||||
)
|
)
|
||||||
|
@ -15,6 +16,7 @@ type ResourceProvider struct {
|
||||||
|
|
||||||
ec2conn *ec2.EC2
|
ec2conn *ec2.EC2
|
||||||
elbconn *elb.ELB
|
elbconn *elb.ELB
|
||||||
|
autoscalingconn *autoscaling.AutoScaling
|
||||||
}
|
}
|
||||||
|
|
||||||
func (p *ResourceProvider) Validate(c *terraform.ResourceConfig) ([]string, []error) {
|
func (p *ResourceProvider) Validate(c *terraform.ResourceConfig) ([]string, []error) {
|
||||||
|
@ -51,6 +53,8 @@ func (p *ResourceProvider) Configure(c *terraform.ResourceConfig) error {
|
||||||
p.ec2conn = ec2.New(auth, region)
|
p.ec2conn = ec2.New(auth, region)
|
||||||
log.Println("Initializing ELB connection")
|
log.Println("Initializing ELB connection")
|
||||||
p.elbconn = elb.New(auth, region)
|
p.elbconn = elb.New(auth, region)
|
||||||
|
log.Println("Initializing AutoScaling connection")
|
||||||
|
p.autoscalingconn = autoscaling.New(auth, region)
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(errs) > 0 {
|
if len(errs) > 0 {
|
||||||
|
|
|
@ -12,6 +12,13 @@ var resourceMap *resource.Map
|
||||||
func init() {
|
func init() {
|
||||||
resourceMap = &resource.Map{
|
resourceMap = &resource.Map{
|
||||||
Mapping: map[string]resource.Resource{
|
Mapping: map[string]resource.Resource{
|
||||||
|
"aws_autoscaling_group": resource.Resource{
|
||||||
|
Create: resource_aws_autoscaling_group_create,
|
||||||
|
Destroy: resource_aws_autoscaling_group_destroy,
|
||||||
|
Diff: resource_aws_autoscaling_group_diff,
|
||||||
|
Refresh: resource_aws_autoscaling_group_refresh,
|
||||||
|
},
|
||||||
|
|
||||||
"aws_elb": resource.Resource{
|
"aws_elb": resource.Resource{
|
||||||
Create: resource_aws_elb_create,
|
Create: resource_aws_elb_create,
|
||||||
Update: resource_aws_elb_update,
|
Update: resource_aws_elb_update,
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
package aws
|
package aws
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"github.com/mitchellh/goamz/autoscaling"
|
||||||
"github.com/mitchellh/goamz/ec2"
|
"github.com/mitchellh/goamz/ec2"
|
||||||
"github.com/mitchellh/goamz/elb"
|
"github.com/mitchellh/goamz/elb"
|
||||||
)
|
)
|
||||||
|
@ -92,6 +93,24 @@ func flattenSecurityGroups(list []ec2.UserSecurityGroup) []string {
|
||||||
return result
|
return result
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Flattens an array of AvailabilityZones into a []string
|
||||||
|
func flattenAvailabilityZones(list []autoscaling.AvailabilityZone) []string {
|
||||||
|
result := make([]string, 0, len(list))
|
||||||
|
for _, g := range list {
|
||||||
|
result = append(result, g.AvailabilityZone)
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
// Flattens an array of LoadBalancerName into a []string
|
||||||
|
func flattenLoadBalancers(list []autoscaling.LoadBalancerName) []string {
|
||||||
|
result := make([]string, 0, len(list))
|
||||||
|
for _, g := range list {
|
||||||
|
result = append(result, g.LoadBalancerName)
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
// Takes the result of flatmap.Expand for an array of strings
|
// Takes the result of flatmap.Expand for an array of strings
|
||||||
// and returns a []string
|
// and returns a []string
|
||||||
func expandStringList(configured []interface{}) []string {
|
func expandStringList(configured []interface{}) []string {
|
||||||
|
|
Loading…
Reference in New Issue