Lambda関数URL(Function URLs)をCloud Automatorの後処理に設定してEC2停止後にインスタンスを終了する

こんにちは。Cloud Automatorの開発・運用をしている尾崎です。

先日発表されたLambda Function URLsは個人的に待望の機能でした。従来LambdaをHTTP(s)で呼び出す際に必要だったAPI Gatewayを不要にできる点にメリットを感じています。

今回はこのLambda Function URLsをCloud AutomatorのWebhook後処理に設定して、Cloud Automatorのジョブで「EC2: インスタンスを停止」した後にLambdaからEC2インスタンスの終了をしてみようと思います。
なお、記事の内容は2022/04/20現在のもので、設定値や設定方法に関しては変わる可能性があります。また、Lambdaに登録するコードに関しても動作保証はしませんので、あらかじめご了承ください。

Cloud AutomatorのWebhook後処理について
Lambda Functionのコード例
Lambda関数をCloud Automatorの後処理に設定する
動作確認
まとめ

Cloud AutomatorのWebhook後処理について

Cloud AutomatorのWebhook後処理についてのドキュメントは「Webhook後処理」にまとまっています。
記事中にある「POSTリクエストの内容」を参考にして、Lambda関数では以下のような流れで処理を行います。

設定しているジョブのWebhook後処理のリクエストかどうかを検証。
Cloud Automatorのログに紐づくリソース操作結果取得APIを利用して、停止したEC2インスタンスIDを取得。
AWSのterminate-instancesを利用して、EC2インスタンスを終了。

次のセクションで実際のコード例を記載します。

Lambda Functionのコード例

const https = require("https");
const util = require("util");

const AWS = require("aws-sdk");

// HTTPSリクエストを行う関数
async function request(url, options, data) {
  return new Promise((resolve, reject) => {
    const req = https.request(url, options, (res) => {
      let body = "";
      res.on("data", (chunk) => (body += chunk.toString()));
      res.on("error", reject);
      res.on("end", () => {
        if (res.statusCode >= 200 && res.statusCode <= 299) {
          resolve({
            statusCode: res.statusCode,
            headers: res.headers,
            body: body,
          });
        } else {
          reject(
            "Request failed. status: " + res.statusCode + ", body: " + body,
          );
        }
      });
    });
    req.on("error", reject);
    if (data != null) {
      req.write(data, "binary");
    }
    req.end();
  });
}

// Lambdaのハンドラー
exports.handler = async (event) => {
  // see: https://support.serverworks.co.jp/hc/ja/articles/115002386953-Webhook後処理
  const webhookPayload = JSON.parse(event.body);
  const result = webhookPayload[0];

  // 1. 設定しているジョブのWebhook後処理のリクエストかどうかを検証する
  // ここではジョブID: 10000を対象としています
  if (result.trigger_job_id !== 10000) {
    return {
      statusCode: 404, // Not Found
      body: JSON.stringify({
        result: false,
      }),
    };
  }

  // 2. 停止したEC2インスタンスIDを取得する
  // see: https://api.cloudautomator.com/#ジョブログ-get-2
  const resourceOperationResultsURL = `https://manager.cloudautomator.com/api/v1/logs/${result.trigger_job_log_id}/resource_operation_results`;
  const resultsResp = await request(resourceOperationResultsURL, {
    method: "GET",
    headers: {
      // 「APIキー」はCloud Automatorのアカウント設定画面から確認できます
      Authorization: "Bearer 「APIキー」",
    },
  });
  const instanceIds = JSON.parse(resultsResp.body)
    .data.map((result) => result.attributes.affected_resource_id)
    .filter((rs) => rs != null);

  // 停止インスタンスが0台の場合はterminateInstancesを実行しない
  if (instanceIds.length === 0) {
    return {
      statusCode: 200, // default value
      body: JSON.stringify({
        result: true,
      }),
    };
  }

  // 3. EC2インスタンスを終了する
  // see: https://docs.aws.amazon.com/AWSJavaScriptSDK/latest/AWS/EC2.html#terminateInstances-property
  // ここではap-northeast-1を指定
  const ec2 = new AWS.EC2({
    apiVersion: "2016-11-15",
    region: "ap-northeast-1",
  });
  await util.promisify(ec2.terminateInstances).bind(ec2)({
    InstanceIds: instanceIds,
  });
  return {
    statusCode: 200, // default value
    body: JSON.stringify({
      result: true,
    }),
  };
};

Lambda関数をCloud Automatorの後処理に設定する

Lambda Functionで関数 URLを設定する方法は公式ドキュメントを参照ください。
その際、認証タイプを NONE で設定する必要があります。
また、上記サンプルコードは実行時間がデフォルトの3秒を超える場合があるので、余裕を持ってLambda関数のタイムアウトを10秒程度に設定しました。

その後、生成された関数URLをCloud Automatorの後処理に登録します。