Skip to content

Commit

Permalink
provider/aws: Finish converting Route53 to aws-sdk-go
Browse files Browse the repository at this point in the history
This migrates Route53 Record to the new sdk and removes goamz/route53.
  • Loading branch information
catsby committed Feb 17, 2015
1 parent 9fe5e13 commit 1b4fd28
Show file tree
Hide file tree
Showing 5 changed files with 120 additions and 89 deletions.
11 changes: 3 additions & 8 deletions builtin/providers/aws/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,11 +12,10 @@ import (
"github.com/mitchellh/goamz/ec2"
"github.com/mitchellh/goamz/elb"
"github.com/mitchellh/goamz/rds"
"github.com/mitchellh/goamz/route53"
"github.com/mitchellh/goamz/s3"

awsGo "github.com/awslabs/aws-sdk-go/aws"
awsr53 "github.com/awslabs/aws-sdk-go/gen/route53"
"github.com/awslabs/aws-sdk-go/gen/route53"
)

type Config struct {
Expand All @@ -31,8 +30,7 @@ type AWSClient struct {
autoscalingconn *autoscaling.AutoScaling
s3conn *s3.S3
rdsconn *rds.Rds
route53 *route53.Route53
awsr53Conn *awsr53.Route53
r53conn *route53.Route53
}

// Client configures and returns a fully initailized AWSClient
Expand Down Expand Up @@ -66,15 +64,12 @@ func (c *Config) Client() (interface{}, error) {
log.Println("[INFO] Initializing RDS connection")
client.rdsconn = rds.New(auth, region)
log.Println("[INFO] Initializing Route53 connection")
client.route53 = route53.New(auth, region)

log.Println("[INFO] Initializing aws-go Route53 connection")
creds := awsGo.Creds(c.AccessKey, c.SecretKey, "")

// aws-sdk-go uses v4 for signing requests, which requires all global
// endpoints to use 'us-east-1'.
// See http://docs.aws.amazon.com/general/latest/gr/sigv4_changes.html
client.awsr53Conn = awsr53.New(creds, "us-east-1", nil)
client.r53conn = route53.New(creds, "us-east-1", nil)
}

if len(errs) > 0 {
Expand Down
108 changes: 71 additions & 37 deletions builtin/providers/aws/resource_aws_route53_record.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,9 @@ import (
"github.com/hashicorp/terraform/helper/hashcode"
"github.com/hashicorp/terraform/helper/resource"
"github.com/hashicorp/terraform/helper/schema"
"github.com/mitchellh/goamz/route53"

"github.com/awslabs/aws-sdk-go/aws"
"github.com/awslabs/aws-sdk-go/gen/route53"
)

func resourceAwsRoute53Record() *schema.Resource {
Expand Down Expand Up @@ -57,11 +59,10 @@ func resourceAwsRoute53Record() *schema.Resource {
}

func resourceAwsRoute53RecordCreate(d *schema.ResourceData, meta interface{}) error {
conn := meta.(*AWSClient).route53

conn := meta.(*AWSClient).r53conn
zone := d.Get("zone_id").(string)

zoneRecord, err := conn.GetHostedZone(zone)
zoneRecord, err := conn.GetHostedZone(&route53.GetHostedZoneRequest{ID: aws.String(zone)})
if err != nil {
return err
}
Expand All @@ -70,7 +71,7 @@ func resourceAwsRoute53RecordCreate(d *schema.ResourceData, meta interface{}) er
// If it does not, add the zone name to form a fully qualified name
// and keep AWS happy.
recordName := d.Get("name").(string)
zoneName := strings.Trim(zoneRecord.HostedZone.Name, ".")
zoneName := strings.Trim(*zoneRecord.HostedZone.Name, ".")
if !strings.HasSuffix(recordName, zoneName) {
d.Set("name", strings.Join([]string{recordName, zoneName}, "."))
}
Expand All @@ -84,16 +85,21 @@ func resourceAwsRoute53RecordCreate(d *schema.ResourceData, meta interface{}) er
// Create the new records. We abuse StateChangeConf for this to
// retry for us since Route53 sometimes returns errors about another
// operation happening at the same time.
req := &route53.ChangeResourceRecordSetsRequest{
Comment: "Managed by Terraform",
changeBatch := &route53.ChangeBatch{
Comment: aws.String("Managed by Terraform"),
Changes: []route53.Change{
route53.Change{
Action: "UPSERT",
Record: *rec,
Action: aws.String("UPSERT"),
ResourceRecordSet: rec,
},
},
}

req := &route53.ChangeResourceRecordSetsRequest{
HostedZoneID: aws.String(CleanZoneID(*zoneRecord.HostedZone.ID)),
ChangeBatch: changeBatch,
}

log.Printf("[DEBUG] Creating resource records for zone: %s, name: %s",
zone, d.Get("name").(string))

Expand All @@ -103,7 +109,7 @@ func resourceAwsRoute53RecordCreate(d *schema.ResourceData, meta interface{}) er
Timeout: 5 * time.Minute,
MinTimeout: 1 * time.Second,
Refresh: func() (interface{}, string, error) {
resp, err := conn.ChangeResourceRecordSets(zone, req)
resp, err := conn.ChangeResourceRecordSets(req)
if err != nil {
if strings.Contains(err.Error(), "PriorRequestNotComplete") {
// There is some pending operation, so just retry
Expand All @@ -114,15 +120,15 @@ func resourceAwsRoute53RecordCreate(d *schema.ResourceData, meta interface{}) er
return nil, "failure", err
}

return resp.ChangeInfo, "accepted", nil
return resp, "accepted", nil
},
}

respRaw, err := wait.WaitForState()
if err != nil {
return err
}
changeInfo := respRaw.(route53.ChangeInfo)
changeInfo := respRaw.(*route53.ChangeResourceRecordSetsResponse).ChangeInfo

// Generate an ID
d.SetId(fmt.Sprintf("%s_%s_%s", zone, d.Get("name").(string), d.Get("type").(string)))
Expand All @@ -135,7 +141,10 @@ func resourceAwsRoute53RecordCreate(d *schema.ResourceData, meta interface{}) er
Timeout: 10 * time.Minute,
MinTimeout: 5 * time.Second,
Refresh: func() (result interface{}, state string, err error) {
return resourceAwsRoute53Wait(conn, changeInfo.ID)
changeRequest := &route53.GetChangeRequest{
ID: aws.String(CleanChangeID(*changeInfo.ID)),
}
return resourceAwsGoRoute53Wait(conn, changeRequest)
},
}
_, err = wait.WaitForState()
Expand All @@ -147,31 +156,33 @@ func resourceAwsRoute53RecordCreate(d *schema.ResourceData, meta interface{}) er
}

func resourceAwsRoute53RecordRead(d *schema.ResourceData, meta interface{}) error {
conn := meta.(*AWSClient).route53
conn := meta.(*AWSClient).r53conn

zone := d.Get("zone_id").(string)
lopts := &route53.ListOpts{
Name: d.Get("name").(string),
Type: d.Get("type").(string),
lopts := &route53.ListResourceRecordSetsRequest{
HostedZoneID: aws.String(CleanZoneID(zone)),
StartRecordName: aws.String(d.Get("name").(string)),
StartRecordType: aws.String(d.Get("type").(string)),
}
resp, err := conn.ListResourceRecordSets(zone, lopts)

resp, err := conn.ListResourceRecordSets(lopts)
if err != nil {
return err
}

// Scan for a matching record
found := false
for _, record := range resp.Records {
if route53.FQDN(record.Name) != route53.FQDN(lopts.Name) {
for _, record := range resp.ResourceRecordSets {
if FQDN(*record.Name) != FQDN(*lopts.StartRecordName) {
continue
}
if strings.ToUpper(record.Type) != strings.ToUpper(lopts.Type) {
if strings.ToUpper(*record.Type) != strings.ToUpper(*lopts.StartRecordType) {
continue
}

found = true

d.Set("records", record.Records)
d.Set("records", record.ResourceRecords)
d.Set("ttl", record.TTL)

break
Expand All @@ -185,7 +196,11 @@ func resourceAwsRoute53RecordRead(d *schema.ResourceData, meta interface{}) erro
}

func resourceAwsRoute53RecordDelete(d *schema.ResourceData, meta interface{}) error {
conn := meta.(*AWSClient).route53
conn := meta.(*AWSClient).r53conn

zone := d.Get("zone_id").(string)
log.Printf("[DEBUG] Deleting resource records for zone: %s, name: %s",
zone, d.Get("name").(string))

// Get the records
rec, err := resourceAwsRoute53RecordBuildSet(d)
Expand All @@ -194,26 +209,28 @@ func resourceAwsRoute53RecordDelete(d *schema.ResourceData, meta interface{}) er
}

// Create the new records
req := &route53.ChangeResourceRecordSetsRequest{
Comment: "Deleted by Terraform",
changeBatch := &route53.ChangeBatch{
Comment: aws.String("Deleted by Terraform"),
Changes: []route53.Change{
route53.Change{
Action: "DELETE",
Record: *rec,
Action: aws.String("DELETE"),
ResourceRecordSet: rec,
},
},
}
zone := d.Get("zone_id").(string)
log.Printf("[DEBUG] Deleting resource records for zone: %s, name: %s",
zone, d.Get("name").(string))

req := &route53.ChangeResourceRecordSetsRequest{
HostedZoneID: aws.String(CleanZoneID(zone)),
ChangeBatch: changeBatch,
}

wait := resource.StateChangeConf{
Pending: []string{"rejected"},
Target: "accepted",
Timeout: 5 * time.Minute,
MinTimeout: 1 * time.Second,
Refresh: func() (interface{}, string, error) {
_, err := conn.ChangeResourceRecordSets(zone, req)
_, err := conn.ChangeResourceRecordSets(req)
if err != nil {
if strings.Contains(err.Error(), "PriorRequestNotComplete") {
// There is some pending operation, so just retry
Expand Down Expand Up @@ -242,17 +259,34 @@ func resourceAwsRoute53RecordDelete(d *schema.ResourceData, meta interface{}) er

func resourceAwsRoute53RecordBuildSet(d *schema.ResourceData) (*route53.ResourceRecordSet, error) {
recs := d.Get("records").(*schema.Set).List()
records := make([]string, 0, len(recs))
records := make([]route53.ResourceRecord, 0, len(recs))

for _, r := range recs {
records = append(records, r.(string))
records = append(records, route53.ResourceRecord{Value: aws.String(r.(string))})
}

rec := &route53.ResourceRecordSet{
Name: d.Get("name").(string),
Type: d.Get("type").(string),
TTL: d.Get("ttl").(int),
Records: records,
Name: aws.String(d.Get("name").(string)),
Type: aws.String(d.Get("type").(string)),
TTL: aws.Long(int64(d.Get("ttl").(int))),
ResourceRecords: records,
}
return rec, nil
}

// CleanZoneID is used to remove the leading /hostedzone/
func CleanZoneID(ID string) string {
if strings.HasPrefix(ID, "/hostedzone/") {
ID = strings.TrimPrefix(ID, "/hostedzone/")
}
return ID
}

func FQDN(name string) string {
n := len(name)
if n == 0 || name[n-1] == '.' {
return name
} else {
return name + "."
}
}
38 changes: 25 additions & 13 deletions builtin/providers/aws/resource_aws_route53_record_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,9 @@ import (

"github.com/hashicorp/terraform/helper/resource"
"github.com/hashicorp/terraform/terraform"
"github.com/mitchellh/goamz/route53"

"github.com/awslabs/aws-sdk-go/aws"
awsr53 "github.com/awslabs/aws-sdk-go/gen/route53"
)

func TestAccRoute53Record(t *testing.T) {
Expand Down Expand Up @@ -43,7 +45,7 @@ func TestAccRoute53Record_generatesSuffix(t *testing.T) {
}

func testAccCheckRoute53RecordDestroy(s *terraform.State) error {
conn := testAccProvider.Meta().(*AWSClient).route53
conn := testAccProvider.Meta().(*AWSClient).r53conn
for _, rs := range s.RootModule().Resources {
if rs.Type != "aws_route53_record" {
continue
Expand All @@ -54,16 +56,21 @@ func testAccCheckRoute53RecordDestroy(s *terraform.State) error {
name := parts[1]
rType := parts[2]

lopts := &route53.ListOpts{Name: name, Type: rType}
resp, err := conn.ListResourceRecordSets(zone, lopts)
lopts := &awsr53.ListResourceRecordSetsRequest{
HostedZoneID: aws.String(CleanZoneID(zone)),
StartRecordName: aws.String(name),
StartRecordType: aws.String(rType),
}

resp, err := conn.ListResourceRecordSets(lopts)
if err != nil {
return err
}
if len(resp.Records) == 0 {
if len(resp.ResourceRecordSets) == 0 {
return nil
}
rec := resp.Records[0]
if route53.FQDN(rec.Name) == route53.FQDN(name) && rec.Type == rType {
rec := resp.ResourceRecordSets[0]
if FQDN(*rec.Name) == FQDN(name) && *rec.Type == rType {
return fmt.Errorf("Record still exists: %#v", rec)
}
}
Expand All @@ -72,7 +79,7 @@ func testAccCheckRoute53RecordDestroy(s *terraform.State) error {

func testAccCheckRoute53RecordExists(n string) resource.TestCheckFunc {
return func(s *terraform.State) error {
conn := testAccProvider.Meta().(*AWSClient).route53
conn := testAccProvider.Meta().(*AWSClient).r53conn
rs, ok := s.RootModule().Resources[n]
if !ok {
return fmt.Errorf("Not found: %s", n)
Expand All @@ -87,16 +94,21 @@ func testAccCheckRoute53RecordExists(n string) resource.TestCheckFunc {
name := parts[1]
rType := parts[2]

lopts := &route53.ListOpts{Name: name, Type: rType}
resp, err := conn.ListResourceRecordSets(zone, lopts)
lopts := &awsr53.ListResourceRecordSetsRequest{
HostedZoneID: aws.String(CleanZoneID(zone)),
StartRecordName: aws.String(name),
StartRecordType: aws.String(rType),
}

resp, err := conn.ListResourceRecordSets(lopts)
if err != nil {
return err
}
if len(resp.Records) == 0 {
if len(resp.ResourceRecordSets) == 0 {
return fmt.Errorf("Record does not exist")
}
rec := resp.Records[0]
if route53.FQDN(rec.Name) == route53.FQDN(name) && rec.Type == rType {
rec := resp.ResourceRecordSets[0]
if FQDN(*rec.Name) == FQDN(name) && *rec.Type == rType {
return nil
}
return fmt.Errorf("Record does not exist: %#v", rec)
Expand Down
Loading

0 comments on commit 1b4fd28

Please sign in to comment.