provider/aws: Convert AWS S3 Bucket to awslabs/aws-sdk-go

This commit is contained in:
Clint Shryock 2015-02-19 15:38:56 -06:00
parent d99df916bc
commit 116e6cac33
3 changed files with 45 additions and 24 deletions

View File

@ -12,10 +12,10 @@ import (
"github.com/mitchellh/goamz/ec2" "github.com/mitchellh/goamz/ec2"
"github.com/mitchellh/goamz/elb" "github.com/mitchellh/goamz/elb"
"github.com/mitchellh/goamz/rds" "github.com/mitchellh/goamz/rds"
"github.com/mitchellh/goamz/s3"
awsGo "github.com/awslabs/aws-sdk-go/aws" awsGo "github.com/awslabs/aws-sdk-go/aws"
"github.com/awslabs/aws-sdk-go/gen/route53" "github.com/awslabs/aws-sdk-go/gen/route53"
"github.com/awslabs/aws-sdk-go/gen/s3"
) )
type Config struct { type Config struct {
@ -31,6 +31,7 @@ type AWSClient struct {
s3conn *s3.S3 s3conn *s3.S3
rdsconn *rds.Rds rdsconn *rds.Rds
r53conn *route53.Route53 r53conn *route53.Route53
region string
} }
// Client configures and returns a fully initailized AWSClient // Client configures and returns a fully initailized AWSClient
@ -53,6 +54,11 @@ func (c *Config) Client() (interface{}, error) {
} }
if len(errs) == 0 { if len(errs) == 0 {
// store AWS region in client struct, for region specific operations such as
// bucket storage in S3
client.region = c.Region
creds := awsGo.Creds(c.AccessKey, c.SecretKey, "")
log.Println("[INFO] Initializing EC2 connection") log.Println("[INFO] Initializing EC2 connection")
client.ec2conn = ec2.New(auth, region) client.ec2conn = ec2.New(auth, region)
log.Println("[INFO] Initializing ELB connection") log.Println("[INFO] Initializing ELB connection")
@ -60,16 +66,14 @@ func (c *Config) Client() (interface{}, error) {
log.Println("[INFO] Initializing AutoScaling connection") log.Println("[INFO] Initializing AutoScaling connection")
client.autoscalingconn = autoscaling.New(auth, region) client.autoscalingconn = autoscaling.New(auth, region)
log.Println("[INFO] Initializing S3 connection") log.Println("[INFO] Initializing S3 connection")
client.s3conn = s3.New(auth, region)
log.Println("[INFO] Initializing RDS connection") log.Println("[INFO] Initializing RDS connection")
client.rdsconn = rds.New(auth, region) client.rdsconn = rds.New(auth, region)
log.Println("[INFO] Initializing Route53 connection") log.Println("[INFO] Initializing Route53 connection")
creds := awsGo.Creds(c.AccessKey, c.SecretKey, "")
// aws-sdk-go uses v4 for signing requests, which requires all global // aws-sdk-go uses v4 for signing requests, which requires all global
// endpoints to use 'us-east-1'. // endpoints to use 'us-east-1'.
// See http://docs.aws.amazon.com/general/latest/gr/sigv4_changes.html // See http://docs.aws.amazon.com/general/latest/gr/sigv4_changes.html
client.r53conn = route53.New(creds, "us-east-1", nil) client.r53conn = route53.New(creds, "us-east-1", nil)
client.s3conn = s3.New(creds, c.Region, nil)
} }
if len(errs) > 0 { if len(errs) > 0 {

View File

@ -5,7 +5,9 @@ import (
"log" "log"
"github.com/hashicorp/terraform/helper/schema" "github.com/hashicorp/terraform/helper/schema"
"github.com/mitchellh/goamz/s3"
"github.com/awslabs/aws-sdk-go/aws"
"github.com/awslabs/aws-sdk-go/gen/s3"
) )
func resourceAwsS3Bucket() *schema.Resource { func resourceAwsS3Bucket() *schema.Resource {
@ -33,14 +35,23 @@ func resourceAwsS3Bucket() *schema.Resource {
func resourceAwsS3BucketCreate(d *schema.ResourceData, meta interface{}) error { func resourceAwsS3BucketCreate(d *schema.ResourceData, meta interface{}) error {
s3conn := meta.(*AWSClient).s3conn s3conn := meta.(*AWSClient).s3conn
awsRegion := meta.(*AWSClient).region
// Get the bucket and acl // Get the bucket and acl
bucket := d.Get("bucket").(string) bucket := d.Get("bucket").(string)
acl := d.Get("acl").(string) acl := d.Get("acl").(string)
log.Printf("[DEBUG] S3 bucket create: %s, ACL: %s", bucket, acl) log.Printf("[DEBUG] S3 bucket create: %s, ACL: %s", bucket, acl)
s3Bucket := s3conn.Bucket(bucket)
err := s3Bucket.PutBucket(s3.ACL(acl)) req := &s3.CreateBucketRequest{
Bucket: aws.String(bucket),
ACL: aws.String(acl),
CreateBucketConfiguration: &s3.CreateBucketConfiguration{
LocationConstraint: aws.String(awsRegion),
},
}
_, err := s3conn.CreateBucket(req)
if err != nil { if err != nil {
return fmt.Errorf("Error creating S3 bucket: %s", err) return fmt.Errorf("Error creating S3 bucket: %s", err)
} }
@ -54,12 +65,12 @@ func resourceAwsS3BucketCreate(d *schema.ResourceData, meta interface{}) error {
func resourceAwsS3BucketRead(d *schema.ResourceData, meta interface{}) error { func resourceAwsS3BucketRead(d *schema.ResourceData, meta interface{}) error {
s3conn := meta.(*AWSClient).s3conn s3conn := meta.(*AWSClient).s3conn
bucket := s3conn.Bucket(d.Id()) err := s3conn.HeadBucket(&s3.HeadBucketRequest{
resp, err := bucket.Head("/") Bucket: aws.String(d.Id()),
})
if err != nil { if err != nil {
return err return err
} }
defer resp.Body.Close()
return nil return nil
} }
@ -67,7 +78,11 @@ func resourceAwsS3BucketDelete(d *schema.ResourceData, meta interface{}) error {
s3conn := meta.(*AWSClient).s3conn s3conn := meta.(*AWSClient).s3conn
log.Printf("[DEBUG] S3 Delete Bucket: %s", d.Id()) log.Printf("[DEBUG] S3 Delete Bucket: %s", d.Id())
bucket := s3conn.Bucket(d.Id()) err := s3conn.DeleteBucket(&s3.DeleteBucketRequest{
Bucket: aws.String(d.Id()),
return bucket.DelBucket() })
if err != nil {
return err
}
return nil
} }

View File

@ -8,10 +8,12 @@ import (
"github.com/hashicorp/terraform/helper/resource" "github.com/hashicorp/terraform/helper/resource"
"github.com/hashicorp/terraform/terraform" "github.com/hashicorp/terraform/terraform"
"github.com/awslabs/aws-sdk-go/aws"
"github.com/awslabs/aws-sdk-go/gen/s3"
) )
func TestAccAWSS3Bucket(t *testing.T) { func TestAccAWSS3Bucket(t *testing.T) {
rand.Seed(time.Now().UnixNano())
resource.Test(t, resource.TestCase{ resource.Test(t, resource.TestCase{
PreCheck: func() { testAccPreCheck(t) }, PreCheck: func() { testAccPreCheck(t) },
@ -35,13 +37,12 @@ func testAccCheckAWSS3BucketDestroy(s *terraform.State) error {
if rs.Type != "aws_s3_bucket" { if rs.Type != "aws_s3_bucket" {
continue continue
} }
err := conn.DeleteBucket(&s3.DeleteBucketRequest{
bucket := conn.Bucket(rs.Primary.ID) Bucket: aws.String(rs.Primary.ID),
resp, err := bucket.Head("/") })
if err == nil { if err != nil {
return fmt.Errorf("S3Bucket still exists") return err
} }
defer resp.Body.Close()
} }
return nil return nil
} }
@ -58,12 +59,13 @@ func testAccCheckAWSS3BucketExists(n string) resource.TestCheckFunc {
} }
conn := testAccProvider.Meta().(*AWSClient).s3conn conn := testAccProvider.Meta().(*AWSClient).s3conn
bucket := conn.Bucket(rs.Primary.ID) err := conn.HeadBucket(&s3.HeadBucketRequest{
resp, err := bucket.Head("/") Bucket: aws.String(rs.Primary.ID),
})
if err != nil { if err != nil {
return fmt.Errorf("S3Bucket error: %v", err) return fmt.Errorf("S3Bucket error: %v", err)
} }
defer resp.Body.Close()
return nil return nil
} }
} }
@ -75,4 +77,4 @@ resource "aws_s3_bucket" "bar" {
bucket = "tf-test-bucket-%d" bucket = "tf-test-bucket-%d"
acl = "public-read" acl = "public-read"
} }
`, rand.Int()) `, rand.New(rand.NewSource(time.Now().UnixNano())).Int())