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/elb"
"github.com/mitchellh/goamz/rds"
"github.com/mitchellh/goamz/s3"
awsGo "github.com/awslabs/aws-sdk-go/aws"
"github.com/awslabs/aws-sdk-go/gen/route53"
"github.com/awslabs/aws-sdk-go/gen/s3"
)
type Config struct {
@ -31,6 +31,7 @@ type AWSClient struct {
s3conn *s3.S3
rdsconn *rds.Rds
r53conn *route53.Route53
region string
}
// Client configures and returns a fully initailized AWSClient
@ -53,6 +54,11 @@ func (c *Config) Client() (interface{}, error) {
}
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")
client.ec2conn = ec2.New(auth, region)
log.Println("[INFO] Initializing ELB connection")
@ -60,16 +66,14 @@ func (c *Config) Client() (interface{}, error) {
log.Println("[INFO] Initializing AutoScaling connection")
client.autoscalingconn = autoscaling.New(auth, region)
log.Println("[INFO] Initializing S3 connection")
client.s3conn = s3.New(auth, region)
log.Println("[INFO] Initializing RDS connection")
client.rdsconn = rds.New(auth, region)
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
// endpoints to use 'us-east-1'.
// See http://docs.aws.amazon.com/general/latest/gr/sigv4_changes.html
client.r53conn = route53.New(creds, "us-east-1", nil)
client.s3conn = s3.New(creds, c.Region, nil)
}
if len(errs) > 0 {

View File

@ -5,7 +5,9 @@ import (
"log"
"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 {
@ -33,14 +35,23 @@ func resourceAwsS3Bucket() *schema.Resource {
func resourceAwsS3BucketCreate(d *schema.ResourceData, meta interface{}) error {
s3conn := meta.(*AWSClient).s3conn
awsRegion := meta.(*AWSClient).region
// Get the bucket and acl
bucket := d.Get("bucket").(string)
acl := d.Get("acl").(string)
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 {
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 {
s3conn := meta.(*AWSClient).s3conn
bucket := s3conn.Bucket(d.Id())
resp, err := bucket.Head("/")
err := s3conn.HeadBucket(&s3.HeadBucketRequest{
Bucket: aws.String(d.Id()),
})
if err != nil {
return err
}
defer resp.Body.Close()
return nil
}
@ -67,7 +78,11 @@ func resourceAwsS3BucketDelete(d *schema.ResourceData, meta interface{}) error {
s3conn := meta.(*AWSClient).s3conn
log.Printf("[DEBUG] S3 Delete Bucket: %s", d.Id())
bucket := s3conn.Bucket(d.Id())
return bucket.DelBucket()
err := s3conn.DeleteBucket(&s3.DeleteBucketRequest{
Bucket: aws.String(d.Id()),
})
if err != nil {
return err
}
return nil
}

View File

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