AWS SDK for Go を使ってS3にファイルをアップロードする

AWS SDK for Goがついにプレビュー版になりましたね！
まだまだ仕様が変わる気がしないでもありませんが現バージョン(v0.6.2)でのS3のアップロードサンプルをやってみます。

CloudFormationでS3bucketをつくる

S3バケットをつくりましょう。作り方は何でもいいです。今回はCloudFormationを使ってみます。

こんな感じのJSONを用意してStackを実行してS3バケットをつくってください。
バケット名は既にある名前だとつくれないので注意してください。

{
  "AWSTemplateFormatVersion": "2010-09-09",
  "Parameters": {
  },
  "Resources": {
    "DemoBucket" : {
      "Type" : "AWS::S3::Bucket",
      "Properties" : {
        "AccessControl" : "PublicReadWrite",
        "BucketName" : "sample-demo",
        "Tags": [
          {
            "Key": "Name",
            "Value": "demo"
          },
          {
            "Key": "Role",
            "Value": "s3"
          }
        ]
      }
    },
    "DemoBucketPolicy" : {
      "Type" : "AWS::S3::BucketPolicy",
      "Properties" : {
        "Bucket" : { "Ref" : "DemoBucket" },
        "PolicyDocument" : {
          "Version": "2008-10-17",
          "Statement": [
            {
              "Sid": "PublicReadGetObject",
              "Effect": "Allow",
              "Principal": {
                "AWS": "*"
              },
              "Action": "s3:GetObject",
              "Resource": "arn:aws:s3:::sample-demo/*"
            }
          ]
        }
      },
      "DependsOn": "DemoBucket"
    }
  },
  "Description": "s3"
}

Gomfileを用意してGomでaws-sdk-goをインストールする

Goマネージャーのgomが便利なのでこれを使ってaws-sdk-goをインストールします。

$ go get github.com/mattn/gom

Gomfile を用意します。

gom 'github.com/aws/aws-sdk-go/aws'
gom 'github.com/aws/aws-sdk-go/service/s3'

パッケージをインストールします。

$ gom install

ここまで準備できたらGoでコードを書いてみましょう。

GoでS3にファイルをアップロードしてみる

コードはこんな感じになります。アクセスキーとシークレットアクセスキーが必要になります。AWSのIAMで生成しておきましょう。

サンプルでアップロードするファイルはGoコードと同じ階層に置きます。
例として ./sample.jpg として用意しました。

goファイルは app.go とします。

package main

import (
	"fmt"
	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/awsutil"
	"github.com/aws/aws-sdk-go/aws/credentials"
	"github.com/aws/aws-sdk-go/service/s3"
	"log"
	"os"
)

func main() {
	fileTransfer := FileTransferToS3{
		AccessKeyId:     "ACCESS_KEY_ID",
		SecretAccessKey: "SECRET_ACCESS_KEY",
		Region:          "us-west-2",
		BucketName:      "bucket-name",
	}

	fileTransfer.PutToS3("./", "sample.jpg")
}

type FileTransferToS3 struct {
	AccessKeyId     string
	SecretAccessKey string
	Region          string
	BucketName      string
}

func (f *FileTransferToS3) PutToS3(path string, filename string) {
	file, err := os.Open(fmt.Sprintf("%s%s", path, filename))
	if err != nil {
		log.Println(err.Error())
	}
	defer file.Close()

	cli := s3.New(&aws.Config{
		Credentials: credentials.NewStaticCredentials(f.AccessKeyId, f.SecretAccessKey, ""),
		Region:      f.Region,
	})

	resp, err := cli.PutObject(&s3.PutObjectInput{
		Bucket: aws.String(f.BucketName),
		Key:    aws.String(filename),
		Body:   file,
	})
	if err != nil {
		log.Println(err.Error())
	}

	log.Println(awsutil.StringValue(resp))
}

パッケージを gom でインストールしたのでgomコマンドで実行してください。

$ gom run app.go

実行してS3に無事にアップロードされたら成功です。