　はじめに

GolangとAWSの勉強がてらに、DynamoDB＋Lambda＋APIGatewayつかってREST APIをつくりました。

　参考

golang + Lambda + API GatewayでREST APIを作ってみた

Amazon API GatewayからAWS Lambdaにパラメータを渡す方法

DynamoDB

今回は簡易的にUserIDとUserNameだけ含むユーザーテーブルを用意します。
テーブルの作り方は”初めてのサーバーレスアプリケーション開発～DynamoDBにテーブルを作成する～”を参考にしてください

Lambda

まずLambdaにDynamoDBへのアクセス権を与えるため、IAMロールを作成しなくてはいけないので
1. [IAM] -> [Roles] -> [Create role] -> AWS servicesからLambdaを選択し[Next Permissions] -> [AmazonDynamoDBFullAccess]と[AWSLambdaBasicExecutionRole]を選択し→[Next Tags]→[Next Review]→[Create role]

Lambdaのメニューから[Create function]を選択して以下のようにLambdaを作成
AWSのGoのSDKをインストール

$ go get -u -v github.com/aws/aws-sdk-go/service/dynamodb/dynamodbattribute
$ go get -u -v github.com/aws/aws-lambda-go/lambda
$ go get -u -v github.com/aws/aws-sdk-go/aws
$ go get -u -v github.com/aws/aws-sdk-go/aws/session
$ go get -u -v github.com/aws/aws-sdk-go/service/dynamodb

もしくは

$ go get -u -v github.com/aws/aws-sdk-go/...

ちなみに -u のフラグで最新のパッケージがインストールできて、-v でインストール経過が確認できます。

/user/{id} - GET

パスパラメータのidと一致するUserIDを持つユーザーデータを返します。
コードはローカルで編集してzipにしてLambdaにアップロードします。

$ GOOS=linux go build -o main
$ zip deployment.zip main

go.getuser.go

package main

import (
    "fmt"
    "github.com/aws/aws-lambda-go/lambda"
    "github.com/aws/aws-sdk-go/aws"
    "github.com/aws/aws-sdk-go/aws/session"
    "github.com/aws/aws-sdk-go/service/dynamodb"
)

type User struct {
    UserID   string `json:"userid"`
    USerName string `json:"username"`
}

type Request struct {
    ID string `json:"id"`
}

type Response struct {
    Data string `json:"body"`
}

func Handler(request Request) (Response, error) {

    sess, err := session.NewSession()
    if err != nil {
        panic(err)
    }

    svc := dynamodb.New(sess)

    getParams := &dynamodb.GetItemInput{
        TableName: aws.String("User"),
        Key: map[string]*dynamodb.AttributeValue{
            "UserID": {
                S: aws.String(request.ID),
            },
        },
    }

    getItem, getErr := svc.GetItem(getParams)
    if getErr != nil {
        panic(getErr)
    }

    return Response{
        Data: fmt.Sprintln(getItem),
        }, nil
}

func main() {
    lambda.Start(Handler)
}

以下のようにテストすると

帰ってきた！

/user/ -POST-

BodyパラメータからIDとUserNameを取得してDynamoDBに書き込む

createuser.go

package main

import (
    "fmt"

    "github.com/aws/aws-lambda-go/lambda"
    "github.com/aws/aws-sdk-go/aws"
    "github.com/aws/aws-sdk-go/aws/session"
    "github.com/aws/aws-sdk-go/service/dynamodb"
)

type User struct {
    UserID   string `json:"userid"`
    USerName string `json:"username"`
}

type Response struct {
    Data string `json:"message"`
}

type Request struct {
    ID string `json:"id"`
    UserName string `json:"username"`
}

func Handler(request Request) (Response, error) {
    sess, err := session.NewSession()
    if err != nil {
        panic(err)
    }
    svc := dynamodb.New(sess)
    putParams := &dynamodb.PutItemInput{
        TableName: aws.String("User"),
        Item: map[string]*dynamodb.AttributeValue{
            "UserID": {
                S : aws.String(request.ID),
            },
            "UserName": {
                S : aws.String(request.UserName),
            },
        },
    }
    putItem, putErr := svc.PutItem(putParams)
    if putErr != nil {
        panic(putErr)
    }
    return Response{
        Data: fmt.Sprintln(putItem),
    }, nil
}
func main() {
    lambda.Start(Handler)
}

/user/{id} DELETE

deleteuser.go

package main

import (
    "github.com/aws/aws-lambda-go/lambda"
    "github.com/aws/aws-sdk-go/aws"
    "github.com/aws/aws-sdk-go/aws/session"
    "github.com/aws/aws-sdk-go/service/dynamodb"
)

type User struct {
    UserID   string `json:"userid"`
    USerName string `json:"username"`
}

type Request struct {
    ID string `json:"id"`
    UserName string `json:"username"`
}

func Handler(request Request) error {

    sess, err := session.NewSession()
    if err != nil {
        panic(err)
    }

    svc := dynamodb.New(sess)

    deleteParams := &dynamodb.DeleteItemInput{
        TableName: aws.String("User"),
        Key: map[string]*dynamodb.AttributeValue{
            "UserID": {
                S: aws.String(request.ID),
            },
        },
    }
    if _, err := svc.DeleteItem(deleteParams); err != nil{
        panic(err)
    }
    return nil
}

func main() {
    lambda.Start(Handler)
}