132 lines
3.4 KiB
Go
132 lines
3.4 KiB
Go
package aws
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/hashicorp/terraform/helper/resource"
|
|
"github.com/hashicorp/terraform/helper/schema"
|
|
)
|
|
|
|
// waitForASGCapacityTimeout gathers the current numbers of healthy instances
|
|
// in the ASG and its attached ELBs and yields these numbers to a
|
|
// capacitySatifiedFunction. Loops for up to wait_for_capacity_timeout until
|
|
// the capacitySatisfiedFunc returns true.
|
|
//
|
|
// See "Waiting for Capacity" in docs for more discussion of the feature.
|
|
func waitForASGCapacity(
|
|
d *schema.ResourceData,
|
|
meta interface{},
|
|
satisfiedFunc capacitySatisfiedFunc) error {
|
|
wait, err := time.ParseDuration(d.Get("wait_for_capacity_timeout").(string))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if wait == 0 {
|
|
log.Printf("[DEBUG] Capacity timeout set to 0, skipping capacity waiting.")
|
|
return nil
|
|
}
|
|
|
|
log.Printf("[DEBUG] Waiting on %s for capacity...", d.Id())
|
|
|
|
return resource.Retry(wait, func() error {
|
|
g, err := getAwsAutoscalingGroup(d.Id(), meta.(*AWSClient).autoscalingconn)
|
|
if err != nil {
|
|
return resource.RetryError{Err: err}
|
|
}
|
|
if g == nil {
|
|
log.Printf("[INFO] Autoscaling Group %q not found", d.Id())
|
|
d.SetId("")
|
|
return nil
|
|
}
|
|
lbis, err := getLBInstanceStates(g, meta)
|
|
if err != nil {
|
|
return resource.RetryError{Err: err}
|
|
}
|
|
|
|
haveASG := 0
|
|
haveELB := 0
|
|
|
|
for _, i := range g.Instances {
|
|
if i.HealthStatus == nil || i.InstanceId == nil || i.LifecycleState == nil {
|
|
continue
|
|
}
|
|
|
|
if !strings.EqualFold(*i.HealthStatus, "Healthy") {
|
|
continue
|
|
}
|
|
|
|
if !strings.EqualFold(*i.LifecycleState, "InService") {
|
|
continue
|
|
}
|
|
|
|
haveASG++
|
|
|
|
inAllLbs := true
|
|
for _, states := range lbis {
|
|
state, ok := states[*i.InstanceId]
|
|
if !ok || !strings.EqualFold(state, "InService") {
|
|
inAllLbs = false
|
|
}
|
|
}
|
|
if inAllLbs {
|
|
haveELB++
|
|
}
|
|
}
|
|
|
|
satisfied, reason := satisfiedFunc(d, haveASG, haveELB)
|
|
|
|
log.Printf("[DEBUG] %q Capacity: %d ASG, %d ELB, satisfied: %t, reason: %q",
|
|
d.Id(), haveASG, haveELB, satisfied, reason)
|
|
|
|
if satisfied {
|
|
return nil
|
|
}
|
|
|
|
return fmt.Errorf("%q: Waiting up to %s: %s", d.Id(), wait, reason)
|
|
})
|
|
}
|
|
|
|
type capacitySatisfiedFunc func(*schema.ResourceData, int, int) (bool, string)
|
|
|
|
// capacitySatifiedCreate treats all targets as minimums
|
|
func capacitySatifiedCreate(d *schema.ResourceData, haveASG, haveELB int) (bool, string) {
|
|
minASG := d.Get("min_size").(int)
|
|
if wantASG := d.Get("desired_capacity").(int); wantASG > 0 {
|
|
minASG = wantASG
|
|
}
|
|
if haveASG < minASG {
|
|
return false, fmt.Sprintf(
|
|
"Need at least %d healthy instances in ASG, have %d", minASG, haveASG)
|
|
}
|
|
minELB := d.Get("min_elb_capacity").(int)
|
|
if wantELB := d.Get("wait_for_elb_capacity").(int); wantELB > 0 {
|
|
minELB = wantELB
|
|
}
|
|
if haveELB < minELB {
|
|
return false, fmt.Sprintf(
|
|
"Need at least %d healthy instances in ELB, have %d", minELB, haveELB)
|
|
}
|
|
return true, ""
|
|
}
|
|
|
|
// capacitySatifiedUpdate only cares about specific targets
|
|
func capacitySatifiedUpdate(d *schema.ResourceData, haveASG, haveELB int) (bool, string) {
|
|
if wantASG := d.Get("desired_capacity").(int); wantASG > 0 {
|
|
if haveASG != wantASG {
|
|
return false, fmt.Sprintf(
|
|
"Need exactly %d healthy instances in ASG, have %d", wantASG, haveASG)
|
|
}
|
|
}
|
|
if wantELB := d.Get("wait_for_elb_capacity").(int); wantELB > 0 {
|
|
if haveELB != wantELB {
|
|
return false, fmt.Sprintf(
|
|
"Need exactly %d healthy instances in ELB, have %d", wantELB, haveELB)
|
|
}
|
|
}
|
|
return true, ""
|
|
}
|