S3にオブジェクトをアップロードして引数をlambdaに送りたいとき

今まではS3をトリガーにLambdaを発火させていたのですが、これだと追加の変数を設定することができず、どうすればいいか悩んでいたときに、InvokeInputに渡せばいいことを知りました。

参考

go server側

S3にアップロードしたあとに、invokeして変数を渡すようにしました。

main.go

package main

import (
	"encoding/json"
	"fmt"

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

// これはLambda関数の完了で返すための構造体定義
type Response struct {
	Message string `json:"message"`
	Ok      bool   `json:"ok"`
}

// 別のLambdaに渡すための構造体定義
type Event struct {
	Records []Record `json:"Records"`
}

type Record struct {
	S3    S3    `json:"s3"`
	Input Input `json:"input"`
}

type Input struct {
	OriginLocation      string `json:"originLocation"`
	DestinationLocation string `json:"destinationLocation"`
	FrameID             string `json:"frameNumber"`
}

type S3 struct {
	Bucket Bucket `json:"bucket"`
	Object Object `json:"object"`
}

type Bucket struct {
	Name string `json:"name"`
}

type Object struct {
	Key string `json:"key"`
}

func Handler() (Response, error) {
	fmt.Println("handler")
	// 呼び出したLambdaに渡すデータ
	payload := Event{
		[]Record{
			{
				S3{
					Bucket{
						Name: "uploaded-photos",
					},
					Object{
						Key: "xxxxxx.jpg",
					},
				},
				Input{
					OriginLocation:      "HND",
					DestinationLocation: "TPE",
					FrameID:             "frame",
				},
			},
		},
	}
	jsonBytes, _ := json.Marshal(payload)

	// このlambdaはaws/aws-lambda-go/lambdaではなく、
	// aws/aws-sdk-go-aws/service/lambdaなので注意
	svc := lambda.New(session.New(&aws.Config{
		Region: aws.String("ap-northeast-1")}))

	input := &lambda.InvokeInput{
		FunctionName:   aws.String("arn:aws:lambda:ap-northeast-1:xxxxxx:function:CreateThumbnail"),
		Payload:        jsonBytes,
		InvocationType: aws.String("Event"),
	}

	_, _ = svc.Invoke(input)

	return Response{
		Message: "success",
		Ok:      true,
	}, nil
}

func main() {
	fmt.Println("start")
	Handler()
}

lambda側

Lambda側ではrecordの中の引数を受け取るようにしました。

lambda_function.py

def lambda_handler(event, context):
    for record in event['Records']:
        bucket = record['s3']['bucket']['name']
        key = unquote_plus(record['s3']['object']['key'])

        selectedImageID = record['input']['frameID']
        originLocation = record['input']['originLocation']
        destinationLocation = record['input']['destinationLocation']