[AWS CDK] Lambda で S3 オブジェクトを読み書きするStackの構築

[AWS CDK] Lambda で S3 オブジェクトを読み書きするStackの構築

2022-03-1815 min read

目次

  1. 概要
  2. 構成の概要
  3. 実装
  4. deploy
  5. バケットポリシーについて

概要

AWS CDK で Lambda を利用して S3 のオブジェクトを読み書きするスタックを構築しました。

読み込み専用 Lambda と書き込み専用 Lmabda を作成してバケットポリシーの定義方法などを説明しています。

※説明について全体的に細かい部分の説明は省いています。

構成の概要

構成の概要は次の通りとします。

  • オブジェクトを管理する S3 バケットを 1 つ作成
  • その S3 のオブジェクトに書き込む Lambda(Writer) と読み込む Lmabda(Reader)を作成
  • 2 つの Lmabda は最低限必要な権限(例: Lambda(Writer)なら書き込み権限 ) をバケットポリシーで定義する

実装

cdk initでプロジェクト初期化し、以下の様にスタックと Lmabda にデプロイするソースを定義します。

CDK Stack

import { Duration, Stack, aws_iam as iam, aws_s3 as s3 } from "aws-cdk-lib";
import { Construct } from "constructs";
import { Runtime } from "aws-cdk-lib/aws-lambda";
import { NodejsFunction } from "aws-cdk-lib/aws-lambda-nodejs";
import { CustomizedProps } from "./customized-props";

export class LambdaAppStack extends Stack {
  constructor(scope: Construct, id: string, props?: CustomizedProps) {
    super(scope, id, props);

    // オブジェクトを読み込むLambda
    const iamRoleForLambdaReader = new iam.Role(
      this,
      "iamRoleForLambdaReader",
      {
        assumedBy: new iam.ServicePrincipal("lambda.amazonaws.com"),
      },
    );
    // オブジェクトを書き込むLambda
    const iamRoleForLambdaWriter = new iam.Role(
      this,
      "iamRoleForLambdaWriter",
      {
        assumedBy: new iam.ServicePrincipal("lambda.amazonaws.com"),
      },
    );
    // コンテンツを配置するS3
    const bucket = new s3.Bucket(this, "TmpBucket");
    // Lambda(Reader)に対するバケットポリシー
    bucket.addToResourcePolicy(
      new iam.PolicyStatement({
        effect: iam.Effect.ALLOW,
        actions: ["s3:GetObject"],
        principals: [iamRoleForLambdaReader],
        resources: [bucket.bucketArn + "/*"],
      }),
    );
    // Lambda(Writer)に対するバケットポリシー
    bucket.addToResourcePolicy(
      new iam.PolicyStatement({
        effect: iam.Effect.ALLOW,
        actions: ["s3:PutObject"],
        principals: [iamRoleForLambdaWriter],
        resources: [bucket.bucketArn + "/*"],
      }),
    );
    // Lambda(Reader)
    const lambdaReader = new NodejsFunction(this, "LambdaReader", {
      entry: "src/lambda-reader/index.ts",
      runtime: Runtime.NODEJS_14_X,
      timeout: Duration.seconds(30),
      role: iamRoleForLambdaReader,
      environment: {
        S3_BUCKET_NAME: bucket.bucketName,
        AWS_NODEJS_CONNECTION_REUSE_ENABLED: "1",
      },
      memorySize: 128,
    });
    // Lambda(Writer)
    const lambdaWriter = new NodejsFunction(this, "LambdaWriter", {
      entry: "src/lambda-writer/index.ts",
      runtime: Runtime.NODEJS_14_X,
      timeout: Duration.seconds(30),
      role: iamRoleForLambdaWriter,
      environment: {
        S3_BUCKET_NAME: bucket.bucketName,
        AWS_NODEJS_CONNECTION_REUSE_ENABLED: "1",
      },
      memorySize: 128,
    });
  }
}

Lambda(Reader)

import { APIGatewayProxyEvent, APIGatewayProxyResult } from 'aws-lambda';
import * as AWS from 'aws-sdk';

const getContent = async () => {
  const s3 = new AWS.S3();
  const params = {
    Bucket: process.env.S3_BUCKET_NAME,
    Key: 'test.txt',
  };
  const res = await s3.getObject(params).promise();
  return res.Body?.toString();
};

export const handler = async (
  event: APIGatewayProxyEvent,
): Promise<APIGatewayProxyResult> => {
  const res = await getContent();
  return {
    statusCode: 200,
    body: JSON.stringify(res),
  };
};

Lambda(Writer)

import { APIGatewayProxyEvent, APIGatewayProxyResult } from 'aws-lambda';
import * as AWS from 'aws-sdk';

const putContent = async (arg: string) => {
  const s3 = new AWS.S3();
  const params = {
    Bucket: process.env.S3_BUCKET_NAME,
    Key: 'test.txt',
    Body: arg,
  };
  await s3.putObject(params).promise();
};

export const handler = async (
  event: APIGatewayProxyEvent,
): Promise<APIGatewayProxyResult> => {
  // 適当な文字列。検証しやすいように日付文字列を渡しておく
  await putContent(new Date().toString());
  return {
    statusCode: 200,
    body: '',
  };
};

Deploy

cdk deployでデプロイします。

デプロイ完了後、AWSのマネジメントコンソールから先にLambda(Writer)をテスト実行すると、 S3にオブジェクトであるtest.txtが作成されます。

次にLambda(Reader)をテスト実行するとtest.txtの内容が読み取れていることが確認できると思います。

バケットポリシーについて

S3に設定されたバケットポリシーは次のように設定されていました。

{
  "Version": "2012-10-17",
  "Statement": [
    {
      "Effect": "Allow",
      "Principal": {
        "AWS": "arn:aws:iam::XXXXXXXXXXXX:role/LambdaAppStack-iamRoleForLambdaReaderXXXXXXXXXXXXXX"
      },
      "Action": "s3:GetObject",
      "Resource": "arn:aws:s3:::lambdaappstack-tmpbucketXXXXXXXXXXXXXX/*"
    },
    {
      "Effect": "Allow",
      "Principal": {
        "AWS": "arn:aws:iam::XXXXXXXXXXXX:role/LambdaAppStack-iamRoleForLambdaWriterXXXXXXXXXXXXXX"
      },
      "Action": "s3:PutObject",
      "Resource": "arn:aws:s3:::lambdaappstack-tmpbucketXXXXXXXXXXXXXX/*"
    }
  ]
}
Tags
javascript(110)
node.js(54)
linux(54)
amazon%20aws(47)
typescript(45)
%E3%82%A2%E3%83%AB%E3%82%B4%E3%83%AA%E3%82%BA%E3%83%A0(36)
%E7%94%BB%E5%83%8F%E5%87%A6%E7%90%86(30)
html5(29)
php(24)
centos(24)
python(22)
%E7%AB%B6%E6%8A%80%E3%83%97%E3%83%AD%E3%82%B0%E3%83%A9%E3%83%9F%E3%83%B3%E3%82%B0(21)
mac(21)
mysql(20)
canvas(19)
opencv(17)
%E9%9B%91%E8%AB%87(16)
docker(16)
wordpress(15)
atcoder(14)
apache(12)
%E6%A9%9F%E6%A2%B0%E5%AD%A6%E7%BF%92(12)
%E3%83%87%E3%83%BC%E3%82%BF%E3%83%99%E3%83%BC%E3%82%B9(12)
amazon%20s3(12)
red%20hat(12)
prisma(12)
ubuntu(11)
github(10)
git(10)
vue.js(10)
%E7%94%BB%E5%83%8F%E5%87%A6%E7%90%86100%E6%9C%AC%E3%83%8E%E3%83%83%E3%82%AF(10)
mariadb(10)
react(9)
aws%20cdk(9)
css3(8)
%E5%8F%AF%E8%A6%96%E5%8C%96(8)
%E5%B0%8F%E3%83%8D%E3%82%BF(8)
nestjs(8)
amazon%20lightsail(7)
next.js(7)
%E3%83%96%E3%83%AD%E3%82%B0(6)
cms(6)
oracle(6)
perl(6)
gitlab(6)
iam(5)
amazon%20ec2(5)
%E8%B3%87%E6%A0%BC%E8%A9%A6%E9%A8%93(5)
aws%20amplify(5)
curl(4)
Author
githubzennqiita
ただの備忘録です。

※外部送信に関する公表事項