Amazon S3にある画像のExif情報の削除&向きを自動調整してアップロードするGoのLambda関数

概要

S3に格納されているオブジェクト(JPEG画像)をコピーした際に起動するLambda関数。
Lambda関数の処理としては、コピーした画像を読み込んで向きを自動調整し、Exif情報を削除した上でS3にアップロードしている。

環境

go1.16.5を使用。
Lambdaのランタイムは go1.x 。

ソースコード

package main

import (
    "bytes"
    "errors"
    "io"
    "log"
    "os"

    "github.com/disintegration/imaging"

    "github.com/aws/aws-lambda-go/events"
    "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/s3"
    "github.com/aws/aws-sdk-go/service/s3/s3manager"
)

// アップロードされた画像のEXIF情報を削除するのと向きを自動調整し、再アップロードする
func HandleRequest(event events.S3Event) error {
    for _, record := range event.Records {
        bucket := record.S3.Bucket.Name
        key := record.S3.Object.Key

        sess := session.Must(session.NewSession(&aws.Config{Region: aws.String("ap-northeast-1")}))
        //fileの読み込み
        obj, err := s3.New(sess).GetObject(&s3.GetObjectInput{
            Bucket: aws.String(bucket),
            Key:    aws.String(key),
        })
        if err != nil {
            log.Fatal(err)
        }
        buf, err := io.ReadAll(obj.Body)
        if err != nil {
            log.Fatal(err)
        }

        // 向きを自動調整
        decoded_image, err := imaging.Decode(bytes.NewReader(buf), imaging.AutoOrientation(true))
        if err != nil {
            log.Fatalln(err)
        }

        //コピー先の空ファイル名
        // Lambdaでは /tmp/ を付けないとエラーになる
        file_name := "/tmp/tmp.jpg"

        err = imaging.Save(decoded_image, file_name)
        if err != nil {
            log.Fatal(err)
        }

        //imagingで作ったファイルをopen
        f, err := os.Open(file_name)
        if err != nil {
            log.Fatal(err)
        }

        //S3にアップロード
        uploader := s3manager.NewUploader(sess)
        _, err = uploader.Upload(&s3manager.UploadInput{
            Bucket: aws.String(bucket),
            Key:    aws.String(key),
            Body:   f,
        })
        if err != nil {
            log.Fatal(err)
        }
    }
    return nil
}

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

ポイント

• https://github.com/disintegration/imaging の AutoOrientation で画像の向きを調整。

• Lambdaでは一時領域として /tmp が提供されていて、画像は /tmp 配下で処理する必要がある。
  　　参考: https://docs.aws.amazon.com/ja_jp/lambda/latest/dg/runtimes-context.html

• ファイルのダウンロード→lambdaで処理→アップロードではなく、ファイルの読み込み→lambdaで処理→アップロードという流れ。
  結局Lambdaのストレージを使用しているので、もしかしたら読み込みにした意味はそこまでなかったかも？

最後に

これによって、スマホで撮影したきゃわわなネコチャン🐈の画像が変な向きで表示される...みたいなこともなくなる（はず）。
ちなみにGoは初めて書きました。難しかったです👶