PowerCMS ブログ

先日の記事では、S3 同期の際に Lambda で自動的に CloudFront のキャッシュをパージする 仕組みをご紹介しましたが、その中で作成する Lambda 関数で、ランタイムに Python を選択していました。今回は Node.js を使って同じ仕組みを構築してみましょう。

全体の流れ

先日の記事 でご紹介している手順とほぼ同じで、「Lambda 関数の作成」のみが今回ご紹介する Node.js に変わります。

また今回、コードを差し替えるだけでは面白くないので、合わせて Node.js の動作に必要なモジュールを追加する手順もご紹介します。

準備

先日の記事 のうち、「前提」から「キャッシュパージの回数を記録するための S3 バケットを作成する」までを行ってください。

Lambda 関数の作成(Node.js 版)

新しい関数を作成します。関数名はここでは「limited-cloudfront-purge-by-file-nodejs」とし、ランタイムに「Node.js 14.x」を指定します。他の項目はデフォルトのまま [関数を作成] ボタンを押します。

関数が作成されたら、コードソースに下記をコピー&ペーストします。DISTRIBUTION_ID, ORIGIN_BUCKET_NAME, COUNT_BUCKET_NAME をご利用環境に合わせて変更してください。また、ここでは、ファイル名が「.purge」のファイルに変更があったら、一日に 10 回までキャッシュのパージを行う設定になっています。状況に合わせて COUNT_MAX, TARGET_FILE を変更してください。

また、念のため、CloudWatch Logs に記録するための console.log を入れています。

require('date-utils');

const AWS = require('aws-sdk');
const s3 = new AWS.S3();
const cloudfront = new AWS.CloudFront();

const DISTRIBUTION_ID = 'CloudFront の ID';
const ORIGIN_BUCKET_NAME = 'オリジンである S3 バケットの名前';
const COUNT_BUCKET_NAME='パージ回数を記録する S3 バケットの名前';
const COUNT_MAX=10;
const TARGET_FILE='.purge';

const pattern = new RegExp('^' + TARGET_FILE + '$');

exports.handler = async (event) => {

  let dt = new Date();
  let key = dt.toFormat("YYYY-MM-DD") + '.txt';

  let count = 0;
  await s3.getObject(
    {
      Bucket: COUNT_BUCKET_NAME,
      Key: key,
    }, (err, data) => {
      if (! err) {
        count = parseInt(data.Body.toString(),10);
        if (isNaN(count)) {
          count = 0;
        }
      }
    }
  ).promise().catch(
    function (error) {
      count = 0;
    }
  );

  console.log(count);

  if ( count < COUNT_MAX ) {
    console.log('NOT MAX');
    for (let i = 0; i < event["Records"].length; i++) {
      let item = event["Records"][i];
      if (item["s3"]["bucket"]["name"] == ORIGIN_BUCKET_NAME ) {
        if (item["s3"]["object"]["key"].match(pattern)) {
          console.log('FILENAME PATTERN MATCHED');
          let params = {
            DistributionId: DISTRIBUTION_ID,
            InvalidationBatch: {
              Paths: {
                Quantity: 1,
                Items: ['/*']
              },
              CallerReference: dt.toFormat("YYYYMMDDHH24MISS")
            }
          };
          await cloudfront.createInvalidation(params).promise();
          console.log('INVALIDATION CREATED');
        }
      }
    }
  } else {
    console.log('IS MAX');
    return;
  }
  var params = {
    Bucket: COUNT_BUCKET_NAME,
    Key: key,
    Body: String(count)
  };
  await s3.putObject(params).promise();
  return;
};

mkdir nodejs
cd nodejs
npm install date-utils
cd ../
zip -r nodejs.zip nodejs